From 845f4b2e702ec29a8740409ec0e7e6e74109b227 Mon Sep 17 00:00:00 2001 From: Guy Arbitman Date: Mon, 18 Nov 2024 01:22:21 +0200 Subject: [PATCH] gpu: fix test after merge --- pkg/gpu/probe_test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pkg/gpu/probe_test.go b/pkg/gpu/probe_test.go index 45a1a6be64e76..9739fc0eccd51 100644 --- 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