Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Optimize nchw MaxPooling #7426

Merged
merged 21 commits into from
Feb 21, 2022
Merged

Optimize nchw MaxPooling #7426

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
7c2d1af
first debug
MARD1NO Feb 4, 2022
f2375f7
fix maxpool
MARD1NO Feb 7, 2022
79ff4db
fix bug
MARD1NO Feb 7, 2022
14c9fd6
remove redundant code
MARD1NO Feb 7, 2022
4f5e6f4
remove redundant read
MARD1NO Feb 7, 2022
d9fb335
remove redundant data_ptr offset
MARD1NO Feb 7, 2022
3b79ba8
use int32 to describe x shape
MARD1NO Feb 8, 2022
4f8bba5
Fix cuda input params for maxpool2d
MARD1NO Feb 8, 2022
6b6a8e3
just for debug
MARD1NO Feb 8, 2022
385c962
just for profile
MARD1NO Feb 8, 2022
1c39ee2
reduce div
MARD1NO Feb 8, 2022
5495891
use int32_t indice
MARD1NO Feb 8, 2022
f62ac8f
revert back to use int64_t
MARD1NO Feb 8, 2022
1d8843f
fix maxpool1d 3d
MARD1NO Feb 9, 2022
36a6fc8
optimize backward
MARD1NO Feb 9, 2022
97cbec0
fix all optimize. TODO: NHWC
MARD1NO Feb 9, 2022
cc31a14
fix comment
MARD1NO Feb 21, 2022
83e0c98
Merge branch 'master' into optimize_nchw_pool
MARD1NO Feb 21, 2022
46a298d
Merge branch 'master' into optimize_nchw_pool
oneflow-ci-bot Feb 21, 2022
80344d4
Merge branch 'master' into optimize_nchw_pool
oneflow-ci-bot Feb 21, 2022
5d09693
Merge branch 'master' into optimize_nchw_pool
oneflow-ci-bot Feb 21, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading