FBGEMM
FBGEMM copied to clipboard
Add CPU group_index_select fwd and bwd impl
Summary: X-link: https://github.com/facebookresearch/FBGEMM/pull/369
The GIS op interfaces are unified for all device backends, but the CPU group_index_select_dim0 is still using the decomposed version due to lack of fused impl for fwd/bwd.
This diff implements the following functions on CPU:
- group_index_select_dim0_forward_impl_cpu
- group_index_select_dim0_backward_impl_cpu
Reviewed By: q10, egienvalue
Differential Revision: D64556901
This pull request was exported from Phabricator. Differential Revision: D64556901
Deploy Preview for pytorch-fbgemm-docs ready!
| Name | Link |
|---|---|
| Latest commit | f09686c06914f46c05962b1ae33166dcab875e9d |
| Latest deploy log | https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/6719e2f5acd4d50008d4a788 |
| Deploy Preview | https://deploy-preview-3273--pytorch-fbgemm-docs.netlify.app |
| Preview on mobile | Toggle QR Code...Use your smartphone camera to open QR code link. |
To edit notification comments on pull requests, go to your Netlify site configuration.
This pull request was exported from Phabricator. Differential Revision: D64556901
This pull request has been merged in pytorch/FBGEMM@fbf0096bd42a84ea01cec86edd15409c84768ac8.