@@ -578,7 +578,7 @@ __device__ __forceinline__ void ReduceModule(
578578 paddle::framework::Array<int , Rank - ReduceRank> left_dim,
579579 paddle::framework::Array<int , Rank - ReduceRank> left_strides) {
580580 // reduce_rank == 1 && reduce_dim[0] == x_dim.size() - 1
581- if (reduceType == ReduceType::kReduceLastDim ) {
581+ if (reduce_type == ReduceType::kReduceLastDim ) {
582582 ReduceLastDim<Tx, Ty, ReduceOp, TransformOp>(x, y, reducer, transformer,
583583 init, reduce_num);
584584
@@ -735,9 +735,8 @@ void TensorReduceFunctorImpl(const framework::Tensor& x, framework::Tensor* y,
735735 return ;
736736 }
737737
738- LaunchReduceKernel<Tx, Ty, ReduceOp<Tx, Ty>, TransformOp>(
739- x_data, y_data, reducer, TransformOp (config.reduce_num ),
740- reducer.initial (), stream, config);
738+ ReduceKernelImpl<Tx, Ty, ReduceOp<Tx, Ty>>(x_data, y_data, reducer,
739+ reducer.initial (), stream, config);
741740}
742741
743742template <typename Tx, template <typename , typename > class ReduceOp >
0 commit comments