From 99ab57123c77190f5021d6d4414e35ed22ef3d49 Mon Sep 17 00:00:00 2001
From: mozga-intel <mateusz.ozga@intel.com>
Date: Tue, 21 May 2019 06:33:51 -0700
Subject: [PATCH] Enabled ngraph elementwise max operator  (#17517)

---
 .../operators/ngraph/ops/elementwise_node.h   |  6 +++++
 .../ngraph/test_elementwise_max_ngraph_op.py  | 22 +++++++++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 python/paddle/fluid/tests/unittests/ngraph/test_elementwise_max_ngraph_op.py

diff --git a/paddle/fluid/operators/ngraph/ops/elementwise_node.h b/paddle/fluid/operators/ngraph/ops/elementwise_node.h
index 1e3f87aabe..38a6f6a269 100644
--- a/paddle/fluid/operators/ngraph/ops/elementwise_node.h
+++ b/paddle/fluid/operators/ngraph/ops/elementwise_node.h
@@ -14,10 +14,13 @@ limitations under the License. */
 
 #pragma once
 
+#include <memory>
 #include <string>
+#include <unordered_map>
 
 #include "ngraph/ngraph.hpp"
 #include "paddle/fluid/operators/ngraph/ops/elementwise_binary_prepare_node.h"
+#include "paddle/fluid/operators/ngraph/ops/op_bridge.h"
 #include "paddle/fluid/platform/ngraph_helper.h"
 
 namespace paddle {
@@ -61,3 +64,6 @@ void BuildElementwiseCompareNode(
 }  // namespace ngraphs
 }  // namespace operators
 }  // namespace paddle
+
+REGISTER_NG_OP(elementwise_max,
+               BuildElementwiseBinaryNode<ngraph::op::Maximum>);
diff --git a/python/paddle/fluid/tests/unittests/ngraph/test_elementwise_max_ngraph_op.py b/python/paddle/fluid/tests/unittests/ngraph/test_elementwise_max_ngraph_op.py
new file mode 100644
index 0000000000..30d6ab4765
--- /dev/null
+++ b/python/paddle/fluid/tests/unittests/ngraph/test_elementwise_max_ngraph_op.py
@@ -0,0 +1,22 @@
+#	Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import print_function
+
+import unittest, sys
+sys.path.append("../")
+from test_elementwise_max_op import *
+
+if __name__ == '__main__':
+    unittest.main()
-- 
GitLab