diff --git a/sycl/unittests/SYCL2020/GetNativeOpenCL.cpp b/sycl/unittests/SYCL2020/GetNativeOpenCL.cpp index 151c56d8e9a8a..343cb522faa6d 100644 --- a/sycl/unittests/SYCL2020/GetNativeOpenCL.cpp +++ b/sycl/unittests/SYCL2020/GetNativeOpenCL.cpp @@ -61,6 +61,13 @@ pi_result redefinedEventGetInfo(pi_event event, pi_event_info param_name, return PI_SUCCESS; } +static pi_result redefinedUSMEnqueueMemset(pi_queue, void *, pi_int32, size_t, + pi_uint32, const pi_event *, + pi_event *event) { + *event = reinterpret_cast(new int{}); + return PI_SUCCESS; +} + TEST(GetNative, GetNativeHandle) { platform Plt{default_selector()}; if (Plt.get_backend() != backend::opencl) { @@ -83,6 +90,8 @@ TEST(GetNative, GetNativeHandle) { Mock.redefine(redefinedDeviceRetain); Mock.redefine(redefinedProgramRetain); Mock.redefine(redefinedEventRetain); + Mock.redefine( + redefinedUSMEnqueueMemset); default_selector Selector; context Context(Plt); diff --git a/sycl/unittests/helpers/CommonRedefinitions.hpp b/sycl/unittests/helpers/CommonRedefinitions.hpp index 274425eb6a6ea..60a35f9b71404 100644 --- a/sycl/unittests/helpers/CommonRedefinitions.hpp +++ b/sycl/unittests/helpers/CommonRedefinitions.hpp @@ -106,12 +106,15 @@ inline pi_result redefinedEventsWaitCommon(pi_uint32 num_events, } inline pi_result redefinedEventReleaseCommon(pi_event event) { + if (event != nullptr) + delete reinterpret_cast(event); return PI_SUCCESS; } inline pi_result redefinedEnqueueKernelLaunchCommon( pi_queue, pi_kernel, pi_uint32, const size_t *, const size_t *, - const size_t *, pi_uint32, const pi_event *, pi_event *) { + const size_t *, pi_uint32, const pi_event *, pi_event *event) { + *event = reinterpret_cast(new int{}); return PI_SUCCESS; }