Update sparse_ops.py to use generic gpu target fbgemm_gpu:input_combine to support both nvidia and AMD

pytorch / FBGEMM

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

Other

1.17k stars 474 forks source link

Closed joebos closed 1 month ago

joebos commented 1 month ago

Summary: Update sparse_ops.py to use generic gpu target fbgemm_gpu:input_combine to support both nvidia and AMD

Differential Revision: D60271751

netlify[bot] commented 1 month ago

Name	Link
Latest commit	687d55ed1cc9d8ef363b436c645bbd8a698214ef
Latest deploy log	https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66a3c1d0dca98900089eb6a1
Deploy Preview	https://deploy-preview-2905--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 1 month ago

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

facebook-github-bot commented 1 month ago

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