Fix: Disable timestamp from wait on multi gpu envs

Related-To: NEO-6948

Signed-off-by: Dominik Dabek <[email protected]>

Author: ddabek-i

shared/source/command_stream/command_stream_receiver_hw_base.inl

@@ -1270,6 +1270,7 @@ inline bool CommandStreamReceiverHw<GfxFamily>::isUpdateTagFromWaitEnabled() {
     auto &hwHelper = HwHelper::get(peekHwInfo().platform.eRenderCoreFamily);
     auto enabled = hwHelper.isUpdateTaskCountFromWaitSupported();
     enabled &= this->isAnyDirectSubmissionEnabled();
+    enabled &= this->peekExecutionEnvironment().rootDeviceEnvironments.size() == 1;
 
     switch (DebugManager.flags.UpdateTaskCountFromWait.get()) {
     case 0:

shared/test/unit_test/command_stream/command_stream_receiver_tests.cpp

@@ -642,6 +642,18 @@ HWTEST_F(CommandStreamReceiverTest, givenUpdateTaskCountFromWaitWhenCheckIfEnabl
     }
 }
 
+HWTEST_F(CommandStreamReceiverTest, givenUpdateTaskCountFromWaitInMultiRootDeviceEnvironmentWhenCheckIfEnabledThenIsDisabled) {
+
+    DebugManagerStateRestore restorer;
+    DebugManager.flags.CreateMultipleRootDevices.set(2);
+    TearDown();
+    SetUp();
+    auto &csr = pDevice->getUltCommandStreamReceiver<FamilyType>();
+
+    csr.directSubmissionAvailable = true;
+    EXPECT_FALSE(csr.isUpdateTagFromWaitEnabled());
+}
+
 struct InitDirectSubmissionFixture {
     void SetUp() { // NOLINT(readability-identifier-naming)
         DebugManager.flags.EnableDirectSubmission.set(1);