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