From 80d6edcf4861666716d3d7d8554384025c1e2964 Mon Sep 17 00:00:00 2001 From: Guy Arbitman Date: Mon, 18 Nov 2024 00:22:11 +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 8e1f690dae0b9..45a1a6be64e76 100644 --- 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()