normalize power limit for jobmanager for multi device jobs

ipatix · ipatix · commit d46857199f3e · 2025-07-18T17:54:13.000+02:00
diff --git a/internal/jobmanager/jobmanager.go b/internal/jobmanager/jobmanager.go
@@ -365,11 +365,13 @@ func (j *JobManager) PowerBudgetWeight() float64 {
 }
 
 func (j *JobManager) PowerBudgetSet(power float64) {
-	if len(j.targetToOptimizer) <= 0 {
+	if len(j.targetToOptimizer) <= 0 || j.deviceCount() <= 0 {
 		j.Debug("Cannot set PowerBudget: No optimizers to account for")
 		return
 	}
 
+	power /= float64(j.deviceCount())
+
 	for _, optimizer := range j.targetToOptimizer {
 		powerBudgetLowerCfg, powerBudgetUpperCfg := optimizer.GetBordersCfg()
 		powerBudgetLowerCur, _ := optimizer.GetBordersCur()

Original file line number	Diff line number	Diff line change
`@@ -365,11 +365,13 @@ func (j *JobManager) PowerBudgetWeight() float64 {`
`365`	`365`	`}`
`366`	`366`
`367`	`367`	`func (j *JobManager) PowerBudgetSet(power float64) {`
`368`		`- if len(j.targetToOptimizer) <= 0 {`
	`368`	`+ if len(j.targetToOptimizer) <= 0 \|\| j.deviceCount() <= 0 {`
`369`	`369`	`j.Debug("Cannot set PowerBudget: No optimizers to account for")`
`370`	`370`	`return`
`371`	`371`	`}`
`372`	`372`
	`373`	`+ power /= float64(j.deviceCount())`
	`374`	`+`
`373`	`375`	`for _, optimizer := range j.targetToOptimizer {`
`374`	`376`	`powerBudgetLowerCfg, powerBudgetUpperCfg := optimizer.GetBordersCfg()`
`375`	`377`	`powerBudgetLowerCur, _ := optimizer.GetBordersCur()`