Skip to content

Commit

Permalink
Reduce the iteration count to make the OSU tests run faster, especial…
Browse files Browse the repository at this point in the history
…ly on slower interconnects
  • Loading branch information
Caspar van Leeuwen committed May 13, 2024
1 parent a1021d7 commit 625a613
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions eessi/testsuite/tests/apps/osu.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,11 @@ class EESSI_OSU_Micro_Benchmarks_pt2pt(osu_benchmark):
# unset num_tasks_per_node from the hpctestlib.
num_tasks_per_node = None

# Set num_warmup_iters to 5 to reduce execution time, especially on slower interconnects
num_warmup_iters = 5
# Set num_iters to 10 to reduce execution time, especially on slower interconnects
num_iters = 10

@run_after('init')
def filter_scales_2gpus(self):
"""Filter out scales with < 2 GPUs if running on GPUs"""
Expand Down Expand Up @@ -169,6 +174,11 @@ class EESSI_OSU_Micro_Benchmarks_coll(osu_benchmark):
# Unset num_tasks_per_node from hpctestlib
num_tasks_per_node = None

# Set num_warmup_iters to 5 to reduce execution time, especially on slower interconnects
num_warmup_iters = 5
# Set num_iters to 10 to reduce execution time, especially on slower interconnects
num_iters = 10

@run_after('init')
def run_after_init(self):
"""hooks to run after init phase"""
Expand Down

0 comments on commit 625a613

Please sign in to comment.