Testing SLURM jobs with ppong

bsc/garlic/default.nix

@@ -28,6 +28,16 @@ let
     ppong-job = srunner { app=ppong; };
 
     exp = {
+      jobs = callPackage ./experiments {
+        apps = map (app: srunner {app=app;}) (
+          genApps [ ppong ] (
+            genConfigs {
+              mpi = [ bsc.intel-mpi pkgs.mpich pkgs.openmpi ];
+            }
+          )
+        );
+      };
+
       mpiImpl = callPackage ./experiments {
         apps = genApps [ ppong ] (
           genConfigs {

bsc/garlic/srunner.nix

@@ -1,5 +1,6 @@
 {
   stdenv
+, numactl
 }:
 
 {
@@ -41,7 +42,8 @@ stdenv.mkDerivation rec {
   dontBuild = true;
 
   installPhase = ''
-    cat > $out <<EOF
+    mkdir -p $out/bin
+    cat > $out/bin/run <<EOF
     #!/bin/bash
     #SBATCH --job-name="${name}"
     ''
@@ -56,13 +58,11 @@ stdenv.mkDerivation rec {
     + optionalString (extra!=null) extra
     +''
     
-    numactl -s
+    ${numactl}/bin/numactl -s
 
-    #!/bin/bash
+    srun ${app}${binary} ${argv}
-    exec ${app}${binary} ${argv}
-    done
     EOF
 
-    chmod +x $out
+    chmod +x $out/bin/run
   '';
 }