Add srun wrapper and use pmi2

2020-08-17 18:50:18 +02:00
parent df18435dfc
commit 01295487d8
7 changed files with 138 additions and 18 deletions
--- a/bsc/garlic/srun.nix
+++ b/bsc/garlic/srun.nix
@@ -0,0 +1,20 @@
+{
+  stdenv
+}:
+{ app , nixPrefix ? "" }:
+
+stdenv.mkDerivation rec {
+  name = "${app.name}-srun";
+  preferLocalBuild = true;
+  phases = [ "installPhase" ];
+  buildInputs = [ app ];
+  dontPatchShebangs = true;
+  installPhase = ''
+    mkdir -p $out/bin
+    cat > $out/bin/run <<EOF
+    #!/bin/sh
+    exec srun --mpi=pmi2 ${nixPrefix}${app}/bin/run
+    EOF
+    chmod +x $out/bin/run
+  '';
+}