test: when heapless then rt buf is created in prolog so manage it carefully

Related-to: NEO-13973

Signed-off-by: Damian Tomczak <damian.tomczak@intel.com>
This commit is contained in:
Damian Tomczak
2025-03-13 21:13:31 +00:00
committed by Compute-Runtime-Automation
parent 948b423ab8
commit e4ab4d2ab7

View File

@@ -1599,6 +1599,8 @@ using RayTracingMatcher = IsAtLeastXeHpCore;
using CommandListAppendLaunchRayTracingKernelTest = Test<CommandListAppendLaunchRayTracingKernelFixture>;
HWTEST2_F(CommandListAppendLaunchRayTracingKernelTest, givenKernelUsingRayTracingWhenAppendLaunchKernelIsCalledThenSuccessIsReturned, RayTracingMatcher) {
VariableBackup<GraphicsAllocation *> rtMemoryBackedBuffer{&neoDevice->rtMemoryBackedBuffer};
Mock<::L0::KernelImp> kernel;
auto pMockModule = std::unique_ptr<Module>(new Mock<Module>(device, nullptr));
kernel.module = pMockModule.get();
@@ -1611,7 +1613,7 @@ HWTEST2_F(CommandListAppendLaunchRayTracingKernelTest, givenKernelUsingRayTracin
kernel.immutableData.kernelDescriptor->kernelAttributes.flags.hasRTCalls = true;
neoDevice->rtMemoryBackedBuffer = nullptr;
rtMemoryBackedBuffer = nullptr;
CmdListKernelLaunchParams launchParams = {};
result = pCommandList->appendLaunchKernel(kernel.toHandle(), groupCount, nullptr, 0, nullptr, launchParams, false);
EXPECT_NE(ZE_RESULT_SUCCESS, result);
@@ -1620,10 +1622,12 @@ HWTEST2_F(CommandListAppendLaunchRayTracingKernelTest, givenKernelUsingRayTracin
result = pCommandList->appendLaunchKernel(kernel.toHandle(), groupCount, nullptr, 0, nullptr, launchParams, false);
EXPECT_EQ(ZE_RESULT_SUCCESS, result);
neoDevice->rtMemoryBackedBuffer = nullptr;
rtMemoryBackedBuffer = buffer1;
}
HWTEST2_F(CommandListAppendLaunchRayTracingKernelTest, givenDcFlushMitigationWhenAppendLaunchKernelWithRayTracingIsCalledThenRequireDcFlush, RayTracingMatcher) {
VariableBackup<GraphicsAllocation *> rtMemoryBackedBuffer{&neoDevice->rtMemoryBackedBuffer, buffer1};
DebugManagerStateRestore restorer;
debugManager.flags.AllowDcFlush.set(0);
@@ -1638,14 +1642,11 @@ HWTEST2_F(CommandListAppendLaunchRayTracingKernelTest, givenDcFlushMitigationWhe
ASSERT_EQ(ZE_RESULT_SUCCESS, result);
kernel.immutableData.kernelDescriptor->kernelAttributes.flags.hasRTCalls = true;
neoDevice->rtMemoryBackedBuffer = buffer1;
CmdListKernelLaunchParams launchParams = {};
result = pCommandList->appendLaunchKernel(kernel.toHandle(), groupCount, nullptr, 0, nullptr, launchParams, false);
EXPECT_EQ(ZE_RESULT_SUCCESS, result);
EXPECT_EQ(pCommandList->requiresDcFlushForDcMitigation, device->getProductHelper().isDcFlushMitigated());
neoDevice->rtMemoryBackedBuffer = nullptr;
}
using RayTracingCmdListTest = Test<RayTracingCmdListFixture>;