Merge pull request #2679 from yingcong-wu/yc/0208-ur-dasan-bugfix-main

[DeviceASAN][bugfix] Not allow concurrent kernel launches across different queue

Author: kbenzie

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

@@ -474,6 +474,13 @@ __urdlllocal ur_result_t UR_APICALL urEnqueueKernelLaunch(
     /// [out][optional] return an event object that identifies this
     /// particular kernel execution instance.
     ur_event_handle_t *phEvent) {
+
+  // This mutex is to prevent concurrent kernel launches across different queues
+  // as the DeviceASAN local/private shadow memory does not support concurrent
+  // kernel launches now.
+  std::scoped_lock<ur_shared_mutex> Guard(
+      getAsanInterceptor()->KernelLaunchMutex);
+
   auto pfnKernelLaunch = getContext()->urDdiTable.Enqueue.pfnKernelLaunch;
 
   if (nullptr == pfnKernelLaunch) {

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

@@ -354,6 +354,8 @@ class AsanInterceptor {
   std::shared_ptr<ShadowMemory>
   getOrCreateShadowMemory(ur_device_handle_t Device, DeviceType Type);
 
+  ur_shared_mutex KernelLaunchMutex;
+
 private:
   ur_result_t updateShadowMemory(std::shared_ptr<ContextInfo> &ContextInfo,
                                  std::shared_ptr<DeviceInfo> &DeviceInfo,