depot/third_party/nixpkgs/pkgs/test/cuda/default.nix
Default email 23b612e36f Project import generated by Copybara.
GitOrigin-RevId: ae5c332cbb5827f6b1f02572496b141021de335f
2024-01-25 23:12:00 +09:00

51 lines
901 B
Nix

{
lib,
recurseIntoAttrs,
cudaPackages,
cudaPackagesGoogle,
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
])