From 8e111ff2f02ed6fd335bedcacd0ec9b883221254 Mon Sep 17 00:00:00 2001 From: Yiannis Papadopoulos Date: Mon, 17 Feb 2025 18:06:55 -0500 Subject: [PATCH] rocr/aie: Handle non-HSA_STATUS_SUCCESS during VisitRegion [ROCm/ROCR-Runtime commit: fb33e2e724308a76468b10b4a6f156f6c73a9968] --- .../runtime/hsa-runtime/core/runtime/amd_aie_agent.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/amd_aie_agent.cpp b/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/amd_aie_agent.cpp index d655d0fa2f..1e3547f003 100644 --- a/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/amd_aie_agent.cpp +++ b/projects/rocr-runtime/runtime/hsa-runtime/core/runtime/amd_aie_agent.cpp @@ -76,7 +76,10 @@ hsa_status_t AieAgent::VisitRegion(bool include_peer, AMD::callback_t 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; }