about summary refs log tree commit diff
path: root/pkgs/applications/science/math/caffe
diff options
context:
space:
mode:
authorMichael Raskin <7c6f434c@mail.ru>2019-02-19 21:54:02 +0000
committerGitHub <noreply@github.com>2019-02-19 21:54:02 +0000
commit446520bb9bafc85c9464f7364ff25512b5fdff99 (patch)
treedfd9d32700d80d34c828a3580c65f81b38c89a1d /pkgs/applications/science/math/caffe
parent5fc8559fbb545a74c62a3d45fa54512de72a7293 (diff)
parent69048c4357fdc3356983bd929dd25cdacbb171b2 (diff)
Merge pull request #56026 from oxij/tree/move-defaults-to-package-files-half-cuda
all-packages.nix: move defaults to package files continues^2
Diffstat (limited to 'pkgs/applications/science/math/caffe')
-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 4e8829f7e31c9..9ee56041ed389 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
 , substituteAll