torch-mlir/python/torch_mlir
Ramiro Leal-Cavazos 11148e60d6
Undo shape lib changes + update function signature of sum + zero (#1035)
This commit does three things:
  1. Reverts some of the shape lib changes merged in
  https://github.com/llvm/torch-mlir/pull/844
  2. Updates the signature of `aten.sum_dim_IntList` that was recently
  updated in
  23bdb570cf
  3. Replaces `aten.zero.functional` with `aten.zero`, updated in 960758b0b7
2022-07-11 10:56:12 -07:00
..
_torch_mlir_custom_op_example Allow torch-mlir to support PyTorch extensions. (#895) 2022-06-13 14:51:30 -07:00
dialects Undo shape lib changes + update function signature of sum + zero (#1035) 2022-07-11 10:56:12 -07:00
eager_mode Link against libtorch (#955) 2022-06-30 12:40:17 -07:00
__init__.py torch_mlir.compile: Allow OutputType as a string. 2022-07-08 17:37:27 -07:00
compiler_utils.py Update debug options in compilation errors (#913) 2022-06-07 10:55:54 -07:00