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

rocmPackages: add gfx1151 to list of broken arches for pytorch

Browse Source
This commit is contained in:
agentelement
2025-09-06 15:24:53 -07:00
parent f781c49241
commit 88325db237
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/torch/source/default.nix
+7 -3
View File
@@ -175,12 +175,16 @@ let
else if cudaSupport then
gpuArchWarner supportedCudaCapabilities unsupportedCudaCapabilities
else if rocmSupport then
# Remove RDNA1 gfx101x archs from default ROCm support list to avoid
# use of undeclared identifier 'CK_BUFFER_RESOURCE_3RD_DWORD'
# TODO: Retest after ROCm 6.4 or torch 2.8
lib.lists.subtractLists [
# Remove RDNA1 gfx101x archs from default ROCm support list to avoid
# use of undeclared identifier 'CK_BUFFER_RESOURCE_3RD_DWORD'
# 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)
else
throw "No GPU targets specified"