apps/basics
1
0
Fork 0
mirror of https://github.com/oneclickvirt/basics.git synced 2025-10-09 18:40:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: 修复GPU使用率检测

Browse Source
This commit is contained in:
spiritlhl
2025-03-30 04:52:49 +00:00
parent 3b9fe7c34a
commit d0a37ca0c8
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
network/network_test.go
View File

@@ -5,6 +5,7 @@ import (
"testing" "testing"
) )
// 本文件夹 network 修改需要同步 https://github.com/oneclickvirt/security 否则 goecs 无法使用
func TestIpv4SecurityCheck(t *testing.T) { func TestIpv4SecurityCheck(t *testing.T) {
// 全项测试 // 全项测试
ipInfo, _, _ := NetworkCheck("both", false, "zh") ipInfo, _, _ := NetworkCheck("both", false, "zh")

3
system/gpu.go
View File

@@ -43,7 +43,6 @@ func updateGPUStat(gpuStat *uint64, wg *sync.WaitGroup) {
return return
} }
defer atomic.StoreInt32(&updateGPUStatus, 0) defer atomic.StoreInt32(&updateGPUStatus, 0)
for statDataFetchAttempts["GPU"] < maxDeviceDataFetchAttempts { for statDataFetchAttempts["GPU"] < maxDeviceDataFetchAttempts {
gs, err := gpustat.GetGPUStat() gs, err := gpustat.GetGPUStat()
if err != nil { if err != nil {
@@ -74,7 +73,9 @@ func getGPUInfo(ret *model.SystemInfo) (*model.SystemInfo, error) {
wg.Add(1) wg.Add(1)
go updateGPUStat(&gpuStat, &wg) go updateGPUStat(&gpuStat, &wg)
wg.Wait() // 等待 updateGPUStat 完成 wg.Wait() // 等待 updateGPUStat 完成
if math.Float64frombits(gpuStat) > 0 {
ret.GpuStats = fmt.Sprintf("%f", math.Float64frombits(gpuStat)) ret.GpuStats = fmt.Sprintf("%f", math.Float64frombits(gpuStat))
}
return ret, nil return ret, nil
} else { } else {
hostDataFetchAttempts["GPU"]++ hostDataFetchAttempts["GPU"]++