torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Replace `constraints` with `dynamic_shapes` in apf

Open BoyuanFeng opened this issue 1 year ago • 1 comments

Summary: constraints argument for torch.export has been deprecated in favor of the dynamic_shapes argument. This PR updates the use of the deprecated API in apf. In addition, this diff registers DefaultRecTrainInput as a pytree node to make it work with torch.export.dynamic_shapes.process_dynamic_shapes().

Differential Revision: D53697239

BoyuanFeng avatar Feb 14 '24 00:02 BoyuanFeng

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

facebook-github-bot avatar Feb 14 '24 03:02 facebook-github-bot