KJZH001/nezhahq-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(gpu): memory leak on windows (#59)

Browse Source
This commit is contained in:
UUBulb 2024-09-05 06:39:17 +08:00 committed by GitHub
parent 64ff1e741d
commit 439be0ddf3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/gpu/stat/stat_windows.go
View File

@ -30,6 +30,7 @@ var (
pdhCollectQueryData = modPdh.NewProc("PdhCollectQueryData") pdhCollectQueryData = modPdh.NewProc("PdhCollectQueryData")
pdhGetFormattedCounterArrayW = modPdh.NewProc("PdhGetFormattedCounterArrayW") pdhGetFormattedCounterArrayW = modPdh.NewProc("PdhGetFormattedCounterArrayW")
pdhAddEnglishCounterW = modPdh.NewProc("PdhAddEnglishCounterW") pdhAddEnglishCounterW = modPdh.NewProc("PdhAddEnglishCounterW")
pdhCloseQuery = modPdh.NewProc("PdhCloseQuery")
) )
type PDH_FMT_COUNTERVALUE_DOUBLE struct { type PDH_FMT_COUNTERVALUE_DOUBLE struct {
@ -131,7 +132,9 @@ func GetGPUStat() (float64, error) {
if err != nil { if err != nil {
return 0, err return 0, err
} }
values, err := getValue(1024, counter) defer pdhCloseQuery.Call(uintptr(counter.Query))
values, err := getValue(8192, counter)
if err != nil { if err != nil {
return 0, err return 0, err
} }