···

       
29
       
29
       
 
       
      short, the Multiplex builder adds multiple versions of a single package to

     

       
30
       
30
       
 
       
      single instance of the CUDA Packages package set. It is used primarily for

     

       
31
       
31
       
 
       
      packages like `cudnn` and `cutensor`.

     

       
32
       
32
       
-
       
- `lib`: A library of functions and data used by and for the CUDA package set.

     

       
33
       
33
       
-
       
    This library is exposed at the top-level as `pkgs.cudaLib`.

     

       
34
       
32
       
 
       
- `modules`: Nixpkgs modules to check the shape and content of CUDA

     

       
35
       
33
       
 
       
    redistributable and feature manifests. These modules additionally use shims

     

       
36
       
34
       
 
       
    provided by some CUDA packages to allow them to re-use the

     

···

       
1
       
1
       
 
       
# The _cuda attribute set is a fixed-point which contains the static functionality required to construct CUDA package

     

       
2
       
2
       
-
       
# sets. For example, `_cuda.cudaData` includes information about NVIDIA's redistributables (such as the names NVIDIA

     

       
3
       
3
       
-
       
# uses for different systems), `_cuda.cudaLib` contains utility functions like `formatCapabilities` (which generate

     

       
4
       
4
       
-
       
# common arguments passed to NVCC and `cmakeFlags`), and `_cuda.cudaFixups` contains `callPackage`-able functions

     

       
5
       
5
       
-
       
# which are provided to the corresponding package's `overrideAttrs` attribute to provide package-specific fixups

     

       
2
       
2
       
+
       
# sets. For example, `_cuda.bootstrapData` includes information about NVIDIA's redistributables (such as the names

     

       
3
       
3
       
+
       
# NVIDIA uses for different systems), `_cuda.lib` contains utility functions like `formatCapabilities` (which generate

     

       
4
       
4
       
+
       
# common arguments passed to NVCC and `cmakeFlags`), and `_cuda.fixups` contains `callPackage`-able functions which

     

       
5
       
5
       
+
       
# are provided to the corresponding package's `overrideAttrs` attribute to provide package-specific fixups

     

       
6
       
6
       
 
       
# out of scope of the generic redistributable builder.

     

       
7
       
7
       
 
       
#

     

       
8
       
8
       
 
       
# Since this attribute set is used to construct the CUDA package sets, it must exist outside the fixed point of the

     

···

       
1
       
1
       
 
       
{

     

       
2
       
2
       
-
       
  cudaData,

     

       
3
       
3
       
-
       
  cudaLib,

     

       
2
       
2
       
+
       
  _cuda,

     

       
4
       
3
       
 
       
  cudaNamePrefix,

     

       
5
       
4
       
 
       
  lib,

     

       
6
       
5
       
 
       
  runCommand,

     

       
7
       
6
       
 
       
}:

     

       
8
       
7
       
 
       
let

     

       
9
       
8
       
 
       
  inherit (builtins) deepSeq toJSON tryEval;

     

       
10
       
10
       
-
       
  inherit (cudaData) cudaCapabilityToInfo;

     

       
11
       
11
       
-
       
  inherit (cudaLib) formatCapabilities;

     

       
9
       
9
       
+
       
  inherit (_cuda.bootstrapData) cudaCapabilityToInfo;

     

       
10
       
10
       
+
       
  inherit (_cuda.lib) formatCapabilities;

     

       
12
       
11
       
 
       
  inherit (lib.asserts) assertMsg;

     

       
13
       
12
       
 
       
in

     

       
14
       
13
       
 
       
# When changing names or formats: pause, validate, and update the assert

     
···

       
62
       
61
       
 
       
        "7.5"

     

       
63
       
62
       
 
       
        "8.6"

     

       
64
       
63
       
 
       
      ];

     

       
64
       
64
       
+
       
      cudaForwardCompat = true;

     

       
65
       
65
       
 
       
    };

     

       
66
       
66
       
 
       
    actualWrapped = (tryEval (deepSeq actual actual)).value;

     

       
67
       
67
       
 
       
  in

     

···

       
14
       
14
       
 
       


     

       
15
       
15
       
 
       
let

     

       
16
       
16
       
 
       
  lib = import ../../lib;

     

       
17
       
17
       
-
       
  inherit (import ../development/cuda-modules/_cuda) cudaLib;

     

       
17
       
17
       
+
       
  cudaLib = (import ../development/cuda-modules/_cuda).lib;

     

       
18
       
18
       
 
       
in

     

       
19
       
19
       
 
       


     

       
20
       
20
       
 
       
{