about summary refs log tree commit diff
path: root/pkgs/test
diff options
context:
space:
mode:
authorSomeone Serge <sergei.kozlukov@aalto.fi>2024-01-12 00:02:07 +0000
committerSomeone Serge <sergei.kozlukov@aalto.fi>2024-01-12 20:24:50 +0000
commit27b277b63c2c8ab5a0e40e88216f1061a80183d0 (patch)
tree3cabd63c866fe547bfd26ab4516e3648cf9a640e /pkgs/test
parentcdcd061e7f312304de5391bf2a01d8c705206eaf (diff)
downloadnixlib-27b277b63c2c8ab5a0e40e88216f1061a80183d0.tar
nixlib-27b277b63c2c8ab5a0e40e88216f1061a80183d0.tar.gz
nixlib-27b277b63c2c8ab5a0e40e88216f1061a80183d0.tar.bz2
nixlib-27b277b63c2c8ab5a0e40e88216f1061a80183d0.tar.lz
nixlib-27b277b63c2c8ab5a0e40e88216f1061a80183d0.tar.xz
nixlib-27b277b63c2c8ab5a0e40e88216f1061a80183d0.tar.zst
nixlib-27b277b63c2c8ab5a0e40e88216f1061a80183d0.zip
tests.cuda: inherit the ready cudaPackages_XX.cuda-samples
...fixing the broken eval
Diffstat (limited to 'pkgs/test')
-rw-r--r--pkgs/test/cuda/default.nix77
1 files changed, 48 insertions, 29 deletions
diff --git a/pkgs/test/cuda/default.nix b/pkgs/test/cuda/default.nix
index 8431e4b9207d..dd9ad8b814dc 100644
--- a/pkgs/test/cuda/default.nix
+++ b/pkgs/test/cuda/default.nix
@@ -1,32 +1,51 @@
-{callPackage}:
+{
+  lib,
+  recurseIntoAttrs,
 
-rec {
-  cuda-samplesPackages = callPackage ./cuda-samples/generic.nix {};
-  inherit (cuda-samplesPackages)
-    cuda-samples_cudatoolkit_10
-    cuda-samples_cudatoolkit_10_0
-    cuda-samples_cudatoolkit_10_1
-    cuda-samples_cudatoolkit_10_2
-    cuda-samples_cudatoolkit_11
-    cuda-samples_cudatoolkit_11_0
-    cuda-samples_cudatoolkit_11_1
-    cuda-samples_cudatoolkit_11_2
-    cuda-samples_cudatoolkit_11_3
-    cuda-samples_cudatoolkit_11_4
-    ;
+  cudaPackages,
+  cudaPackagesGoogle,
 
-  cuda-library-samplesPackages = callPackage ./cuda-library-samples/generic.nix {};
-  inherit (cuda-library-samplesPackages)
-    cuda-library-samples_cudatoolkit_10
-    cuda-library-samples_cudatoolkit_10_1
-    cuda-library-samples_cudatoolkit_10_2
-    cuda-library-samples_cudatoolkit_11
-    cuda-library-samples_cudatoolkit_11_0
-    cuda-library-samples_cudatoolkit_11_1
-    cuda-library-samples_cudatoolkit_11_2
-    cuda-library-samples_cudatoolkit_11_3
-    cuda-library-samples_cudatoolkit_11_4
-    ;
+  cudaPackages_10_0,
+  cudaPackages_10_1,
+  cudaPackages_10_2,
+  cudaPackages_10,
 
-  __attrsFailEvaluation = true;
-}
+  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
+])