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

Jump to bottom

Add aten::avg_pool2d/avg_pool2d_backward #434

Merged

fengyuan14 merged 23 commits into main from add_avg_pool2d_op

Jun 30, 2024

Contributor

chunhuanMeng commented Jun 19, 2024

ops need to support in this pr

avg_pool2d
avg_pool2d_backward

Contributor Author

chunhuanMeng commented Jun 19, 2024

fengyuan14 changed the title ~~Add op AvgPool2d forward/backward~~ Add aten::avg_pool2d/avg_pool2d_backward

ZhiweiYan-96 force-pushed the add_avg_pool2d_op branch from fc5014d to 989eb44 Compare

June 20, 2024 08:53

huaiyuzh closed this

huaiyuzh force-pushed the add_avg_pool2d_op branch from 88c6385 to dfd679b Compare

June 24, 2024 02:36


          add AveragePool2d op

674f58a

chunhuanMeng reopened this

chunhuanMeng and others added 13 commits

June 24, 2024 02:54


          undo for merge

d89e1f3


          Merge branch 'main' into add_avg_pool2d_op

89fb9c3


          move file

5ce93ea


          Update xpu_functions.yaml

bd6bcfe


          Update xpu_functions.yaml


          Update xpu_functions.yaml

01783ef


          enable ut for averagepool2d

01c28d4


          add skip ut

e9da9df


          Remove spaces

90f63c6


          Merge branch 'main' into add_avg_pool2d_op

b8539d8


          correct file location

6326a7a


          header file location

ce6dc1e


          Merge branch 'main' into add_avg_pool2d_op

b59ae69

dvrogozh reviewed

View reviewed changes

yaml/xpu_functions.yaml Show resolved Hide resolved

chunhuanMeng added 2 commits

June 26, 2024 13:05


          Merge branch 'main' into add_avg_pool2d_op

5cf145d


          Update AveragePool2dKernels.cpp

f61c635

majing921201 reviewed

View reviewed changes

src/ATen/native/xpu/AveragePool2d.cpp Outdated Show resolved Hide resolved

majing921201 reviewed

View reviewed changes

src/ATen/native/xpu/sycl/AveragePool2dKernels.cpp Outdated Show resolved Hide resolved

majing921201 reviewed

View reviewed changes

src/ATen/native/xpu/sycl/AveragePool2dKernels.cpp Outdated Show resolved Hide resolved

majing921201 reviewed

View reviewed changes

src/ATen/native/xpu/sycl/AveragePool2dKernels.cpp Show resolved Hide resolved

majing921201 reviewed

View reviewed changes

src/ATen/native/xpu/sycl/AveragePool2dKernels.cpp Outdated Show resolved Hide resolved

majing921201 reviewed

View reviewed changes

src/ATen/native/xpu/sycl/AveragePool2dKernels.cpp Outdated Show resolved Hide resolved


          use syclMaxWorkItemsPerEU() directly

19faada

fengyuan14 marked this pull request as draft

June 29, 2024 05:43

fengyuan14 marked this pull request as ready for review

June 29, 2024 06:04

fengyuan14 added 4 commits

June 29, 2024 21:34


          Merge branch 'main' into add_avg_pool2d_op

608ca13


          Coding style

cb5b094


          Fixing compilation issues


          Merge branch 'main' into add_avg_pool2d_op

b0c3052

Contributor

fengyuan14 commented Jun 30, 2024

Aligned with CUDA. Will skip the fail case,

Contributor

fengyuan14 commented Jun 30, 2024

Aligned with CUDA. Will skip the fail case,

avg_pool1d is not listed in supported list, so the dtypes in test follow CPU definition. Retrieve the case, once avg_pool1d is enabled.

fengyuan14 approved these changes

View reviewed changes

fengyuan14 added this pull request to the merge queue

Merged via the queue into main with commit c0292ac

2 checks passed

fengyuan14 deleted the add_avg_pool2d_op branch

June 30, 2024 03:04

fengyuan14 added 2 commits

June 30, 2024 16:57


          Merge branch 'main' into add_avg_pool2d_op

ba9d6df


          Remove cases

cc5122f

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet