#64 fix 'GPU' of ProcessInfo in GetProcessInfo (#65)

Signed-off-by: Jerry <[email protected]>
NVIDIA · Apr 9, 2024 · eeff1dd · eeff1dd
1 parent 5e46e1b
commit eeff1dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/dcgm/process_info.go b/pkg/dcgm/process_info.go
@@ -178,7 +178,7 @@ func getProcessInfo(groupId GroupHandle, pid uint) (processInfo []ProcessInfo, e
 		}
 
 		pInfo := ProcessInfo{
-			GPU:                uint(pidInfo.summary.gpuId),
+			GPU:                uint(pidInfo.gpus[i].gpuId),
 			PID:                uint(pidInfo.pid),
 			Name:               name,
 			ProcessUtilization: processUtil,