[Fix] Add test case for cornell-zhang#458

zzzDavid · May 3, 2022 · cd0affe · cd0affe
1 parent 62efe4f
commit cd0affe
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/issues/test_issue_458.py b/tests/issues/test_issue_458.py
@@ -0,0 +1,15 @@
+import heterocl as hcl
+
+def test():
+    # generates: assert ((uint32)0 < uint32(A[0]));
+    # the second uint32 is oddly specified ...
+    def func(A):
+        hcl.assert_(A[0] > 0, "")
+
+    A = hcl.placeholder((2,), "A", dtype=hcl.UInt(16))
+    s = hcl.create_schedule([A], func)
+    m = hcl.build(s, "shls")
+    print(m)
+
+if __name__ == "__main__":
+    test()