intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-12-30 09:58:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add ults for kernel with printf buffer

Browse Source 
Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>
This commit is contained in:
Mateusz Hoppe
2021-03-31 15:20:22 +00:00
committed by Compute-Runtime-Automation
parent 68c0df32bc
commit dc9b856628
1 changed files with 73 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

73
level_zero/core/test/unit_tests/sources/kernel/test_kernel.cpp
View File

@@ -1505,5 +1505,78 @@ TEST_F(KernelPrintHandlerTest, whenPrintPrintfOutputIsCalledThenPrintfBufferIsUs
EXPECT_EQ(buffer, MyPrintfHandler::getPrintfSurfaceInitialDataSize());
}
using PrintfTest = Test<DeviceFixture>;
TEST_F(PrintfTest, givenKernelWithPrintfThenPrintfBufferIsCreated) {
Mock<Module> mockModule(this->device, nullptr);
Mock<Kernel> mockKernel;
mockKernel.descriptor.kernelAttributes.flags.usesPrintf = true;
mockKernel.module = &mockModule;
EXPECT_TRUE(mockKernel.getImmutableData()->getDescriptor().kernelAttributes.flags.usesPrintf);
ze_kernel_desc_t kernelDesc = {};
kernelDesc.pKernelName = "mock";
mockKernel.createPrintfBuffer();
EXPECT_NE(nullptr, mockKernel.getPrintfBufferAllocation());
}
TEST_F(PrintfTest, GivenKernelNotUsingPrintfWhenCreatingPrintfBufferThenAllocationIsNotCreated) {
Mock<Module> mockModule(this->device, nullptr);
Mock<Kernel> mockKernel;
mockKernel.descriptor.kernelAttributes.flags.usesPrintf = false;
mockKernel.module = &mockModule;
ze_kernel_desc_t kernelDesc = {};
kernelDesc.pKernelName = "mock";
mockKernel.createPrintfBuffer();
EXPECT_EQ(nullptr, mockKernel.getPrintfBufferAllocation());
}
TEST_F(PrintfTest, WhenCreatingPrintfBufferThenAllocationAddedToResidencyContainer) {
Mock<Module> mockModule(this->device, nullptr);
Mock<Kernel> mockKernel;
mockKernel.descriptor.kernelAttributes.flags.usesPrintf = true;
mockKernel.module = &mockModule;
ze_kernel_desc_t kernelDesc = {};
kernelDesc.pKernelName = "mock";
mockKernel.createPrintfBuffer();
auto printfBufferAllocation = mockKernel.getPrintfBufferAllocation();
EXPECT_NE(nullptr, printfBufferAllocation);
EXPECT_NE(0u, mockKernel.residencyContainer.size());
EXPECT_EQ(mockKernel.residencyContainer[mockKernel.residencyContainer.size() - 1], printfBufferAllocation);
}
TEST_F(PrintfTest, WhenCreatingPrintfBufferThenCrossThreadDataIsPatched) {
Mock<Module> mockModule(this->device, nullptr);
Mock<Kernel> mockKernel;
mockKernel.descriptor.kernelAttributes.flags.usesPrintf = true;
mockKernel.module = &mockModule;
ze_kernel_desc_t kernelDesc = {};
kernelDesc.pKernelName = "mock";
auto crossThreadData = std::make_unique<uint32_t[]>(4);
mockKernel.descriptor.payloadMappings.implicitArgs.printfSurfaceAddress.stateless = 0;
mockKernel.descriptor.payloadMappings.implicitArgs.printfSurfaceAddress.pointerSize = sizeof(uintptr_t);
mockKernel.crossThreadData.reset(reinterpret_cast<uint8_t *>(crossThreadData.get()));
mockKernel.crossThreadDataSize = sizeof(uint32_t[4]);
mockKernel.createPrintfBuffer();
auto printfBufferAllocation = mockKernel.getPrintfBufferAllocation();
EXPECT_NE(nullptr, printfBufferAllocation);
auto printfBufferAddressPatched = *reinterpret_cast<uintptr_t *>(crossThreadData.get());
auto printfBufferGpuAddressOffset = static_cast<uintptr_t>(printfBufferAllocation->getGpuAddressToPatch());
EXPECT_EQ(printfBufferGpuAddressOffset, printfBufferAddressPatched);
mockKernel.crossThreadData.release();
}
} // namespace ult
} // namespace L0