From 44a5636316b287272911cffd394c1505c6d6127c Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Thu, 18 Apr 2019 09:55:30 +0200
Subject: [PATCH] Added 'debug' jobclass to pizdaint scripts

---
 python/waLBerla/tools/jobscripts/pizdaint_hybrid.job | 2 +-
 python/waLBerla/tools/jobscripts/pizdaint_hybrid.py  | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/python/waLBerla/tools/jobscripts/pizdaint_hybrid.job b/python/waLBerla/tools/jobscripts/pizdaint_hybrid.job
index 14f298ea5..c01605b8f 100644
--- a/python/waLBerla/tools/jobscripts/pizdaint_hybrid.job
+++ b/python/waLBerla/tools/jobscripts/pizdaint_hybrid.job
@@ -7,7 +7,7 @@
 #SBATCH --ntasks-per-core={tasks_per_core}
 #SBATCH --ntasks-per-node={tasks_per_node}
 #SBATCH --cpus-per-task={cpus_per_task}
-#SBATCH --partition=normal
+#SBATCH --partition={partition}
 #SBATCH --constraint=gpu
 {additional_lines}
 
diff --git a/python/waLBerla/tools/jobscripts/pizdaint_hybrid.py b/python/waLBerla/tools/jobscripts/pizdaint_hybrid.py
index 8fed29efd..4abbc9942 100644
--- a/python/waLBerla/tools/jobscripts/pizdaint_hybrid.py
+++ b/python/waLBerla/tools/jobscripts/pizdaint_hybrid.py
@@ -37,6 +37,10 @@ def createJobscript(wall_time=None, nodes=None, cores=None, initial_dir=None, jo
     if not error_file:
         error_file = job_name
 
+    partition = 'normal'
+    if nodes <= 4 and wall_time.total_seconds() < 30 * 60:
+        partition = 'debug'
+
     tasks_per_node = min(CORES_PER_NODE, cores)
     additional_lines = ""
     if account:
@@ -52,10 +56,11 @@ def createJobscript(wall_time=None, nodes=None, cores=None, initial_dir=None, jo
                                                output_file=output_file,
                                                additional_lines=additional_lines,
                                                error_file=error_file,
+                                               partition=partition,
                                                job_name=job_name,
                                                wall_time=wall_time)
 
-    exec_line = "srun %s %s \n"
+    exec_line = "srun -n %d %s %s \n"
 
     if exe_name is not None:
         for param_file in parameter_files:
-- 
GitLab