Skip to content

Commit

Permalink
tensorflow: disable XLA without CUDA
Browse files Browse the repository at this point in the history
This configuration is sadly broken.
  • Loading branch information
abbradar committed Feb 27, 2018
1 parent d86999e commit 94ebc13
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions pkgs/development/python-modules/tensorflow/default.nix
Expand Up @@ -3,8 +3,9 @@
, which, swig, binutils, glibcLocales
, python, jemalloc, openmpi
, numpy, six, protobuf, tensorflow-tensorboard, backports_weakref, mock, enum34, absl-py
, xlaSupport ? true
, cudaSupport ? false, nvidia_x11 ? null, cudatoolkit ? null, cudnn ? null
# XLA without CUDA is broken
, xlaSupport ? cudaSupport
# Default from ./configure script
, cudaCapabilities ? [ "3.5" "5.2" ]
, sse42Support ? false
Expand Down Expand Up @@ -145,9 +146,10 @@ in buildPythonPackage rec {

meta = with stdenv.lib; {
description = "Computation using data flow graphs for scalable machine learning";
homepage = "http://tensorflow.org";
homepage = http://tensorflow.org;
license = licenses.asl20;
maintainers = with maintainers; [ jyp abbradar ];
platforms = with platforms; if cudaSupport then linux else linux ++ darwin;
broken = !(xlaSupport -> cudaSupport);
};
}

0 comments on commit 94ebc13

Please sign in to comment.