diff options
Diffstat (limited to 'pkgs/development/python-modules/libgpuarray')
| -rw-r--r-- | pkgs/development/python-modules/libgpuarray/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/libgpuarray/default.nix b/pkgs/development/python-modules/libgpuarray/default.nix index 2be115ae3658..89a83df42684 100644 --- a/pkgs/development/python-modules/libgpuarray/default.nix +++ b/pkgs/development/python-modules/libgpuarray/default.nix @@ -9,7 +9,7 @@ , six , nose , Mako -, cudaSupport ? false, cudatoolkit +, cudaSupport ? false, cudaPackages , openclSupport ? true, ocl-icd, clblas }: @@ -30,7 +30,7 @@ buildPythonPackage rec { configurePhase = "cmakeConfigurePhase"; libraryPath = lib.makeLibraryPath ( - lib.optionals cudaSupport [ cudatoolkit.lib cudatoolkit.out ] + lib.optionals cudaSupport (with cudaPackages; [ cudatoolkit.lib cudatoolkit.out ]) ++ lib.optionals openclSupport ([ clblas ] ++ lib.optional (!stdenv.isDarwin) ocl-icd) ); |