added a test that crashes due to allocate using an undefined variable

Author: bulanova-huawei

tests/python/unittest/test_pass_split_host_device.py

@@ -29,5 +29,21 @@ def test_loop_dependent_allocate():
     # referencing undefined variable
     tvm.lower(s, [A,C])
 
+def test_loop_dependent_allocate2():
+    import tvm
+    m = tvm.convert(12)
+    l = tvm.convert(6)
+    A = tvm.placeholder((m, l), name='A')
+    A1 = tvm.compute((m, l), lambda i, j: A[i, j], name='A1')
+    A2 = tvm.compute((m, l), lambda i, j: A1[i, j] + 3, name='A2')
+    s = tvm.create_schedule(A2.op)
+
+    fused_axes = s[A2].fuse(A2.op.axis[0], A2.op.axis[1])
+    xo, xi = s[A2].split(fused_axes, 10)
+    s[A1].compute_at(s[A2], xo)
+
+    print(tvm.lower(s, [A, A2]))
+
 if __name__ == "__main__":
     test_loop_dependent_allocate()
+    test_loop_dependent_allocate2()