From 221026d0260bafa696daea577ff0901fad5e53e8 Mon Sep 17 00:00:00 2001 From: Josh Fromm Date: Tue, 19 Jul 2022 09:11:25 -0700 Subject: [PATCH] Fix test type. --- tests/python/topi/python/test_topi_transform.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/python/topi/python/test_topi_transform.py b/tests/python/topi/python/test_topi_transform.py index da4a8c033afc0..c3155c948a8df 100644 --- 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)