diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix index cae1d47971e9..960a5c2a109d 100644 --- a/pkgs/development/python-modules/torch/source/default.nix +++ b/pkgs/development/python-modules/torch/source/default.nix @@ -521,6 +521,9 @@ buildPythonPackage.override { inherit stdenv; } rec { } // lib.optionalAttrs rocmSupport { 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 = [ @@ -760,6 +763,7 @@ buildPythonPackage.override { inherit stdenv; } rec { license = lib.licenses.bsd3; maintainers = with lib.maintainers; [ GaetanLepage + LunNova # esp. for ROCm teh thoughtpolice tscholak