torchrec
torchrec copied to clipboard
Implement unlift_and_swap function
Summary: Implement a new swapping API which does the following:
- Takes an exported program and torchrec serializer
- Constructs torchrec modules based on serialized metadata stored in the exported program
- Unlifts the exported program to a normal nn.Module
- Partitions out nodes and swaps them out with calls to the torchrec modules
Differential Revision: D63683421