We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9a3ec23 commit cd3e107Copy full SHA for cd3e107
python/tvm/contrib/cutlass/build.py
@@ -28,6 +28,7 @@
28
from tvm import relax, relay, runtime
29
from tvm._ffi.registry import register_func
30
from tvm.contrib.nvcc import get_cuda_version
31
+from tvm.topi.utils import get_const_tuple
32
33
from .gen_conv2d import CutlassConv2DProfiler
34
from .gen_gemm import CutlassGemmProfiler
@@ -545,7 +546,7 @@ def _extract_relax_function_signature(f):
545
546
547
for i, arg in enumerate(f.params):
548
sinfo = arg.struct_info
- signature["arg%d_shape" % i] = list(sinfo.shape)
549
+ signature["arg%d_shape" % i] = get_const_tuple(sinfo.shape)
550
signature["arg%d_dtype" % i] = sinfo.dtype
551
552
ret_sinfo = f.ret_struct_info
0 commit comments