torch-mlir/python/torch_mlir
Vivek Khandelwal abf4f207cd [MLIR][TORCH] Add canonicalizer for aten.new_empty_strided op
Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2023-01-19 13:37:32 +05:30
..
_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 Unrevert #1724 (#1737) 2022-12-20 11:17:21 -05:00
dialects [MLIR][TORCH] Add canonicalizer for aten.new_empty_strided op 2023-01-19 13:37:32 +05:30
__init__.py Allow passing traced `torch.nn.Module`s into `torch_mlir.compile` (#1743) 2022-12-22 08:39:55 -08:00
compiler_utils.py Update torch-mlir-opt error message. 2022-10-05 15:02:10 -04:00
dynamo.py [torchdynamo] Add native_group_norm and split op to the decomp list 2023-01-18 10:40:46 +05:30