BoomMicrophone/nixpkgs-suyu
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

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

Browse source 
python311Packages.tensorflow: llvmPackages_11 -> llvmPackages
This commit is contained in:
a-n-n-a-l-e-e 2024-02-19 19:15:57 -08:00 committed by GitHub
commit cb298310ab
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pkgs/development/python-modules/tensorflow/default.nix
View file

@ -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)