Unverified Commit 6dc66364 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.torch: fix ROCm build (#457612)

parents 5ea05c1d 0610e52b

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

+4 −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 = [
@@ -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