Skip to content

Commit

Permalink
Docstrings and skip test if corecnt > 256
Browse files Browse the repository at this point in the history
  • Loading branch information
Crivella committed Aug 29, 2024
1 parent 06f90c9 commit 77fa436
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions eessi/testsuite/tests/apps/MetalWalls.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
"""
Expand Down

0 comments on commit 77fa436

Please sign in to comment.