Skip to content

Commit

Permalink
Move GRUCellFusion and SequenceFusion transformations before NopElimi…
Browse files Browse the repository at this point in the history
…nation (#14213)
  • Loading branch information
itikhono authored Nov 25, 2022
1 parent 6954838 commit 4f0b846
Showing 1 changed file with 4 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -146,14 +146,17 @@ bool ov::pass::MOCTransformations::run_on_model(const std::shared_ptr<ngraph::Fu
REGISTER_PASS(manager, BroadcastElementwiseFusion)
REGISTER_PASS(manager, PullThroughReduce)

// GRUCellFusion and SequenceFusion should be before NopElimination
REGISTER_PASS(manager, GRUCellFusion)
REGISTER_PASS(manager, SequenceFusion)

auto transpose_sinking = manager.register_pass<ov::pass::GraphRewrite>();
ADD_MATCHER(transpose_sinking, TransposeSinking)

// SplitSqueezeConcatFusion should work in same GraphRewrite as TransposesSinking,
// because it replaces pattern that may contain Transposes which must be optimized before
// the transformation and it also inserts Transpose that can be optimized by TransposeSinking
ADD_MATCHER(transpose_sinking, SplitSqueezeConcatFusion)

auto eliminations = manager.register_pass<ov::pass::GraphRewrite>();
ADD_MATCHER(eliminations, EliminateUnsqueezeGather)
ADD_MATCHER(eliminations, NopElimination, m_use_shapes)
Expand All @@ -178,8 +181,6 @@ bool ov::pass::MOCTransformations::run_on_model(const std::shared_ptr<ngraph::Fu
ADD_MATCHER(common_fusions, GeluFusion)
ADD_MATCHER(common_fusions, LeakyReluFusion)
ADD_MATCHER(common_fusions, RandomUniformFusion)
ADD_MATCHER(common_fusions, GRUCellFusion)
ADD_MATCHER(common_fusions, SequenceFusion)
ADD_MATCHER(common_fusions, ConvertTensorIteratorToSequence)
ADD_MATCHER(common_fusions, SplitConcatPairToInterpolateFusion, m_use_shapes)
if (m_use_shapes) {
Expand Down

0 comments on commit 4f0b846

Please sign in to comment.