[compiler-rt] [mlir] Add ContractionOpInterface utility functions for vector matrix multiplication (PR #68945)

Diego Caballero via llvm-commits llvm-commits at lists.llvm.org
Mon Oct 16 11:18:40 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 llvm-commits mailing list