Update linalg.generic syntax (#27)

spcl · Oct 17, 2023 · 5bff437 · 5bff437
1 parent e16a6a2
commit 5bff437
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mlir/dialects/linalg.py b/mlir/dialects/linalg.py
@@ -142,6 +142,10 @@ class LinalgGeneric(DialectOp):
                  " ins( {inargs.ssa_id_list} : {in_types.type_list_no_parens} )"
                  " outs( {outargs.ssa_id_list} : {out_types.type_list_no_parens} )"
                  " {region.region}"),
+                ("linalg.generic {attr.attribute_value} "
+                 " ins( {inargs.ssa_id_list} : {in_types.type_list_no_parens} )"
+                 " outs( {outargs.ssa_id_list} : {out_types.type_list_no_parens} )"
+                 " {region.region} -> {out_type.type}"),
                 ("linalg.generic {attr.attribute_value} "
                  " ins( {inargs.ssa_id_list} : {in_types.type_list_no_parens} )"
                  " init( {init_args.ssa_id_list} : {init_types.type_list_no_parens} )"