torch-mlir/python/torch_mlir
Vivek Khandelwal fd236b2c89 [MLIR][TORCH] Add decomposition for prims.var and prims.sqrt op
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>
2023-01-11 17:39:10 +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 decomposition for prims.var and prims.sqrt op 2023-01-11 17:39:10 +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] Move to aot_autograd instead of raw make_fx 2022-12-15 01:55:50 -08:00