llvm-project
072ce70d - [MLIR][AArch64] Check indexing maps before checking for dimensions compatibility

Commit
177 days ago
[MLIR][AArch64] Check indexing maps before checking for dimensions compatibility In `LowerContractionToSVEI8MMPattern` check we have the expected indexing maps before deciding which operand dimension must match with which. For example, with indexing map like: lhs: (m, n, k) -> (m, k) rhs: (m, n, k) -> (n, k) acc: (m, n, k) -> (m, n) we would like the second `lhs` dimension (columns) to match with the second `rhs` (rows, transposed) whereas with indexing maps like lhs: (m, n, k) -> (m, k) rhs: (m, n, k) -> (k, n) acc: (m, n, k) -> (m, n) we would like the second `lhs` dimension (columns) to match with the first `rhs` (rows, canonical matrix multiplication). Since only the first kind of indexing maps is supported, the patch does not change anything of significance, just the notification message when the pattern would fail to apply anyway.
Parents
Loading