torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Update PositionWeightedModule to make it jit trace compitable

Open ZhengkaiZ opened this issue 1 year ago • 1 comments

Summary: In

torch.ops.fbgemm.offsets_range(features[key].offsets().long(), torch.numel(features[key].values())

This part

torch.numel(features[key].values()

will be traced into constant

Reviewed By: snabelkabiya

Differential Revision: D53744703

ZhengkaiZ avatar Feb 23 '24 23:02 ZhengkaiZ

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

facebook-github-bot avatar Feb 23 '24 23:02 facebook-github-bot