Skip to content

Commit 6550dca

Browse files
extend command list unit test
Signed-off-by: Zbigniew Zdanowicz <[email protected]>
1 parent 8924024 commit 6550dca

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

level_zero/core/test/unit_tests/sources/cmdlist/test_cmdlist_append_launch_kernel_3.cpp

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -745,14 +745,21 @@ HWTEST2_F(MultiTileImmediateCommandListAppendLaunchFunctionXeHpCoreTest, givenIm
745745
auto result = immediateCmdList->initialize(device, NEO::EngineGroupType::Compute, 0u);
746746
ASSERT_EQ(ZE_RESULT_SUCCESS, result);
747747

748+
CmdListKernelLaunchParams launchParams = {};
749+
result = immediateCmdList->appendLaunchKernelWithParams(kernel.get(), &groupCount, nullptr, launchParams);
750+
ASSERT_EQ(ZE_RESULT_SUCCESS, result);
751+
748752
auto cmdStream = immediateCmdList->commandContainer.getCommandStream();
749753

750754
auto sizeBefore = cmdStream->getUsed();
751-
CmdListKernelLaunchParams launchParams = {};
752755
result = immediateCmdList->appendLaunchKernelWithParams(kernel.get(), &groupCount, nullptr, launchParams);
753756
ASSERT_EQ(ZE_RESULT_SUCCESS, result);
754757
auto sizeAfter = cmdStream->getUsed();
755758

759+
uint64_t bbStartGpuAddress = cmdStream->getGraphicsAllocation()->getGpuAddress() + sizeBefore;
760+
bbStartGpuAddress += sizeof(WALKER_TYPE) + sizeof(PIPE_CONTROL) + sizeof(MI_ATOMIC) + sizeof(MI_SEMAPHORE_WAIT) +
761+
sizeof(MI_BATCH_BUFFER_START) + 3 * sizeof(uint32_t);
762+
756763
GenCmdList cmdList;
757764
ASSERT_TRUE(FamilyType::PARSE::parseCommandBuffer(
758765
cmdList,
@@ -771,6 +778,8 @@ HWTEST2_F(MultiTileImmediateCommandListAppendLaunchFunctionXeHpCoreTest, givenIm
771778
EXPECT_EQ(itorPipeControl, itorStoreDataImm);
772779

773780
auto itorBbStart = find<MI_BATCH_BUFFER_START *>(itorPipeControl, cmdList.end());
781+
auto cmdBbStart = genCmdCast<MI_BATCH_BUFFER_START *>(*itorBbStart);
782+
EXPECT_EQ(bbStartGpuAddress, cmdBbStart->getBatchBufferStartAddress());
774783
ASSERT_NE(cmdList.end(), itorBbStart);
775784

776785
auto itorMiAtomic = find<MI_ATOMIC *>(itorBbStart, cmdList.end());

0 commit comments

Comments
 (0)