test: extract duplicated code to functions

Signed-off-by: Naklicki, Mateusz <mateusz.naklicki@intel.com>
This commit is contained in:
Naklicki, Mateusz
2025-08-22 15:36:43 +00:00
committed by Compute-Runtime-Automation
parent 31bcea128f
commit 71238d4124
2 changed files with 35 additions and 42 deletions

View File

@@ -25,51 +25,13 @@ void MutableCommandListFixtureInit::setUp(bool createInOrder) {
mutableCommandList = createMutableCmdList();
mockKernelImmData2 = std::make_unique<MockImmutableData>(0u);
mockKernelImmData2->kernelDescriptor->kernelAttributes.crossThreadDataSize = crossThreadInitSize;
mockKernelImmData2->crossThreadDataSize = crossThreadInitSize;
mockKernelImmData2->crossThreadDataTemplate.reset(new uint8_t[crossThreadInitSize]);
mockKernelImmData2->kernelDescriptor->payloadMappings.implicitArgs.indirectDataPointerAddress.offset = 0;
mockKernelImmData2->kernelDescriptor->payloadMappings.implicitArgs.indirectDataPointerAddress.pointerSize = sizeof(void *);
mockKernelImmData2->kernelDescriptor->payloadMappings.implicitArgs.scratchPointerAddress.offset = 8;
mockKernelImmData2->kernelDescriptor->payloadMappings.implicitArgs.scratchPointerAddress.pointerSize = sizeof(void *);
{
std::initializer_list<ZebinTestData::AppendElfAdditionalSection> additionalSections = {};
zebinData2 = std::make_unique<ZebinTestData::ZebinWithL0TestCommonModule>(device->getHwInfo(), additionalSections);
const auto &src = zebinData2->storage;
ze_module_desc_t moduleDesc = {};
moduleDesc.format = ZE_MODULE_FORMAT_NATIVE;
moduleDesc.pInputModule = reinterpret_cast<const uint8_t *>(src.data());
moduleDesc.inputSize = src.size();
ModuleBuildLog *moduleBuildLog = nullptr;
module2 = std::make_unique<MockModule>(device,
moduleBuildLog,
ModuleType::user,
0,
mockKernelImmData2.get());
module2->type = ModuleType::user;
ze_result_t result = ZE_RESULT_ERROR_MODULE_BUILD_FAILURE;
result = module2->initialize(&moduleDesc, device->getNEODevice());
EXPECT_EQ(ZE_RESULT_SUCCESS, result);
}
mockKernelImmData2 = prepareKernelImmData();
module2 = prepareModule(mockKernelImmData2.get());
kernel2 = std::make_unique<ModuleImmutableDataFixture::MockKernel>(module2.get());
createKernel(kernel2.get());
module2->mockKernelImmData->kernelDescriptor->kernelMetadata.kernelName = "test2";
mockKernelImmData = std::make_unique<MockImmutableData>(0u);
mockKernelImmData->kernelDescriptor->kernelAttributes.crossThreadDataSize = crossThreadInitSize;
mockKernelImmData->crossThreadDataSize = crossThreadInitSize;
mockKernelImmData->crossThreadDataTemplate.reset(new uint8_t[crossThreadInitSize]);
mockKernelImmData->kernelDescriptor->payloadMappings.implicitArgs.indirectDataPointerAddress.offset = 0;
mockKernelImmData->kernelDescriptor->payloadMappings.implicitArgs.indirectDataPointerAddress.pointerSize = sizeof(void *);
mockKernelImmData->kernelDescriptor->payloadMappings.implicitArgs.scratchPointerAddress.offset = 8;
mockKernelImmData->kernelDescriptor->payloadMappings.implicitArgs.scratchPointerAddress.pointerSize = sizeof(void *);
mockKernelImmData = prepareKernelImmData();
createModuleFromMockBinary(0u, false, mockKernelImmData.get());
kernel = std::make_unique<ModuleImmutableDataFixture::MockKernel>(module.get());
createKernel(kernel.get());
@@ -107,6 +69,36 @@ void MutableCommandListFixtureInit::tearDown() {
ModuleImmutableDataFixture::tearDown();
}
std::unique_ptr<ModuleImmutableDataFixture::MockImmutableData> MutableCommandListFixtureInit::prepareKernelImmData() {
auto immData = std::make_unique<MockImmutableData>(0u);
immData->kernelDescriptor->kernelAttributes.crossThreadDataSize = crossThreadInitSize;
immData->crossThreadDataSize = crossThreadInitSize;
immData->crossThreadDataTemplate.reset(new uint8_t[crossThreadInitSize]);
immData->kernelDescriptor->payloadMappings.implicitArgs.indirectDataPointerAddress.offset = 0;
immData->kernelDescriptor->payloadMappings.implicitArgs.indirectDataPointerAddress.pointerSize = sizeof(void *);
immData->kernelDescriptor->payloadMappings.implicitArgs.scratchPointerAddress.offset = 8;
immData->kernelDescriptor->payloadMappings.implicitArgs.scratchPointerAddress.pointerSize = sizeof(void *);
return immData;
}
std::unique_ptr<ModuleImmutableDataFixture::MockModule> MutableCommandListFixtureInit::prepareModule(MockImmutableData *immData) {
std::initializer_list<ZebinTestData::AppendElfAdditionalSection> additionalSections = {};
auto zebinDataOut = std::make_unique<ZebinTestData::ZebinWithL0TestCommonModule>(device->getHwInfo(), additionalSections);
const auto &src = zebinDataOut->storage;
ze_module_desc_t moduleDesc = {};
moduleDesc.format = ZE_MODULE_FORMAT_NATIVE;
moduleDesc.pInputModule = reinterpret_cast<const uint8_t *>(src.data());
moduleDesc.inputSize = src.size();
ModuleBuildLog *moduleBuildLog = nullptr;
auto mod = std::make_unique<ModuleImmutableDataFixture::MockModule>(device, moduleBuildLog, ModuleType::user, 0, immData);
mod->type = ModuleType::user;
ze_result_t result = mod->initialize(&moduleDesc, device->getNEODevice());
EXPECT_EQ(ZE_RESULT_SUCCESS, result);
return mod;
}
std::unique_ptr<MutableCommandList> MutableCommandListFixtureInit::createMutableCmdList() {
ze_result_t returnValue;

View File

@@ -40,6 +40,8 @@ struct MutableCommandListFixtureInit : public ModuleImmutableDataFixture {
void setUp(bool createInOrder);
void tearDown();
std::unique_ptr<MockImmutableData> prepareKernelImmData();
std::unique_ptr<MockModule> prepareModule(MockImmutableData *immData);
std::unique_ptr<MutableCommandList> createMutableCmdList();
Event *createTestEvent(bool cbEvent, bool signalScope, bool timestamp, bool external);
void *allocateUsm(size_t size);
@@ -69,7 +71,6 @@ struct MutableCommandListFixtureInit : public ModuleImmutableDataFixture {
std::unique_ptr<MockKernel> kernel;
std::unique_ptr<MockKernel> kernel2;
std::unique_ptr<MockModule> module2;
std::unique_ptr<ZebinTestData::ZebinWithL0TestCommonModule> zebinData2;
std::unique_ptr<VariableBackup<::NEO::HardwareInfo>> backupHwInfo;
uint64_t commandId = 0;