nixpkgs/pkgs/test/cuda/default.nix

50 lines
827 B
Nix

{
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)
recurseIntoAttrs
]
))
recurseIntoAttrs
])