frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[submodule] upgrade cutlass version to 4.2.1 and completely resolved python/cutlass name collision (#164156)

Browse Source 
Differential Revision: D83489362

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164156
Approved by: https://github.com/Skylion007, https://github.com/mlazos
This commit is contained in:
Henry Tsang
2025-09-30 17:04:57 +00:00
committed by PyTorch MergeBot
parent 906fe7b120
commit 7d7ae4d7b2
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
third_party/cutlass vendored

Submodule third_party/cutlass updated: 57e3cfb47a...f3fde58372

8
torch/_inductor/codegen/cuda/cutlass_utils.py
View File

@ -112,13 +112,13 @@ def try_import_cutlass() -> bool:
) )
cutlass_library_src_path = path_join(cutlass_python_path, "cutlass_library") 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") pycute_src_path = path_join(cutlass_python_path, "pycute")
tmp_cutlass_full_path = os.path.abspath(os.path.join(cache_dir(), "torch_cutlass")) 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_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") dst_link_pycute = path_join(tmp_cutlass_full_path, "pycute")
# mock modules to import cutlass # mock modules to import cutlass
@ -145,7 +145,9 @@ def try_import_cutlass() -> bool:
link_and_append( link_and_append(
dst_link_library, cutlass_library_src_path, tmp_cutlass_full_path 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) link_and_append(dst_link_pycute, pycute_src_path, tmp_cutlass_full_path)
for module in mock_modules: for module in mock_modules: