torch-mlir/frontends/pytorch/examples
Sean Silva f5dfa02523 Add `aten.mm` to linalg lowering.
This is our first op with error semantics, and stresses the system.

There are a few design notes of special interest:
- RefineTypes.cpp's note about shape inference in the presence of code
  that dynamically produces and error, and it is provable statically.
- ATenToLinalg.cpp's notes about future automation of the ATen->linalg
  path.
- The notes in Passes.td about using low-tech `std.assert` ops instead
  of `shape.assuming`.

Note: Doesn't work on IREE yet due to the `std.assert` op (needs to be
lowered to `vm.fail` on the IREE side).
2021-04-16 12:03:31 -07:00
..
__init__.py Add cos_e2e.py, test_utils and support for tensor inputs (#134) 2020-11-24 19:02:50 -08:00
cos_e2e.py Add support for compiling through IREE. 2021-04-09 13:15:07 -07:00
div_inplace_e2e.py Add support for compiling through IREE. 2021-04-09 13:15:07 -07:00
mm_e2e.py Add support for compiling through IREE. 2021-04-09 13:15:07 -07:00
mul_maximum_e2e.py Add support for compiling through IREE. 2021-04-09 13:15:07 -07:00
tanh_out_e2e.py Add support for compiling through IREE. 2021-04-09 13:15:07 -07:00
test_utils.py MILESTONE: TorchScript unary tanh runs on RefBackend 2021-04-07 11:06:34 -07:00
torchscript_mm_e2e.py Add `aten.mm` to linalg lowering. 2021-04-16 12:03:31 -07:00
torchscript_tanh_e2e.py Add support for compiling through IREE. 2021-04-09 13:15:07 -07:00
torchscript_tanh_e2e_iree.py Add support for compiling through IREE. 2021-04-09 13:15:07 -07:00