python37Packages.cupy: Add cutensor optional dep

2021-03-15 23:09:15 +00:00
parent a680b02816
commit 13b5995f06
2 changed files with 3 additions and 1 deletions
--- a/pkgs/development/python-modules/cupy/default.nix
+++ b/pkgs/development/python-modules/cupy/default.nix
@@ -1,7 +1,7 @@
 { lib, buildPythonPackage
 , fetchPypi, isPy3k, linuxPackages
 , fastrlock, numpy, six, wheel, pytest, mock, setuptools
-, cudatoolkit, cudnn, nccl
+, cudatoolkit, cudnn, cutensor, nccl
 }:

 buildPythonPackage rec {
@@ -26,6 +26,7 @@ buildPythonPackage rec {
  propagatedBuildInputs = [
    cudatoolkit
    cudnn
+    cutensor
    linuxPackages.nvidia_x11
    nccl
    fastrlock