torchrec
torchrec copied to clipboard
Published
20 hours ago •
pytorch
pytorch
Small syntactic changes for dynamo compatibility
Summary: Dynamo has some gaps in support of generators, list comprehension etc. Avoiding them for now with syntactic changes
Previous diff was reverted because recat was created on the target device from the start. Then with per-item manipulations it was writing directly to device (which broke freya training as it looks like freya does not support per-item changes).
In this diff recat is created on "cpu", the same as List[int] in original version.
Differential Revision: D54192498
