Track GPU discovery failure information (#5820)

* Expose GPU discovery failure information

* Remove exposed API for now
This commit is contained in:
Daniel Hiltgen
2024-10-14 16:26:45 -07:00
committed by GitHub
parent 5dd0477fd4
commit f3c8b898cd
5 changed files with 242 additions and 92 deletions

View File

@@ -66,3 +66,15 @@ func (l GpuInfoList) GetVisibleDevicesEnv() (string, string) {
// No-op on darwin
return "", ""
}
func GetSystemInfo() SystemInfo {
mem, _ := GetCPUMem()
return SystemInfo{
System: CPUInfo{
GpuInfo: GpuInfo{
memInfo: mem,
},
},
GPUs: GetGPUInfo(),
}
}