MirrorHub/nixpkgs
0
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-16 23:03:40 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

This allows cupy to find the thrust library which is needed among other things for cupy.argsort

Browse source 
Removed unnecessary dependencies

Remove extra whitespace
This commit is contained in:
Christoph Stich 2020-01-14 12:26:09 +00:00 committed by Frederik Rietdijk
parent 019b637fb1
commit eaeb87add1
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pkgs/development/python-modules/cupy/default.nix
View file

@ -1,6 +1,6 @@
{ stdenv, buildPythonPackage
, fetchPypi, isPy3k, linuxPackages
, fastrlock, numpy, six, wheel, pytest, mock
, fastrlock, numpy, six, wheel, pytest, mock, setuptools
, cudatoolkit, cudnn, nccl
}:
@ -18,6 +18,10 @@ buildPythonPackage rec {
mock
];
preConfigure = ''
export CUDA_PATH=${cudatoolkit}
'';
propagatedBuildInputs = [
cudatoolkit
cudnn
@ -26,12 +30,15 @@ buildPythonPackage rec {
fastrlock
numpy
six
setuptools
wheel
];
# In python3, test was failed...
doCheck = !isPy3k;
enableParallelBuilding = true;
meta = with stdenv.lib; {
description = "A NumPy-compatible matrix library accelerated by CUDA";
homepage = https://cupy.chainer.org/;