torch-mlir/python/test
Ramiro Leal-Cavazos 3260a1ea6e
Allow passing traced `torch.nn.Module`s into `torch_mlir.compile` (#1743)
This commit adds support for passing to `torch_mlir.compile` the
result of running `torch.jit.trace` on a model by relaxing the
condition that checks if the model is already in JIT IR to allow any
`torch.jit.ScriptModule`.

Fixes https://github.com/llvm/torch-mlir/issues/1739
2022-12-22 08:39:55 -08:00
..
compile_api Allow passing traced `torch.nn.Module`s into `torch_mlir.compile` (#1743) 2022-12-22 08:39:55 -08:00
debug [torchdynamo] Move to aot_autograd instead of raw make_fx 2022-12-15 01:55:50 -08:00
lazy_backend Fix LTC lib_torch_mlir_ltc.so import error (#1283) 2022-08-25 18:25:01 -04:00
torchscript_e2e_test Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00
annotations-sugar.py Remove "torchscript" association from the e2e framework. 2022-08-29 14:10:03 -07:00
lit.cfg.py Miscellaneous fixes for Windows builds (#1376) 2022-09-29 12:07:43 -05:00
lit.site.cfg.py.in Dual license the torch-mlir project. 2021-10-01 10:46:08 -07:00