Skip to content

Commit 6517d84

Browse files
committed
fix
1 parent 17bdb4f commit 6517d84

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

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

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -407,9 +407,6 @@ struct ReduceConfig {
407407
grid_dim->x = grid_num;
408408
grid_dim->y = std::max(std::min(input_split_num_1, input_split_num_3),
409409
input_split_num_2);
410-
411-
blocking_size = detail::AlignUp(reduce_num_per_thread, grid_dim->y);
412-
413410
// if grid.y > 1, we need launch reduce kernel again.
414411
if (grid_dim->y > 1) {
415412
should_reduce_again = true;
@@ -621,7 +618,7 @@ __device__ void ReduceHigherDim(const Tx* x, Ty* y, ReduceOp reducer,
621618
template <typename Tx, typename Ty, typename ReduceOp, typename TransformOp>
622619
__device__ void ReduceAny(const Tx* x, Ty* y, ReduceOp reducer,
623620
TransformOp transformer, Ty init, int reduce_num,
624-
int left_num, int block_size, bool reduce_lastdim,
621+
int left_num, bool reduce_lastdim,
625622
const IndexCalculator& reduce_index_calculator,
626623
const IndexCalculator& left_index_calculator) {
627624
int input_idx, left_idx, stride;
@@ -721,8 +718,8 @@ __device__ void ReduceModule(const Tx* x, Ty* y, ReduceOp reducer,
721718
// reduce_rank >= 2
722719
} else {
723720
ReduceAny<Tx, Ty, ReduceOp, TransformOp>(
724-
x, y, reducer, transformer, init, reduce_num, left_num, blocking_size,
725-
reduce_lastdim, reduce_index_calculator, left_index_calculator);
721+
x, y, reducer, transformer, init, reduce_num, left_num, reduce_lastdim,
722+
reduce_index_calculator, left_index_calculator);
726723
}
727724
}
728725

0 commit comments

Comments
 (0)