Skip to content

Commit

Permalink
Fix linter issues
Browse files Browse the repository at this point in the history
  • Loading branch information
Caspar van Leeuwen committed Jun 28, 2024
1 parent 8167eee commit 04a8997
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions eessi/testsuite/hooks.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,8 @@ def assign_tasks_per_compute_unit(test: rfm.RegressionTest, compute_unit: str, n
Total task count is determined based on the number of nodes to be used in the test.
Behaviour of this function is (usually) sensible for MPI tests.
WARNING: when using COMPUTE_UNIT[HWTHREAD] and invoking a hook for process binding, please verify that process binding happens correctly
WARNING: when using COMPUTE_UNIT[HWTHREAD] and invoking a hook for process binding, please verify that process
binding happens correctly.
Arguments:
- test: the ReFrame test to which this hook should apply
Expand Down Expand Up @@ -230,7 +231,9 @@ def _assign_one_task_per_cpu(test: rfm.RegressionTest):
# neither num_tasks_per_node nor num_cpus_per_task are set
if not test.num_tasks_per_node and not test.num_cpus_per_task:
check_proc_attribute_defined(test, 'num_cpus_per_core')
test.num_tasks_per_node = int(test.default_num_cpus_per_node / test.current_partition.processor.num_cpus_per_core)
test.num_tasks_per_node = int(
test.default_num_cpus_per_node / test.current_partition.processor.num_cpus_per_core
)
test.num_cpus_per_task = int(test.default_num_cpus_per_node / test.num_tasks_per_node)

# num_tasks_per_node is not set, but num_cpus_per_task is
Expand Down

0 comments on commit 04a8997

Please sign in to comment.