@@ -526,15 +526,22 @@ ur_result_t ur_platform_handle_t_::initialize() {
526526 ZeMemGetPitchFor2dImageExt.Supported =
527527 ZeMemGetPitchFor2dImageExt.zeMemGetPitchFor2dImage != nullptr ;
528528
529- ZE_CALL_NOCHECK (zeDriverGetExtensionFunctionAddress,
530- (ZeDriver, " zeCommandListAppendLaunchKernelWithArguments" ,
531- reinterpret_cast <void **>(
532- &ZeCommandListAppendLaunchKernelWithArgumentsExt
533- .zeCommandListAppendLaunchKernelWithArguments )));
529+ if (zeCommandListAppendLaunchKernelWithArguments && this ->isDriverVersionNewerOrSimilar (1 , 14 , 35917 )) {
530+ ZeCommandListAppendLaunchKernelWithArgumentsExt
531+ .zeCommandListAppendLaunchKernelWithArgumentsFunctionPtr =
532+ zeCommandListAppendLaunchKernelWithArguments;
533+ ZeCommandListAppendLaunchKernelWithArgumentsExt.Supported = true ;
534+ } else {
535+ ZeCommandListAppendLaunchKernelWithArgumentsExt.Supported = false ;
536+ }
534537
538+ printf (" ur_platform_handle_t_::initialize: "
539+ " zeCommandListAppendLaunchKernelWithArguments ptr = %p\n " ,
540+ ZeCommandListAppendLaunchKernelWithArgumentsExt
541+ .zeCommandListAppendLaunchKernelWithArgumentsFunctionPtr );
535542 ZeCommandListAppendLaunchKernelWithArgumentsExt.Supported =
536543 ZeCommandListAppendLaunchKernelWithArgumentsExt
537- .zeCommandListAppendLaunchKernelWithArguments != nullptr ;
544+ .zeCommandListAppendLaunchKernelWithArgumentsFunctionPtr != nullptr ;
538545
539546 // Check if the driver supports zeCommandListAppendLaunchKernelWithArguments()
540547 // with cooperative mode (version >= 1.6.35005)
0 commit comments