From 1ca5d57ab06e62957b4b612597ee40a2be6d6f16 Mon Sep 17 00:00:00 2001
From: Zbigniew Zdanowicz <zbigniew.zdanowicz@intel.com>
Date: Mon, 19 Sep 2022 13:09:56 +0000
Subject: [PATCH] Add systolic getter method to unit test helper

Signed-off-by: Zbigniew Zdanowicz <zbigniew.zdanowicz@intel.com>
---
 shared/test/common/gen12lp/unit_test_helper_gen12lp.cpp      | 5 +++++
 shared/test/common/helpers/unit_test_helper.h                | 1 +
 .../test/common/helpers/unit_test_helper_bdw_and_later.inl   | 5 +++++
 .../test/common/helpers/unit_test_helper_xehp_and_later.inl  | 5 +++++
 4 files changed, 16 insertions(+)

diff --git a/shared/test/common/gen12lp/unit_test_helper_gen12lp.cpp b/shared/test/common/gen12lp/unit_test_helper_gen12lp.cpp
index 04a155b332..9342262321 100644
--- a/shared/test/common/gen12lp/unit_test_helper_gen12lp.cpp
+++ b/shared/test/common/gen12lp/unit_test_helper_gen12lp.cpp
@@ -56,5 +56,10 @@ bool UnitTestHelper<Family>::getDisableFusionStateFromFrontEndCommand(const type
     return feCmd.getDisableSlice0Subslice2();
 }
 
+template <>
+bool UnitTestHelper<Family>::getSystolicFlagValueFromPipelineSelectCommand(const typename Family::PIPELINE_SELECT &pipelineSelectCmd) {
+    return pipelineSelectCmd.getSpecialModeEnable();
+}
+
 template struct UnitTestHelper<Family>;
 } // namespace NEO
diff --git a/shared/test/common/helpers/unit_test_helper.h b/shared/test/common/helpers/unit_test_helper.h
index 065339a9ca..deda301271 100644
--- a/shared/test/common/helpers/unit_test_helper.h
+++ b/shared/test/common/helpers/unit_test_helper.h
@@ -90,6 +90,7 @@ struct UnitTestHelper {
 
     static bool getDisableFusionStateFromFrontEndCommand(const typename GfxFamily::VFE_STATE_TYPE &feCmd);
     static bool getComputeDispatchAllWalkerFromFrontEndCommand(const typename GfxFamily::VFE_STATE_TYPE &feCmd);
+    static bool getSystolicFlagValueFromPipelineSelectCommand(const typename GfxFamily::PIPELINE_SELECT &pipelineSelectCmd);
 };
 
 } // namespace NEO
diff --git a/shared/test/common/helpers/unit_test_helper_bdw_and_later.inl b/shared/test/common/helpers/unit_test_helper_bdw_and_later.inl
index 127886f4d0..2a7d321e40 100644
--- a/shared/test/common/helpers/unit_test_helper_bdw_and_later.inl
+++ b/shared/test/common/helpers/unit_test_helper_bdw_and_later.inl
@@ -87,4 +87,9 @@ std::vector<GenCmdList::iterator> UnitTestHelper<GfxFamily>::findAllMidThreadPre
     return findAll<typename GfxFamily::GPGPU_CSR_BASE_ADDRESS *>(begin, end);
 }
 
+template <typename GfxFamily>
+bool UnitTestHelper<GfxFamily>::getSystolicFlagValueFromPipelineSelectCommand(const typename GfxFamily::PIPELINE_SELECT &pipelineSelectCmd) {
+    return false;
+}
+
 } // namespace NEO
diff --git a/shared/test/common/helpers/unit_test_helper_xehp_and_later.inl b/shared/test/common/helpers/unit_test_helper_xehp_and_later.inl
index 62fc554db6..24eebbe837 100644
--- a/shared/test/common/helpers/unit_test_helper_xehp_and_later.inl
+++ b/shared/test/common/helpers/unit_test_helper_xehp_and_later.inl
@@ -116,4 +116,9 @@ std::vector<GenCmdList::iterator> UnitTestHelper<GfxFamily>::findAllMidThreadPre
     return emptyList;
 }
 
+template <typename GfxFamily>
+bool UnitTestHelper<GfxFamily>::getSystolicFlagValueFromPipelineSelectCommand(const typename GfxFamily::PIPELINE_SELECT &pipelineSelectCmd) {
+    return pipelineSelectCmd.getSystolicModeEnable();
+}
+
 } // namespace NEO