diff options
| author | volth <volth@volth.com> | 2020-08-05 02:32:41 +0000 |
|---|---|---|
| committer | volth <volth@volth.com> | 2020-08-05 11:18:26 +0000 |
| commit | cf7b63df5b9efdef4e8e1b3261d7040199f7e671 (patch) | |
| tree | 74dbf7a1731896d48e90671f2b3dfb4dff91ec40 /pkgs/development/python-modules/tensorflow/2/default.nix | |
| parent | 463db72e631e0bd4a835796324eb1981071c6ee4 (diff) | |
| download | nixpkgs-cf7b63df5b9efdef4e8e1b3261d7040199f7e671.tar nixpkgs-cf7b63df5b9efdef4e8e1b3261d7040199f7e671.tar.gz nixpkgs-cf7b63df5b9efdef4e8e1b3261d7040199f7e671.tar.bz2 nixpkgs-cf7b63df5b9efdef4e8e1b3261d7040199f7e671.tar.lz nixpkgs-cf7b63df5b9efdef4e8e1b3261d7040199f7e671.tar.xz nixpkgs-cf7b63df5b9efdef4e8e1b3261d7040199f7e671.tar.zst nixpkgs-cf7b63df5b9efdef4e8e1b3261d7040199f7e671.zip | |
gcc.arch: refactor, move tables under lib/
Diffstat (limited to 'pkgs/development/python-modules/tensorflow/2/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/tensorflow/2/default.nix | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/pkgs/development/python-modules/tensorflow/2/default.nix b/pkgs/development/python-modules/tensorflow/2/default.nix index 8f30ecbcc98..eedd6e6d0df 100644 --- a/pkgs/development/python-modules/tensorflow/2/default.nix +++ b/pkgs/development/python-modules/tensorflow/2/default.nix @@ -23,9 +23,9 @@ , xlaSupport ? cudaSupport # Default from ./configure script , cudaCapabilities ? [ "3.5" "5.2" ] -, sse42Support ? builtins.elem (stdenv.hostPlatform.platform.gcc.arch or "default") ["westmere" "sandybridge" "ivybridge" "haswell" "broadwell" "skylake" "skylake-avx512" "btver2" "bdver1" "bdver2" "bdver3" "bdver4" "znver1"] -, avx2Support ? builtins.elem (stdenv.hostPlatform.platform.gcc.arch or "default") [ "haswell" "broadwell" "skylake" "skylake-avx512" "bdver4" "znver1"] -, fmaSupport ? builtins.elem (stdenv.hostPlatform.platform.gcc.arch or "default") [ "haswell" "broadwell" "skylake" "skylake-avx512" "bdver2" "bdver3" "bdver4" "znver1"] +, sse42Support ? stdenv.hostPlatform.sse4_2Support +, avx2Support ? stdenv.hostPlatform.avx2Support +, fmaSupport ? stdenv.hostPlatform.fmaSupport # Darwin deps , Foundation, Security }: |