Dev maxpool series op 123d (#5244)
* add torch maxpool2d cpu kernel implementation * refine cpu implementation * op gpu implemetation * bug-fix * refine * add pooling kernel util * reimplementation of cpu kernel * gpu kernel implementation * gpu kernel implementation * bug fix * change indice index * change indice index of grad compute * add flag: return_indices * format code * update inteface param * add param dilation * bug-fix * update padding param * add maxpool2d test case * format code * bug fix * update tese case * update sbp of indice * add code sample * add new line * use oneflow_api.BlobDesc * add op torch maxpool1d * add sample codes * change dtype of indice * add torch maxpool3d interface * add implementation of maxpool3d * reformat codes * add atomic add * bug-fix * backward bug fix * update search index * format code * update imterface params * add sample code of maxpool3d * refine and add test case * support ellipsis type slice * refine * refine * support slice assign ellipsis type * refine * register fn to localtensor * merge master * update maxpooling ops * refactor maxpool2d op * add ceil mode testcase * rm useless code * refine * refactor maxpool3d op and test cases * refine * add maxpool1d implementation * simplify params using struct * fix sbp fn * del notes * refine * refine * add pooling functional api impl * refine * update sbp fn * refin thread num * refine * auto format by CI * rm bn test for debug * limit registers num * remove double dtype * . * auto format by CI * update data type * refine * add cuda synchronize and error string capture * move synchronize placement * debug all kernel compute * debug * debug * debug * debug * auto format by CI * debug std infinity * debug * add numeric utils * format * refine * add reference * remove useless header * move namespace to detail * fix doctest error Co-authored-by:oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com> Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org>
Showing
- oneflow/core/autograd/gradient_funcs/pooling.cpp 120 additions, 0 deletionsoneflow/core/autograd/gradient_funcs/pooling.cpp
- oneflow/core/functional/functional_api.yaml 23 additions, 0 deletionsoneflow/core/functional/functional_api.yaml
- oneflow/core/functional/impl/nn_functor.cpp 45 additions, 0 deletionsoneflow/core/functional/impl/nn_functor.cpp
- oneflow/core/functional/impl/nn_grad_functor.cpp 52 additions, 0 deletionsoneflow/core/functional/impl/nn_grad_functor.cpp
- oneflow/core/kernel/util/numeric_limits.cuh 128 additions, 0 deletionsoneflow/core/kernel/util/numeric_limits.cuh
- oneflow/core/kernel/util/numerics.cuh 249 additions, 0 deletionsoneflow/core/kernel/util/numerics.cuh
- oneflow/python/nn/modules/pooling.py 110 additions, 59 deletionsoneflow/python/nn/modules/pooling.py
- oneflow/python/ops/nn_ops.py 420 additions, 0 deletionsoneflow/python/ops/nn_ops.py
- oneflow/python/test/modules/test_pooling.py 269 additions, 17 deletionsoneflow/python/test/modules/test_pooling.py
- oneflow/user/kernels/pooling_kernel.cpp 252 additions, 0 deletionsoneflow/user/kernels/pooling_kernel.cpp
- oneflow/user/kernels/pooling_kernel.cu 148 additions, 0 deletionsoneflow/user/kernels/pooling_kernel.cu
- oneflow/user/kernels/pooling_kernel_util.cpp 115 additions, 0 deletionsoneflow/user/kernels/pooling_kernel_util.cpp
- oneflow/user/kernels/pooling_kernel_util.h 277 additions, 0 deletionsoneflow/user/kernels/pooling_kernel_util.h
- oneflow/user/ops/pooling_op.cpp 219 additions, 0 deletionsoneflow/user/ops/pooling_op.cpp
Please register or sign in to comment