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`. |
||
|---|---|---|
| .. | ||
| Backend/RefJIT | ||
| Conversion | ||
| Dialect | ||
| Python | ||
| RefBackend | ||
| Typing | ||
| CMakeLists.txt | ||
| InitAll.h | ||