/* * Copyright (C) 2017-2018 Intel Corporation * * SPDX-License-Identifier: MIT * */ #include "unit_tests/fixtures/device_fixture.h" #include "unit_tests/fixtures/device_host_queue_fixture.h" #include "unit_tests/mocks/mock_kernel.h" #include "unit_tests/mocks/mock_program.h" #include "unit_tests/mocks/mock_buffer.h" using namespace OCLRT; using namespace DeviceHostQueue; struct KernelArgDevQueueTest : public DeviceFixture, public DeviceHostQueueFixture { protected: void SetUp() override { DeviceFixture::SetUp(); DeviceHostQueueFixture::SetUp(); pDeviceQueue = createQueueObject(); pKernelInfo = std::make_unique(); pKernelInfo->kernelArgInfo.resize(1); pKernelInfo->kernelArgInfo[0].isDeviceQueue = true; kernelArgPatchInfo.crossthreadOffset = 0x4; kernelArgPatchInfo.size = 0x4; kernelArgPatchInfo.sourceOffset = 0; pKernelInfo->kernelArgInfo[0].kernelArgPatchInfoVector.push_back(kernelArgPatchInfo); program = std::make_unique(*pDevice->getExecutionEnvironment()); pKernel = new MockKernel(program.get(), *pKernelInfo, *pDevice); ASSERT_EQ(CL_SUCCESS, pKernel->initialize()); uint8_t pCrossThreadData[crossThreadDataSize]; memset(pCrossThreadData, crossThreadDataInit, sizeof(pCrossThreadData)); pKernel->setCrossThreadData(pCrossThreadData, sizeof(pCrossThreadData)); } void TearDown() override { delete pKernel; delete pDeviceQueue; DeviceHostQueueFixture::TearDown(); DeviceFixture::TearDown(); } bool crossThreadDataUnchanged() { for (uint32_t i = 0; i < crossThreadDataSize; i++) { if (pKernel->mockCrossThreadData[i] != crossThreadDataInit) { return false; } } return true; } static const uint32_t crossThreadDataSize = 0x10; static const char crossThreadDataInit = 0x7e; std::unique_ptr program; DeviceQueue *pDeviceQueue = nullptr; MockKernel *pKernel = nullptr; std::unique_ptr pKernelInfo; KernelArgPatchInfo kernelArgPatchInfo; }; TEST_F(KernelArgDevQueueTest, GIVENkernelWithDevQueueArgWHENsetArgHandleTHENsetsProperHandle) { EXPECT_EQ(pKernel->kernelArgHandlers[0], &Kernel::setArgDevQueue); } TEST_F(KernelArgDevQueueTest, GIVENdevQueueArgHandlerWHENpassDevQueueTHENacceptObjAndPatch) { auto clDeviceQueue = static_cast(pDeviceQueue); auto ret = pKernel->setArgDevQueue(0, sizeof(cl_command_queue), &clDeviceQueue); EXPECT_EQ(ret, CL_SUCCESS); auto gpuAddress = static_cast(pDeviceQueue->getQueueBuffer()->getGpuAddressToPatch()); auto patchLocation = ptrOffset(pKernel->mockCrossThreadData.data(), kernelArgPatchInfo.crossthreadOffset); EXPECT_EQ(*(reinterpret_cast(patchLocation)), gpuAddress); } TEST_F(KernelArgDevQueueTest, GIVENdevQueueArgHandlerWHENpassNormalQueueTHENrejectObjAndReturnError) { auto clCmdQueue = static_cast(pCommandQueue); auto ret = pKernel->setArgDevQueue(0, sizeof(cl_command_queue), &clCmdQueue); EXPECT_EQ(ret, CL_INVALID_DEVICE_QUEUE); EXPECT_EQ(crossThreadDataUnchanged(), true); } TEST_F(KernelArgDevQueueTest, GIVENdevQueueArgHandlerWHENpassNonQueueObjTHENrejectObjAndReturnError) { Buffer *buffer = new MockBuffer(); auto clBuffer = static_cast(buffer); auto ret = pKernel->setArgDevQueue(0, sizeof(cl_command_queue), &clBuffer); EXPECT_EQ(ret, CL_INVALID_DEVICE_QUEUE); EXPECT_EQ(crossThreadDataUnchanged(), true); delete buffer; } TEST_F(KernelArgDevQueueTest, GIVENdevQueueArgHandlerWHENpassFakeQueueTHENrejectObjAndReturnError) { char *pFakeDeviceQueue = new char[sizeof(DeviceQueue)]; auto clFakeDeviceQueue = reinterpret_cast(pFakeDeviceQueue); auto ret = pKernel->setArgDevQueue(0, sizeof(cl_command_queue), &clFakeDeviceQueue); EXPECT_EQ(ret, CL_INVALID_DEVICE_QUEUE); EXPECT_EQ(crossThreadDataUnchanged(), true); delete[] pFakeDeviceQueue; } TEST_F(KernelArgDevQueueTest, GIVENdevQueueArgHandlerWHENpassNullptrTHENrejectObjAndReturnError) { auto ret = pKernel->setArgDevQueue(0, sizeof(cl_command_queue), nullptr); EXPECT_EQ(ret, CL_INVALID_ARG_VALUE); EXPECT_EQ(crossThreadDataUnchanged(), true); } TEST_F(KernelArgDevQueueTest, GIVENdevQueueArgHandlerWHENpassWrongSizeTHENrejectObjAndReturnError) { auto clDeviceQueue = static_cast(pDeviceQueue); auto ret = pKernel->setArgDevQueue(0, sizeof(cl_command_queue) - 1, &clDeviceQueue); EXPECT_EQ(ret, CL_INVALID_ARG_SIZE); EXPECT_EQ(crossThreadDataUnchanged(), true); }