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

Add bounds check in prefetch in TBE training #3015

Closed sryap closed 3 weeks ago

sryap commented 3 weeks ago

Summary: This diff adds bounds checking in SplitTBE training's prefetch. The operator is wrapped inside the prepare_inputs function which does input tensor type casting and VBE metadata generation.

Differential Revision: D61524189

netlify[bot] commented 3 weeks ago

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
Latest commit 97d69f062496cac734a6e902941dd9384d0855d7
Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66c4d1e05d131000080704a7
Deploy Preview https://deploy-preview-3015--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 3 weeks ago

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

facebook-github-bot commented 3 weeks ago

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

facebook-github-bot commented 3 weeks ago

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

facebook-github-bot commented 3 weeks ago

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

facebook-github-bot commented 3 weeks ago

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

facebook-github-bot commented 3 weeks ago

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