added cmd.wait to cleanup when running the bare process

DataDog · Nov 28, 2024 · eed8dfc · eed8dfc
1 parent 4441f12
commit eed8dfc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/gpu/testutil/samplebins.go b/pkg/gpu/testutil/samplebins.go
@@ -106,6 +106,7 @@ func runCommandAndPipeOutput(t *testing.T, command []string, args SampleArgs) (c
 	t.Cleanup(func() {
 		if cmd.Process != nil {
 			_ = cmd.Process.Kill()
+			_ = cmd.Wait()
 		}
 	})