Files
compute-runtime/opencl/test/unit_test/fixtures/kernel_data_fixture.h
Warchulski, Jaroslaw 3d59dce80c Cleanup includes 27
Cleaned up files:
opencl/source/command_queue/command_queue.h
shared/source/built_ins/registry/built_ins_registry.h
shared/source/kernel/kernel_descriptor.h

Related-To: NEO-5548
Signed-off-by: Warchulski, Jaroslaw <jaroslaw.warchulski@intel.com>
2023-01-11 16:10:28 +01:00

99 lines
2.6 KiB
C++

/*
* Copyright (C) 2018-2023 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/helpers/aligned_memory.h"
#include "shared/source/memory_manager/memory_manager.h"
#include "shared/source/program/kernel_info.h"
#include "shared/test/common/mocks/mock_kernel_info.h"
#include "opencl/source/cl_device/cl_device.h"
#include "opencl/test/unit_test/mocks/mock_context.h"
#include "opencl/test/unit_test/mocks/mock_program.h"
#include "gtest/gtest.h"
using namespace NEO;
using namespace iOpenCL;
class KernelDataTest : public testing::Test {
public:
KernelDataTest() {
memset(&kernelBinaryHeader, 0x00, sizeof(SKernelBinaryHeaderCommon));
pCurPtr = nullptr;
pKernelData = nullptr;
kernelName = "test";
pDsh = nullptr;
pGsh = nullptr;
pKernelHeap = nullptr;
pSsh = nullptr;
pPatchList = nullptr;
kernelDataSize = 0;
kernelNameSize = (uint32_t)alignUp(strlen(kernelName.c_str()) + 1, sizeof(uint32_t));
dshSize = 0;
gshSize = 0;
kernelHeapSize = 0;
sshSize = 0;
patchListSize = 0;
checkSum = 0;
shaderHashCode = 0;
kernelUnpaddedSize = 0;
pKernelInfo = nullptr;
}
void buildAndDecode();
protected:
void SetUp() override {
kernelBinaryHeader.KernelNameSize = kernelNameSize;
pContext = new MockContext;
rootDeviceIndex = pContext->getDevice(0)->getRootDeviceIndex();
program = std::make_unique<MockProgram>(pContext, false, toClDeviceVector(*pContext->getDevice(0)));
}
void TearDown() override {
if (pKernelInfo->kernelAllocation) {
pContext->getDevice(0)->getMemoryManager()->freeGraphicsMemory(pKernelInfo->kernelAllocation);
const_cast<KernelInfo *>(pKernelInfo)->kernelAllocation = nullptr;
}
program.reset();
delete pContext;
alignedFree(pKernelData);
}
char *pCurPtr;
char *pKernelData;
SKernelBinaryHeaderCommon kernelBinaryHeader;
std::string kernelName;
void *pDsh;
void *pGsh;
void *pKernelHeap;
void *pSsh;
void *pPatchList;
uint32_t kernelDataSize;
uint32_t kernelNameSize;
uint32_t dshSize;
uint32_t gshSize;
uint32_t kernelHeapSize;
uint32_t sshSize;
uint32_t patchListSize;
uint32_t checkSum;
uint64_t shaderHashCode;
uint32_t kernelUnpaddedSize;
std::unique_ptr<MockProgram> program;
MockContext *pContext;
const KernelInfo *pKernelInfo;
uint32_t rootDeviceIndex = std::numeric_limits<uint32_t>::max();
};