test: move ocl test with async thread to mt tests

Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2024-06-17 14:03:02 +00:00
committed by Compute-Runtime-Automation
parent 296d2bb148
commit 21a74fcbef
3 changed files with 75 additions and 56 deletions

View File

@ -1,5 +1,5 @@
/*
* Copyright (C) 2018-2023 Intel Corporation
* Copyright (C) 2018-2024 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
@ -15,6 +15,13 @@
using namespace NEO;
struct CallbackData {
cl_kernel kernel;
cl_command_queue queue;
bool callbackCalled = false;
UserEvent *signalCallbackDoneEvent = nullptr;
};
class ScenarioTest : public ::testing::Test,
public PlatformFixture {
using PlatformFixture::setUp;

View File

@ -7,8 +7,10 @@
#include "shared/source/memory_manager/memory_manager.h"
#include "opencl/source/event/async_events_handler.h"
#include "opencl/test/unit_test/event/event_fixture.h"
#include "opencl/test/unit_test/fixtures/buffer_fixture.h"
#include "opencl/test/unit_test/fixtures/scenario_test_fixture.h"
#include "opencl/test/unit_test/mocks/mock_command_queue.h"
#include <memory>
@ -150,3 +152,67 @@ HWTEST_F(EventTests, givenOneThreadUpdatingUserEventAnotherWaitingOnFinishWhenFi
clReleaseEvent(returnedEvent);
}
}
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, givenAsyncHandlerEnabledAndUserEventBlockingEnqueueAndOutputEventWithCallbackWhenUserEventIsSetCompleteThenCallbackIsExecuted) {
debugManager.flags.EnableAsyncEventsHandler.set(true);
cl_command_queue clCommandQ = nullptr;
cl_queue_properties properties = 0;
cl_kernel clKernel = kernelInternals->mockMultiDeviceKernel;
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);
context->getAsyncEventsHandler().closeThread();
}

View File

@ -1,5 +1,5 @@
/*
* Copyright (C) 2018-2023 Intel Corporation
* Copyright (C) 2018-2024 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
@ -8,7 +8,6 @@
#include "shared/test/common/mocks/mock_device.h"
#include "shared/test/common/test_macros/test.h"
#include "opencl/source/event/async_events_handler.h"
#include "opencl/source/event/user_event.h"
#include "opencl/test/unit_test/fixtures/scenario_test_fixture.h"
#include "opencl/test/unit_test/mocks/mock_command_queue.h"
@ -18,13 +17,6 @@
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};
@ -82,49 +74,3 @@ TEST_F(ScenarioTest, givenAsyncHandlerDisabledAndUserEventBlockingEnqueueAndOutp
clReleaseEvent(eventOut);
clReleaseCommandQueue(clCommandQ);
}
TEST_F(ScenarioTest, givenAsyncHandlerEnabledAndUserEventBlockingEnqueueAndOutputEventWithCallbackWhenUserEventIsSetCompleteThenCallbackIsExecuted) {
debugManager.flags.EnableAsyncEventsHandler.set(true);
cl_command_queue clCommandQ = nullptr;
cl_queue_properties properties = 0;
cl_kernel clKernel = kernelInternals->mockMultiDeviceKernel;
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);
context->getAsyncEventsHandler().closeThread();
}