torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

add bag reverse mapping for block_bucketize kernel

Open YazhiGao opened this issue 1 year ago • 2 comments

Summary: X-link: https://github.com/pytorch/FBGEMM/pull/2477

  • we add the per indices bucketize bag reverse index return for inference batching needs.

Differential Revision: D55492793

YazhiGao avatar Apr 03 '24 04:04 YazhiGao

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

facebook-github-bot avatar Apr 03 '24 04:04 facebook-github-bot