torchrec
torchrec copied to clipboard
Faster KJT init
Summary: To improve inference, we want to make creating a KJT as cheap as possible, which means the init method is nothing more than a attribute setter. All other fields are calculated lazily. This is practicularly important wrt jit script and moving between compilation units.
Differential Revision: D59765149
This pull request was exported from Phabricator. Differential Revision: D59765149
This pull request was exported from Phabricator. Differential Revision: D59765149
This pull request was exported from Phabricator. Differential Revision: D59765149
This pull request was exported from Phabricator. Differential Revision: D59765149
This pull request was exported from Phabricator. Differential Revision: D59765149
This pull request was exported from Phabricator. Differential Revision: D59765149
This pull request was exported from Phabricator. Differential Revision: D59765149
This pull request was exported from Phabricator. Differential Revision: D59765149
This pull request has been reverted by f94e27bde8c3e75f37a4086c71f4744dfba737d5.