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

@@ -463,6 +463,12 @@ __urdlllocal ur_result_t UR_APICALL urEnqueueKernelLaunch(
         phEvent ///< [out][optional] return an event object that identifies this particular
                 ///< kernel execution instance.
 ) {
+    // 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

@@ -360,6 +360,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,