KJZH001/nezhahq-agent
1
0
Fork 0
Code Issues Pull Requests 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
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/gpu/stat/stat_windows.go
+4 -1
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
} }