File pytorch-2.7.1-cpp-extension-multilib.patch of Package python-torch
Fixes multilib (%LIB_DIR% should be replaced in ebuild)
--- a/torch/utils/cpp_extension.py
+++ b/torch/utils/cpp_extension.py
@@ -1478,10 +1477,10 @@ def library_paths(device_type: str = "cpu") -> list[str]:
paths = [TORCH_LIB_PATH]
if device_type == "cuda" and IS_HIP_EXTENSION:
- lib_dir = 'lib'
+ lib_dir = '@LIB_DIR@'
paths.append(_join_rocm_home(lib_dir))
if HIP_HOME is not None:
- paths.append(os.path.join(HIP_HOME, 'lib'))
+ paths.append(os.path.join(HIP_HOME, '@LIB_DIR@'))
elif device_type == "cuda":
if IS_WINDOWS:
lib_dir = os.path.join('lib', 'x64')
@@ -2295,7 +2294,7 @@ def _prepare_ldflags(extra_ldflags, with_cuda, verbose, is_standalone):
if CUDNN_HOME is not None:
extra_ldflags.append(f'-L{os.path.join(CUDNN_HOME, "lib64")}')
elif IS_HIP_EXTENSION:
- extra_ldflags.append(f'-L{_join_rocm_home("lib")}')
+ extra_ldflags.append(f'-L{_join_rocm_home("@LIB_DIR@")}')
extra_ldflags.append('-lamdhip64')
return extra_ldflags