mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-31 20:13:04 +08:00
5d01677454
Functions: fileExists and loadDataToFile use IO functions from namespace IoFunctions Now tests that use these functions are mocked by default, but some still require access to real files and have been restored the ability to read files. They will be mocked in next PRs. Related-To: NEO-7006 Signed-off-by: Marcel Skierkowski <marcel.skierkowski@intel.com>
99 lines
3.6 KiB
C++
99 lines
3.6 KiB
C++
/*
|
|
* Copyright (C) 2018-2025 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "shared/source/helpers/file_io.h"
|
|
#include "shared/test/common/mocks/mock_device.h"
|
|
|
|
#include "opencl/test/unit_test/mocks/mock_cl_device.h"
|
|
#include "opencl/test/unit_test/mocks/mock_context.h"
|
|
#include "opencl/test/unit_test/mocks/mock_kernel.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace NEO;
|
|
|
|
class PatchedKernelTest : public ::testing::Test {
|
|
public:
|
|
void SetUp() override {
|
|
USE_REAL_FILE_SYSTEM();
|
|
device = std::make_unique<MockClDevice>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(defaultHwInfo.get(), rootDeviceIndex));
|
|
context.reset(new MockContext(device.get()));
|
|
program.reset(Program::createBuiltInFromSource<MockProgram>("FillBufferBytes", context.get(), context->getDevices(), &retVal));
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
program->build(program->getDevices(), nullptr);
|
|
kernel.reset(Kernel::create(program.get(), program->getKernelInfoForKernel("FillBufferBytes"), *device, retVal));
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
}
|
|
void TearDown() override {
|
|
context.reset();
|
|
}
|
|
|
|
const uint32_t rootDeviceIndex = 0u;
|
|
std::unique_ptr<MockContext> context;
|
|
std::unique_ptr<MockClDevice> device;
|
|
std::unique_ptr<MockProgram> program;
|
|
std::unique_ptr<Kernel> kernel;
|
|
cl_int retVal = CL_SUCCESS;
|
|
};
|
|
|
|
TEST_F(PatchedKernelTest, givenKernelWithoutPatchedArgsWhenIsPatchedIsCalledThenReturnsFalse) {
|
|
EXPECT_FALSE(kernel->isPatched());
|
|
}
|
|
|
|
TEST_F(PatchedKernelTest, givenKernelWithAllArgsSetWithBufferWhenIsPatchedIsCalledThenReturnsTrue) {
|
|
auto buffer = clCreateBuffer(context.get(), CL_MEM_READ_ONLY, sizeof(int), nullptr, &retVal);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
auto argsNum = kernel->getKernelArgsNumber();
|
|
for (uint32_t i = 0; i < argsNum; i++) {
|
|
kernel->setArg(i, buffer);
|
|
}
|
|
EXPECT_TRUE(kernel->isPatched());
|
|
clReleaseMemObject(buffer);
|
|
}
|
|
|
|
TEST_F(PatchedKernelTest, givenKernelWithoutAllArgsSetWhenIsPatchedIsCalledThenReturnsFalse) {
|
|
auto buffer = clCreateBuffer(context.get(), CL_MEM_READ_ONLY, sizeof(int), nullptr, &retVal);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
auto argsNum = kernel->getKernelArgsNumber();
|
|
for (uint32_t i = 0; i < argsNum; i++) {
|
|
kernel->setArg(0, buffer);
|
|
}
|
|
EXPECT_FALSE(kernel->isPatched());
|
|
clReleaseMemObject(buffer);
|
|
}
|
|
|
|
TEST_F(PatchedKernelTest, givenArgSvmAllocWhenArgIsSetThenArgIsPatched) {
|
|
const ClDeviceInfo &devInfo = device->getDeviceInfo();
|
|
if (devInfo.svmCapabilities == 0) {
|
|
GTEST_SKIP();
|
|
}
|
|
EXPECT_FALSE(kernel->getKernelArguments()[0].isPatched);
|
|
kernel->setArgSvmAlloc(0, nullptr, nullptr, 0u);
|
|
EXPECT_TRUE(kernel->getKernelArguments()[0].isPatched);
|
|
}
|
|
|
|
TEST_F(PatchedKernelTest, givenArgSvmWhenArgIsSetThenArgIsPatched) {
|
|
uint32_t size = sizeof(int);
|
|
EXPECT_FALSE(kernel->getKernelArguments()[0].isPatched);
|
|
kernel->setArgSvm(0, size, nullptr, nullptr, 0);
|
|
EXPECT_TRUE(kernel->getKernelArguments()[0].isPatched);
|
|
}
|
|
|
|
TEST_F(PatchedKernelTest, givenKernelWithOneArgumentToPatchWhichIsNonzeroIndexedWhenThatArgumentIsSetThenKernelIsPatched) {
|
|
uint32_t size = sizeof(int);
|
|
MockKernelWithInternals mockKernel(*device.get(), context.get());
|
|
mockKernel.kernelInfo.kernelDescriptor.kernelAttributes.numArgsToPatch = 1;
|
|
mockKernel.kernelInfo.addArgBuffer(1, 0);
|
|
|
|
kernel.reset(mockKernel.mockKernel);
|
|
kernel->initialize();
|
|
EXPECT_FALSE(kernel->Kernel::isPatched());
|
|
kernel->setArgSvm(1, size, nullptr, nullptr, 0u);
|
|
EXPECT_TRUE(kernel->Kernel::isPatched());
|
|
kernel.release();
|
|
}
|