gpu: fix test after merge

DataDog · Nov 17, 2024 · 80d6edc · 80d6edc
1 parent 352dd75
commit 80d6edc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/gpu/probe_test.go b/pkg/gpu/probe_test.go
@@ -158,7 +158,7 @@ func (s *probeTestSuite) TestMultiGPUSupport() {
 	// Wait until the process finishes and we can get the stats. Run this instead of waiting for the process to finish
 	// so that we can time out correctly
 	require.Eventually(t, func() bool {
-		return !utils.IsProgramTraced(gpuAttacherName, cmd.Process.Pid)
+		return !utils.IsProgramTraced(gpuModuleName, gpuAttacherName, cmd.Process.Pid)
 	}, 60*time.Second, 500*time.Millisecond, "process not stopped")
 
 	stats, err := probe.GetAndFlush()