@@ -662,6 +662,7 @@ struct EnqueueAuxKernelTests : public EnqueueKernelTest {
662
662
void dispatchAuxTranslation (MultiDispatchInfo &multiDispatchInfo, MemObjsForAuxTranslation &memObjsForAuxTranslation,
663
663
AuxTranslationDirection auxTranslationDirection) override {
664
664
CommandQueueHw<FamilyType>::dispatchAuxTranslation (multiDispatchInfo, memObjsForAuxTranslation, auxTranslationDirection);
665
+ auxTranslationDirections.push_back (auxTranslationDirection);
665
666
Kernel *lastKernel = nullptr ;
666
667
for (const auto &dispatchInfo : multiDispatchInfo) {
667
668
lastKernel = dispatchInfo.getKernel ();
@@ -675,6 +676,7 @@ struct EnqueueAuxKernelTests : public EnqueueKernelTest {
675
676
CommandQueueHw<FamilyType>::waitUntilComplete (taskCountToWait, flushStampToWait, useQuickKmdSleep);
676
677
}
677
678
679
+ std::vector<AuxTranslationDirection> auxTranslationDirections;
678
680
std::vector<DispatchInfo> dispatchInfos;
679
681
std::vector<std::tuple<Kernel *, size_t , MemObjsForAuxTranslation, AuxTranslationDirection>> dispatchAuxTranslationInputs;
680
682
uint32_t waitCalled = 0 ;
@@ -742,6 +744,9 @@ HWTEST_F(EnqueueAuxKernelTests, givenMultipleArgsWhenAuxTranslationIsRequiredThe
742
744
}
743
745
744
746
EXPECT_EQ (4u , pipeControlCount);
747
+ ASSERT_EQ (2u , cmdQ.auxTranslationDirections .size ());
748
+ EXPECT_EQ (AuxTranslationDirection::AuxToNonAux, cmdQ.auxTranslationDirections [0 ]);
749
+ EXPECT_EQ (AuxTranslationDirection::NonAuxToAux, cmdQ.auxTranslationDirections [1 ]);
745
750
}
746
751
747
752
HWTEST_F (EnqueueAuxKernelTests, givenKernelWithRequiredAuxTranslationWhenEnqueuedThenDispatchAuxTranslationBuiltin) {
0 commit comments