[MLIR:HLO] Change LHLO Outfeed to support multiple inputs.

PiperOrigin-RevId: 351447034
2021-01-12 14:12:43 -08:00 · 2021-01-12 14:12:43 -08:00 · 0651d58edf
parent 0b85d5c510
commit 0651d58edf
1 changed files with 1 additions and 1 deletions
--- 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
  );
 }