Skip to content

Commit

Permalink
Merge pull request EESSI#143 from casparvl/reduce_osu_iteration_count
Browse files Browse the repository at this point in the history
Reduce the iteration count to make the OSU tests run faster, especially on slower interconnects
  • Loading branch information
satishskamath authored May 13, 2024
2 parents a1021d7 + 625a613 commit 59938bd
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 59938bd

Please sign in to comment.