test fix

tile-ai · Feb 19, 2025 · 1a62699 · 1a62699
1 parent f10eed5
commit 1a62699
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/testing/python/jit/test_tilelang_jit_gemm_ctypes.py b/testing/python/jit/test_tilelang_jit_gemm_ctypes.py
@@ -88,7 +88,7 @@ def run_gemm(
 
     stramp = "&*(XS)"
 
-    @tvm.register_func
+    @tvm.register_func(override=True)
     def tilelang_callback_cuda_postproc(code, _):
         code = f"// {stramp}\n" + code
         return code