add missed lock

Author: AllanZyne

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

@@ -696,6 +696,7 @@ ur_result_t AsanInterceptor::prepareLaunch(
     std::shared_ptr<DeviceInfo> &DeviceInfo, ur_queue_handle_t Queue,
     ur_kernel_handle_t Kernel, LaunchInfo &LaunchInfo) {
     auto &KernelInfo = getOrCreateKernelInfo(Kernel);
+    std::shared_lock<ur_shared_mutex> Guard(KernelInfo.Mutex);
 
     auto ArgNums = GetKernelNumArgs(Kernel);
     auto LocalMemoryUsage =

source/loader/layers/sanitizer/msan/msan_interceptor.cpp

@@ -371,6 +371,7 @@ ur_result_t MsanInterceptor::prepareLaunch(
 
     // Set membuffer arguments
     auto &KernelInfo = getOrCreateKernelInfo(Kernel);
+    std::shared_lock<ur_shared_mutex> Guard(KernelInfo.Mutex);
 
     for (const auto &[ArgIndex, MemBuffer] : KernelInfo.BufferArgs) {
         char *ArgPointer = nullptr;