Skip to content

Commit

Permalink
Optimize nchw MaxPooling (#7426)
Browse files Browse the repository at this point in the history
* first debug

* fix maxpool

* fix bug

* remove redundant code

* remove redundant read

* remove redundant data_ptr offset

* use int32 to describe x shape

* Fix cuda input params for maxpool2d

* just for debug

* just for profile

* reduce div

* use int32_t indice

* revert back to use int64_t

* fix maxpool1d 3d

* optimize backward

* fix all optimize. TODO: NHWC

* fix comment

Co-authored-by: oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
  • Loading branch information
2 people authored and marigoold committed Mar 15, 2022
1 parent 2b4fea6 commit 1bc08d3
Show file tree
Hide file tree
Showing 4 changed files with 460 additions and 431 deletions.
Loading

0 comments on commit 1bc08d3

Please sign in to comment.