SWDEV-518356: added check to avoid out of range hip host to device. (#267)

added check to avoid out of range.

Co-authored-by: Venkateshwar Reddy Kandula <vkandula@amd.com>
Cette révision appartient à :
Kandula, Venkateshwar reddy
2025-03-11 15:37:59 -05:00
révisé par GitHub
Parent f7e94c1ee8
révision 8735ae4eb0
+11 -1
Voir le fichier
@@ -909,8 +909,18 @@ executable_freeze(hsa_executable_t executable, const char* options)
CHECK_NOTNULL(get_hip_register_data())
->rlock([device_name](
const hip::hip_register_data& register_data) {
return register_data.host_function_map.at(device_name);
// Add check for out of range here
const auto it =
register_data.host_function_map.find(device_name);
if(it == register_data.host_function_map.end())
{
return rocprofiler_callback_tracing_code_object_host_kernel_symbol_register_data_t{};
}
return it->second;
});
// when kernel_symbol_device_map kernels are not present in
// host_function_map, skip.
if(host_data.device_function == nullptr) continue;
host_data.code_object_id = sym_data.code_object_id;
host_data.kernel_id = sym_data.kernel_id;
host_data.host_function_id = ++get_host_function_id();