Fix missing mutex on get_alloc_info_entry

coldav · Aug 15, 2024 · 562f212 · 562f212
1 parent d7e0fad
commit 562f212
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/source/adapters/native_cpu/context.hpp b/source/adapters/native_cpu/context.hpp
@@ -109,8 +109,10 @@ struct ur_context_handle_t_ : RefCounted {
     return UR_RESULT_SUCCESS;
   }
 
+  // Note this is made non-const to access the mutex
   const native_cpu::usm_alloc_info &
-  get_alloc_info_entry(const void *ptr) const {
+  get_alloc_info_entry(const void *ptr) {
+    std::lock_guard<std::mutex> lock(alloc_mutex);
     auto it = allocations.find(ptr);
     if (it == allocations.end()) {
       return native_cpu::usm_alloc_info_null_entry;