Skip to content

Commit d74838b

Browse files
Prune dims supported by reduce op. (#11113)
1 parent 2381249 commit d74838b

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

paddle/fluid/operators/reduce_op.h

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -135,15 +135,16 @@ class ReduceKernel : public framework::OpKernel<T> {
135135
} else {
136136
int ndim = context.Input<Tensor>("X")->dims().size();
137137
int rdim = context.Attr<std::vector<int>>("dim").size();
138-
HANDLE_DIM(6, 5);
139-
HANDLE_DIM(6, 4);
140-
HANDLE_DIM(6, 3);
141-
HANDLE_DIM(6, 2);
142-
HANDLE_DIM(6, 1);
143-
HANDLE_DIM(5, 4);
144-
HANDLE_DIM(5, 3);
145-
HANDLE_DIM(5, 2);
146-
HANDLE_DIM(5, 1);
138+
// comments for accelerating compiling temporarily.
139+
// HANDLE_DIM(6, 5);
140+
// HANDLE_DIM(6, 4);
141+
// HANDLE_DIM(6, 3);
142+
// HANDLE_DIM(6, 2);
143+
// HANDLE_DIM(6, 1);
144+
// HANDLE_DIM(5, 4);
145+
// HANDLE_DIM(5, 3);
146+
// HANDLE_DIM(5, 2);
147+
// HANDLE_DIM(5, 1);
147148
HANDLE_DIM(4, 3);
148149
HANDLE_DIM(4, 2);
149150
HANDLE_DIM(4, 1);

0 commit comments

Comments
 (0)