localPackages.hdf5-nvhpc: limit build

2026-01-12 04:39:23 +08:00 · 2024-02-27 12:52:04 +08:00
parent 4ea44a179e
commit 5aeca3451d
1 changed files with 3 additions and 1 deletions
--- a/local/pkgs/hdf5-nvhpc/default.nix
+++ b/local/pkgs/hdf5-nvhpc/default.nix
@@ -1,7 +1,8 @@
 {
  buildFHSEnv, writeScript, stdenvNoCC,
  src,
-  nvhpc, lmod, cmake, gfortran
+  nvhpc, lmod, cmake, gfortran,
+  config, nvhpcArch ? config.nvhpcArch or "px"
 }:
 let
  buildEnv = buildFHSEnv
@@ -35,4 +36,5 @@ in stdenvNoCC.mkDerivation
    ${buildEnv}/bin/buildEnv ${buildScript}
  '';
  dontInstall = true;
+  requiredSystemFeatures = [ "nvhpcarch-${nvhpcArch}" ];
 }