mirror of https://github.com/llvm/torch-mlir
af9e8a5e63
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! |
||
|---|---|---|
| .. | ||
| test | ||
| torch_mlir | ||
| torch_mlir_e2e_test | ||
| CMakeLists.txt | ||
| TorchMLIRModule.cpp | ||