MKLDNN conv + elementwise_add fusion: fusion options added

7 years ago · 415b261555
parent 7c64aa0fdc
commit 415b261555
1 changed files with 2 additions and 0 deletions
--- a/paddle/fluid/framework/ir/conv_elementwise_add_mkldnn_fuse_pass.cc
+++ b/paddle/fluid/framework/ir/conv_elementwise_add_mkldnn_fuse_pass.cc
@ -81,6 +81,8 @@ graph_ptr ConvElementwiseAddMKLDNNFusePass::ApplyImpl(graph_ptr graph) const {
    GET_IR_NODE_FROM_SUBGRAPH(elementwise_add_out, elementwise_add_out,
                              elementwise_add_pattern);

+    if (FindFuseOption(conv_op, elementwise_add_op) != FUSE_MKLDNN) return;
+
    OpDesc op_desc;
    op_desc.SetType("conv2d");