support gpu to cpu gather in merge pooled emb

pytorch / FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

Other

1.18k stars 486 forks source link

Closed 842974287 closed 3 months ago

842974287 commented 3 months ago

Summary: Previously we don't support gathering gpu tensors into cpu. This diff adds the support.

Differential Revision: D58954707

netlify[bot] commented 3 months ago

Name	Link
Latest commit	5da54fd22f7cd4a53919918fa69763bca6b848af
Latest deploy log	https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/6679ae5c40e7c300098893f6
Deploy Preview	https://deploy-preview-2776--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.

facebook-github-bot commented 3 months ago

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

facebook-github-bot commented 3 months ago

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