131 lines
4.0 KiB
C++
131 lines
4.0 KiB
C++
/*
|
|
* Copyright (C) 2017-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "runtime/event/async_events_handler.h"
|
|
#include "runtime/event/user_event.h"
|
|
#include "runtime/helpers/options.h"
|
|
#include "runtime/platform/platform.h"
|
|
#include "test.h"
|
|
#include "unit_tests/fixtures/scenario_test_fixture.h"
|
|
#include "unit_tests/mocks/mock_command_queue.h"
|
|
#include "unit_tests/mocks/mock_kernel.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace NEO;
|
|
|
|
struct CallbackData {
|
|
cl_kernel kernel;
|
|
cl_command_queue queue;
|
|
bool callbackCalled = false;
|
|
UserEvent *signalCallbackDoneEvent = nullptr;
|
|
};
|
|
|
|
void CL_CALLBACK callback(cl_event event, cl_int status, void *data) {
|
|
CallbackData *callbackData = (CallbackData *)data;
|
|
size_t offset[] = {0, 0, 0};
|
|
size_t gws[] = {1, 1, 1};
|
|
|
|
clEnqueueNDRangeKernel(callbackData->queue, callbackData->kernel, 1, offset, gws, nullptr, 0, nullptr, nullptr);
|
|
clFinish(callbackData->queue);
|
|
|
|
callbackData->callbackCalled = true;
|
|
|
|
if (callbackData->signalCallbackDoneEvent) {
|
|
cl_event callbackEvent = callbackData->signalCallbackDoneEvent;
|
|
clSetUserEventStatus(callbackEvent, CL_COMPLETE);
|
|
// No need to reatin and release this synchronization event
|
|
//clReleaseEvent(callbackEvent);
|
|
}
|
|
}
|
|
|
|
TEST_F(ScenarioTest, givenAsyncHandlerDisabledAndUserEventBlockingEnqueueAndOutputEventWithCallbackWhenUserEventIsSetCompleteThanCallbackIsExecuted) {
|
|
DebugManager.flags.EnableAsyncEventsHandler.set(false);
|
|
|
|
cl_command_queue clCommandQ = nullptr;
|
|
cl_queue_properties properties = 0;
|
|
cl_kernel clKernel = kernel;
|
|
size_t offset[] = {0, 0, 0};
|
|
size_t gws[] = {1, 1, 1};
|
|
|
|
cl_int retVal = CL_SUCCESS;
|
|
cl_int success = CL_SUCCESS;
|
|
|
|
UserEvent *userEvent = new UserEvent(context);
|
|
cl_event eventBlocking = userEvent;
|
|
cl_event eventOut = nullptr;
|
|
|
|
clCommandQ = clCreateCommandQueue(context, devices[0], properties, &retVal);
|
|
|
|
retVal = clEnqueueNDRangeKernel(clCommandQ, clKernel, 1, offset, gws, nullptr, 1, &eventBlocking, &eventOut);
|
|
EXPECT_EQ(success, retVal);
|
|
ASSERT_NE(nullptr, eventOut);
|
|
|
|
CallbackData data;
|
|
data.kernel = clKernel;
|
|
data.queue = clCommandQ;
|
|
data.callbackCalled = false;
|
|
|
|
clSetEventCallback(eventOut, CL_COMPLETE, callback, &data);
|
|
EXPECT_FALSE(data.callbackCalled);
|
|
|
|
clSetUserEventStatus(eventBlocking, CL_COMPLETE);
|
|
userEvent->release();
|
|
|
|
clWaitForEvents(1, &eventOut);
|
|
EXPECT_TRUE(data.callbackCalled);
|
|
|
|
clReleaseEvent(eventOut);
|
|
clReleaseCommandQueue(clCommandQ);
|
|
}
|
|
|
|
TEST_F(ScenarioTest, givenAsyncHandlerEnabledAndUserEventBlockingEnqueueAndOutputEventWithCallbackWhenUserEventIsSetCompleteThanCallbackIsExecuted) {
|
|
DebugManager.flags.EnableAsyncEventsHandler.set(true);
|
|
|
|
cl_command_queue clCommandQ = nullptr;
|
|
cl_queue_properties properties = 0;
|
|
cl_kernel clKernel = kernel;
|
|
size_t offset[] = {0, 0, 0};
|
|
size_t gws[] = {1, 1, 1};
|
|
cl_int retVal = CL_SUCCESS;
|
|
cl_int success = CL_SUCCESS;
|
|
UserEvent *userEvent = new UserEvent(context);
|
|
cl_event eventBlocking = userEvent;
|
|
cl_event eventOut = nullptr;
|
|
|
|
clCommandQ = clCreateCommandQueue(context, devices[0], properties, &retVal);
|
|
retVal = clEnqueueNDRangeKernel(clCommandQ, clKernel, 1, offset, gws, nullptr, 1, &eventBlocking, &eventOut);
|
|
|
|
EXPECT_EQ(success, retVal);
|
|
ASSERT_NE(nullptr, eventOut);
|
|
|
|
CallbackData data;
|
|
data.kernel = clKernel;
|
|
data.queue = clCommandQ;
|
|
data.callbackCalled = false;
|
|
data.signalCallbackDoneEvent = new UserEvent(context);
|
|
cl_event callbackEvent = data.signalCallbackDoneEvent;
|
|
|
|
clSetEventCallback(eventOut, CL_COMPLETE, callback, &data);
|
|
EXPECT_FALSE(data.callbackCalled);
|
|
|
|
clSetUserEventStatus(eventBlocking, CL_COMPLETE);
|
|
userEvent->release();
|
|
|
|
clWaitForEvents(1, &eventOut);
|
|
clWaitForEvents(1, &callbackEvent);
|
|
|
|
EXPECT_TRUE(data.callbackCalled);
|
|
|
|
data.signalCallbackDoneEvent->release();
|
|
|
|
clReleaseEvent(eventOut);
|
|
clReleaseCommandQueue(clCommandQ);
|
|
|
|
platform()->getAsyncEventsHandler()->closeThread();
|
|
}
|