Merge commit 'b0519e8441273e20bbcbe1bb7b56cdf55369f514' into develop

2025-08-18 18:08:07 +00:00
parent c95526c57e b0519e8441
commit bc455989d9
52 changed files with 1439 additions and 6178 deletions
@@ -109,6 +109,7 @@ TEST_CASE("test_svm_byte_granularity") {
  // get all the devices going simultaneously
  for(unsigned int d = 0; d < num_devices; d++)  // device ids starting at 1.
  {
+    HIP_CHECK(hipSetDevice(d));
    write_owned_locations<<<num_elements, 1, 0, streams[d]>>>(pA, num_devices_plus_host, d);
    HIP_CHECK(hipGetLastError());
  }
@@ -125,6 +126,7 @@ TEST_CASE("test_svm_byte_granularity") {
  size_t adjusted_num_elements = num_elements - num_devices;
  for(unsigned int d = 0; d < num_devices; d++)
  {
+    HIP_CHECK(hipSetDevice(d));
    sum_neighbor_locations<<<adjusted_num_elements, 1, 0, streams[d]>>>(pA, num_devices_plus_host,
                                                                     error_counts[d]);
    HIP_CHECK(hipGetLastError());