From a27a8c5da8384a8d3d6a4334a412cf54ad9eec1b Mon Sep 17 00:00:00 2001
From: Tomasz Patejko <tomasz.patejko@intel.com>
Date: Fri, 28 Sep 2018 13:41:41 +0200
Subject: [PATCH] MKLDNN conv + elementwise_add fusion: bias in test marked as
 persistable

---
 .../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 3d37398076d..ce79a465ca7 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
@@ -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);
       }
     }
-- 
GitLab