use least amount of transformations

openvinotoolkit · Dec 24, 2024 · 1a867fb · 1a867fb
1 parent fb472cd
commit 1a867fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/plugins/intel_cpu/src/transformations/cpu_opset/common/pass/stateful_sdpa_fusion.cpp b/src/plugins/intel_cpu/src/transformations/cpu_opset/common/pass/stateful_sdpa_fusion.cpp
@@ -306,7 +306,7 @@ bool SDPASubgraphFusion::run_on_model(const std::shared_ptr<ov::Model>& f) {
     ov::pass::Manager manager(get_pass_config(), "SDPASubgraphFusion");
     manager.set_per_pass_validation(false);
 
-    CPU_REGISTER_PASS_COMMON(manager, ov::pass::SimplifyShapeOfSubGraph, true);
+    CPU_REGISTER_PASS_COMMON(manager, ov::pass::SimplifyGatherShapeOf);
     CPU_REGISTER_PASS_COMMON(manager, ov::pass::transpose_sinking::TSShapeOfForward);
     CPU_REGISTER_PASS_COMMON(manager, StatefulSDPAFusion);
     CPU_REGISTER_PASS_COMMON(manager, ov::pass::Validate);