Skip to content

Commit

Permalink
[GPU] Fuse MVN before ConvertPrecision (openvinotoolkit#18845)
Browse files Browse the repository at this point in the history
  • Loading branch information
pavel-esir authored Jul 28, 2023
1 parent 4c1fc5f commit 5113900
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/plugins/intel_gpu/src/plugin/transformations_pipeline.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -70,6 +70,7 @@
#include <transformations/op_conversions/lstm_cell_decomposition.hpp>
#include <transformations/op_conversions/rnn_cell_decomposition.hpp>
#include <transformations/op_conversions/mvn6_decomposition.hpp>
#include <transformations/common_optimizations/mvn_fusion.hpp>
#include <transformations/op_conversions/normalize_l2_decomposition.hpp>
#include <transformations/op_conversions/bidirectional_sequences_decomposition.hpp>
#include <transformations/op_conversions/convert_previous_nms_to_nms_9.hpp>
Expand Down Expand Up @@ -193,9 +194,10 @@ void TransformationsPipeline::apply(std::shared_ptr<ov::Model> func) {
type_to_fuse_map empty_fuse_map = {};
manager.register_pass<ov::pass::Validate>();

// fuse softmax patterns so that they will not be marked as precision sensitive in ConvertPrecision
// fuse softmax, MVN patterns, so that they will not be marked as precision sensitive in ConvertPrecision
manager.register_pass<ov::pass::SoftmaxFusion>();
// decompose MVNs that sre not supported in GPU, so the they will be marked as precision sensitive in ConvertPrecision
manager.register_pass<ov::pass::MVNFusion>();
// decompose MVNs that sre not supported in GPU, so that they will be marked as precision sensitive in ConvertPrecision
manager.register_pass<ov::pass::MVN6Decomposition>();
manager.register_pass<ov::pass::BroadcastTransition>();

Expand Down

0 comments on commit 5113900

Please sign in to comment.