gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

python3Packages.torchWithRocm: allow gfx1151 (Strix Halo)

Browse Source 
gfx1151 support in torch was fixed by the ROCm 6.4 bump
This commit is contained in:
Luna Nova
2025-09-14 17:45:24 -07:00
parent 5e31e13667
commit 3c2cbd04cb
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/torch/source/default.nix
+2 -6
View File
@@ -187,11 +187,7 @@ let
# TODO: Retest after ROCm 6.4 or torch 2.8
"gfx1010"
"gfx1012"
# Strix Halo seems to be broken as well, see
# https://github.com/NixOS/nixpkgs/pull/440359.
"gfx1151"
] (rocmPackages.clr.localGpuTargets or rocmPackages.clr.gpuTargets)
] rocmPackages.clr.localGpuTargets or rocmPackages.clr.gpuTargets
else
throw "No GPU targets specified"
);
@@ -412,7 +408,7 @@ buildPythonPackage rec {
export ROCM_PATH=${rocmtoolkit_joined}
export ROCM_SOURCE_DIR=${rocmtoolkit_joined}
export PYTORCH_ROCM_ARCH="${gpuTargetString}"
export CMAKE_CXX_FLAGS="-I${rocmtoolkit_joined}/include -I${rocmtoolkit_joined}/include/rocblas"
export CMAKE_CXX_FLAGS="-I${rocmtoolkit_joined}/include"
python tools/amd_build/build_amd.py
'';