commit 4d308324196fa395a8032e31c179e3bfb1c8b1fe · pyrox.dev/nixpkgs

+144

pkgs/development/cuda-modules/cusparselt/extension.nix

···

       1
       1
       +
       # Support matrix can be found at

     

       2
       2
       +
       # https://docs.nvidia.com/deeplearning/cudnn/archives/cudnn-880/support-matrix/index.html

     

       3
       3
       +
       {

     

       4
       4
       +
         lib,

     

       5
       5
       +
         stdenv,

     

       6
       6
       +
         cudaVersion,

     

       7
       7
       +
         flags,

     

       8
       8
       +
         mkVersionedPackageName,

     

       9
       9
       +
       }:

     

       10
       10
       +
       let

     

       11
       11
       +
         inherit (lib)

     

       12
       12
       +
           attrsets

     

       13
       13
       +
           lists

     

       14
       14
       +
           modules

     

       15
       15
       +
           versions

     

       16
       16
       +
           strings

     

       17
       17
       +
           trivial

     

       18
       18
       +
           ;

     

       19
       19
       +
       

     

       20
       20
       +
         inherit (stdenv) hostPlatform;

     

       21
       21
       +
       

     

       22
       22
       +
         redistName = "cusparselt";

     

       23
       23
       +
         pname = "libcusparse_lt";

     

       24
       24
       +
       

     

       25
       25
       +
         cusparseltVersions = [

     

       26
       26
       +
           "0.7.1"

     

       27
       27
       +
         ];

     

       28
       28
       +
       

     

       29
       29
       +
         # Manifests :: { redistrib, feature }

     

       30
       30
       +
       

     

       31
       31
       +
         # Each release of cusparselt gets mapped to an evaluated module for that release.

     

       32
       32
       +
         # From there, we can get the min/max CUDA versions supported by that release.

     

       33
       33
       +
         # listOfManifests :: List Manifests

     

       34
       34
       +
         listOfManifests =

     

       35
       35
       +
           let

     

       36
       36
       +
             configEvaluator =

     

       37
       37
       +
               fullCusparseltVersion:

     

       38
       38
       +
               modules.evalModules {

     

       39
       39
       +
                 modules = [

     

       40
       40
       +
                   ../modules

     

       41
       41
       +
                   # We need to nest the manifests in a config.cusparselt.manifests attribute so the

     

       42
       42
       +
                   # module system can evaluate them.

     

       43
       43
       +
                   {

     

       44
       44
       +
                     cusparselt.manifests = {

     

       45
       45
       +
                       redistrib = trivial.importJSON (./manifests + "/redistrib_${fullCusparseltVersion}.json");

     

       46
       46
       +
                       feature = trivial.importJSON (./manifests + "/feature_${fullCusparseltVersion}.json");

     

       47
       47
       +
                     };

     

       48
       48
       +
                   }

     

       49
       49
       +
                 ];

     

       50
       50
       +
               };

     

       51
       51
       +
             # Un-nest the manifests attribute set.

     

       52
       52
       +
             releaseGrabber = evaluatedModules: evaluatedModules.config.cusparselt.manifests;

     

       53
       53
       +
           in

     

       54
       54
       +
           lists.map (trivial.flip trivial.pipe [

     

       55
       55
       +
             configEvaluator

     

       56
       56
       +
             releaseGrabber

     

       57
       57
       +
           ]) cusparseltVersions;

     

       58
       58
       +
       

     

       59
       59
       +
         # Our cudaVersion tells us which version of CUDA we're building against.

     

       60
       60
       +
         # The subdirectories in lib/ tell us which versions of CUDA are supported.

     

       61
       61
       +
         # Typically the names will look like this:

     

       62
       62
       +
         #

     

       63
       63
       +
         # - 10.2

     

       64
       64
       +
         # - 11

     

       65
       65
       +
         # - 11.0

     

       66
       66
       +
         # - 12

     

       67
       67
       +
       

     

       68
       68
       +
         # libPath :: String

     

       69
       69
       +
         libPath =

     

       70
       70
       +
           let

     

       71
       71
       +
             cudaMajorMinor = versions.majorMinor cudaVersion;

     

       72
       72
       +
             cudaMajor = versions.major cudaVersion;

     

       73
       73
       +
           in

     

       74
       74
       +
           if cudaMajorMinor == "10.2" then cudaMajorMinor else cudaMajor;

     

       75
       75
       +
       

     

       76
       76
       +
         # A release is supported if it has a libPath that matches our CUDA version for our platform.

     

       77
       77
       +
         # LibPath are not constant across the same release -- one platform may support fewer

     

       78
       78
       +
         # CUDA versions than another.

     

       79
       79
       +
         # redistArch :: String

     

       80
       80
       +
         redistArch = flags.getRedistArch hostPlatform.system;

     

       81
       81
       +
         # platformIsSupported :: Manifests -> Boolean

     

       82
       82
       +
         platformIsSupported =

     

       83
       83
       +
           { feature, redistrib, ... }:

     

       84
       84
       +
           (attrsets.attrByPath [

     

       85
       85
       +
             pname

     

       86
       86
       +
             redistArch

     

       87
       87
       +
           ] null feature) != null;

     

       88
       88
       +
       

     

       89
       89
       +
         # TODO(@connorbaker): With an auxiliary file keeping track of the CUDA versions each release supports,

     

       90
       90
       +
         # we could filter out releases that don't support our CUDA version.

     

       91
       91
       +
         # However, we don't have that currently, so we make a best-effort to try to build TensorRT with whatever

     

       92
       92
       +
         # libPath corresponds to our CUDA version.

     

       93
       93
       +
         # supportedManifests :: List Manifests

     

       94
       94
       +
         supportedManifests = builtins.filter platformIsSupported listOfManifests;

     

       95
       95
       +
       

     

       96
       96
       +
         # Compute versioned attribute name to be used in this package set

     

       97
       97
       +
         # Patch version changes should not break the build, so we only use major and minor

     

       98
       98
       +
         # computeName :: RedistribRelease -> String

     

       99
       99
       +
         computeName = { version, ... }: mkVersionedPackageName redistName version;

     

       100
       100
       +
       in

     

       101
       101
       +
       final: _:

     

       102
       102
       +
       let

     

       103
       103
       +
         # buildCusparseltPackage :: Manifests -> AttrSet Derivation

     

       104
       104
       +
         buildCusparseltPackage =

     

       105
       105
       +
           { redistrib, feature }:

     

       106
       106
       +
           let

     

       107
       107
       +
             drv = final.callPackage ../generic-builders/manifest.nix {

     

       108
       108
       +
               inherit pname redistName;

     

       109
       109
       +
               redistribRelease = redistrib.${pname};

     

       110
       110
       +
               featureRelease = feature.${pname};

     

       111
       111
       +
             };

     

       112
       112
       +
             fixedDrv = drv.overrideAttrs (prevAttrs: {

     

       113
       113
       +
               buildInputs =

     

       114
       114
       +
                 prevAttrs.buildInputs

     

       115
       115
       +
                 ++ lists.optionals (strings.versionOlder cudaVersion "11.4") [ final.cudatoolkit ]

     

       116
       116
       +
                 ++ lists.optionals (strings.versionAtLeast cudaVersion "11.4") (

     

       117
       117
       +
                   [ final.libcublas.lib ]

     

       118
       118
       +
                   # For some reason, the 1.4.x release of cusparselt requires the cudart library.

     

       119
       119
       +
                   ++ lists.optionals (strings.hasPrefix "1.4" redistrib.${pname}.version) [ final.cuda_cudart.lib ]

     

       120
       120
       +
                 );

     

       121
       121
       +
               meta = prevAttrs.meta // {

     

       122
       122
       +
                 description = "cuSPARSELt: A High-Performance CUDA Library for Sparse Matrix-Matrix Multiplication";

     

       123
       123
       +
                 homepage = "https://developer.nvidia.com/cusparselt-downloads";

     

       124
       124
       +
       

     

       125
       125
       +
                 maintainers = prevAttrs.meta.maintainers ++ [ lib.maintainers.sepiabrown ];

     

       126
       126
       +
                 license = lib.licenses.unfreeRedistributable // {

     

       127
       127
       +
                   shortName = "cuSPARSELt EULA";

     

       128
       128
       +
                   fullName = "cuSPARSELt SUPPLEMENT TO SOFTWARE LICENSE AGREEMENT FOR NVIDIA SOFTWARE DEVELOPMENT KITS";

     

       129
       129
       +
                   url = "https://docs.nvidia.com/cuda/cusparselt/license.html";

     

       130
       130
       +
                 };

     

       131
       131
       +
               };

     

       132
       132
       +
             });

     

       133
       133
       +
           in

     

       134
       134
       +
           attrsets.nameValuePair (computeName redistrib.${pname}) fixedDrv;

     

       135
       135
       +
       

     

       136
       136
       +
         extension =

     

       137
       137
       +
           let

     

       138
       138
       +
             nameOfNewest = computeName (lists.last supportedManifests).redistrib.${pname};

     

       139
       139
       +
             drvs = builtins.listToAttrs (lists.map buildCusparseltPackage supportedManifests);

     

       140
       140
       +
             containsDefault = attrsets.optionalAttrs (drvs != { }) { cusparselt = drvs.${nameOfNewest}; };

     

       141
       141
       +
           in

     

       142
       142
       +
           drvs // containsDefault;

     

       143
       143
       +
       in

     

       144
       144
       +
       extension

