Fix request sum returning non-totaled GPU

michaelmdresser · michaelmdresser · commit 0076061459e5 · 2023-02-15T13:04:49.000-08:00
Signed-off-by: Michael Dresser &lt;michaelmdresser@gmail.com&gt;
diff --git a/pkg/cmd/predict.go b/pkg/cmd/predict.go
@@ -159,8 +159,8 @@ func sumContainerResources(replicas int, spec v1.PodSpec) v1.ResourceList {
 
 	// Only include GPU data if we have any. Put under the generic "gpu"
 	// name because we don't (currently) distinguish between providers.
-	if !podGPU.IsZero() {
-		result[resourceGPUKey] = *podGPU
+	if !totalGPU.IsZero() {
+		result[resourceGPUKey] = *totalGPU
 	}
 
 	return result

Original file line number	Diff line number	Diff line change
`@@ -159,8 +159,8 @@ func sumContainerResources(replicas int, spec v1.PodSpec) v1.ResourceList {`
`159`	`159`
`160`	`160`	`// Only include GPU data if we have any. Put under the generic "gpu"`
`161`	`161`	`// name because we don't (currently) distinguish between providers.`
`162`		`- if !podGPU.IsZero() {`
`163`		`- result[resourceGPUKey] = *podGPU`
	`162`	`+ if !totalGPU.IsZero() {`
	`163`	`+ result[resourceGPUKey] = *totalGPU`
`164`	`164`	`}`
`165`	`165`
`166`	`166`	`return result`