fix id of gpu devices never delete when number gpu decrease

volcano-sh · May 7, 2022 · 5846a5d · 5846a5d
1 parent 6ea2341
commit 5846a5d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/scheduler/api/node_info.go b/pkg/scheduler/api/node_info.go
@@ -298,6 +298,7 @@ func (ni *NodeInfo) setNodeGPUInfo(node *v1.Node) {
 	}
 
 	memoryPerCard := uint(totalMemory / gpuNumber)
+	ni.GPUDevices = make(map[int]*GPUDevice)
 	for i := 0; i < int(gpuNumber); i++ {
 		ni.GPUDevices[i] = NewGPUDevice(i, memoryPerCard)
 	}