diff --git a/zeus/optimizer/power_limit.py b/zeus/optimizer/power_limit.py index e232c571..0ac2c2d1 100644 --- a/zeus/optimizer/power_limit.py +++ b/zeus/optimizer/power_limit.py @@ -263,9 +263,7 @@ def __init__( pls.append(gpus.getPowerManagementLimitConstraints(index)) if not all(pls[0] == pl for pl in pls): raise ValueError("Power limits ranges are not uniform across GPUs.") - self.power_limits = list( - range(pls[0][1], pls[0][0] - self.pl_step, -self.pl_step) - ) + self.power_limits = list(range(pls[0][1], pls[0][0] - 1, -self.pl_step)) # Turn on persistence mode and set to the highest power limit. try: