feat: move data tmp in reduce sum to buf (#896)
* move data tmp in reduce sum to buf * change var * change way to calc buf size in reduce sum
Showing
- oneflow/core/kernel/reduce_sum_kernel.cpp 3 additions, 4 deletionsoneflow/core/kernel/reduce_sum_kernel.cpp
- oneflow/core/operator/reduce_sum_op.cpp 4 additions, 7 deletionsoneflow/core/operator/reduce_sum_op.cpp
- oneflow/core/operator/reduce_sum_op.h 2 additions, 1 deletiononeflow/core/operator/reduce_sum_op.h
Please register or sign in to comment