about summary refs log tree commit diff
path: root/pkgs/applications
diff options
context:
space:
mode:
authorJan Malakhovski <oxij@oxij.org>2019-02-03 15:32:15 +0000
committerJan Malakhovski <oxij@oxij.org>2019-02-03 15:32:15 +0000
commit31aa884b7f9108103e244ade50fe3c9ce4d6b9c3 (patch)
treed5128861ff53b8bffb2a70fc8c2422b6290b5194 /pkgs/applications
parent49148a3ea264bf89db13879c6f3bbeba43dad1bb (diff)
downloadnixlib-31aa884b7f9108103e244ade50fe3c9ce4d6b9c3.tar
nixlib-31aa884b7f9108103e244ade50fe3c9ce4d6b9c3.tar.gz
nixlib-31aa884b7f9108103e244ade50fe3c9ce4d6b9c3.tar.bz2
nixlib-31aa884b7f9108103e244ade50fe3c9ce4d6b9c3.tar.lz
nixlib-31aa884b7f9108103e244ade50fe3c9ce4d6b9c3.tar.xz
nixlib-31aa884b7f9108103e244ade50fe3c9ce4d6b9c3.tar.zst
nixlib-31aa884b7f9108103e244ade50fe3c9ce4d6b9c3.zip
caffe: move defaults to package file
Diffstat (limited to 'pkgs/applications')
-rw-r--r--pkgs/applications/science/math/caffe/default.nix6
1 files changed, 3 insertions, 3 deletions
diff --git a/pkgs/applications/science/math/caffe/default.nix b/pkgs/applications/science/math/caffe/default.nix
index e56c63e01bf0..c0c26d6e453c 100644
--- a/pkgs/applications/science/math/caffe/default.nix
+++ b/pkgs/applications/science/math/caffe/default.nix
@@ -1,4 +1,4 @@
-{ stdenv, lib, runCommand
+{ config, stdenv, lib, runCommand
 , fetchFromGitHub
 , fetchurl
 , cmake
@@ -13,8 +13,8 @@
 , Accelerate, CoreGraphics, CoreVideo
 , lmdbSupport ? true, lmdb
 , leveldbSupport ? true, leveldb, snappy
-, cudaSupport ? stdenv.isLinux, cudatoolkit
-, cudnnSupport ? false, cudnn ? null
+, cudaSupport ? config.cudaSupport or false, cudatoolkit
+, cudnnSupport ? cudaSupport, cudnn ? null
 , ncclSupport ? false, nccl ? null
 , pythonSupport ? false, python ? null, numpy ? null
 }: