python3Packages.torch: fix ROCm build by disabling USE_FBGEMM_GENAI

This commit is contained in:
Luna Nova
2025-11-01 11:02:43 -07:00
parent b1bf0af247
commit b59b5bae80

View File

@@ -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 = [