Skip to content

Commit 43c7804

Browse files
committed
add SDPAFuseTransposeReshape back
1 parent 7a74399 commit 43c7804

File tree

2 files changed

+209
-208
lines changed

2 files changed

+209
-208
lines changed

src/plugins/intel_cpu/src/transformations/cpu_opset/common/pass/stateful_sdpa_fusion.cpp

+2-1
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
#include "ov_ops/type_relaxed.hpp"
2424
#include "transformations/common_optimizations/simplify_shape_of_sub_graph.hpp"
2525
#include "transformations/cpu_opset/common/op/sdpa.hpp"
26+
#include "transformations/cpu_opset/x64/pass/sdpa_fuse_transpose_reshape.hpp"
2627
#include "transformations/defs.hpp"
2728
#include "transformations/op_conversions/convert_broadcast3.hpp"
2829
#include "transformations/transpose_sinking/ts_shape_of.hpp"
@@ -307,7 +308,7 @@ bool SDPASubgraphFusion::run_on_model(const std::shared_ptr<ov::Model>& f) {
307308
CPU_REGISTER_PASS_COMMON(manager, ov::pass::SimplifyGatherShapeOf);
308309
CPU_REGISTER_PASS_COMMON(manager, ov::pass::transpose_sinking::TSShapeOfForward);
309310
CPU_REGISTER_PASS_COMMON(manager, StatefulSDPAFusion);
310-
CPU_REGISTER_PASS_COMMON(manager, ov::pass::Validate);
311+
CPU_REGISTER_PASS_X64(manager, SDPAFuseTransposeReshape);
311312

312313
manager.run_passes(f);
313314
return false;

0 commit comments

Comments
 (0)