test: add findWalkerTypeCmd function

Related-To: NEO-10641

Signed-off-by: Kamil Kopryk <kamil.kopryk@intel.com>
This commit is contained in:
Kamil Kopryk 2024-06-25 15:47:26 +00:00 committed by Compute-Runtime-Automation
parent 2800282bdb
commit 1702526c40
4 changed files with 17 additions and 6 deletions

View File

@ -1445,7 +1445,7 @@ void CommandListScratchPatchFixtureInit::testScratchInline(bool useImmediate) {
ptrOffset(cmdListStream->getCpuBase(), usedBefore),
usedAfter - usedBefore));
auto walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerCmd(cmdList.begin(), cmdList.end(), true);
auto walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerTypeCmd(cmdList.begin(), cmdList.end());
ASSERT_NE(cmdList.end(), walkerIterator);
void *walkerPtrWithScratch = *walkerIterator;
@ -1462,7 +1462,7 @@ void CommandListScratchPatchFixtureInit::testScratchInline(bool useImmediate) {
ptrOffset(cmdListStream->getCpuBase(), usedBefore),
usedAfter - usedBefore));
walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerCmd(cmdList.begin(), cmdList.end(), true);
walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerTypeCmd(cmdList.begin(), cmdList.end());
ASSERT_NE(cmdList.end(), walkerIterator);
void *walkerPtrWithoutScratch = *walkerIterator;
@ -1522,7 +1522,7 @@ void CommandListScratchPatchFixtureInit::testScratchGrowingPatching() {
ptrOffset(cmdListStream->getCpuBase(), usedBefore),
usedAfter - usedBefore));
auto walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerCmd(cmdList.begin(), cmdList.end(), true);
auto walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerTypeCmd(cmdList.begin(), cmdList.end());
ASSERT_NE(cmdList.end(), walkerIterator);
void *walkerPtrWithScratch = *walkerIterator;
@ -1563,7 +1563,7 @@ void CommandListScratchPatchFixtureInit::testScratchGrowingPatching() {
ptrOffset(cmdListStream->getCpuBase(), usedBefore),
usedAfter - usedBefore));
walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerCmd(cmdList.begin(), cmdList.end(), true);
walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerTypeCmd(cmdList.begin(), cmdList.end());
ASSERT_NE(cmdList.end(), walkerIterator);
void *walkerPtrWithSlot1Scratch = *walkerIterator;
@ -1624,7 +1624,7 @@ void CommandListScratchPatchFixtureInit::testScratchSameNotPatching() {
ptrOffset(cmdListStream->getCpuBase(), usedBefore),
usedAfter - usedBefore));
auto walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerCmd(cmdList.begin(), cmdList.end(), true);
auto walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerTypeCmd(cmdList.begin(), cmdList.end());
ASSERT_NE(cmdList.end(), walkerIterator);
void *walkerPtrWithScratch = *walkerIterator;
@ -1683,7 +1683,7 @@ void CommandListScratchPatchFixtureInit::testScratchImmediatePatching() {
ptrOffset(cmdListStream->getCpuBase(), usedBefore),
usedAfter - usedBefore));
auto walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerCmd(cmdList.begin(), cmdList.end(), true);
auto walkerIterator = NEO::UnitTestHelper<FamilyType>::findWalkerTypeCmd(cmdList.begin(), cmdList.end());
ASSERT_NE(cmdList.end(), walkerIterator);
void *walkerPtrWithScratch = *walkerIterator;

View File

@ -101,6 +101,7 @@ struct UnitTestHelper {
static bool findStateCacheFlushPipeControl(CommandStreamReceiver &csr, LinearStream &csrStream);
static void verifyDummyBlitWa(const RootDeviceEnvironment *rootDeviceEnvironment, GenCmdList::iterator &cmdIterator);
static GenCmdList::iterator findWalkerCmd(GenCmdList::iterator begin, GenCmdList::iterator end, bool heapless);
static GenCmdList::iterator findWalkerTypeCmd(GenCmdList::iterator begin, GenCmdList::iterator end);
};
} // namespace NEO

View File

@ -159,4 +159,9 @@ GenCmdList::iterator UnitTestHelper<GfxFamily>::findWalkerCmd(GenCmdList::iterat
return find<typename GfxFamily::GPGPU_WALKER *>(begin, end);
}
template <typename GfxFamily>
GenCmdList::iterator UnitTestHelper<GfxFamily>::findWalkerTypeCmd(GenCmdList::iterator begin, GenCmdList::iterator end) {
return find<typename GfxFamily::GPGPU_WALKER *>(begin, end);
}
} // namespace NEO

View File

@ -143,4 +143,9 @@ GenCmdList::iterator UnitTestHelper<GfxFamily>::findWalkerCmd(GenCmdList::iterat
return find<typename GfxFamily::COMPUTE_WALKER *>(begin, end);
}
template <typename GfxFamily>
GenCmdList::iterator UnitTestHelper<GfxFamily>::findWalkerTypeCmd(GenCmdList::iterator begin, GenCmdList::iterator end) {
return find<typename GfxFamily::COMPUTE_WALKER *>(begin, end);
}
} // namespace NEO