Add infrastructure to request ULLS restart

Signed-off-by: Lukasz Jobczyk <lukasz.jobczyk@intel.com>

Author: LukaszJobczyk

shared/source/command_stream/submissions_aggregator.h

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2018-2021 Intel Corporation
+ * Copyright (C) 2018-2022 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -48,6 +48,7 @@ struct BatchBuffer {
     void *endCmdPtr = nullptr;
 
     bool useSingleSubdevice = false;
+    bool ringBufferRestartRequest = false;
 };
 
 struct CommandBuffer : public IDNode<CommandBuffer> {

shared/source/direct_submission/direct_submission_hw.inl

@@ -429,6 +429,10 @@ bool DirectSubmissionHw<GfxFamily, Dispatcher>::dispatchCommandBuffer(BatchBuffe
     //for now workloads requiring cache coherency are not supported
     UNRECOVERABLE_IF(batchBuffer.requiresCoherency);
 
+    if (batchBuffer.ringBufferRestartRequest) {
+        this->stopRingBuffer();
+    }
+
     this->startRingBuffer();
 
     size_t dispatchSize = getSizeDispatch();

shared/test/unit_test/direct_submission/direct_submission_tests_2.cpp

@@ -830,6 +830,23 @@ HWTEST_F(DirectSubmissionDispatchBufferTest,
     EXPECT_EQ(expectedValue, directSubmission.reserved);
 }
 
+HWTEST_F(DirectSubmissionDispatchBufferTest, givenRingBufferRestartRequestWhenDispatchCommandBuffer) {
+    FlushStampTracker flushStamp(true);
+    MockDirectSubmissionHw<FamilyType, BlitterDispatcher<FamilyType>> directSubmission(*pDevice->getDefaultEngine().commandStreamReceiver);
+    bool ret = directSubmission.initialize(true, true);
+    EXPECT_TRUE(ret);
+    EXPECT_EQ(directSubmission.submitCount, 1u);
+
+    ret = directSubmission.dispatchCommandBuffer(batchBuffer, flushStamp);
+    EXPECT_TRUE(ret);
+    EXPECT_EQ(directSubmission.submitCount, 1u);
+
+    batchBuffer.ringBufferRestartRequest = true;
+    ret = directSubmission.dispatchCommandBuffer(batchBuffer, flushStamp);
+    EXPECT_TRUE(ret);
+    EXPECT_EQ(directSubmission.submitCount, 2u);
+}
+
 HWTEST_F(DirectSubmissionDispatchBufferTest, givenDebugFlagSetWhenDispatchingWorkloadThenProgramSfenceInstruction) {
     DebugManagerStateRestore restorer{};