torch-mlir/python/torch_mlir
Vivek Khandelwal 68f568b704 [MLIR][TORCH] Add E2E support for prims.convert_element_type op
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-22 09:36:36 +05:30
..
_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 Update TorchMlirBackendImpl Methods (#1580) 2022-11-14 00:37:49 -05:00
dialects [MLIR][TORCH] Add E2E support for prims.convert_element_type op 2022-11-22 09:36:36 +05:30
eager_mode Fix https://github.com/llvm/torch-mlir/issues/1618 by stripping `requires_grad` from results of view ops. (#1624) 2022-11-21 19:15:53 -06:00
__init__.py [torch_mlir.compile] Handle the case of already-scripted models better 2022-11-16 10:47:13 -08:00
compiler_utils.py Update torch-mlir-opt error message. 2022-10-05 15:02:10 -04:00