Skip to content

Commit 415b261

Browse files
author
Tomasz Patejko
committed
MKLDNN conv + elementwise_add fusion: fusion options added
1 parent 7c64aa0 commit 415b261

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/fluid/framework/ir/conv_elementwise_add_mkldnn_fuse_pass.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,8 @@ graph_ptr ConvElementwiseAddMKLDNNFusePass::ApplyImpl(graph_ptr graph) const {
8181
GET_IR_NODE_FROM_SUBGRAPH(elementwise_add_out, elementwise_add_out,
8282
elementwise_add_pattern);
8383

84+
if (FindFuseOption(conv_op, elementwise_add_op) != FUSE_MKLDNN) return;
85+
8486
OpDesc op_desc;
8587
op_desc.SetType("conv2d");
8688

0 commit comments

Comments
 (0)