rocr/aie: Handle non-HSA_STATUS_SUCCESS during VisitRegion

2025-02-17 18:06:55 -05:00
parent de8f8f076d
commit fb33e2e724
1 changed files with 4 additions and 1 deletions
@@ -76,7 +76,10 @@ hsa_status_t AieAgent::VisitRegion(bool include_peer,
  AMD::callback_t<decltype(callback)> call(callback);
  for (const auto r : regions_) {
    hsa_region_t region_handle(core::MemoryRegion::Convert(r));
-    call(region_handle, data);
+    hsa_status_t status = call(region_handle, data);
+    if (status != HSA_STATUS_SUCCESS) {
+      return status;
+    }
  }
  return HSA_STATUS_SUCCESS;
 }