From 77fa436428373ae10bc916890918153106aa7a33 Mon Sep 17 00:00:00 2001 From: crivella Date: Thu, 29 Aug 2024 13:41:39 +0200 Subject: [PATCH] Docstrings and skip test if corecnt > 256 --- eessi/testsuite/tests/apps/MetalWalls.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/eessi/testsuite/tests/apps/MetalWalls.py b/eessi/testsuite/tests/apps/MetalWalls.py index 94c9fed9..9a2af792 100644 --- a/eessi/testsuite/tests/apps/MetalWalls.py +++ b/eessi/testsuite/tests/apps/MetalWalls.py @@ -94,16 +94,27 @@ def run_after_setup(self): @run_after('setup') def set_binding(self): + """Set binding to compact to improve performance reproducibility.""" hooks.set_compact_process_binding(self) @run_after('setup') def request_mem(self): + """Request memory per node based on the benchmark.""" mem_per_task = 0.4 if self.benchmark_info[0] == 'hackathonGPU/benchmark5': mem_per_task = 1.2 memory_required = self.num_tasks_per_node * mem_per_task + 2 hooks.req_memory_per_node(test=self, app_mem_req=memory_required * 1024) + @run_after('setup') + def skip_max_corecnt(self): + """Skip tests if number of tasks per node exceeds maximum core count.""" + max_corecnt = 256 + self.skip_if( + self.num_tasks > max_corecnt, + f'Number of tasks per node {self.num_tasks} exceeds maximum core count {max_corecnt} for {self.bench_name}' + ) + @run_after('setup') def set_omp_num_threads(self): """