Sean Silva af9e8a5e63 [torchdynamo] Move to aot_autograd instead of raw make_fx ... As [@ezyang suggested](https://github.com/pytorch/pytorch/issues/90276#issuecomment-1339791275), use `torch._dynamo.optimizations.training.aot_autograd` instead of raw `make_fx`. This is more future proof and gives us the backward pass and functionalization. We don't currently get functionalization because of https://github.com/pytorch/pytorch/issues/90759 This also incidentally fixes the source location handling, which makes `lockstep_basic.py` give an accurate source location!		2022-12-15 01:55:50 -08:00
..
lockstep_basic.py	[torchdynamo] Move to aot_autograd instead of raw make_fx	2022-12-15 01:55:50 -08:00