Enable task count from wait on DG2 and later

Signed-off-by: Karol Trojanowski <[email protected]>

Author: Trojanowski, Karol

opencl/test/unit_test/helpers/hw_helper_tests_dg2_and_later.cpp

@@ -154,6 +154,12 @@ HWTEST2_F(HwHelperTestsDg2AndLater, givenXeHPGAndLaterPlatformWhenCheckingIfUnTy
     EXPECT_TRUE(hwHelper.unTypedDataPortCacheFlushRequired());
 }
 
+HWTEST2_F(HwHelperTestsDg2AndLater, givenHwHelperWhenCheckIsUpdateTaskCountFromWaitSupportedThenReturnsTrue, IsAtLeastXeHpgCore) {
+    auto &hwHelper = HwHelper::get(hardwareInfo.platform.eRenderCoreFamily);
+
+    EXPECT_TRUE(hwHelper.isUpdateTaskCountFromWaitSupported());
+}
+
 using HwInfoConfigTestDg2AndLater = ::testing::Test;
 
 HWTEST2_F(HwInfoConfigTestDg2AndLater, givenDg2AndLaterPlatformWhenAskedIfHeapInLocalMemThenTrueIsReturned, IsAtLeastXeHpgCore) {

shared/source/helpers/hw_helper_dg2_and_later.inl

@@ -11,6 +11,11 @@
 
 namespace NEO {
 
+template <>
+bool HwHelperHw<Family>::isUpdateTaskCountFromWaitSupported() const {
+    return true;
+}
+
 template <typename GfxFamily>
 inline void MemorySynchronizationCommands<GfxFamily>::setPipeControlExtraProperties(PIPE_CONTROL &pipeControl, PipeControlArgs &args) {
     pipeControl.setHdcPipelineFlush(args.hdcPipelineFlush);

shared/source/helpers/hw_helper_pvc_and_later.inl

@@ -36,11 +36,6 @@ bool HwHelperHw<Family>::isCooperativeDispatchSupported(const EngineGroupType en
     return true;
 }
 
-template <>
-bool HwHelperHw<Family>::isUpdateTaskCountFromWaitSupported() const {
-    return true;
-}
-
 template <>
 uint32_t HwHelperHw<Family>::adjustMaxWorkGroupCount(uint32_t maxWorkGroupCount, const EngineGroupType engineGroupType,
                                                      const HardwareInfo &hwInfo, bool isEngineInstanced) const {