FBGEMM icon indicating copy to clipboard operation
FBGEMM copied to clipboard

Add CPU group_index_select fwd and bwd impl

Open spcyppt opened this issue 1 year ago • 2 comments

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

spcyppt avatar Oct 23 '24 17:10 spcyppt

This pull request was exported from Phabricator. Differential Revision: D64556901

facebook-github-bot avatar Oct 23 '24 17:10 facebook-github-bot

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...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site configuration.

netlify[bot] avatar Oct 23 '24 17:10 netlify[bot]

This pull request was exported from Phabricator. Differential Revision: D64556901

facebook-github-bot avatar Oct 24 '24 06:10 facebook-github-bot

This pull request has been merged in pytorch/FBGEMM@fbf0096bd42a84ea01cec86edd15409c84768ac8.

facebook-github-bot avatar Oct 24 '24 17:10 facebook-github-bot