[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