pkgs/test/cuda/default.nix at devShellTools-shell · tjh.dev/nixpkgs

tjh.dev / nixpkgs

Clone of https://github.com/NixOS/nixpkgs.git (to stress-test knotserver)

nixpkgs / pkgs / test / cuda / default.nix

at devShellTools-shell 849 B view raw

 1{
 2  lib,
 3  recurseIntoAttrs,
 4
 5  cudaPackages,
 6
 7  cudaPackages_11_0,
 8  cudaPackages_11_1,
 9  cudaPackages_11_2,
10  cudaPackages_11_3,
11  cudaPackages_11_4,
12  cudaPackages_11_5,
13  cudaPackages_11_6,
14  cudaPackages_11_7,
15  cudaPackages_11_8,
16  cudaPackages_11,
17
18  cudaPackages_12_0,
19  cudaPackages_12_1,
20  cudaPackages_12_2,
21  cudaPackages_12_3,
22  cudaPackages_12_4,
23  cudaPackages_12_5,
24  cudaPackages_12_6,
25  cudaPackages_12_8,
26  cudaPackages_12_9,
27  cudaPackages_12,
28}@args:
29
30let
31  isTest =
32    name: package:
33    builtins.elem (package.pname or null) [
34      "cuda-samples"
35      "cuda-library-samples"
36      "saxpy"
37    ];
38in
39(lib.trivial.pipe args [
40  (lib.filterAttrs (name: _: lib.hasPrefix "cudaPackages" name))
41  (lib.mapAttrs (
42    _: ps:
43    lib.pipe ps [
44      (lib.filterAttrs isTest)
45      recurseIntoAttrs
46    ]
47  ))
48  recurseIntoAttrs
49])