Success sbatch launch in MN4 with nbody seq

This commit is contained in:
Rodrigo Arias 2020-08-10 14:13:28 +02:00
parent 9631f4c223
commit b9e9409a59
4 changed files with 26 additions and 8 deletions

View File

@ -65,7 +65,13 @@ let
};
nbodyBSjob = callPackage ./dispatcher.nix {
jobs = map (app: srunner {app=app;}) (
jobs = map (app: srunner {
app=app;
prefix="/gpfs/projects/bsc15/nix";
exclusive=false;
ntasks = "1";
}
) (
genApp nbody (
genConfigs {
cc = [ bsc.icc ];

View File

@ -19,7 +19,7 @@ stdenv.mkDerivation {
mkdir -p $out/bin
cat > $out/bin/execute-all-jobs <<EOF
#!/bin/bash
#!/bin/sh
for j in $out/jobs/*; do
echo "sbatch \$j"

View File

@ -29,12 +29,25 @@ stdenv.mkDerivation {
"BS=${toString blocksize}"
];
dontPatchShebangs = true;
installPhase = ''
mkdir -p $out/bin
cp nbody $out/bin/
cat > $out/bin/run <<EOF
#!/bin/bash
#!/bin/sh
# We need to enter the nix namespace first, in order to have /nix
# available, so we use this hack:
if [ ! -e /nix ]; then
echo "running nix-setup \$0"
exec nix-setup \$0
fi
ls -l /nix
pwd
exec $out/bin/nbody -p ${toString particles} -t ${toString timesteps}
EOF

View File

@ -5,6 +5,7 @@
{
app
, prefix ? ""
, argv ? ""
, binary ? "/bin/run"
, ntasks ? null
@ -55,11 +56,9 @@ stdenv.mkDerivation rec {
+ sbatchOpt "time" time
+ sbatchOpt "qos" qos
+ optionalString (extra!=null) extra
+''
${numactl}/bin/numactl -s
srun ${app}${binary} ${argv}
+
''
srun ${prefix}${app}${binary} ${argv}
EOF
'';
}