replace podreqgpu with containerreqgpu

Sakuralbj · Sakuralbj · commit 475e1a399fab · 2019-07-09T16:49:58.000+08:00
Signed-off-by: 630445639 &lt;630445639@qq.com&gt;
diff --git a/pkg/gpu/nvidia/allocate.go b/pkg/gpu/nvidia/allocate.go
@@ -34,7 +34,7 @@ func (m *NvidiaDevicePlugin) buildErrResponse(reqs *pluginapi.AllocateRequest, p
 			},
 		}
 		if m.mps {
-			response.Envs[EnvPercentage] = fmt.Sprintf("%d", 100*podReqGPU/getGPUMemory())
+			response.Envs[EnvPercentage] = fmt.Sprintf("%d", 100*uint(len(req.DevicesIDs))/getGPUMemory()) //
 		}
 		responses.ContainerResponses = append(responses.ContainerResponses, &response)
 	}
@@ -125,7 +125,7 @@ func (m *NvidiaDevicePlugin) Allocate(ctx context.Context,
 				},
 			}
 			if m.mps {
-				response.Envs[EnvPercentage] = fmt.Sprintf("%d", 100*podReqGPU/getGPUMemory())
+				response.Envs[EnvPercentage] = fmt.Sprintf("%d", 100*reqGPU/getGPUMemory())
 			}
 			responses.ContainerResponses = append(responses.ContainerResponses, &response)
 		}

Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,7 @@ func (m NvidiaDevicePlugin) buildErrResponse(reqs pluginapi.AllocateRequest, p`
`34`	`34`	`},`
`35`	`35`	`}`
`36`	`36`	`if m.mps {`
`37`		`- response.Envs[EnvPercentage] = fmt.Sprintf("%d", 100*podReqGPU/getGPUMemory())`
	`37`	`+ response.Envs[EnvPercentage] = fmt.Sprintf("%d", 100*uint(len(req.DevicesIDs))/getGPUMemory()) //`
`38`	`38`	`}`
`39`	`39`	`responses.ContainerResponses = append(responses.ContainerResponses, &response)`
`40`	`40`	`}`
`@@ -125,7 +125,7 @@ func (m *NvidiaDevicePlugin) Allocate(ctx context.Context,`
`125`	`125`	`},`
`126`	`126`	`}`
`127`	`127`	`if m.mps {`
`128`		`- response.Envs[EnvPercentage] = fmt.Sprintf("%d", 100*podReqGPU/getGPUMemory())`
	`128`	`+ response.Envs[EnvPercentage] = fmt.Sprintf("%d", 100*reqGPU/getGPUMemory())`
`129`	`129`	`}`
`130`	`130`	`responses.ContainerResponses = append(responses.ContainerResponses, &response)`
`131`	`131`	`}`