pkgs/test/cuda/default.nix

   1 {
   2   lib,
   3   recurseIntoAttrs,
   4
   5   cudaPackages,
   6
   7   cudaPackages_10_0,
   8   cudaPackages_10_1,
   9   cudaPackages_10_2,
  10   cudaPackages_10,
  11
  12   cudaPackages_11_0,
  13   cudaPackages_11_1,
  14   cudaPackages_11_2,
  15   cudaPackages_11_3,
  16   cudaPackages_11_4,
  17   cudaPackages_11_5,
  18   cudaPackages_11_6,
  19   cudaPackages_11_7,
  20   cudaPackages_11_8,
  21   cudaPackages_11,
  22
  23   cudaPackages_12_0,
  24   cudaPackages_12_1,
  25   cudaPackages_12_2,
  26   cudaPackages_12_3,
  27   cudaPackages_12,
  28 }@args:
  29
  30 let
  31   isTest =
  32     name: package:
  33     builtins.elem (package.pname or null) [
  34       "cuda-samples"
  35       "cuda-library-samples"
  36       "saxpy"
  37     ];
  38 in
  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 ])