SWDEV-519722 - Update test to work on MGpu machines without P2P access (#145)

[ROCm/hip-tests commit: 161ecddafe]
2025-05-14 20:52:58 +05:30
parent 0ad7f9c137
commit 63376e2b32
1 changed files with 1 additions and 0 deletions
@@ -446,6 +446,7 @@ void TestCore(const TestParams& p) {

  // Copy results back to Host
  for (auto i = 0u; i < p.num_devices; ++i) {
+    HIP_CHECK(hipSetDevice(i));
    const auto device_offset = i * p.kernel_count * p.ThreadCount();
    HIP_CHECK(hipMemcpy(old_vals.data() + device_offset, old_vals_devs[i].ptr(),
                        old_vals_alloc_size, hipMemcpyDeviceToHost));