search

pytorch / FBGEMM

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

Allow `block_bucketize_sparse_features` to return original idx #2925

Closed dracifer closed 1 month ago

dracifer commented 1 month ago

Summary: current operator will do new_idx = orig_idx / world_size even when we are in 0-collision hashing path. Add the capability of just returning the original idx in this case with a flag to control. default should not change existing logic.

Reviewed By: bixue2010

Differential Revision: D60526038

netlify[bot] commented 1 month ago

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
Latest commit d67ee6d3d2a2b795e617cf4a86aae0e092930ba8
Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66b13a650cebbc0008c61692
Deploy Preview https://deploy-preview-2925--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: D60526038

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

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