update hardware info is nil

f30e0637 · duanjinfei · 2e8f683d · f30e0637
Commit f30e0637 authored May 31, 2024 by duanjinfei
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

msg_resp.go nm/msg_resp.go +3 -0

No files found.
--- a/nm/msg_resp.go
+++ b/nm/msg_resp.go
@@ -179,6 +179,9 @@ func DeviceUsageResp(params ...interface{}) *nodemanagerV2.WorkerMessage {
 func GpuUsageResp(params ...interface{}) *nodemanagerV2.WorkerMessage {
 	log.Info("DeviceUsageResp params :", params)
 	hardwareInfo := GetHardwareInfo()
+	if hardwareInfo == nil {
+		return nil
+	}
 	gpusUsage := make([]*nodemanagerV2.GPUUsage, 0)
 	for _, gpuInfo := range hardwareInfo.GPU {
 		usage := &nodemanagerV2.GPUUsage{