Skip to content

Commit 980a675

Browse files
committed
fix typo to pass the ci
test=develop
1 parent e0d4e04 commit 980a675

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

paddle/fluid/framework/ir/fc_fuse_pass_tester.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ void SetOp(ProgramDesc* prog, const std::string& type,
2929
if (type == "mul") {
3030
op->SetInput("X", {inputs[0]});
3131
op->SetInput("Y", {inputs[1]});
32+
op->SetAttr("x_num_col_dims", {1});
3233
} else if (type == "elementwise_add") {
3334
op->SetInput("X", inputs);
3435
}

paddle/fluid/operators/fc_op.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ void FCOpMaker::Make() {
121121
AddInput("W", "(Tensor), The weight fc op with shape (I, O).");
122122
AddInput("Bias", "(Tensor, optional) Bias vector with shape (1 x O")
123123
.AsDispensable();
124-
AddAttr<int>("x_num_col_dims",
124+
AddAttr<int>("in_num_col_dims",
125125
"(int, default 1), The fc op can take tensors with more than "
126126
"two dimensions as its inputs.")
127127
.SetDefault(1)

0 commit comments

Comments
 (0)