torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Add SimpleFSDP to leaf node

Open Microve opened this issue 8 months ago • 2 comments

Summary: Torchrec should not trace into SimpleFSDP which will cause failures. We thus added it to the leaf node

Differential Revision: D71450848

Microve avatar Mar 19 '25 19:03 Microve