Skip to content
Snippets Groups Projects
Unverified Commit a5bee493 authored by Luyang's avatar Luyang Committed by GitHub
Browse files

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: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Co-authored-by: default avataroneflow-ci-bot <ci-bot@oneflow.org>
parent 097f5f99
No related branches found
No related tags found
No related merge requests found
Showing
with 2427 additions and 76 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment