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

Revert "[ROCm] [CK] Composable Kernel integration for inductor backend (#158747)"

Browse Source 
This reverts commit 019fed39aa6b2dd8c69347378d53423e5efae8d4.

Reverted https://github.com/pytorch/pytorch/pull/158747 on behalf of https://github.com/jithunnair-amd due to Broke linux-binary-manywheel-rocm / manywheel-py3_9-rocm6_4-test: 019fed39aa/1 ... PR didn't have this job run successfully due to CI outage ([comment](https://github.com/pytorch/pytorch/pull/158747#issuecomment-3259212343))
This commit is contained in:
PyTorch MergeBot
2025-09-05 17:27:45 +00:00
parent 261a84a176
commit d711f27845
6 changed files with 35 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
torch/_inductor/utils.py
View File

@ -1979,7 +1979,16 @@ def use_ck_template(layout: Layout) -> bool:
log.warning("Please pip install Composable Kernel package")
return False
config.rocm.ck_dir = ck_package_dirname
if config.is_fbcode():
config.rocm.ck_dir = ck_package_dirname
if not config.rocm.ck_dir:
log.warning("Please set TORCHINDUCTOR_CK_DIR env variable")
return False
if ck_package_dirname != config.rocm.ck_dir:
log.warning("Invalid path to CK library")
return False
return True