lock directly

Author: yingcong-wu

source/loader/layers/sanitizer/asan/asan_interceptor.cpp

@@ -310,21 +310,19 @@ ur_result_t AsanInterceptor::postLaunchKernel(ur_kernel_handle_t Kernel,
 std::shared_ptr<ShadowMemory>
 AsanInterceptor::getOrCreateShadowMemory(ur_device_handle_t Device,
                                          DeviceType Type) {
+  std::scoped_lock<ur_shared_mutex> Guard(m_ShadowMapMutex);
   if (m_ShadowMap.find(Type) == m_ShadowMap.end()) {
-    std::scoped_lock<ur_shared_mutex> Guard(m_ShadowMapMutex);
-    if (m_ShadowMap.find(Type) == m_ShadowMap.end()) {
-      ur_context_handle_t InternalContext;
-      auto Res = getContext()->urDdiTable.Context.pfnCreate(1, &Device, nullptr,
-                                                            &InternalContext);
-      if (Res != UR_RESULT_SUCCESS) {
-        getContext()->logger.error("Failed to create shadow context");
-        return nullptr;
-      }
-      std::shared_ptr<ContextInfo> CI;
-      insertContext(InternalContext, CI);
-      m_ShadowMap[Type] = GetShadowMemory(InternalContext, Device, Type);
-      m_ShadowMap[Type]->Setup();
+    ur_context_handle_t InternalContext;
+    auto Res = getContext()->urDdiTable.Context.pfnCreate(1, &Device, nullptr,
+                                                          &InternalContext);
+    if (Res != UR_RESULT_SUCCESS) {
+      getContext()->logger.error("Failed to create shadow context");
+      return nullptr;
     }
+    std::shared_ptr<ContextInfo> CI;
+    insertContext(InternalContext, CI);
+    m_ShadowMap[Type] = GetShadowMemory(InternalContext, Device, Type);
+    m_ShadowMap[Type]->Setup();
   }
   return m_ShadowMap[Type];
 }