@@ -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,
621618template <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