diff --git a/third_party/cutlass b/third_party/cutlass index 57e3cfb47a2d..f3fde58372d3 160000 --- a/third_party/cutlass +++ b/third_party/cutlass @@ -1 +1 @@ -Subproject commit 57e3cfb47a2d9e0d46eb6335c3dc411498efa198 +Subproject commit f3fde58372d33e9a5650ba7b80fc48b3b49d40c8 diff --git a/torch/_inductor/codegen/cuda/cutlass_utils.py b/torch/_inductor/codegen/cuda/cutlass_utils.py index bdbbbe58a3b1..6b4119952d32 100644 --- a/torch/_inductor/codegen/cuda/cutlass_utils.py +++ b/torch/_inductor/codegen/cuda/cutlass_utils.py @@ -112,13 +112,13 @@ def try_import_cutlass() -> bool: ) cutlass_library_src_path = path_join(cutlass_python_path, "cutlass_library") - cutlass_src_path = path_join(cutlass_python_path, "cutlass") + cutlass_cppgen_src_path = path_join(cutlass_python_path, "cutlass_cppgen") pycute_src_path = path_join(cutlass_python_path, "pycute") tmp_cutlass_full_path = os.path.abspath(os.path.join(cache_dir(), "torch_cutlass")) dst_link_library = path_join(tmp_cutlass_full_path, "cutlass_library") - dst_link_cutlass = path_join(tmp_cutlass_full_path, "cutlass_cppgen") + dst_link_cutlass_cppgen = path_join(tmp_cutlass_full_path, "cutlass_cppgen") dst_link_pycute = path_join(tmp_cutlass_full_path, "pycute") # mock modules to import cutlass @@ -145,7 +145,9 @@ def try_import_cutlass() -> bool: link_and_append( dst_link_library, cutlass_library_src_path, tmp_cutlass_full_path ) - link_and_append(dst_link_cutlass, cutlass_src_path, tmp_cutlass_full_path) + link_and_append( + dst_link_cutlass_cppgen, cutlass_cppgen_src_path, tmp_cutlass_full_path + ) link_and_append(dst_link_pycute, pycute_src_path, tmp_cutlass_full_path) for module in mock_modules: