python3Packages.torch: fix ROCm build (#457612)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user