Move pollForCompletionMask for TBX to new method

Change-Id: I1ce760ea185064ec9122eb5bf11d9b99c9c700e2
Signed-off-by: Dunajski, Bartosz <[email protected]>

Author: BartoszDunajski

runtime/command_stream/tbx_command_stream_receiver_hw.h

@@ -33,6 +33,9 @@ class TbxCommandStreamReceiverHw : public CommandStreamReceiverSimulatedHw<GfxFa
     using BaseClass::engineIndex;
     using BaseClass::osContext;
 
+    uint32_t getMaskAndValueForPollForCompletion() const;
+    bool getpollNotEqualValueForPollForCompletion() const;
+
   public:
     using CommandStreamReceiverSimulatedCommonHw<GfxFamily>::initAdditionalMMIO;
     using CommandStreamReceiverSimulatedCommonHw<GfxFamily>::aubManager;

runtime/command_stream/tbx_command_stream_receiver_hw.inl

@@ -350,11 +350,13 @@ void TbxCommandStreamReceiverHw<GfxFamily>::pollForCompletion() {
     typedef typename AubMemDump::CmdServicesMemTraceRegisterPoll CmdServicesMemTraceRegisterPoll;
 
     auto mmioBase = this->getCsTraits(osContext->getEngineType()).mmioBase;
-    bool pollNotEqual = false;
+    bool pollNotEqual = getpollNotEqualValueForPollForCompletion();
+    uint32_t mask = getMaskAndValueForPollForCompletion();
+    uint32_t value = mask;
     tbxStream.registerPoll(
         AubMemDump::computeRegisterOffset(mmioBase, 0x2234), //EXECLIST_STATUS
-        0x100,
-        0x100,
+        mask,
+        value,
         pollNotEqual,
         CmdServicesMemTraceRegisterPoll::TimeoutActionValues::Abort);
 }
@@ -431,4 +433,14 @@ void TbxCommandStreamReceiverHw<GfxFamily>::waitBeforeMakingNonResidentWhenRequi
         this->makeCoherent(*allocation);
     }
 }
+
+template <typename GfxFamily>
+uint32_t TbxCommandStreamReceiverHw<GfxFamily>::getMaskAndValueForPollForCompletion() const {
+    return 0x100;
+}
+
+template <typename GfxFamily>
+bool TbxCommandStreamReceiverHw<GfxFamily>::getpollNotEqualValueForPollForCompletion() const {
+    return false;
+}
 } // namespace OCLRT