Skip to content

Commit 453b8c2

Browse files
committed
Re-use MetadataPtr
1 parent 16ec6a6 commit 453b8c2

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

source/loader/layers/sanitizer/asan_interceptor.cpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -522,8 +522,8 @@ SanitizerInterceptor::registerDeviceGlobals(ur_context_handle_t Context,
522522
for (auto Device : Devices) {
523523
ManagedQueue Queue(Context, Device);
524524

525-
[[maybe_unused]] size_t MetadataSize;
526-
[[maybe_unused]] void *MetadataPtr;
525+
size_t MetadataSize;
526+
void *MetadataPtr;
527527
auto Result = context.urDdiTable.Program.pfnGetGlobalVariablePointer(
528528
Device, Program, kSPIR_AsanDeviceGlobalMetadata, &MetadataSize,
529529
&MetadataPtr);
@@ -535,10 +535,9 @@ SanitizerInterceptor::registerDeviceGlobals(ur_context_handle_t Context,
535535
const uint64_t NumOfDeviceGlobal =
536536
MetadataSize / sizeof(DeviceGlobalInfo);
537537
std::vector<DeviceGlobalInfo> GVInfos(NumOfDeviceGlobal);
538-
Result = context.urDdiTable.Enqueue.pfnDeviceGlobalVariableRead(
539-
Queue, Program, kSPIR_AsanDeviceGlobalMetadata, true,
540-
sizeof(DeviceGlobalInfo) * NumOfDeviceGlobal, 0, &GVInfos[0], 0,
541-
nullptr, nullptr);
538+
Result = context.urDdiTable.Enqueue.pfnUSMMemcpy(
539+
Queue, true, &GVInfos[0], MetadataPtr,
540+
sizeof(DeviceGlobalInfo) * NumOfDeviceGlobal, 0, nullptr, nullptr);
542541
if (Result != UR_RESULT_SUCCESS) {
543542
context.logger.error("Device Global[{}] Read Failed: {}",
544543
kSPIR_AsanDeviceGlobalMetadata, Result);

0 commit comments

Comments
 (0)