diff --git a/runtime/command_stream/tbx_command_stream_receiver_hw.inl b/runtime/command_stream/tbx_command_stream_receiver_hw.inl index 3d3f8fd406..0e51aebbda 100644 --- a/runtime/command_stream/tbx_command_stream_receiver_hw.inl +++ b/runtime/command_stream/tbx_command_stream_receiver_hw.inl @@ -199,8 +199,6 @@ FlushStamp TbxCommandStreamReceiverHw::flush(BatchBuffer &batchBuffer processResidency(allocationsForResidency); submitBatchBuffer(batchBufferGpuAddress, pBatchBuffer, sizeBatchBuffer, this->getMemoryBank(batchBuffer.commandBufferAllocation), this->getPPGTTAdditionalBits(batchBuffer.commandBufferAllocation)); - - pollForCompletion(); return 0; } diff --git a/unit_tests/command_stream/tbx_command_stream_tests.cpp b/unit_tests/command_stream/tbx_command_stream_tests.cpp index 0c0edf8097..d0df1a1908 100644 --- a/unit_tests/command_stream/tbx_command_stream_tests.cpp +++ b/unit_tests/command_stream/tbx_command_stream_tests.cpp @@ -367,7 +367,7 @@ HWTEST_F(TbxCommandStreamTests, givenTbxCommandStreamReceiverWhenFlushIsCalledTh EXPECT_TRUE(mockHardwareContext->initializeCalled); EXPECT_TRUE(mockHardwareContext->submitCalled); - EXPECT_TRUE(mockHardwareContext->pollForCompletionCalled); + EXPECT_FALSE(mockHardwareContext->pollForCompletionCalled); EXPECT_TRUE(tbxCsr.writeMemoryWithAubManagerCalled); pDevice->executionEnvironment->memoryManager->freeGraphicsMemory(commandBuffer);