[clang] [mlir] Add ContractionOpInterface utility functions for vector matrix multiplication (PR #68945)
Diego Caballero via cfe-commits
cfe-commits at lists.llvm.org
Mon Oct 16 11:18:42 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",
----------------
dcaballe wrote:
what row-major/col-major vecmat would mean?
https://github.com/llvm/llvm-project/pull/68945
More information about the cfe-commits
mailing list