91 lines
2.2 KiB
C++
91 lines
2.2 KiB
C++
/*
|
|
* Copyright (C) 2017-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "runtime/program/kernel_info.h"
|
|
#include "unit_tests/mocks/mock_device.h"
|
|
#include "unit_tests/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;
|
|
pDevice = MockDevice::createWithNewExecutionEnvironment<MockDevice>(nullptr);
|
|
program = std::make_unique<MockProgram>(*pDevice->getExecutionEnvironment());
|
|
}
|
|
|
|
void TearDown() override {
|
|
if (pKernelInfo->kernelAllocation) {
|
|
pDevice->getMemoryManager()->freeGraphicsMemory(pKernelInfo->kernelAllocation);
|
|
const_cast<KernelInfo *>(pKernelInfo)->kernelAllocation = nullptr;
|
|
}
|
|
delete pDevice;
|
|
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;
|
|
MockDevice *pDevice;
|
|
const KernelInfo *pKernelInfo;
|
|
};
|