torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Permute export fix

Open PaulZhang12 opened this issue 9 months ago • 1 comments

Summary: Make KJT permute work in edge cases for torch.export: https://fb.workplace.com/groups/6829516587176185/posts/7206415192819654/?comment_id=7207715762689597&reply_comment_id=7210658785728628.

A cleaner solution than before as well. Necessary for enabling PT2 eager model processing on AIMP

Differential Revision: D57162331

PaulZhang12 avatar May 10 '24 19:05 PaulZhang12

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

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