diff --git a/scripts/launchSlurmArray.py b/scripts/launchSlurmArray.py index eac17b70767902c52a639500b3af0fdf39eb6765..d6aff7bc785d6bd3fc968afe5f1d556765bc3970 100644 --- a/scripts/launchSlurmArray.py +++ b/scripts/launchSlurmArray.py @@ -12,7 +12,7 @@ def launchSlurmArray(names, commands, jobName, device, nbHours, limit) : commandList = " ".join(commands) - if hostname == "jean-zay1" : + if "jean-zay" in hostname : print("""#! /usr/bin/env bash #SBATCH --array=0-{}%{} @@ -24,8 +24,8 @@ def launchSlurmArray(names, commands, jobName, device, nbHours, limit) : #SBATCH --cpus-per-task=10 #SBATCH --gres=gpu:1 #SBATCH --hint=nomultithread -#SBATCH --partition=gpu_p1 #SBATCH --qos={} +#SBATCH --constraint=v100-16g #SBATCH --time={}:00:00 module purge @@ -56,7 +56,7 @@ cat $tmpFile >> $newErr rm $tmpFile eval "${{commands[$SLURM_ARRAY_TASK_ID]}}" -""".format(len(names)-1, limit, jobName, "qos_gpu-t4" if nbHours > 20 else "qos_gpu-t3", nbHours, " ".join(names), " ".join(commands)), file=sFile) +""".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() elif hostname == "sms.liscluster" : print('''#! /usr/bin/env bash