From 94ebc13a6ac5c6448a932ca48ae9e2bd9ce755ea Mon Sep 17 00:00:00 2001 From: Nikolay Amiantov Date: Tue, 27 Feb 2018 03:16:33 +0300 Subject: [PATCH] tensorflow: disable XLA without CUDA This configuration is sadly broken. --- pkgs/development/python-modules/tensorflow/default.nix | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pkgs/development/python-modules/tensorflow/default.nix b/pkgs/development/python-modules/tensorflow/default.nix index e51ceb313ef2..85dd30118cc3 100644 --- a/pkgs/development/python-modules/tensorflow/default.nix +++ b/pkgs/development/python-modules/tensorflow/default.nix @@ -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 @@ -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); }; }