From 5f5015f9da282a487b3b60ded0f9124c1fa29e38 Mon Sep 17 00:00:00 2001 From: Luca Date: Mon, 17 Feb 2020 14:51:16 +0100 Subject: [PATCH] Adjust num_cpus_per_task in on Arolla and Tsa --- cscs-checks/microbenchmarks/dgemm/dgemm.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/cscs-checks/microbenchmarks/dgemm/dgemm.py b/cscs-checks/microbenchmarks/dgemm/dgemm.py index e6efd535e3..2bdd9cf7ad 100644 --- a/cscs-checks/microbenchmarks/dgemm/dgemm.py +++ b/cscs-checks/microbenchmarks/dgemm/dgemm.py @@ -67,11 +67,17 @@ def setup(self, partition, environ, **job_opts): elif partition.fullname in ['arolla:cn', 'arolla:pn', 'kesch:cn', 'kesch:pn', 'tsa:cn', 'tsa:pn']: - self.num_cpus_per_task = 12 self.build_system.cflags += ['-I$EBROOTOPENBLAS/include'] self.build_system.ldflags = ['-L$EBROOTOPENBLAS/lib', '-lopenblas', '-lpthread', '-lgfortran'] + if partition.fullname in ['arolla:cn', 'tsa:cn']: + self.num_cpus_per_task = 16 + elif partition.fullname in ['arolla:pn', 'tsa:pn']: + self.num_cpus_per_task = 40 + elif partition.fullname in ['kesch:cn', 'kesch:pn']: + self.num_cpus_per_task = 12 + if self.num_cpus_per_task: self.variables = { 'OMP_NUM_THREADS': str(self.num_cpus_per_task)