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

python3Packages.torch: fix ROCm build (#457612)

Browse Source
This commit is contained in:
Gaétan Lepage
2025-11-01 23:06:09 +00:00
committed by GitHub
parent 5ea05c1ddd 0610e52b10
commit 6dc66364c3
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
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 = [
@@ -760,6 +763,7 @@ buildPythonPackage.override { inherit stdenv; } rec {
license = lib.licenses.bsd3; license = lib.licenses.bsd3;
maintainers = with lib.maintainers; [ maintainers = with lib.maintainers; [
GaetanLepage GaetanLepage
LunNova # esp. for ROCm
teh teh
thoughtpolice thoughtpolice
tscholak tscholak