Skip to content

Commit 8ba58eb

Browse files
authored
add AsExtra to partial_sum op (#35381)
add AsExtra to the following attribute of partial_sum op: 1. use_mkldnn
1 parent 42d3650 commit 8ba58eb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddle/fluid/operators/partial_sum_op.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,8 @@ class PartialSumOpMaker : public framework::OpProtoAndCheckerMaker {
143143
AddAttr<bool>(
144144
"use_mkldnn",
145145
"(bool, default false) Indicates if MKL-DNN kernel will be used")
146-
.SetDefault(false);
146+
.SetDefault(false)
147+
.AsExtra();
147148
AddAttr<int>("start_index", "The start index of tensor wanted to be added.")
148149
.SetDefault(0);
149150
AddAttr<int>("length", "The length of tensor wanted to be added.")

0 commit comments

Comments
 (0)