Update linalg fill syntax

spcl · Nov 10, 2023 · 41b18fe · 41b18fe
1 parent bddedc2
commit 41b18fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_linalg.py b/tests/test_linalg.py
@@ -68,8 +68,8 @@ def test_dot():
 
 def test_fill():
     assert_roundtrip_equivalence("""module {
-  func.func @fill_view(%arg0: memref<?xf32, strided<[1], offset: ?>>, %arg1: f32) {
-    linalg.fill( %arg0 , %arg1 )  : memref<?xf32, strided<[1], offset: ?>> , f32
+  func.func @fill_view(%arg0: f32, %arg1: tensor<?x?xf32>) {
+    linalg.fill ins( %arg0 : f32 ) outs( %arg1 : tensor<?x?xf32> ) -> tensor<?x?xf32>
     return
   }
 }""")