update worker info

b90c22d8 · vicotor · 20a90af6 · b90c22d8
Commit b90c22d8 authored May 23, 2024 by vicotor
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

worker_registry.go server/worker_registry.go +5 -0

No files found.
--- a/server/worker_registry.go
+++ b/server/worker_registry.go
@@ -103,6 +103,11 @@ func (w workerRegistry) DetailInfo() (json.RawMessage, error) {
 			info.CpuCore = 8
 		}
 		info.GPUModel = generateAGpuModel()
+		for i := 0; i < len(w.worker.info.deviceInfo.Devices); i++ {
+			if strings.HasPrefix(w.worker.info.deviceInfo.Devices[i].DeviceType, "gpu") {
+				info.GPUCount++
+			}
+		}
 		info.GPURam = generateAGpuRam()
 		ramTotal, ramUsed := generateARam()