@@ -86,6 +86,39 @@ def LinalgContractionOpInterface :
OpInterface<"ContractionOpInterface"> {
/*methodBody=*/[{
return mlir::isRowMajorBatchMatmul($_op.getIndexingMaps());
}]>,
+InterfaceMethod<
+/*desc=*/[{
+ Returns whether the given op has index
@@ -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(indexingMaps[0]).getValue();
d
https://github.com/dcaballe commented:
Awesome! Just a couple of minor comments from my side!
https://github.com/llvm/llvm-project/pull/68945
___
cfe-commits mailing list
cfe-commits@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-co
