torch-mlir/include/npcomp/Dialect/ATen/IR
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
..
.gitignore Add a number of kernels and new patterns. 2020-11-04 14:36:59 -08:00
ATenDialect.h Bump llvm-project to 444822d77a7fea28aa49edf24533c987efa1b2ee 2020-12-11 14:43:38 -08:00
ATenDialect.td Bump llvm-project to bc556e5685c0f97e79fb7b3c6f15cc5062db8e36 2021-01-08 14:18:09 -08:00
ATenOpInterface.td NFC: Re-organize ATen directory structure and fix warnings. 2020-10-22 14:13:26 -07:00
ATenOpInterfaces.h NFC: Re-organize ATen directory structure and fix warnings. 2020-10-22 14:13:26 -07:00
ATenOpStatisticsUtils.h Bump llvm-project to 444822d77a7fea28aa49edf24533c987efa1b2ee 2020-12-11 14:43:38 -08:00
ATenOps.td Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
CMakeLists.txt NFC: Re-organize ATen directory structure and fix warnings. 2020-10-22 14:13:26 -07:00
GeneratedATenOps.cpp.inc Add support for "trailing_" and "out" variants of various ops. 2021-03-19 10:34:50 -07:00
GeneratedATenOps.td Add `aten.mm` to linalg lowering. 2021-04-16 12:03:31 -07:00
LegacyGeneratedATenOps.td Add aten.mm op and "test" it e2e. 2020-11-20 17:21:24 -08:00