-
Notifications
You must be signed in to change notification settings - Fork 26.3k
adaptive_max_pool2d: port to structured kernel
#56317
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.
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
adaptive_max_pool2d: port to structured kernel
#56317
Conversation
[ghstack-poisoned]
💊 CI failures summary and remediationsAs of commit f4c6902 (more details on the Dr. CI page): 💚 💚 Looks good so far! There are no failures yet. 💚 💚 This comment was automatically generated by Dr. CI (expand for details).Follow this link to opt-out of these comments for your Pull Requests.Please report bugs/suggestions to the (internal) Dr. CI Users group. |
[ghstack-poisoned]
| { | ||
| Tensor output = at::empty({0}, input.options()); | ||
| Tensor indices = at::empty({0}, input.options().dtype(kLong)); | ||
| adaptive_max_pool2d_out_cpu_template( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@SplitInfinity should delete this function? same as cuda.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
and should use git rebase -i to squash these two commits into one?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'll delete it. It has already been deleted in the CUDA version.
As for squashing commits, the merging automation will take care of that.
[ghstack-poisoned]
|
@SplitInfinity merged this pull request in 0d43947. |
Summary: Pull Request resolved: pytorch#56317 Test Plan: Imported from OSS Reviewed By: ailzhang Differential Revision: D27862930 Pulled By: SplitInfinity fbshipit-source-id: e2d199df0ebaf585698f26fcfda5a0301ba67ade
adaptive_max_pool2d: port to structured kernel
Stack from ghstack:
adaptive_max_pool3d: port to structured kernel #56320 Port adaptive_max_pool3d to structured kerneladaptive_max_pool2d: port to structured kernel #56317 Port adaptive_max_pool2d to structured kernelDifferential Revision: D27862930