gpu: fix test after merge

DataDog · Nov 17, 2024 · 845f4b2 · 845f4b2
1 parent 80d6edc
commit 845f4b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/gpu/probe_test.go b/pkg/gpu/probe_test.go
@@ -153,7 +153,7 @@ func (s *probeTestSuite) TestMultiGPUSupport() {
 	selectedGPU := testutil.GPUUUIDs[2]
 
 	cmd := testutil.RunSampleWithArgs(t, testutil.CudaSample, sampleArgs)
-	utils.WaitForProgramsToBeTraced(t, gpuAttacherName, cmd.Process.Pid, utils.ManualTracingFallbackEnabled)
+	utils.WaitForProgramsToBeTraced(t, gpuModuleName, gpuAttacherName, cmd.Process.Pid, utils.ManualTracingFallbackEnabled)
 
 	// 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