[Mlir-commits] [mlir] [mlir] Add ContractionOpInterface utility functions for vector matrix multiplication (PR #68945)
Diego Caballero
llvmlistbot at llvm.org
Mon Oct 16 11:18:42 PDT 2023
================
@@ -96,6 +96,79 @@ bool mlir::isRowMajorBatchMatmul(ArrayAttr indexingMaps) {
return indexingMaps == maps;
}
+bool mlir::isVecmat(ArrayAttr indexingMaps) {
+ if (indexingMaps.size() != 3)
+ return false;
+ auto map0 = cast<AffineMapAttr>(indexingMaps[0]).getValue();
----------------
dcaballe wrote:
nit: for readability, we try to minimize the use of `auto` to some extent, leaving it only for cases where the actual type class is explicit on the right-hand side. I would use the type name here and for the mlir contexts below.
https://github.com/llvm/llvm-project/pull/68945
More information about the Mlir-commits
mailing list