Fix node gpu info init

volcano-sh · Jun 23, 2020 · 70ad2b2 · 70ad2b2
1 parent ca7818c
commit 70ad2b2
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/pkg/scheduler/api/node_info.go b/pkg/scheduler/api/node_info.go
@@ -153,17 +153,19 @@ func (ni *NodeInfo) setNodeGPUInfo(node *v1.Node) {
 	if node == nil {
 		return
 	}
-	if memory, ok := node.Status.Capacity[VolcanoGPUNumber]; ok {
-		ni.GPUTotalMemory = memory.Value()
+	memory, ok := node.Status.Capacity[VolcanoGPUResource]
+	if !ok {
+		return
 	}
+	ni.GPUTotalMemory = memory.Value()
 
 	gpuNumber, ok := node.Status.Capacity[VolcanoGPUNumber]
-	if ok {
-		ni.GPUNumber = gpuNumber.Value()
+	if !ok {
+		return
 	}
+	ni.GPUNumber = gpuNumber.Value()
 
 	memoryPerCard := uint(ni.GPUTotalMemory / ni.GPUNumber)
-
 	for i := 0; i < int(ni.GPUNumber); i++ {
 		ni.GPUDevices[i] = NewGPUDevice(i, memoryPerCard)
 	}