cudaPackages.saxpy: passthru: test gpu/runtime

2023-11-21 20:48:09 +00:00 · 2023-11-21 20:48:09 +00:00 · 9aa0403154
commit 9aa0403154
parent 927b15ed6d
1 changed files with 11 additions and 0 deletions
--- a/pkgs/development/cuda-modules/saxpy/default.nix
+++ b/pkgs/development/cuda-modules/saxpy/default.nix
@ -3,6 +3,7 @@
  cmake,
  cudaPackages,
  lib,
+  saxpy,
 }:
 let
  inherit (cudaPackages)
@ -58,6 +59,16 @@ backendStdenv.mkDerivation {
    (lib.cmakeFeature "CMAKE_CUDA_ARCHITECTURES" flags.cmakeCudaArchitecturesString)
  ];

+  passthru.tests.withCuda = saxpy.overrideAttrs (
+    _: {
+      requiredSystemFeatures = ["cuda"];
+      doInstallCheck = true;
+      postInstallCheck = ''
+        $out/bin/saxpy
+      '';
+    }
+  );
+
  meta = rec {
    description = "Simple (Single-precision AX Plus Y) FindCUDAToolkit.cmake example for testing cross-compilation";
    license = lib.licenses.mit;