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