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 |
||
---|---|---|
.. | ||
already_scripted.py | ||
already_traced.py | ||
backend_legal_ops.py | ||
basic.py | ||
make_fx.py | ||
multiple_methods.py | ||
output_type_spec.py | ||
tracing.py |