fix local memory tests

Author: AllanZyne

source/loader/layers/sanitizer/asan_interceptor.cpp

@@ -743,7 +743,8 @@ ur_result_t SanitizerInterceptor::prepareLaunch(
         // Write shadow memory offset for local memory
         if (cl_DetectLocals) {
             // CPU needn't this
-            if (DeviceInfo->Type == DeviceType::GPU_PVC) {
+            if (DeviceInfo->Type == DeviceType::GPU_PVC ||
+                DeviceInfo->Type == DeviceType::GPU_DG2) {
                 size_t LocalMemorySize = GetLocalMemorySize(DeviceInfo->Handle);
                 size_t LocalShadowMemorySize =
                     (NumWG * LocalMemorySize) >> ASAN_SHADOW_SCALE;
@@ -762,6 +763,14 @@ ur_result_t SanitizerInterceptor::prepareLaunch(
                 context.logger.info("ShadowMemory(Local, {} - {})",
                                     (void *)LaunchInfo.LocalShadowOffset,
                                     (void *)LaunchInfo.LocalShadowOffsetEnd);
+
+                // Write shadow memory offset for local memory
+                EnqueueWriteGlobal(kSPIR_AsanShadowMemoryLocalStart,
+                                   &LaunchInfo.LocalShadowOffset,
+                                   sizeof(LaunchInfo.LocalShadowOffset));
+                EnqueueWriteGlobal(kSPIR_AsanShadowMemoryLocalEnd,
+                                   &LaunchInfo.LocalShadowOffsetEnd,
+                                   sizeof(LaunchInfo.LocalShadowOffsetEnd));
             }
         }
     } while (false);

source/loader/layers/sanitizer/asan_libdevice.hpp

@@ -90,6 +90,10 @@ constexpr auto kSPIR_AsanShadowMemoryGlobalStart =
     "__AsanShadowMemoryGlobalStart";
 constexpr auto kSPIR_AsanShadowMemoryGlobalEnd = "__AsanShadowMemoryGlobalEnd";
 
+constexpr auto kSPIR_AsanShadowMemoryLocalStart =
+    "__AsanShadowMemoryLocalStart";
+constexpr auto kSPIR_AsanShadowMemoryLocalEnd = "__AsanShadowMemoryLocalEnd";
+
 constexpr auto kSPIR_DeviceType = "__DeviceType";
 constexpr auto kSPIR_AsanDebug = "__AsanDebug";