rocr/aie: Handle non-HSA_STATUS_SUCCESS during VisitRegion

[ROCm/ROCR-Runtime commit: fb33e2e724]
Цей коміт міститься в:
Yiannis Papadopoulos
2025-02-17 18:06:55 -05:00
зафіксовано Yat Sin, David
джерело 77f4bbfdf1
коміт 8e111ff2f0
+4 -1
Переглянути файл
@@ -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;
}