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

test=develop

MKLDNN conv + elementwise_add fusion: bias in tests made persistent.
test=develop
b73b8683 · Tomasz Patejko · a1fa2032 · b73b8683
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/fluid/framework/ir/conv_elementwise_add_mkldnn_fuse_pass_tester.cc ...mework/ir/conv_elementwise_add_mkldnn_fuse_pass_tester.cc +1 -1

未找到文件。
--- 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);
      }
    }