diff --git a/source/adapters/native_cpu/context.hpp b/source/adapters/native_cpu/context.hpp index 0595f211d9..c59ab4eafb 100644 --- a/source/adapters/native_cpu/context.hpp +++ b/source/adapters/native_cpu/context.hpp @@ -109,8 +109,9 @@ struct ur_context_handle_t_ : RefCounted { return UR_RESULT_SUCCESS; } - const native_cpu::usm_alloc_info & - get_alloc_info_entry(const void *ptr) const { + // Note this is made non-const to access the mutex + const native_cpu::usm_alloc_info &get_alloc_info_entry(const void *ptr) { + std::lock_guard lock(alloc_mutex); auto it = allocations.find(ptr); if (it == allocations.end()) { return native_cpu::usm_alloc_info_null_entry;