From 51a00fb4c8b3d3f0b8c235c99e18ec4c7bc45a1e Mon Sep 17 00:00:00 2001
From: Lukasz Jobczyk <lukasz.jobczyk@intel.com>
Date: Wed, 2 Aug 2023 10:33:50 +0000
Subject: [PATCH] fix: Enable timestamp wait for events on MTL

Related-To: NEO-8067

Signed-off-by: Lukasz Jobczyk <lukasz.jobczyk@intel.com>
---
 .../xe_hpg_core/mtl/os_agnostic_product_helper_mtl.inl       | 5 +++++
 .../unit_test/xe_hpg_core/mtl/excludes_xe_hpg_core_mtl.cpp   | 1 +
 .../xe_hpg_core/mtl/product_config_helper_tests_mtl.cpp      | 4 ++++
 3 files changed, 10 insertions(+)

diff --git a/shared/source/xe_hpg_core/mtl/os_agnostic_product_helper_mtl.inl b/shared/source/xe_hpg_core/mtl/os_agnostic_product_helper_mtl.inl
index 55cac66d0b..92510e5a45 100644
--- a/shared/source/xe_hpg_core/mtl/os_agnostic_product_helper_mtl.inl
+++ b/shared/source/xe_hpg_core/mtl/os_agnostic_product_helper_mtl.inl
@@ -47,6 +47,11 @@ uint32_t ProductHelperHw<gfxProduct>::getSteppingFromHwRevId(const HardwareInfo
     return CommonConstants::invalidStepping;
 }
 
+template <>
+bool ProductHelperHw<gfxProduct>::isTimestampWaitSupportedForEvents() const {
+    return true;
+}
+
 template <>
 bool ProductHelperHw<gfxProduct>::isDirectSubmissionConstantCacheInvalidationNeeded(const HardwareInfo &hwInfo) const {
     return true;
diff --git a/shared/test/unit_test/xe_hpg_core/mtl/excludes_xe_hpg_core_mtl.cpp b/shared/test/unit_test/xe_hpg_core/mtl/excludes_xe_hpg_core_mtl.cpp
index 5ae901379c..4d3a68f06e 100644
--- a/shared/test/unit_test/xe_hpg_core/mtl/excludes_xe_hpg_core_mtl.cpp
+++ b/shared/test/unit_test/xe_hpg_core/mtl/excludes_xe_hpg_core_mtl.cpp
@@ -24,3 +24,4 @@ HWTEST_EXCLUDE_PRODUCT(ProductHelperTest, whenCheckIsCachingOnCpuAvailableThenAl
 HWTEST_EXCLUDE_PRODUCT(ProductHelperTest, givenProductHelperWhenCheckingIsBufferPoolAllocatorSupportedThenCorrectValueIsReturned, IGFX_METEORLAKE);
 HWTEST_EXCLUDE_PRODUCT(ProductHelperTest, givenDebugFlagWhenCheckingIsResolveDependenciesByPipeControlsSupportedThenCorrectValueIsReturned, IGFX_METEORLAKE);
 HWTEST_EXCLUDE_PRODUCT(ProductHelperCommonTest, givenPatIndexAndAllocationTypeWhenCallOverridePatIndexThenSameIndexIsReturned, IGFX_METEORLAKE);
+HWTEST_EXCLUDE_PRODUCT(ProductHelperTestXeHpgCore, givenProductHelperWhenCheckTimestampWaitSupportForEventsThenReturnFalse, IGFX_METEORLAKE);
diff --git a/shared/test/unit_test/xe_hpg_core/mtl/product_config_helper_tests_mtl.cpp b/shared/test/unit_test/xe_hpg_core/mtl/product_config_helper_tests_mtl.cpp
index 4b2feefd68..14106e7c98 100644
--- a/shared/test/unit_test/xe_hpg_core/mtl/product_config_helper_tests_mtl.cpp
+++ b/shared/test/unit_test/xe_hpg_core/mtl/product_config_helper_tests_mtl.cpp
@@ -46,4 +46,8 @@ MTLTEST_F(ProductConfigHelperMtlTests, givenMtlConfigsWhenSearchForDeviceAcronym
 
 MTLTEST_F(ProductHelperMtlTests, givenProductHelperWhenCheckingIsBufferPoolAllocatorSupportedThenCorrectValueIsReturned) {
     EXPECT_TRUE(productHelper->isBufferPoolAllocatorSupported());
+}
+
+MTLTEST_F(ProductHelperMtlTests, givenMtlWhenCheckingIsTimestampWaitSupportedForEventsThenReturnTrue) {
+    EXPECT_TRUE(productHelper->isTimestampWaitSupportedForEvents());
 }
\ No newline at end of file