@@ -561,7 +561,7 @@ def BroadcastOp : LinalgStructuredBase_Op<"broadcast", [
561
561
def MatmulOp : LinalgStructuredBase_Op<"matmul", [
562
562
AttrSizedOperandSegments,
563
563
LinalgContractionOpInterface]> {
564
-
564
+
565
565
let summary = [{
566
566
Performs a matrix multiplication of two 2D inputs without broadcast or transpose.
567
567
}];
@@ -593,17 +593,16 @@ def MatmulOp : LinalgStructuredBase_Op<"matmul", [
593
593
]
594
594
ins(%arg0, %arg1 : memref<3xf32>, memref<5x7xf32>)
595
595
outs(%arg2: memref<3x7xf32>)
596
- ```
596
+ ```
597
597
598
- Example Broadcast and transpose:
599
- ```
600
- linalg.matmul indexing_maps = [
601
- affine_map<(d0, d1, d2) -> (d2, d0)>, // transpose
602
- affine_map<(d0, d1, d2) -> (d2)>, // broadcast
603
- affine_map<(d0, d1, d2) -> (d0, d1)>
604
- ]
605
- ins(%arg0, %arg1 : memref<5x3xf32>, memref<7xf32>) outs(%arg2: memref<3x7xf32>)
606
- ```
598
+ Example Broadcast and transpose:
599
+ ```
600
+ linalg.matmul indexing_maps = [
601
+ affine_map<(d0, d1, d2) -> (d2, d0)>, // transpose
602
+ affine_map<(d0, d1, d2) -> (d2)>, // broadcast
603
+ affine_map<(d0, d1, d2) -> (d0, d1)>
604
+ ]
605
+ ins(%arg0, %arg1 : memref<5x3xf32>, memref<7xf32>) outs(%arg2: memref<3x7xf32>)
607
606
}];
608
607
609
608
let arguments = (ins
0 commit comments