Skip to content

Commit 968a56b

Browse files
authored
Merge pull request #13373 from sfraczek/conv-relu-pass-hotfix
hotfix for conv-relu pass
2 parents 323d017 + dd149d4 commit 968a56b

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

paddle/fluid/framework/ir/conv_relu_mkldnn_fuse_pass.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ std::unique_ptr<ir::Graph> ConvReLUFusePass::ApplyImpl(
5858
desc.SetInput("Input", std::vector<std::string>({conv_relu_i_in}));
5959
desc.SetInput("Filter", std::vector<std::string>({conv_relu_w_in}));
6060
desc.SetInput("Bias", std::vector<std::string>({conv_relu_b_in}));
61-
desc.SetOutput("Out", std::vector<std::string>({conv_relu_out}));
61+
desc.SetOutput("Output", std::vector<std::string>({conv_relu_out}));
6262
desc.SetType("conv2d");
6363
for (auto& attr : conv->Op()->GetAttrMap()) {
6464
desc.SetAttr(attr.first, attr.second);

paddle/fluid/inference/analysis/analyzer.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,9 @@ class Analyzer : public OrderedRegistry<PassManager> {
7272
"mul_gru_fuse_pass", //
7373
"seq_concat_fc_fuse_pass", //
7474
"fc_fuse_pass", //
75+
#ifdef PADDLE_WITH_MKLDNN
76+
"conv_relu_mkldnn_fuse_pass", //
77+
#endif
7578
}};
7679

7780
std::unordered_set<std::string> disabled_ir_passes_;

0 commit comments

Comments
 (0)