Skip to content

Commit

Permalink
prepend check_always_request_gpus with underscore
Browse files Browse the repository at this point in the history
  • Loading branch information
Samuel Moors committed Feb 29, 2024
1 parent a13e910 commit 82891ba
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions eessi/testsuite/hooks.py
Original file line number Diff line number Diff line change
Expand Up @@ -114,7 +114,7 @@ def assign_tasks_per_compute_unit(test: rfm.RegressionTest, compute_unit: str, n
else:
raise ValueError(f'compute unit {compute_unit} is currently not supported')

check_always_request_gpus(test)
_check_always_request_gpus(test)


def _assign_num_tasks_per_node(test: rfm.RegressionTest, num_per: int = 1):
Expand Down Expand Up @@ -466,7 +466,7 @@ def set_compact_thread_binding(test: rfm.RegressionTest):
log(f'Set environment variable KMP_AFFINITY to {test.env_vars["KMP_AFFINITY"]}')


def check_always_request_gpus(test: rfm.RegressionTest):
def _check_always_request_gpus(test: rfm.RegressionTest):
"""
Make sure we always request enough GPUs if required for the current GPU partition (cluster-specific policy)
"""
Expand Down

0 comments on commit 82891ba

Please sign in to comment.