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

Jerry Wu via llvm-commits llvm-commits at lists.llvm.org
Mon Oct 16 10:24:58 PDT 2023


================
@@ -86,6 +86,39 @@ def LinalgContractionOpInterface : OpInterface<"ContractionOpInterface"> {
     /*methodBody=*/[{
         return mlir::isRowMajorBatchMatmul($_op.getIndexingMaps());
     }]>,
+    InterfaceMethod<
+    /*desc=*/[{
+      Returns whether the given op has indexing maps that correspond to a
+      vector-matrix multiplication.
+    }],
+    /*retTy=*/"bool",
+    /*methodName=*/"isVecmat",
----------------
pzread wrote:

According to the existing methods, do we need to specify these are raw major vecmat?

https://github.com/llvm/llvm-project/pull/68945


More information about the llvm-commits mailing list