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

python3Packages.torch: fix ROCm build by disabling USE_FBGEMM_GENAI

Browse Source
This commit is contained in:
Luna Nova
2025-11-01 11:02:43 -07:00
parent b1bf0af247
commit b59b5bae80
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pkgs/development/python-modules/torch/source/default.nix
View File

@@ -521,6 +521,9 @@ buildPythonPackage.override { inherit stdenv; } rec {
} }
// lib.optionalAttrs rocmSupport { // lib.optionalAttrs rocmSupport {
AOTRITON_INSTALLED_PREFIX = "${rocmPackages.aotriton}"; AOTRITON_INSTALLED_PREFIX = "${rocmPackages.aotriton}";
# Broken HIP flag setup, fails to compile due to not finding rocthrust
# Only supports gfx942 so let's turn it off for now
USE_FBGEMM_GENAI = setBool false;
}; };
nativeBuildInputs = [ nativeBuildInputs = [