@@ -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