@@ -434,18 +434,6 @@ ur_result_t MsanInterceptor::prepareLaunch(
434
434
ur_kernel_handle_t Kernel, USMLaunchInfo &LaunchInfo) {
435
435
auto Program = GetProgram (Kernel);
436
436
437
- auto EnqueueWriteGlobal = [&Queue, &Program](const char *Name,
438
- const void *Value, size_t Size ) {
439
- auto Result = getContext ()->urDdiTable .Enqueue .pfnDeviceGlobalVariableWrite (
440
- Queue, Program, Name, false , Size , 0 , Value, 0 , nullptr , nullptr );
441
- if (Result != UR_RESULT_SUCCESS) {
442
- getContext ()->logger .error (" Failed to write device global \" {}\" : {}" ,
443
- Name, Result);
444
- return Result;
445
- }
446
- return UR_RESULT_SUCCESS;
447
- };
448
-
449
437
// Set membuffer arguments
450
438
auto &KernelInfo = getOrCreateKernelInfo (Kernel);
451
439
getContext ()->logger .info (" KernelInfo {} (Name=<{}>, IsInstrumented={}, "
@@ -578,8 +566,11 @@ ur_result_t MsanInterceptor::prepareLaunch(
578
566
(void *)LaunchInfo.Data .Host .LocalArgs , LaunchInfo.Data .Host .NumLocalArgs ,
579
567
ToString (LaunchInfo.Data .Host .DeviceTy ), LaunchInfo.Data .Host .Debug );
580
568
581
- ur_result_t URes = EnqueueWriteGlobal (
582
- " __MsanLaunchInfo" , &LaunchInfo.Data .DevicePtr , sizeof (uptr));
569
+ ur_result_t URes =
570
+ getContext ()->urDdiTable .Enqueue .pfnDeviceGlobalVariableWrite (
571
+ Queue, Program, " __MsanLaunchInfo" , false ,
572
+ sizeof (LaunchInfo.Data .DevicePtr ), 0 , &LaunchInfo.Data .DevicePtr , 0 ,
573
+ nullptr , nullptr );
583
574
if (URes != UR_RESULT_SUCCESS) {
584
575
getContext ()->logger .info (" EnqueueWriteGlobal(__MsanLaunchInfo) "
585
576
" failed, maybe empty kernel: {}" ,
0 commit comments