fix gpu stats panic error (#769)

baetyl · Jun 8, 2023 · 7cb19cd · 7cb19cd
1 parent 8e8d18b
commit 7cb19cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ami/kube/kube_collect.go b/ami/kube/kube_collect.go
@@ -153,12 +153,12 @@ func (k *kubeImpl) CollectNodeStats() (map[string]interface{}, error) {
 
 		var nodeStatsMerge map[string]interface{}
 		if len(gpuExts) > 0 {
-			if ext, ok := gpuExts[node.Name]; ok {
+			if ext, ok := gpuExts[node.Name]; ok && ext != nil {
 				nodeStatsMerge = ext.(map[string]interface{})
 			}
 		}
 		if len(nodeExts) > 0 {
-			if ext, ok := nodeExts[node.Name]; ok {
+			if ext, ok := nodeExts[node.Name]; ok && ext != nil {
 				if nodeStatsMerge == nil {
 					nodeStatsMerge = make(map[string]interface{}, 0)
 				}

diff --git a/ami/native/native.go b/ami/native/native.go
@@ -827,7 +827,7 @@ func (impl *nativeImpl) CollectNodeStats() (map[string]interface{}, error) {
 	}
 	var nodeStatsMerge map[string]interface{}
 	if len(gpuExts) > 0 {
-		if ext, ok := gpuExts[ho.Hostname]; ok {
+		if ext, ok := gpuExts[ho.Hostname]; ok && ext != nil {
 			nodeStatsMerge = ext.(map[string]interface{})
 		}
 	}