From d8262de5218dab02426766777917640af1ff20b9 Mon Sep 17 00:00:00 2001 From: Vladislav Golubev Date: Mon, 20 Jan 2025 18:51:31 +0100 Subject: [PATCH] [CPU] Move ConvertBroadcast3 after LPT --- .../src/transformations/transformation_pipeline.cpp | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/plugins/intel_cpu/src/transformations/transformation_pipeline.cpp b/src/plugins/intel_cpu/src/transformations/transformation_pipeline.cpp index da61917a146db0..13a10c722284b5 100644 --- a/src/plugins/intel_cpu/src/transformations/transformation_pipeline.cpp +++ b/src/plugins/intel_cpu/src/transformations/transformation_pipeline.cpp @@ -52,6 +52,7 @@ #include "transformations/op_conversions/convert_avgpool_downgrade.hpp" #include "transformations/op_conversions/convert_batch_to_space.hpp" #include "transformations/op_conversions/convert_bitwise_to_logical_bool.hpp" +#include "transformations/op_conversions/convert_broadcast3.hpp" #include "transformations/op_conversions/convert_broadcast_to_tiles.hpp" #include "transformations/op_conversions/convert_depth_to_space.hpp" #include "transformations/op_conversions/convert_gather_downgrade.hpp" @@ -71,6 +72,7 @@ #include "transformations/op_conversions/convert_roi_align_v9_to_v3.hpp" #include "transformations/op_conversions/convert_scatter_nd_update15_downgrade.hpp" #include "transformations/op_conversions/convert_sequences_to_tensor_iterator.hpp" +#include "transformations/op_conversions/convert_shapeof3.hpp" #include "transformations/op_conversions/convert_shuffle_channels3.hpp" #include "transformations/op_conversions/convert_slice_to_strided_slice.hpp" #include "transformations/op_conversions/convert_slicescatter.hpp" @@ -446,8 +448,8 @@ void Transformations::PreLpt(const std::vector& defaultPrecis CPU_REGISTER_PASS_COMMON(manager, ov::pass::WrapInterpolateIntoTransposes); CPU_REGISTER_PASS_COMMON(manager, ov::pass::TransposeSinking); CPU_REGISTER_PASS_COMMON(manager, ov::pass::ConvertSequenceToTensorIterator); - CPU_REGISTER_PASS_COMMON(manager, ov::pass::ConvertOpSet3ToOpSet2); CPU_REGISTER_PASS_COMMON(manager, ov::pass::ConvertOpSet2ToOpSet1); + CPU_REGISTER_PASS_COMMON(manager, ov::pass::ConvertShapeOf3); CPU_REGISTER_PASS_COMMON(manager, ov::pass::LSTMCellDecomposition); CPU_REGISTER_PASS_COMMON(manager, ov::pass::GRUCellDecomposition); CPU_REGISTER_PASS_COMMON(manager, ov::pass::RNNCellDecomposition); @@ -666,11 +668,8 @@ void Transformations::PreLpt(const std::vector& defaultPrecis CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertCompressedOnlyToLegacy); CPU_DISABLE_PASS_COMMON(manager, ov::pass::EyeDecomposition); CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertGELU); - CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertShuffleChannels3); CPU_DISABLE_PASS_COMMON(manager, ov::pass::Gelu7Downgrade); - CPU_DISABLE_PASS_COMMON(manager, ov::pass::SoftPlusDecomposition); CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertMod); - CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertShuffleChannels3); CPU_DISABLE_PASS_COMMON(manager, ov::pass::WeightsDequantizeToFakeQuantize); CPU_DISABLE_PASS_COMMON(manager, ov::pass::SimplifyCTCGreedyDecoderSeqLen); CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertGather7ToGather1); @@ -685,7 +684,6 @@ void Transformations::PreLpt(const std::vector& defaultPrecis CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertROIAlign9To3); CPU_DISABLE_PASS_COMMON(manager, ov::pass::SoftSignDecomposition); CPU_DISABLE_PASS_COMMON(manager, ov::pass::UniqueDecomposition); - CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertTopK3); CPU_DISABLE_PASS_COMMON(manager, ov::pass::ConvertTopK11ToTopK3); CPU_DISABLE_PASS_COMMON(manager, ov::pass::HSwishDecomposition); CPU_DISABLE_PASS_COMMON(manager, ov::pass::MatMulConstTransposesExtraction); @@ -851,6 +849,7 @@ void Transformations::PostLpt() { ov::pass::Manager postLPTPassManager("CPU:PostLPT"); postLPTPassManager.set_per_pass_validation(false); + CPU_REGISTER_PASS_COMMON(postLPTPassManager, ov::pass::ConvertBroadcast3); CPU_REGISTER_PASS_COMMON(postLPTPassManager, ov::pass::UnrollTensorIterator); CPU_REGISTER_PASS_COMMON(postLPTPassManager, ov::pass::ReshapePRelu); CPU_SET_CALLBACK_COMMON(