diff --git a/bitblas/tl/tuner.py b/bitblas/tl/tuner.py index e3a7b5fd..fd3c98ef 100644 --- a/bitblas/tl/tuner.py +++ b/bitblas/tl/tuner.py @@ -122,6 +122,7 @@ def tvm_callback_cuda_postproc(code, _): code = tensor_remove_make_int4(code) code = tensor_remove_make_int2(code) return code + # check only have one function in the module if len(mod.functions) > 1: raise ValueError("Only support one function in the module")