From 08ceff0182f4f95902c38a2d3f637f4187e6444d Mon Sep 17 00:00:00 2001 From: tomdol <tomasz.dolbniak@intel.com> Date: Fri, 28 Jun 2019 12:06:56 +0200 Subject: [PATCH] Decompose fused ops in PlaidML backend --- src/ngraph/runtime/plaidml/plaidml_compiler.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/ngraph/runtime/plaidml/plaidml_compiler.cpp b/src/ngraph/runtime/plaidml/plaidml_compiler.cpp index 0150788fb..60b793e71 100644 --- a/src/ngraph/runtime/plaidml/plaidml_compiler.cpp +++ b/src/ngraph/runtime/plaidml/plaidml_compiler.cpp @@ -20,6 +20,7 @@ #include "ngraph/pass/algebraic_simplification.hpp" #include "ngraph/pass/core_fusion.hpp" #include "ngraph/pass/cse.hpp" +#include "ngraph/pass/fused_op_decomposition.hpp" #include "ngraph/pass/get_output_element_elimination.hpp" #include "ngraph/pass/like_replacement.hpp" #include "ngraph/pass/liveness.hpp" @@ -87,6 +88,7 @@ std::shared_ptr<ngraph::runtime::plaidml::PlaidML_Executable> ngraph::pass::Manager pass_manager; // We apply the same general-purposes passes as the CPU backend. + pass_manager.register_pass<ngraph::pass::FusedOpDecomposition>(); pass_manager.register_pass<ngraph::pass::LikeReplacement>(); pass_manager.register_pass<ngraph::pass::NopElimination>(); pass_manager.register_pass<ngraph::pass::ZeroDimTensorElimination>(); -- 2.18.0