+44

pkgs/development/cuda-modules/cusparselt/manifests/feature_0.7.1.json

···

       1
       1
       +
       {

     

       2
       2
       +
         "libcusparse_lt": {

     

       3
       3
       +
           "linux-aarch64": {

     

       4
       4
       +
             "outputs": {

     

       5
       5
       +
               "bin": false,

     

       6
       6
       +
               "dev": true,

     

       7
       7
       +
               "doc": false,

     

       8
       8
       +
               "lib": true,

     

       9
       9
       +
               "sample": false,

     

       10
       10
       +
               "static": true

     

       11
       11
       +
             }

     

       12
       12
       +
           },

     

       13
       13
       +
           "linux-sbsa": {

     

       14
       14
       +
             "outputs": {

     

       15
       15
       +
               "bin": false,

     

       16
       16
       +
               "dev": true,

     

       17
       17
       +
               "doc": false,

     

       18
       18
       +
               "lib": true,

     

       19
       19
       +
               "sample": false,

     

       20
       20
       +
               "static": true

     

       21
       21
       +
             }

     

       22
       22
       +
           },

     

       23
       23
       +
           "linux-x86_64": {

     

       24
       24
       +
             "outputs": {

     

       25
       25
       +
               "bin": false,

     

       26
       26
       +
               "dev": true,

     

       27
       27
       +
               "doc": false,

     

       28
       28
       +
               "lib": true,

     

       29
       29
       +
               "sample": false,

     

       30
       30
       +
               "static": true

     

       31
       31
       +
             }

     

       32
       32
       +
           },

     

       33
       33
       +
           "windows-x86_64": {

     

       34
       34
       +
             "outputs": {

     

       35
       35
       +
               "bin": false,

     

       36
       36
       +
               "dev": true,

     

       37
       37
       +
               "doc": false,

     

       38
       38
       +
               "lib": false,

     

       39
       39
       +
               "sample": false,

     

       40
       40
       +
               "static": false

     

       41
       41
       +
             }

     

       42
       42
       +
           }

     

       43
       43
       +
         }

     

       44
       44
       +
       }

