frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +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

8
setup.py
View File

@ -324,7 +324,6 @@ from tools.setup_helpers.env import (
IS_WINDOWS,
)
from tools.setup_helpers.generate_linker_script import gen_linker_script
from tools.setup_helpers.rocm_env import get_ck_dependency_string, IS_ROCM
def str2bool(value: str | None) -> bool:
@ -507,6 +506,7 @@ else:
sysconfig.get_config_var("LIBDIR")
) / sysconfig.get_config_var("INSTSONAME")
################################################################################
# Version, create_version_file, and package_name
################################################################################
@ -1494,12 +1494,6 @@ def configure_extension_build() -> tuple[
map(str.strip, pytorch_extra_install_requires.split("|"))
)
# Adding extra requirements for ROCm builds
if IS_ROCM and platform.system() == "Linux":
extra_install_requires.append(
f"rocm-composable-kernel {get_ck_dependency_string()}"
)
# Cross-compile for M1
if IS_DARWIN:
macos_target_arch = os.getenv("CMAKE_OSX_ARCHITECTURES", "")