Unverified Commit b59b5bae authored by Luna Nova's avatar Luna Nova
Browse files

python3Packages.torch: fix ROCm build by disabling USE_FBGEMM_GENAI

parent b1bf0af2

pkgs/development/python-modules/torch/source/default.nix

+3 −0
Original line number Diff line number Diff line
@@ -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 = [