pytorch / FBGEMM

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

Support original indices for FBGEMM block bucketization flag #2999

Closed dstaay-fb closed 2 months ago

dstaay-fb commented 2 months ago

Summary: For ZCH we want to operate in global id space; which requires expanding fbgemm block bucketization kernels to operate on global ids, not local. Purposed 'maybe_keep_orig_idx' -> 'keep_orig_idx'

Differential Revision: D61102970

netlify[bot] commented 2 months ago

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
Latest commit c9a55ce36c92fb9a3ab8f25d7dd1fd5b3a6ecc61
Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66bd536fb59108000762b517
Deploy Preview https://deploy-preview-2999--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 2 months ago

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

facebook-github-bot commented 2 months ago

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

facebook-github-bot commented 2 months ago

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

facebook-github-bot commented 2 months ago

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