torch-mlir/python/torch_mlir
Sean Silva 6403c0e56f torch_mlir.compile: allow custom backend_legal_ops set
Allow customizing `backend_legal_ops` for "torch" output type, since we
don't know which backend will be used (it might be a custom backend).
We don't allow customizing the `backend_legal_ops` for the other output
types (Linalg, TOSA, MHLO) since those backends control their set of
legal ops directly.

Fixes #1418
2022-10-12 04:21:22 -07:00
..
_torch_mlir_custom_op_example Re-enable custom op support 2022-08-16 22:49:08 +05:30
cmake/modules Add initial LTC backend (#610) 2022-07-30 09:40:02 -04:00
csrc Fix `fromIntArrayRef` call (#1479) 2022-10-11 13:29:07 -04:00
dialects [MLIR][TORCH] Add e2e support for `aten.Mish` op (#1470) 2022-10-11 14:03:10 -07:00
eager_mode Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
__init__.py torch_mlir.compile: allow custom backend_legal_ops set 2022-10-12 04:21:22 -07:00
compiler_utils.py Update torch-mlir-opt error message. 2022-10-05 15:02:10 -04:00