torch-mlir/python/torch_mlir
Henry Tu a605fe279c Add example Torch MLIR LTC Backend (#725) 2022-07-30 09:40:02 -04:00
..
_torch_mlir_custom_op_example Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
cmake/modules Add initial LTC backend (#610) 2022-07-30 09:40:02 -04:00
csrc Add example Torch MLIR LTC Backend (#725) 2022-07-30 09:40:02 -04:00
dialects Added JIT to MLIR lowering (#724) 2022-07-30 09:40:02 -04:00
eager_mode Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
__init__.py [MHLO] Init MHLO integration. (#1083) 2022-07-20 16:18:16 -07:00
compiler_utils.py Add named exception TorchMlirCompilerError. (#1064) 2022-07-15 16:32:36 -05:00