From d0cf420bffa17728f574a51f947e3ee21c42fde6 Mon Sep 17 00:00:00 2001 From: Yaxing Cai Date: Sun, 31 Jul 2022 11:25:04 -0700 Subject: [PATCH] [Fix] Minor modification in unittests (#12247) Update unittests to align with the expected behavior --- tests/python/unittest/test_arith_domain_touched.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/unittest/test_arith_domain_touched.py b/tests/python/unittest/test_arith_domain_touched.py index 4e508b283df1..3641f06ab8a2 100644 --- 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))