MKLDNN conv + elementwise_add fusion: bias in test marked as persistable

6 years ago · a27a8c5da8
parent af8c71317c
commit a27a8c5da8
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/framework/ir/conv_elementwise_add_mkldnn_fuse_pass_tester.cc
+++ b/paddle/fluid/framework/ir/conv_elementwise_add_mkldnn_fuse_pass_tester.cc
@ -103,7 +103,7 @@ TEST(ConvElementwiseAddMKLDNNFusePass, ConvolutionWithElementwiseAddRelu) {
             {"a", "b", "bias", "weights", "c", "d", "e", "f"})) {
      auto* var = prog.MutableBlock(0)->Var(v);
      var->SetType(proto::VarType::LOD_TENSOR);
-      if (v == "weights") {
+      if (v == "weights" || v == "bias") {
        var->SetPersistable(true);
      }
    }