Skip to content

Commit 66ec9bf

Browse files
committed
update
1 parent 3e10bd2 commit 66ec9bf

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

paddle/fluid/operators/reduce_ops/reduce_op.cu.h

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -344,8 +344,7 @@ struct ReduceConfig {
344344
if (rank == reduce_rank ||
345345
rank == 2 && reduce_rank == 1 && reduce_dim[0] == 1) {
346346
reduce_type = static_cast<int>(ReduceType::kReduceLastDim);
347-
} else if (reduce_rank == 1 &&
348-
((rank == 2 && is_large_enough) || rank != 2)) {
347+
} else if (reduce_rank == 1) {
349348
// ReduceFirstDim and reduceSecondDim
350349
reduce_type = static_cast<int>(ReduceType::kReduceHigherDim);
351350
} else {

0 commit comments

Comments
 (0)