fix build

Author: AllanZyne

source/loader/layers/sanitizer/asan_shadow_setup.cpp

@@ -134,13 +134,13 @@ ur_result_t SetupShadowMemory(ur_context_handle_t Context, uptr &ShadowBegin,
     // multiple contexts. Therefore, we just create one shadow memory here.
     static ur_result_t Result = [&Context]() {
         // TODO: Protect Bad Zone
-        auto Result = context.urDdiTable.VirtualMem.pfnReserve(
+        auto Result = getContext()->urDdiTable.VirtualMem.pfnReserve(
             Context, nullptr, SHADOW_SIZE, (void **)&LOW_SHADOW_BEGIN);
         if (Result == UR_RESULT_SUCCESS) {
             HIGH_SHADOW_END = LOW_SHADOW_BEGIN + SHADOW_SIZE;
             // Retain the context which reserves shadow memory
             ShadowContext = Context;
-            context.urDdiTable.Context.pfnRetain(Context);
+            getContext()->urDdiTable.Context.pfnRetain(Context);
         }
         return Result;
     }();
@@ -154,9 +154,9 @@ ur_result_t DestroyShadowMemory() {
         if (!ShadowContext) {
             return UR_RESULT_SUCCESS;
         }
-        auto Result = context.urDdiTable.VirtualMem.pfnFree(
+        auto Result = getContext()->urDdiTable.VirtualMem.pfnFree(
             ShadowContext, (const void *)LOW_SHADOW_BEGIN, SHADOW_SIZE);
-        context.urDdiTable.Context.pfnRelease(ShadowContext);
+        getContext()->urDdiTable.Context.pfnRelease(ShadowContext);
         return Result;
     }();
     return Result;

source/loader/layers/sanitizer/ur_sanitizer_utils.cpp

@@ -129,9 +129,9 @@ DeviceType GetDeviceType(ur_context_handle_t Context,
     case UR_DEVICE_TYPE_GPU: {
         uptr Ptr;
         [[maybe_unused]] ur_result_t Result =
-            context.urDdiTable.USM.pfnDeviceAlloc(Context, Device, nullptr,
-                                                  nullptr, 4, (void **)&Ptr);
-        context.logger.debug("GetDeviceType: {}", (void *)Ptr);
+            getContext()->urDdiTable.USM.pfnDeviceAlloc(
+                Context, Device, nullptr, nullptr, 4, (void **)&Ptr);
+        getContext()->logger.debug("GetDeviceType: {}", (void *)Ptr);
         assert(Result == UR_RESULT_SUCCESS &&
                "getDeviceType() failed at allocating device USM");
         // FIXME: There's no API querying the address bits of device, so we guess it by the
@@ -142,7 +142,7 @@ DeviceType GetDeviceType(ur_context_handle_t Context,
         } else {
             Type = DeviceType::GPU_DG2;
         }
-        Result = context.urDdiTable.USM.pfnFree(Context, (void *)Ptr);
+        Result = getContext()->urDdiTable.USM.pfnFree(Context, (void *)Ptr);
         assert(Result == UR_RESULT_SUCCESS &&
                "getDeviceType() failed at releasing device USM");
         return Type;