[Fix] Out of Bounds Power Limit in GlobalPowerLimitOptimizer (#143)

ml-energy · Dec 10, 2024 · 6c6b79c · 6c6b79c
1 parent b73bd7f
commit 6c6b79c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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: