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`. |
||
|---|---|---|
| .. | ||
| torch_mlir | ||
| torch_mlir_utils/codegen | ||
| CMakeLists.txt | ||