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
commit e73558e41de6c731a6cd9d3eb3df15bc99782a4c Author: Masahiro Masuda <masahi...@gmail.com> AuthorDate: Sat Dec 2 01:27:46 2023 +0000 WA for H100 --- python/tvm/contrib/cutlass/gen_conv2d.py | 2 +- python/tvm/contrib/cutlass/gen_gemm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/contrib/cutlass/gen_conv2d.py b/python/tvm/contrib/cutlass/gen_conv2d.py index 3d14a427b1..37121eb8b5 100644 --- a/python/tvm/contrib/cutlass/gen_conv2d.py +++ b/python/tvm/contrib/cutlass/gen_conv2d.py @@ -185,7 +185,7 @@ class CutlassConv2DProfiler: def __init__(self, sm, cutlass_path, binary_path): self.gemm_profiler = CutlassGemmProfiler(sm, cutlass_path, binary_path) self.sm = sm - assert sm in GENERATOR_FUNC_TABLE, f"sm{sm} not supported yet." + # assert sm in GENERATOR_FUNC_TABLE, f"sm{sm} not supported yet." self.engine = ProfilerEngine(sm, cutlass_path, binary_path) self.cache_path = os.path.join(binary_path, "cutlass_conv2d_cache.pickle") if os.path.exists(self.cache_path): diff --git a/python/tvm/contrib/cutlass/gen_gemm.py b/python/tvm/contrib/cutlass/gen_gemm.py index a3204791f0..123385cefd 100644 --- a/python/tvm/contrib/cutlass/gen_gemm.py +++ b/python/tvm/contrib/cutlass/gen_gemm.py @@ -194,7 +194,7 @@ class CutlassGemmProfiler: """Profile all candidate kernels and select the best one.""" def __init__(self, sm, cutlass_path, binary_path): - assert sm in GENERATOR_FUNC_TABLE and sm in DEFAULT_KERNELS, f"sm{sm} not supported yet." + # assert sm in GENERATOR_FUNC_TABLE and sm in DEFAULT_KERNELS, f"sm{sm} not supported yet." self.engine = ProfilerEngine(sm, cutlass_path, binary_path) self.sm = sm self.cache_path = os.path.join(binary_path, "cutlass_gemm_cache.pickle")