torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Prefetch for ITEP v8

Open henrylhtsang opened this issue 1 year ago • 2 comments

Summary: design doc: https://docs.google.com/document/d/17_nqdEtH6B_ev9Gnuw2mpgtFq4dqzC6-XUdw18R4F8Q/

changes

in ITEP

make remap a callback of input dist

in torchrec

copy callbacks from input dist to fused input dist

impact

For non-ITEP modules, this should not change anything

Differential Revision: D57012790

henrylhtsang avatar May 10 '24 17:05 henrylhtsang

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

facebook-github-bot avatar May 10 '24 17:05 facebook-github-bot

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

facebook-github-bot avatar May 10 '24 21:05 facebook-github-bot