torch-mlir/e2e_testing
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
..
main.py [cleanup] Use `"` instead of `'` for string literals 2022-12-12 02:40:09 -08:00
xfail_sets.py [torchdynamo] Move to aot_autograd instead of raw make_fx 2022-12-15 01:55:50 -08:00