[tvm] branch unity updated: [Unity] Fix getting shapes for cutlass BYOC kernels (#14411)

[masahi]

This is an automated email from the ASF dual-hosted git repository.

masahi pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/unity by this push:
     new cd3e10786d [Unity] Fix getting shapes for cutlass BYOC kernels (#14411)
cd3e10786d is described below

commit cd3e10786da6820fe878f618c323db113f056bda
Author: Wuwei Lin <wu...@apache.org>
AuthorDate: Mon Mar 27 19:09:10 2023 -0700

    [Unity] Fix getting shapes for cutlass BYOC kernels (#14411)
---
 python/tvm/contrib/cutlass/build.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/tvm/contrib/cutlass/build.py 
b/python/tvm/contrib/cutlass/build.py
index 47bdcaa790..d9eefd34a3 100644
--- a/python/tvm/contrib/cutlass/build.py
+++ b/python/tvm/contrib/cutlass/build.py
@@ -28,6 +28,7 @@ import tvm
 from tvm import relax, relay, runtime
 from tvm._ffi.registry import register_func
 from tvm.contrib.nvcc import get_cuda_version
+from tvm.topi.utils import get_const_tuple
 
 from .gen_conv2d import CutlassConv2DProfiler
 from .gen_gemm import CutlassGemmProfiler
@@ -545,7 +546,7 @@ def _extract_relax_function_signature(f):
 
     for i, arg in enumerate(f.params):
         sinfo = arg.struct_info
-        signature["arg%d_shape" % i] = list(sinfo.shape)
+        signature["arg%d_shape" % i] = get_const_tuple(sinfo.shape)
         signature["arg%d_dtype" % i] = sinfo.dtype
 
     ret_sinfo = f.ret_struct_info