Bläddra i källkod

Change how gpu request/limits are parsed to support fractional gpus

Kaelan Patel 4 år sedan
förälder
incheckning
9a488e06b1
1 ändrade filer med 4 tillägg och 4 borttagningar
  1. 4 4
      pkg/costmodel/costmodel.go

+ 4 - 4
pkg/costmodel/costmodel.go

@@ -501,14 +501,14 @@ func (cm *CostModel) ComputeCostData(cli prometheusClient.Client, cp costAnalyze
 
 				gpuReqCount := 0.0
 				if g, ok := container.Resources.Requests["nvidia.com/gpu"]; ok {
-					gpuReqCount = float64(g.Value())
+					gpuReqCount = g.AsApproximateFloat64()
 				} else if g, ok := container.Resources.Limits["nvidia.com/gpu"]; ok {
-					gpuReqCount = float64(g.Value())
+					gpuReqCount = g.AsApproximateFloat64()
 				} else if g, ok := container.Resources.Requests["k8s.amazonaws.com/vgpu"]; ok {
 					// divide vgpu request/limits by total vgpus to get the portion of physical gpus requested
-					gpuReqCount = float64(g.Value()) / vgpuCoeff
+					gpuReqCount = g.AsApproximateFloat64() / vgpuCoeff
 				} else if g, ok := container.Resources.Limits["k8s.amazonaws.com/vgpu"]; ok {
-					gpuReqCount = float64(g.Value()) / vgpuCoeff
+					gpuReqCount = g.AsApproximateFloat64() / vgpuCoeff
 				}
 				GPUReqV := []*util.Vector{
 					{