We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e10bd2 commit 66ec9bfCopy full SHA for 66ec9bf
paddle/fluid/operators/reduce_ops/reduce_op.cu.h
@@ -344,8 +344,7 @@ struct ReduceConfig {
344
if (rank == reduce_rank ||
345
rank == 2 && reduce_rank == 1 && reduce_dim[0] == 1) {
346
reduce_type = static_cast<int>(ReduceType::kReduceLastDim);
347
- } else if (reduce_rank == 1 &&
348
- ((rank == 2 && is_large_enough) || rank != 2)) {
+ } else if (reduce_rank == 1) {
349
// ReduceFirstDim and reduceSecondDim
350
reduce_type = static_cast<int>(ReduceType::kReduceHigherDim);
351
} else {
0 commit comments