MKLDNN conv + elementwise_add fusion: fusion options added

415b2615 · Tomasz Patejko · 7c64aa0f · 415b2615
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/framework/ir/conv_elementwise_add_mkldnn_fuse_pass.cc ...uid/framework/ir/conv_elementwise_add_mkldnn_fuse_pass.cc +2 -0

未找到文件。
--- 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");