update memToshadow for pvc

Author: AllanZyne

libdevice/sanitizer/msan_rtl.cpp

@@ -132,10 +132,15 @@ inline uptr __msan_get_shadow_cpu(uptr addr) {
 
 inline uptr __msan_get_shadow_pvc(uptr addr, uint32_t as) {
   // Device USM only
-  uptr shadow_ptr = ((__SYCL_GLOBAL__ MsanLaunchInfo *)__MsanLaunchInfo.get())
-                        ->GlobalShadowOffset +
-                    (addr & 0x3FFF'FFFF'FFFFULL);
-  return shadow_ptr;
+  auto shadow_begin = ((__SYCL_GLOBAL__ MsanLaunchInfo *)__MsanLaunchInfo.get())
+                          ->GlobalShadowOffset;
+  auto shadow_end = ((__SYCL_GLOBAL__ MsanLaunchInfo *)__MsanLaunchInfo.get())
+                        ->GlobalShadowOffsetEnd;
+  if (addr < shadow_begin) {
+    return Ptr + (shadow_begin - 0xff00'0000'0000'0000ULL);
+  } else {
+    return Ptr - (0xff00'ffff'ffff'ffffULL - shadow_end);
+  }
 }
 
 } // namespace