pytorch / FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Other
1.17k stars 474 forks source link

Add masked_index_select and refactor masked_index_put #2910

Closed sryap closed 1 month ago

sryap commented 1 month ago

Summary: This diff adds torch.ops.fbgemm.masked_index_select (will be used in subsequent diffs) and refactors torch.ops.fbgemm.masked_index_put.

Differential Revision: D60362812

netlify[bot] commented 1 month ago

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
Latest commit 348425d1f4b740a42a17c9d9de9c57aa37f75b3d
Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66a825095d7ddf0008584413
Deploy Preview https://deploy-preview-2910--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.

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

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