From 8fb29b2ca98164c15e6253001a5fd906ef90f792 Mon Sep 17 00:00:00 2001 From: Tomasz Patejko Date: Fri, 28 Sep 2018 13:57:33 +0200 Subject: [PATCH] MKLDNN conv + elementwise_add fusion: new nodes marked as input or output test=develop --- paddle/fluid/framework/ir/graph_pattern_detector.cc | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/framework/ir/graph_pattern_detector.cc b/paddle/fluid/framework/ir/graph_pattern_detector.cc index f6c8609fd71..6d524651e0b 100644 --- a/paddle/fluid/framework/ir/graph_pattern_detector.cc +++ b/paddle/fluid/framework/ir/graph_pattern_detector.cc @@ -1003,15 +1003,19 @@ PDNode *patterns::Conv::operator()() { auto conv_op = pattern->NewNode(conv_op_repr())->assert_is_op("conv2d"); auto input_var = pattern->NewNode(conv_input_repr()) + ->AsInput() ->assert_is_op_input("conv2d", "Input"); - auto bias_var = - pattern->NewNode(conv_bias_repr())->assert_is_op_input("conv2d", "Bias"); + auto bias_var = pattern->NewNode(conv_bias_repr()) + ->AsInput() + ->assert_is_op_input("conv2d", "Bias"); auto filter_var = pattern->NewNode(conv_filter_repr()) + ->AsInput() ->assert_is_op_input("conv2d", "Filter"); auto output_var = pattern->NewNode(conv_output_repr()) + ->AsOutput() ->assert_is_op_output("conv2d", "Output"); conv_op->LinksFrom({input_var, bias_var, filter_var}); @@ -1025,6 +1029,7 @@ PDNode *patterns::ElementwiseAdd::operator()(PDNode *y_var) { ->assert_is_op("elementwise_add"); auto x_var = pattern->NewNode(elementwise_add_x_repr()) + ->AsInput() ->assert_is_op_input("elementwise_add", "X"); y_var->assert_is_op_input("elementwise_add", "Y"); -- GitLab