[Fix] Minor modification in unittests (apache#12247)

Update unittests to align with the expected behavior
daobook · Nov 25, 2022 · d0cf420 · d0cf420
1 parent 8888b18
commit d0cf420
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_arith_domain_touched.py b/tests/python/unittest/test_arith_domain_touched.py
@@ -22,7 +22,7 @@
 @T.prim_func
 def scalar_func(a: T.handle, b: T.handle):
     m = T.var("int32")
-    n = T.int32(100)
+    n = 100
     A = T.match_buffer(a, (n, m))
     B = T.match_buffer(b, (n, m))
 
@@ -33,7 +33,7 @@ def scalar_func(a: T.handle, b: T.handle):
 @T.prim_func
 def vector_func(a: T.handle, b: T.handle):
     n = T.var("int32")
-    m = T.int32(128)
+    m = 128
     A = T.match_buffer(a, (n, m))
     B = T.match_buffer(b, (n, m))