diff options
| author | Jean-Philippe Bernardy <jeanphilippe.bernardy@gmail.com> | 2017-10-25 10:56:51 +0200 |
|---|---|---|
| committer | Nikolay Amiantov <ab@fmap.me> | 2017-11-04 13:18:11 +0300 |
| commit | 6269306a8dfa0744ddc4609946a8105b996cd23a (patch) | |
| tree | 2816a4ef380e570679a4d58a7156f0b3ff35fd46 /pkgs/development/python-modules/tensorflow | |
| parent | bb87d2c81a6405032d23cc46c4e859d7818b46cd (diff) | |
| download | nixlib-6269306a8dfa0744ddc4609946a8105b996cd23a.tar nixlib-6269306a8dfa0744ddc4609946a8105b996cd23a.tar.gz nixlib-6269306a8dfa0744ddc4609946a8105b996cd23a.tar.bz2 nixlib-6269306a8dfa0744ddc4609946a8105b996cd23a.tar.lz nixlib-6269306a8dfa0744ddc4609946a8105b996cd23a.tar.xz nixlib-6269306a8dfa0744ddc4609946a8105b996cd23a.tar.zst nixlib-6269306a8dfa0744ddc4609946a8105b996cd23a.zip | |
pythonPackages.tensorflow: add flags for efficent math on CPU
Diffstat (limited to 'pkgs/development/python-modules/tensorflow')
| -rw-r--r-- | pkgs/development/python-modules/tensorflow/default.nix | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tensorflow/default.nix b/pkgs/development/python-modules/tensorflow/default.nix index a0a2f3a24afc..2ce419fa7fb6 100644 --- a/pkgs/development/python-modules/tensorflow/default.nix +++ b/pkgs/development/python-modules/tensorflow/default.nix @@ -7,6 +7,9 @@ , cudaSupport ? false, nvidia_x11 ? null, cudatoolkit ? null, cudnn ? null # Default from ./configure script , cudaCapabilities ? [ "3.5" "5.2" ] +, sse42Support ? false +, avx2Support ? false +, fmaSupport ? false }: assert cudaSupport -> cudatoolkit != null @@ -77,6 +80,9 @@ let hardeningDisable = [ "all" ]; bazelFlags = [ "--config=opt" ] + ++ lib.optional sse42Support "--copt=-msse4.2" + ++ lib.optional avx2Support "--copt=-mavx2" + ++ lib.optional fmaSupport "--copt=-mfma" ++ lib.optional cudaSupport "--config=cuda"; bazelTarget = "//tensorflow/tools/pip_package:build_pip_package"; |