mirror of https://github.com/llvm/torch-mlir
1dfcfa9cd1
Note that unlike aten.matmul which has dynamic behavior depending on the argument ranks (can do matrix-matrix, matrix-vector, batch matmul, etc.), aten.mm is just a vanilla matrix multiply, which can be lowered precisely to tcf.matmul. The "test" is really just an example that I stared at while getting my feet wet with this. We probably want something that actually tests this as part of `ninja check-npcomp`. |
||
|---|---|---|
| .. | ||
| .gitignore | ||
| ATenDialect.h | ||
| ATenDialect.td | ||
| ATenOpInterface.td | ||
| ATenOpInterfaces.h | ||
| ATenOpStatisticsUtils.h | ||
| ATenOps.td | ||
| CMakeLists.txt | ||
| GeneratedATenOps.cpp.inc | ||
| GeneratedATenOps.td | ||
| LegacyGeneratedATenOps.td | ||