Remove EngineInstance parameter from pollForCompletion call

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

Author: BartoszDunajski

runtime/command_stream/aub_command_stream_receiver_hw.h

@@ -60,7 +60,7 @@ class AUBCommandStreamReceiverHw : public CommandStreamReceiverSimulatedHw<GfxFa
 
     // Family specific version
     MOCKABLE_VIRTUAL void submitBatchBuffer(size_t engineIndex, uint64_t batchBufferGpuAddress, const void *batchBuffer, size_t batchBufferSize, uint32_t memoryBank, uint64_t entryBits);
-    MOCKABLE_VIRTUAL void pollForCompletion(EngineInstanceT engineInstance);
+    MOCKABLE_VIRTUAL void pollForCompletion();
 
     uint32_t getDumpHandle();
     MOCKABLE_VIRTUAL void addContextToken(uint32_t dumpHandle);

runtime/command_stream/aub_command_stream_receiver_hw.inl

@@ -345,7 +345,7 @@ FlushStamp AUBCommandStreamReceiverHw<GfxFamily>::flush(BatchBuffer &batchBuffer
     submitBatchBuffer(engineIndex, batchBufferGpuAddress, pBatchBuffer, sizeBatchBuffer, this->getMemoryBank(batchBuffer.commandBufferAllocation), this->getPPGTTAdditionalBits(batchBuffer.commandBufferAllocation));
 
     if (!DebugManager.flags.AUBDumpConcurrentCS.get()) {
-        pollForCompletion(osContext->getEngineType());
+        pollForCompletion();
     }
 
     if (this->standalone) {
@@ -563,13 +563,13 @@ void AUBCommandStreamReceiverHw<GfxFamily>::submitBatchBuffer(size_t engineIndex
 }
 
 template <typename GfxFamily>
-void AUBCommandStreamReceiverHw<GfxFamily>::pollForCompletion(EngineInstanceT engineInstance) {
+void AUBCommandStreamReceiverHw<GfxFamily>::pollForCompletion() {
     if (hardwareContext) {
         hardwareContext->pollForCompletion();
         return;
     }
 
-    const auto mmioBase = this->getCsTraits(engineInstance).mmioBase;
+    const auto mmioBase = this->getCsTraits(osContext->getEngineType()).mmioBase;
     const bool pollNotEqual = false;
     const uint32_t mask = getMaskAndValueForPollForCompletion();
     const uint32_t value = mask;

runtime/command_stream/tbx_command_stream_receiver_hw.h

@@ -49,7 +49,7 @@ class TbxCommandStreamReceiverHw : public CommandStreamReceiverSimulatedHw<GfxFa
 
     // Family specific version
     MOCKABLE_VIRTUAL void submitBatchBuffer(size_t engineIndex, uint64_t batchBufferGpuAddress, const void *batchBuffer, size_t batchBufferSize, uint32_t memoryBank, uint64_t entryBits);
-    MOCKABLE_VIRTUAL void pollForCompletion(EngineInstanceT engineInstance);
+    MOCKABLE_VIRTUAL void pollForCompletion();
 
     static CommandStreamReceiver *create(const HardwareInfo &hwInfoIn, bool withAubDump, ExecutionEnvironment &executionEnvironment);
 

runtime/command_stream/tbx_command_stream_receiver_hw.inl

@@ -210,7 +210,7 @@ FlushStamp TbxCommandStreamReceiverHw<GfxFamily>::flush(BatchBuffer &batchBuffer
 
     submitBatchBuffer(engineIndex, batchBufferGpuAddress, pBatchBuffer, sizeBatchBuffer, this->getMemoryBank(batchBuffer.commandBufferAllocation), this->getPPGTTAdditionalBits(batchBuffer.commandBufferAllocation));
 
-    pollForCompletion(osContext->getEngineType());
+    pollForCompletion();
     return 0;
 }
 
@@ -344,15 +344,15 @@ void TbxCommandStreamReceiverHw<GfxFamily>::submitBatchBuffer(size_t engineIndex
 }
 
 template <typename GfxFamily>
-void TbxCommandStreamReceiverHw<GfxFamily>::pollForCompletion(EngineInstanceT engineInstance) {
+void TbxCommandStreamReceiverHw<GfxFamily>::pollForCompletion() {
     if (hardwareContext) {
         hardwareContext->pollForCompletion();
         return;
     }
 
     typedef typename AubMemDump::CmdServicesMemTraceRegisterPoll CmdServicesMemTraceRegisterPoll;
 
-    auto mmioBase = this->getCsTraits(engineInstance).mmioBase;
+    auto mmioBase = this->getCsTraits(osContext->getEngineType()).mmioBase;
     bool pollNotEqual = false;
     tbxStream.registerPoll(
         AubMemDump::computeRegisterOffset(mmioBase, 0x2234), //EXECLIST_STATUS

unit_tests/aub_tests/command_stream/aub_command_stream_fixture.h

@@ -84,7 +84,7 @@ class AUBCommandStreamFixture : public CommandStreamFixture {
         }
 
         auto aubCsr = reinterpret_cast<AUBCommandStreamReceiverHw<FamilyType> *>(csr);
-        aubCsr->pollForCompletion(csr->getOsContext().getEngineType());
+        aubCsr->pollForCompletion();
     }
 
     GraphicsAllocation *createResidentAllocationAndStoreItInCsr(const void *address, size_t size) {

unit_tests/mocks/mock_aub_csr.h

@@ -86,8 +86,8 @@ struct MockAubCsr : public AUBCommandStreamReceiverHw<GfxFamily> {
         AUBCommandStreamReceiverHw<GfxFamily>::submitBatchBuffer(engineIndex, batchBufferGpuAddress, batchBuffer, batchBufferSize, memoryBank, entryBits);
         submitBatchBufferCalled = true;
     }
-    void pollForCompletion(EngineInstanceT engineInstance) override {
-        AUBCommandStreamReceiverHw<GfxFamily>::pollForCompletion(engineInstance);
+    void pollForCompletion() override {
+        AUBCommandStreamReceiverHw<GfxFamily>::pollForCompletion();
         pollForCompletionCalled = true;
     }
     void expectMemoryEqual(void *gfxAddress, const void *srcAddress, size_t length) {

unit_tests/mocks/mock_tbx_csr.h

@@ -50,8 +50,8 @@ class MockTbxCsr : public TbxCommandStreamReceiverHw<GfxFamily> {
         TbxCommandStreamReceiverHw<GfxFamily>::submitBatchBuffer(engineIndex, batchBufferGpuAddress, batchBuffer, batchBufferSize, memoryBank, entryBits);
         submitBatchBufferCalled = true;
     }
-    void pollForCompletion(EngineInstanceT engineInstance) override {
-        TbxCommandStreamReceiverHw<GfxFamily>::pollForCompletion(engineInstance);
+    void pollForCompletion() override {
+        TbxCommandStreamReceiverHw<GfxFamily>::pollForCompletion();
         pollForCompletionCalled = true;
     }
     void makeCoherent(GraphicsAllocation &gfxAllocation) override {