From 980a6753a85b4e5a962a2662864c5abf351502b5 Mon Sep 17 00:00:00 2001 From: Tao Luo Date: Wed, 14 Nov 2018 18:46:23 +0800 Subject: [PATCH] fix typo to pass the ci test=develop --- paddle/fluid/framework/ir/fc_fuse_pass_tester.cc | 1 + paddle/fluid/operators/fc_op.cc | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/framework/ir/fc_fuse_pass_tester.cc b/paddle/fluid/framework/ir/fc_fuse_pass_tester.cc index 2db7d95cae1..4e1e4e27f9b 100644 --- a/paddle/fluid/framework/ir/fc_fuse_pass_tester.cc +++ b/paddle/fluid/framework/ir/fc_fuse_pass_tester.cc @@ -29,6 +29,7 @@ void SetOp(ProgramDesc* prog, const std::string& type, if (type == "mul") { op->SetInput("X", {inputs[0]}); op->SetInput("Y", {inputs[1]}); + op->SetAttr("x_num_col_dims", {1}); } else if (type == "elementwise_add") { op->SetInput("X", inputs); } diff --git a/paddle/fluid/operators/fc_op.cc b/paddle/fluid/operators/fc_op.cc index 1f1c5823df2..e80249fc878 100644 --- a/paddle/fluid/operators/fc_op.cc +++ b/paddle/fluid/operators/fc_op.cc @@ -121,7 +121,7 @@ void FCOpMaker::Make() { AddInput("W", "(Tensor), The weight fc op with shape (I, O)."); AddInput("Bias", "(Tensor, optional) Bias vector with shape (1 x O") .AsDispensable(); - AddAttr("x_num_col_dims", + AddAttr("in_num_col_dims", "(int, default 1), The fc op can take tensors with more than " "two dimensions as its inputs.") .SetDefault(1) -- GitLab