about summary refs log tree commit diff
path: root/pkgs/test/cuda/default.nix
blob: b463da7c55c56e76be6357445b967759c4dff1d7 (plain) (blame)
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
{
  lib,
  recurseIntoAttrs,

  cudaPackages,

  cudaPackages_10_0,
  cudaPackages_10_1,
  cudaPackages_10_2,
  cudaPackages_10,

  cudaPackages_11_0,
  cudaPackages_11_1,
  cudaPackages_11_2,
  cudaPackages_11_3,
  cudaPackages_11_4,
  cudaPackages_11_5,
  cudaPackages_11_6,
  cudaPackages_11_7,
  cudaPackages_11_8,
  cudaPackages_11,

  cudaPackages_12_0,
  cudaPackages_12_1,
  cudaPackages_12_2,
  cudaPackages_12_3,
  cudaPackages_12,
}@args:

let
  isTest =
    name: package:
    builtins.elem (package.pname or null) [
      "cuda-samples"
      "cuda-library-samples"
      "saxpy"
    ];
in
(lib.trivial.pipe args [
  (lib.filterAttrs (name: _: lib.hasPrefix "cudaPackages" name))
  (lib.mapAttrs (
    _: ps:
    lib.pipe ps [
      (lib.filterAttrs isTest)
      (as: as // { __attrsFailEvaluation = true; })
      recurseIntoAttrs
    ]
  ))
  recurseIntoAttrs
])