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

mxnet: fix cuda support enabled build (#48781)

Browse Source
This commit is contained in:
Julien Dehos 2019-04-12 12:31:22 +02:00 committed by Joachim F
parent 728c6b7582
commit a13f5d0da1
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pkgs/applications/science/math/mxnet/default.nix
View File

@ -26,6 +26,7 @@ stdenv.mkDerivation rec {
cmakeFlags = cmakeFlags =
(if cudaSupport then [ (if cudaSupport then [
"-DUSE_OLDCMAKECUDA=ON" # see https://github.com/apache/incubator-mxnet/issues/10743
"-DCUDA_ARCH_NAME=All" "-DCUDA_ARCH_NAME=All"
"-DCUDA_HOST_COMPILER=${cudatoolkit.cc}/bin/cc" "-DCUDA_HOST_COMPILER=${cudatoolkit.cc}/bin/cc"
] else [ "-DUSE_CUDA=OFF" ]) ] else [ "-DUSE_CUDA=OFF" ])