Skip to content

Commit

Permalink
remove unused arguments and address pylint error
Browse files Browse the repository at this point in the history
  • Loading branch information
sjsprecious committed Oct 6, 2024
1 parent 0b307d8 commit 0d7ce8b
Showing 1 changed file with 3 additions and 10 deletions.
13 changes: 3 additions & 10 deletions CIME/XML/env_mach_pes.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,10 +41,6 @@ def get_value(
attribute=None,
resolved=True,
subgroup=None,
max_mpitasks_per_node=None,
max_cputasks_per_gpu_node=None,
ngpus_per_node=None,
oversubscribe_gpu=False,
): # pylint: disable=arguments-differ
# Special variable NINST_MAX is used to determine the number of
# drivers in multi-driver mode.
Expand All @@ -59,12 +55,9 @@ def get_value(
value = EnvBase.get_value(self, vid, attribute, resolved, subgroup)

if "NTASKS" in vid or "ROOTPE" in vid:
if max_mpitasks_per_node is None:
max_mpitasks_per_node = self.get_value("MAX_MPITASKS_PER_NODE")
if max_cputasks_per_gpu_node is None:
max_cputasks_per_gpu_node = self.get_value("MAX_CPUTASKS_PER_GPU_NODE")
if ngpus_per_node is None:
ngpus_per_node = self.get_value("NGPUS_PER_NODE")
max_mpitasks_per_node = self.get_value("MAX_MPITASKS_PER_NODE")
max_cputasks_per_gpu_node = self.get_value("MAX_CPUTASKS_PER_GPU_NODE")
ngpus_per_node = self.get_value("NGPUS_PER_NODE")
if (ngpus_per_node and value) and value < 0:
value = -1 * value * max_cputasks_per_gpu_node
elif value and value < 0:
Expand Down

0 comments on commit 0d7ce8b

Please sign in to comment.