torch-mlir/python/torch_mlir
Jiawei Wu c7fa42b7d3
[Torch Dialect] Add canonicalizer for aten.to.other op (#2273)
Canonicalize aten.to.other to prim.device + prim.dtype + aten.to.device
Co-authored-by: wujiawei.aml <wujiawei.aml@bytedance.com>
2023-06-30 09:43:08 +08:00
..
_torch_mlir_custom_op_example [custom op] Generalize shape library logic to work with dtypes (#1594) 2022-12-13 08:25:41 -08:00
cmake/modules Add initial LTC backend (#610) 2022-07-30 09:40:02 -04:00
csrc add mse_loss_backward (#2111) 2023-05-12 14:29:13 -07:00
dialects [Torch Dialect] Add canonicalizer for aten.to.other op (#2273) 2023-06-30 09:43:08 +08:00
__init__.py Pure-Python FX importer. (#2098) 2023-05-12 00:46:33 -05:00
_dynamo_fx_importer.py Bump LLVM (#2176) 2023-06-13 16:17:23 +02:00
_version.py Fix version comparison against stable (#2209) 2023-06-07 10:19:38 +02:00
compiler_utils.py [Dynamo] turn on `no_python=True` for dynamo tests (#2040) 2023-04-28 18:05:17 -05:00
dynamo.py Fix version comparison against stable (#2209) 2023-06-07 10:19:38 +02:00