From 136245ce321120ed7337fb4fa9b47f165c151d40 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Thu, 14 Jan 2021 10:57:55 +0100
Subject: [PATCH] updated slurm array script for jeanZay cluster
---
scripts/launchSlurmArray.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/scripts/launchSlurmArray.py b/scripts/launchSlurmArray.py
index eac17b7..d6aff7b 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
--
GitLab