also include them in the convert-linalg-to-memref-stream pass

xdslproject · Jul 23, 2024 · 9a38bd5 · 9a38bd5
1 parent ffccfe1
commit 9a38bd5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/filecheck/transforms/convert_linalg_to_memref_stream.mlir b/tests/filecheck/transforms/convert_linalg_to_memref_stream.mlir
@@ -15,7 +15,9 @@ linalg.generic {
         affine_map<() -> ()>,
         affine_map<() -> ()>
     ],
-    iterator_types = []
+    iterator_types = [],
+    doc = "documentation string",
+    library_call = "library call"
 } ins(%A, %B : memref<f64>, memref<f64>) outs(%C : memref<f64>) {
 ^0(%a : f64, %b : f64, %acc_old : f64):
     %prod = arith.mulf %a, %b : f64
@@ -29,7 +31,9 @@ linalg.generic {
 // CHECK-NEXT:        affine_map<() -> ()>,
 // CHECK-NEXT:        affine_map<() -> ()>
 // CHECK-NEXT:      ],
-// CHECK-NEXT:      iterator_types = []
+// CHECK-NEXT:      iterator_types = [],
+// CHECK-NEXT:      doc = "documentation string",
+// CHECK-NEXT:      library_call = "library call"
 // CHECK-NEXT:    } ins(%A, %B : memref<f64>, memref<f64>) outs(%C : memref<f64>) {
 // CHECK-NEXT:    ^0(%a : f64, %b : f64, %acc_old : f64):
 // CHECK-NEXT:      %prod = arith.mulf %a, %b : f64

diff --git a/xdsl/transforms/convert_linalg_to_memref_stream.py b/xdsl/transforms/convert_linalg_to_memref_stream.py
@@ -49,6 +49,8 @@ def match_and_rewrite(self, op: linalg.Generic, rewriter: PatternRewriter) -> No
                 iterator_types,
                 bounds,
                 ArrayAttr(()),
+                op.doc,
+                op.library_call,
             )
         )
-Original file line number
+Diff line change
@@ Expand Up @@
                     iterator_types,
                     bounds,
                     ArrayAttr(()),
+                    op.doc,
+                    op.library_call,
                 )
             )
@@ Expand Down @@