pkgs/test/cuda/default.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

{
  lib,
  pkgs,
}:
let
  getTests =
    cps:
    lib.recurseIntoAttrs {
      inherit (cps) saxpy;
      inherit (cps.tests) cuda-library-samples;
    };
in
lib.recurseIntoAttrs (
  lib.mapAttrs (_: getTests) {
    inherit (pkgs)
      cudaPackages

      cudaPackages_12
      cudaPackages_12_6
      cudaPackages_12_8
      cudaPackages_12_9

      cudaPackages_13
      cudaPackages_13_0
      ;
  }
)