From 0651d58edf3bcdd65de86a2f25e97560022a3fa4 Mon Sep 17 00:00:00 2001
From: Rahul Joshi <jurahul@google.com>
Date: Tue, 12 Jan 2021 14:12:43 -0800
Subject: [PATCH] [MLIR:HLO] Change LHLO Outfeed to support multiple inputs.

PiperOrigin-RevId: 351447034
---
 include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td b/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
index c549368..b5c2a73 100644
--- a/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
+++ b/include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
@@ -584,7 +584,7 @@ def LHLO_InfeedOp: LHLO_Op<"infeed", []>, BASE_HLO_InfeedOp {
 
 def LHLO_OutfeedOp: LHLO_Op<"outfeed", []> {
   let arguments = (ins
-    Arg<LHLO_Buffer, "", [MemRead]>:$operand,
+    Arg<Variadic<LHLO_Buffer>, "", [MemRead]>:$operands,
     DefaultValuedAttr<StrAttr, "">:$config
   );
 }