@@ -2258,6 +2258,8 @@ piEnqueueKernelLaunch(pi_queue Queue, pi_kernel Kernel, pi_uint32 WorkDim,
2258
2258
2259
2259
ze_event_handle_t *ZeEventWaitList =
2260
2260
_pi_event::createZeEventList (NumEventsInWaitList, EventWaitList);
2261
+ if (!ZeEventWaitList)
2262
+ return PI_OUT_OF_HOST_MEMORY;
2261
2263
2262
2264
// Add the command to the command list
2263
2265
ZE_CALL (zeCommandListAppendLaunchKernel (
@@ -2708,6 +2710,8 @@ enqueueMemCopyHelper(pi_command_type CommandType, pi_queue Queue, void *Dst,
2708
2710
2709
2711
ze_event_handle_t *ZeEventWaitList =
2710
2712
_pi_event::createZeEventList (NumEventsInWaitList, EventWaitList);
2713
+ if (!ZeEventWaitList)
2714
+ return PI_OUT_OF_HOST_MEMORY;
2711
2715
2712
2716
ZE_CALL (zeCommandListAppendWaitOnEvents (ZeCommandList, NumEventsInWaitList,
2713
2717
ZeEventWaitList));
@@ -2766,6 +2770,8 @@ static pi_result enqueueMemCopyRectHelper(
2766
2770
2767
2771
ze_event_handle_t *ZeEventWaitList =
2768
2772
_pi_event::createZeEventList (NumEventsInWaitList, EventWaitList);
2773
+ if (!ZeEventWaitList)
2774
+ return PI_OUT_OF_HOST_MEMORY;
2769
2775
2770
2776
ZE_CALL (zeCommandListAppendWaitOnEvents (ZeCommandList, NumEventsInWaitList,
2771
2777
ZeEventWaitList));
@@ -2926,6 +2932,8 @@ enqueueMemFillHelper(pi_command_type CommandType, pi_queue Queue, void *Ptr,
2926
2932
2927
2933
ze_event_handle_t *ZeEventWaitList =
2928
2934
_pi_event::createZeEventList (NumEventsInWaitList, EventWaitList);
2935
+ if (!ZeEventWaitList)
2936
+ return PI_OUT_OF_HOST_MEMORY;
2929
2937
2930
2938
ZE_CALL (zeCommandListAppendWaitOnEvents (ZeCommandList, NumEventsInWaitList,
2931
2939
ZeEventWaitList));
0 commit comments