add a test for taichi-dev#813

tests/python/test_offload_cross.py

@@ -45,7 +45,7 @@ def ker():
 
 
 @ti.all_archs
-def test_offload_with_cross_block_locals2():
+def test_offload_with_cross_block_locals3():
     ret = ti.var(ti.f32, shape=())
 
     @ti.kernel
@@ -61,6 +61,23 @@ def ker():
     assert ret[None] == 1
 
 
+@ti.all_archs
+def test_offload_with_cross_block_locals4():
+    ret = ti.var(ti.f32, shape=())
+
+    @ti.kernel
+    def ker():
+        a = 1
+        b = 0
+        for i in range(10):
+            b += a
+        ret[None] = b
+
+    ker()
+
+    assert ret[None] == 10
+
+
 @ti.archs_excluding(ti.opengl)  # OpenGL doesn't support dynamic range for now
 def test_offload_with_flexible_bounds():
     s = ti.var(ti.i32, shape=())