Fix test type.

apache · Jul 19, 2022 · 221026d · 221026d
1 parent 698e978
commit 221026d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/python/topi/python/test_topi_transform.py b/tests/python/topi/python/test_topi_transform.py
@@ -814,7 +814,8 @@ def check_device(target, dev):
 
 def verify_trilu(input_shape, upper, k=0):
     x = te.placeholder(shape=input_shape, name="x", dtype="float32")
-    trilu_result = topi.transform.trilu(x, k, upper)
+    k_tir = tvm.tir.const(k, dtype="int32")
+    trilu_result = topi.transform.trilu(x, k_tir, upper)
 
     def check_device(target, dev):
         print("Running on target: %s" % target)