mirror of https://github.com/llvm/torch-mlir
3260a1ea6e
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 |
||
---|---|---|
.. | ||
compile_api | ||
debug | ||
lazy_backend | ||
torchscript_e2e_test | ||
CMakeLists.txt | ||
annotations-sugar.py | ||
lit.cfg.py | ||
lit.site.cfg.py.in |