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

redo of [Inductor][Cutlass] verify cutlass has cache_file attribute before moving...resolves cutlass cute exception (#158206)

Browse Source 
trying to land https://github.com/pytorch/pytorch/pull/156672

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158206
Approved by: https://github.com/lessw2020, https://github.com/Skylion007
This commit is contained in:
henrylhtsang
2025-07-13 17:58:25 -07:00
committed by PyTorch MergeBot
parent 5633283574
commit f87d117939
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -45,7 +45,10 @@ def move_cutlass_compiled_cache() -> None:
else:
import cutlass as python_cutlass # type: ignore[import-not-found] # noqa: F401
if not os.path.exists(python_cutlass.CACHE_FILE):
# Check if the CACHE_FILE attribute exists in python_cutlass and if the file exists
if not hasattr(python_cutlass, "CACHE_FILE") or not os.path.exists(
python_cutlass.CACHE_FILE
):
return
try: