torch-mlir/lib/Dialect/Torch
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
..
IR Basic infra for annotate shapes and dtypes on arguments. 2021-04-01 18:40:03 -07:00
Transforms Add `aten.mm` to linalg lowering. 2021-04-16 12:03:31 -07:00
CMakeLists.txt Implement GlobalizeObjectGraph transformation. 2021-02-18 18:18:47 -08:00