Transpose Sinking before SDPA fusion (#25504)

jane-intel · web-flow · commit 9a3f65618e10 · 2024-07-11T14:34:25.000Z
### Details: - *Transpose Sinking before SDPA fusion* ### Tickets: - *CVS-144513*
diff --git a/src/plugins/intel_cpu/src/transformations/transformation_pipeline.cpp b/src/plugins/intel_cpu/src/transformations/transformation_pipeline.cpp
@@ -94,6 +94,7 @@
 #include "transformations/symbolic_transformations/symbolic_optimizations.hpp"
 #include "transformations/init_node_info.hpp"
 #include "transformations/rt_info/keep_const_precision.hpp"
+#include "transformations/transpose_sinking/ts_shape_of.hpp"
 #include "utils/ngraph_transformation.hpp"
 #include "utils/print_model.hpp"
 
@@ -837,6 +838,7 @@ void Transformations::PostLpt() {
             }
         }
     }
+    CPU_REGISTER_PASS_COMMON(postLPTPassManager, ov::pass::transpose_sinking::TSShapeOfForward);
     CPU_REGISTER_PASS_COMMON(postLPTPassManager, StatefulSDPAFusion);
 
     // Should be before Snippets pipeline because Ngram pattern contains eltwise nodes that can be tokenized by Snippets.

Original file line number	Diff line number	Diff line change
`@@ -94,6 +94,7 @@`
`94`	`94`	`#include "transformations/symbolic_transformations/symbolic_optimizations.hpp"`
`95`	`95`	`#include "transformations/init_node_info.hpp"`
`96`	`96`	`#include "transformations/rt_info/keep_const_precision.hpp"`
	`97`	`+#include "transformations/transpose_sinking/ts_shape_of.hpp"`
`97`	`98`	`#include "utils/ngraph_transformation.hpp"`
`98`	`99`	`#include "utils/print_model.hpp"`
`99`	`100`
`@@ -837,6 +838,7 @@ void Transformations::PostLpt() {`
`837`	`838`	`}`
`838`	`839`	`}`
`839`	`840`	`}`
	`841`	`+ CPU_REGISTER_PASS_COMMON(postLPTPassManager, ov::pass::transpose_sinking::TSShapeOfForward);`
`840`	`842`	`CPU_REGISTER_PASS_COMMON(postLPTPassManager, StatefulSDPAFusion);`
`841`	`843`
`842`	`844`	`// Should be before Snippets pipeline because Ngram pattern contains eltwise nodes that can be tokenized by Snippets.`