[MLIR] Add missing visitor for Relu in compiler.cpp (#3308)

f54e9159 · Diego Caballero · Scott Cyphers · 0ca40376 · f54e9159
Commit f54e9159 authored Jul 26, 2019 by Diego Caballero Committed by Scott Cyphers Jul 26, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

compiler.cpp src/contrib/mlir/compiler.cpp +6 -0

No files found.
--- a/src/contrib/mlir/compiler.cpp
+++ b/src/contrib/mlir/compiler.cpp
@@ -398,6 +398,12 @@ namespace ngraph
                            compiler.m_builder->getI64IntegerAttr(ng_node_gather->get_axis()));
                return op;
            }
+
+            template <>
+            mlir::Operation* MLIRCompiler::COMPILE_OP_DECL(ngraph::op::Relu)
+            {
+                return compiler.create_generic_op<mlir::NGReluOp>(ng_node);
+            }
        }
    }
 }