Unverified Commit cb298310 authored by a-n-n-a-l-e-e's avatar a-n-n-a-l-e-e Committed by GitHub
Browse files

Merge pull request #288960 from a-n-n-a-l-e-e/tensorflow-llvm11-unpin 

python311Packages.tensorflow: llvmPackages_11 -> llvmPackages
parents 3c0e8a15 64820eca

pkgs/development/python-modules/tensorflow/default.nix

+2 −2
Original line number Diff line number Diff line
@@ -29,7 +29,7 @@ 
, avx2Support  ? stdenv.hostPlatform.avx2Support

, fmaSupport   ? stdenv.hostPlatform.fmaSupport

# Darwin deps

, Foundation, Security, cctools, llvmPackages_11

, Foundation, Security, cctools, llvmPackages

}:



let
@@ -51,7 +51,7 @@ let 
  # translation units, so the build fails at link time

  stdenv =

    if cudaSupport then cudaPackagesGoogle.backendStdenv

    else if originalStdenv.isDarwin then llvmPackages_11.stdenv

    else if originalStdenv.isDarwin then llvmPackages.stdenv

    else originalStdenv;

  inherit (cudaPackagesGoogle) cudatoolkit nccl;

  # use compatible cuDNN (https://www.tensorflow.org/install/source#gpu)