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`. |
||
|---|---|---|
| .. | ||
| ATenToTCF | ||
| BasicpyToStd | ||
| NumpyToTCF | ||
| TCFToLinalg | ||
| TCFToStd | ||
| TCFToTCP | ||
| CMakeLists.txt | ||
| PassDetail.h | ||
| Passes.cpp | ||