torch-mlir

Commit Graph

Author	SHA1	Message	Date
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

Author

SHA1

Message

Date

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

1 Commits (4199feffed3f9b89e321a8bd65f2b78bc5be2700)