MKLDNN conv + elementwise_add fusion: bias in tests made persistent.

test=develop
6 years ago · b73b868366
parent a1fa203287
commit b73b868366
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
@ -204,7 +204,7 @@ TEST(ConvElementwiseAddMKLDNNFusePass, SigmoidConvolutionAddElementwiseRelu) {
             {"a", "b", "bias", "weights", "c", "d", "e", "f"})) {
      auto* var = prog.MutableBlock(0)->Var(v);
      var->SetType(proto::VarType::LOD_TENSOR);
-      if (v.find("weights")) {
+      if (v.find("weights") || v.find("bias")) {
        var->SetPersistable(true);
      }
    }