+35

pkgs/development/cuda-modules/cusparselt/manifests/redistrib_0.7.1.json

···

       1
       1
       +
       {

     

       2
       2
       +
           "release_date": "2025-02-25",

     

       3
       3
       +
           "release_label": "0.7.1",

     

       4
       4
       +
           "release_product": "cusparselt",

     

       5
       5
       +
           "libcusparse_lt": {

     

       6
       6
       +
               "name": "NVIDIA cuSPARSELt",

     

       7
       7
       +
               "license": "cuSPARSELt",

     

       8
       8
       +
               "license_path": "libcusparse_lt/LICENSE.txt",

     

       9
       9
       +
               "version": "0.7.1.0",

     

       10
       10
       +
               "linux-x86_64": {

     

       11
       11
       +
                   "relative_path": "libcusparse_lt/linux-x86_64/libcusparse_lt-linux-x86_64-0.7.1.0-archive.tar.xz",

     

       12
       12
       +
                   "sha256": "a0d885837887c73e466a31b4e86aaae2b7d0cc9c5de0d40921dbe2a15dbd6a88",

     

       13
       13
       +
                   "md5": "b2e5f3c9b9d69e1e0b55b16de33fdc6e",

     

       14
       14
       +
                   "size": "353151840"

     

       15
       15
       +
               },

     

       16
       16
       +
               "linux-sbsa": {

     

       17
       17
       +
                   "relative_path": "libcusparse_lt/linux-sbsa/libcusparse_lt-linux-sbsa-0.7.1.0-archive.tar.xz",

     

       18
       18
       +
                   "sha256": "4a131d0a54728e53ba536b50bb65380603456f1656e7df8ee52e285618a0b57c",

     

       19
       19
       +
                   "md5": "612a712c7da6e801ee773687e99af87e",

     

       20
       20
       +
                   "size": "352406784"

     

       21
       21
       +
               },

     

       22
       22
       +
               "windows-x86_64": {

     

       23
       23
       +
                   "relative_path": "libcusparse_lt/windows-x86_64/libcusparse_lt-windows-x86_64-0.7.1.0-archive.zip",

     

       24
       24
       +
                   "sha256": "004bcb1b700c24ca8d60a8ddd2124640f61138a6c29914d2afaa0bfa0d0e3cf2",

     

       25
       25
       +
                   "md5": "a1d8df8dc8ff4b3bd0e859f992f8f392",

     

       26
       26
       +
                   "size": "268594665"

     

       27
       27
       +
               },

     

       28
       28
       +
               "linux-aarch64": {

     

       29
       29
       +
                   "relative_path": "libcusparse_lt/linux-aarch64/libcusparse_lt-linux-aarch64-0.7.1.0-archive.tar.xz",

     

       30
       30
       +
                   "sha256": "d3b0a660fd552e0bd9a4491b15299d968674833483d5f164cfea35e70646136c",

     

       31
       31
       +
                   "md5": "54e3f3b28c94118991ce54ec38f531fb",

     

       32
       32
       +
                   "size": "5494380"

     

       33
       33
       +
               }

     

       34
       34
       +
           }

     

       35
       35
       +
       }

