public/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tensorflow: disable XLA without CUDA

Browse Source 
This configuration is sadly broken.
This commit is contained in:
Nikolay Amiantov 2018-02-27 03:16:33 +03:00
parent d86999ee25
commit 94ebc13a6a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkgs/development/python-modules/tensorflow/default.nix
View File

@ -3,8 +3,9 @@
, which, swig, binutils, glibcLocales , which, swig, binutils, glibcLocales
, python, jemalloc, openmpi , python, jemalloc, openmpi
, numpy, six, protobuf, tensorflow-tensorboard, backports_weakref, mock, enum34, absl-py , numpy, six, protobuf, tensorflow-tensorboard, backports_weakref, mock, enum34, absl-py
, xlaSupport ? true
, cudaSupport ? false, nvidia_x11 ? null, cudatoolkit ? null, cudnn ? null , cudaSupport ? false, nvidia_x11 ? null, cudatoolkit ? null, cudnn ? null
# XLA without CUDA is broken
, xlaSupport ? cudaSupport
# Default from ./configure script # Default from ./configure script
, cudaCapabilities ? [ "3.5" "5.2" ] , cudaCapabilities ? [ "3.5" "5.2" ]
, sse42Support ? false , sse42Support ? false
@ -145,9 +146,10 @@ in buildPythonPackage rec {
meta = with stdenv.lib; { meta = with stdenv.lib; {
description = "Computation using data flow graphs for scalable machine learning"; description = "Computation using data flow graphs for scalable machine learning";
homepage = "http://tensorflow.org"; homepage = http://tensorflow.org;
license = licenses.asl20; license = licenses.asl20;
maintainers = with maintainers; [ jyp abbradar ]; maintainers = with maintainers; [ jyp abbradar ];
platforms = with platforms; if cudaSupport then linux else linux ++ darwin; platforms = with platforms; if cudaSupport then linux else linux ++ darwin;
broken = !(xlaSupport -> cudaSupport);
}; };
} }