@@ -662,6 +662,7 @@ struct EnqueueAuxKernelTests : public EnqueueKernelTest {
662662 void dispatchAuxTranslation (MultiDispatchInfo &multiDispatchInfo, MemObjsForAuxTranslation &memObjsForAuxTranslation,
663663 AuxTranslationDirection auxTranslationDirection) override {
664664 CommandQueueHw<FamilyType>::dispatchAuxTranslation (multiDispatchInfo, memObjsForAuxTranslation, auxTranslationDirection);
665+ auxTranslationDirections.push_back (auxTranslationDirection);
665666 Kernel *lastKernel = nullptr ;
666667 for (const auto &dispatchInfo : multiDispatchInfo) {
667668 lastKernel = dispatchInfo.getKernel ();
@@ -675,6 +676,7 @@ struct EnqueueAuxKernelTests : public EnqueueKernelTest {
675676 CommandQueueHw<FamilyType>::waitUntilComplete (taskCountToWait, flushStampToWait, useQuickKmdSleep);
676677 }
677678
679+ std::vector<AuxTranslationDirection> auxTranslationDirections;
678680 std::vector<DispatchInfo> dispatchInfos;
679681 std::vector<std::tuple<Kernel *, size_t , MemObjsForAuxTranslation, AuxTranslationDirection>> dispatchAuxTranslationInputs;
680682 uint32_t waitCalled = 0 ;
@@ -742,6 +744,9 @@ HWTEST_F(EnqueueAuxKernelTests, givenMultipleArgsWhenAuxTranslationIsRequiredThe
742744 }
743745
744746 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 ]);
745750}
746751
747752HWTEST_F (EnqueueAuxKernelTests, givenKernelWithRequiredAuxTranslationWhenEnqueuedThenDispatchAuxTranslationBuiltin) {
0 commit comments