From 9f4c9946569744a1ba0269996244701e8fb346b6 Mon Sep 17 00:00:00 2001
From: Amanda Tang <amandatang88888@gmail.com>
Date: Tue, 17 Oct 2023 11:01:34 -0400
Subject: [PATCH] Update linalg.generic syntax

---
 mlir/dialects/linalg.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mlir/dialects/linalg.py b/mlir/dialects/linalg.py
index 4d82f8c..e348694 100644
--- 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} )"