From b73b86836678271774790ed2d7facd1f5b1ebe5d Mon Sep 17 00:00:00 2001
From: Tomasz Patejko <tomasz.patejko@intel.com>
Date: Tue, 9 Oct 2018 10:51:51 +0200
Subject: [PATCH] MKLDNN conv + elementwise_add fusion: bias in tests made
 persistent.

test=develop
---
 .../ir/conv_elementwise_add_mkldnn_fuse_pass_tester.cc          | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git 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
index ce79a465ca..08c3b23cf3 100644
--- 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);
       }
     }
-- 
GitLab