test: large grf test parser issue

Related-To: NEO-13585

Signed-off-by: Wojciech Konior <wojciech.konior@intel.com>

Author: WojciechKonior

opencl/test/unit_test/aub_tests/command_queue/large_grf_aub_tests_xehp_and_later.cpp

@@ -464,9 +464,8 @@ HWTEST2_P(LargeGrfTest, givenMixedLargeGrfAndSmallGrfKernelsWhenExecutedThenResu
             EXPECT_FALSE(largeGrfValues[0]);
         } else {
             EXPECT_EQ(0u, largeGrfValues.size());
-            largeGrfValues = NEO::UnitTestHelper<FamilyType>::getProgrammedLargeGrfValues(*this->csr,
-                                                                                          this->csr->getCS(0));
-            ASSERT_NE(0u, largeGrfValues.size());
+            largeGrfValues = NEO::UnitTestHelper<FamilyType>::getProgrammedLargeGrfValues(this->csr->getCS(0));
+            ASSERT_EQ(1u, largeGrfValues.size());
             EXPECT_FALSE(largeGrfValues[0]);
         }
     }

shared/test/common/gen12lp/unit_test_helper_gen12lp.cpp

@@ -126,6 +126,11 @@ std::vector<bool> UnitTestHelper<GfxFamily>::getProgrammedLargeGrfValues(Command
     return {};
 }
 
+template <typename GfxFamily>
+std::vector<bool> UnitTestHelper<GfxFamily>::getProgrammedLargeGrfValues(LinearStream &linearStream) {
+    return {};
+}
+
 template <typename GfxFamily>
 inline bool UnitTestHelper<GfxFamily>::getWorkloadPartitionForStoreRegisterMemCmd(typename GfxFamily::MI_STORE_REGISTER_MEM &storeRegisterMem) {
     return false;

shared/test/common/helpers/unit_test_helper.h

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2018-2024 Intel Corporation
+ * Copyright (C) 2018-2025 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -93,6 +93,8 @@ struct UnitTestHelper {
 
     static std::vector<bool> getProgrammedLargeGrfValues(CommandStreamReceiver &csr, LinearStream &linearStream);
 
+    static std::vector<bool> getProgrammedLargeGrfValues(LinearStream &linearStream);
+
     static uint32_t getProgrammedGrfValue(CommandStreamReceiver &csr, LinearStream &linearStream);
 
     static bool getWorkloadPartitionForStoreRegisterMemCmd(typename GfxFamily::MI_STORE_REGISTER_MEM &storeRegisterMem);

shared/test/common/helpers/unit_test_helper_xehp_and_later.inl

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2021-2024 Intel Corporation
+ * Copyright (C) 2021-2025 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -111,6 +111,20 @@ std::vector<bool> UnitTestHelper<GfxFamily>::getProgrammedLargeGrfValues(Command
     return largeGrfValues;
 }
 
+template <typename GfxFamily>
+std::vector<bool> UnitTestHelper<GfxFamily>::getProgrammedLargeGrfValues(LinearStream &linearStream) {
+    using STATE_COMPUTE_MODE = typename GfxFamily::STATE_COMPUTE_MODE;
+
+    std::vector<bool> largeGrfValues;
+    HardwareParse hwParser;
+    hwParser.parseCommands<GfxFamily>(linearStream);
+    auto commands = hwParser.getCommandsList<STATE_COMPUTE_MODE>();
+    for (auto &cmd : commands) {
+        largeGrfValues.push_back(reinterpret_cast<STATE_COMPUTE_MODE *>(cmd)->getLargeGrfMode());
+    }
+    return largeGrfValues;
+}
+
 template <typename GfxFamily>
 inline bool UnitTestHelper<GfxFamily>::getWorkloadPartitionForStoreRegisterMemCmd(typename GfxFamily::MI_STORE_REGISTER_MEM &storeRegisterMem) {
     return storeRegisterMem.getWorkloadPartitionIdOffsetEnable();