Fixed dcgm.WatchPidFields using wrong time unit

Signed-off-by: helinfan.hlf <[email protected]>
NVIDIA · Oct 27, 2023 · 875019b · 875019b
1 parent d898cc7
commit 875019b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/dcgm/api.go b/pkg/dcgm/api.go
@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"os"
 	"sync"
+	"time"
 )
 
 var (
@@ -89,7 +90,7 @@ func GetDeviceTopology(gpuId uint) ([]P2PLink, error) {
 // WatchPidFields lets DCGM start recording stats for GPU process
 // It needs to be called before calling GetProcessInfo
 func WatchPidFields() (GroupHandle, error) {
-	return watchPidFields(defaultUpdateFreq, defaultMaxKeepAge, defaultMaxKeepSamples)
+	return watchPidFields(time.Microsecond * time.Duration(defaultUpdateFreq), time.Second * time.Duration(defaultMaxKeepAge), defaultMaxKeepSamples)
 }
 
 // GetProcessInfo provides detailed per GPU stats for this process