commit 376664bb45f0aec8c1cf6b332542d912d675bb85 · pyrox.dev/nixpkgs

+11 -12
pkgs/development/python-modules/cupy/default.nix
···

       3
       3
        
         stdenv,

     

       4
       4
        
         buildPythonPackage,

     

       5
       5
        
         fetchFromGitHub,

     

       6
       6
       -
         cython_0,

     

       6
       6
       +
         cython,

     

       7
       7
        
         fastrlock,

     

       8
       8
        
         numpy,

     

       9
       9
       -
         wheel,

     

       10
       9
        
         pytestCheckHook,

     

       11
       10
        
         mock,

     

       12
       11
        
         setuptools,

     

       13
       12
        
         cudaPackages,

     

       14
       13
        
         addDriverRunpath,

     

       15
       15
       -
         pythonOlder,

     

       16
       14
        
         symlinkJoin,

     

       17
       15
        
       }:

     

       18
       16
        
       

     
···

       58
       56
        
       in

     

       59
       57
        
       buildPythonPackage rec {

     

       60
       58
        
         pname = "cupy";

     

       61
       61
       -
         version = "13.5.1";

     

       62
       62
       -
         format = "setuptools";

     

       63
       63
       -
       

     

       64
       64
       -
         disabled = pythonOlder "3.7";

     

       59
       59
       +
         version = "13.6.0";

     

       60
       60
       +
         pyproject = true;

     

       65
       61
        
       

     

       66
       62
        
         stdenv = cudaPackages.backendStdenv;

     

       67
       63
        
       

     
···

       69
       65
        
           owner = "cupy";

     

       70
       66
        
           repo = "cupy";

     

       71
       67
        
           tag = "v${version}";

     

       72
       72
       -
           hash = "sha256-8RgyvsU3lnt6nO0J1tiLBOdYsX0jJkjPH/SpKQz4o7s=";

     

       68
       68
       +
           hash = "sha256-nU3VL0MSCN+mI5m7C5sKAjBSL6ybM6YAk5lJiIDY0ck=";

     

       73
       69
        
           fetchSubmodules = true;

     

       74
       70
        
         };

     

       75
       71
        
       

     
···

       83
       79
        
           export CUPY_NUM_NVCC_THREADS="$NIX_BUILD_CORES"

     

       84
       80
        
         '';

     

       85
       81
        
       

     

       82
       82
       +
         build-system = [

     

       83
       83
       +
           cython

     

       84
       84
       +
           fastrlock

     

       85
       85
       +
           setuptools

     

       86
       86
       +
         ];

     

       87
       87
       +
       

     

       86
       88
        
         nativeBuildInputs = [

     

       87
       87
       -
           setuptools

     

       88
       88
       -
           wheel

     

       89
       89
        
           addDriverRunpath

     

       90
       90
       -
           cython_0

     

       91
       90
        
           cudaPackages.cuda_nvcc

     

       92
       91
        
         ];

     

       93
       92
        
       

     
···

       101
       100
        
         NVCC = "${lib.getExe cudaPackages.cuda_nvcc}"; # FIXME: splicing/buildPackages

     

       102
       101
        
         CUDA_PATH = "${cudatoolkit-joined}";

     

       103
       102
        
       

     

       104
       104
       -
         propagatedBuildInputs = [

     

       103
       103
       +
         dependencies = [

     

       105
       104
        
           fastrlock

     

       106
       105
        
           numpy

     

       107
       106
        
         ];