[Mlir-commits] [mlir] [mlir][linalg] Add patterns to convert matmul to transposed variants (PR #89075)
Cullen Rhodes
llvmlistbot at llvm.org
Mon Apr 22 04:00:17 PDT 2024
================
@@ -73,6 +73,21 @@ def ApplyTilingCanonicalizationPatternsOp : Op<Transform_Dialect,
let assemblyFormat = "attr-dict";
}
+def ApplyTransposeMatmulPatternsOp : Op<Transform_Dialect,
+ "apply_patterns.linalg.transpose_matmul",
+ [DeclareOpInterfaceMethods<PatternDescriptorOpInterface>]> {
+ let description = [{
+ Collects patterns to convert Linalg matmul ops to transposed variants.
+
+ By default the A matrix is transposed, set `transpose_a=false` to transpose
+ B matrix.
+ }];
+
+ let arguments = (ins DefaultValuedAttr<BoolAttr, "true">:$transpose_a);
----------------
c-rhodes wrote:
I've added a enum
https://github.com/llvm/llvm-project/pull/89075
More information about the Mlir-commits
mailing list