pkgs/development/cuda-modules/modules/cusparselt/default.nix

···

       1
       1
       +
       { options, ... }:

     

       2
       2
       +
       {

     

       3
       3
       +
         options.cusparselt.manifests = options.generic.manifests;

     

       4
       4
       +
       }

pkgs/development/cuda-modules/modules/default.nix

···

       4
       4
        
           # Always after generic

     

       5
       5
        
           ./cuda

     

       6
       6
        
           ./cudnn

     

       7
       7
       +
           ./cusparselt

     

       7
       8
        
           ./cutensor

     

       8
       9
        
           ./tensorrt

     

       9
       10
        
         ];

pkgs/top-level/cuda-packages.nix

···

       148
       148
        
             (callPackage ../development/cuda-modules/cutensor/extension.nix {

     

       149
       149
        
               inherit cudaVersion flags mkVersionedPackageName;

     

       150
       150
        
             })

     

       151
       151
       +
             (callPackage ../development/cuda-modules/cusparselt/extension.nix {

     

       152
       152
       +
               inherit cudaVersion flags mkVersionedPackageName;

     

       153
       153
       +
             })

     

       151
       154
        
             (callPackage ../development/cuda-modules/generic-builders/multiplex.nix {

     

       152
       155
        
               inherit cudaVersion flags mkVersionedPackageName;

     

       153
       156
        
               pname = "tensorrt";

pkgs/top-level/release-cuda.nix

···

       23
       23
        
             || builtins.elem license.shortName [

     

       24
       24
        
               "CUDA EULA"

     

       25
       25
        
               "cuDNN EULA"

     

       26
       26
       +
               "cuSPARSELt EULA"

     

       26
       27
        
               "cuTENSOR EULA"

     

       27
       28
        
               "NVidia OptiX EULA"

     

       28
       29
        
             ]