diff --git a/scripts/launchSlurmArray.py b/scripts/launchSlurmArray.py index 5a4c9960996c47ca0aa202fc83467de3f687ef1d..c08ceee4df4c0968a25f57e509ca6b1ed7a0edf2 100644 --- a/scripts/launchSlurmArray.py +++ b/scripts/launchSlurmArray.py @@ -54,6 +54,7 @@ cat $tmpFile >> $newErr rm $tmpFile +>&2 echo "Job ID : " "$SLURM_ARRAY_JOB_ID"_"$SLURM_ARRAY_TASK_ID" eval "${{commands[$SLURM_ARRAY_TASK_ID]}}" """.format(len(names)-1, limit, jobName, nbCPU, nbHours, " ".join(names), " ".join(commands)), file=sFile) sFile.close() @@ -99,6 +100,7 @@ cat $tmpFile >> $newErr rm $tmpFile +>&2 echo "Job ID : " "$SLURM_ARRAY_JOB_ID"_"$SLURM_ARRAY_TASK_ID" eval "${{commands[$SLURM_ARRAY_TASK_ID]}}" """.format(len(names)-1, limit, jobName, "qos_gpu-t4" if int(nbHours) > 20 else "qos_gpu-t3", nbHours, " ".join(names), " ".join(commands)), file=sFile) sFile.close() @@ -140,8 +142,9 @@ cat $tmpFile >> $newErr rm $tmpFile +>&2 echo "Job ID : " "$SLURM_ARRAY_JOB_ID"_"$SLURM_ARRAY_TASK_ID" eval "${{commands[$SLURM_ARRAY_TASK_ID]}}" -'''.format(len(names)-1, limit, jobName, nbCPU, "cpu" if device == "cpu" else "gpu\n#SBATCH --gres=gpu", nbHours, " ".join(names), commandList), file=sFile) +'''.format(len(names)-1, limit, jobName, nbCPU, "cpu" if device == "cpu" else "all\n#SBATCH --gres=gpu", nbHours, " ".join(names), commandList), file=sFile) sFile.close() else : print("ERROR : Unknown hostname \'%s\'"%hostname)