Skip to content

Commit 508e6b9

Browse files
committed
move mutex
1 parent fd60baf commit 508e6b9

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

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

+2-2
Original file line numberDiff line numberDiff line change
@@ -478,8 +478,8 @@ __urdlllocal ur_result_t UR_APICALL urEnqueueKernelLaunch(
478478
// This mutex is to prevent concurrent kernel launches across different queues
479479
// as the DeviceASAN local/private shadow memory does not support concurrent
480480
// kernel launches now.
481-
static ur_shared_mutex KernelLaunchMutex;
482-
std::scoped_lock<ur_shared_mutex> Guard(KernelLaunchMutex);
481+
std::scoped_lock<ur_shared_mutex> Guard(
482+
getAsanInterceptor()->KernelLaunchMutex);
483483

484484
auto pfnKernelLaunch = getContext()->urDdiTable.Enqueue.pfnKernelLaunch;
485485

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

+2
Original file line numberDiff line numberDiff line change
@@ -354,6 +354,8 @@ class AsanInterceptor {
354354
std::shared_ptr<ShadowMemory>
355355
getOrCreateShadowMemory(ur_device_handle_t Device, DeviceType Type);
356356

357+
ur_shared_mutex KernelLaunchMutex;
358+
357359
private:
358360
ur_result_t updateShadowMemory(std::shared_ptr<ContextInfo> &ContextInfo,
359361
std::shared_ptr<DeviceInfo> &DeviceInfo,

0 commit comments

Comments
 (0)