[MLIR:HLO] Change LHLO Infeed to support multiple outputs.

PiperOrigin-RevId: 348851612
2020-12-23 14:55:15 -08:00 · 2020-12-23 14:55:15 -08:00 · e3754d7b5c
parent bd8e768e23
commit e3754d7b5c
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
@ -577,7 +577,7 @@ def LHLO_CholeskyOp: LHLO_Op<"cholesky", [SameOperandsElementType]>, BASE_HLO_Ch

 def LHLO_InfeedOp: LHLO_Op<"infeed", []>, BASE_HLO_InfeedOp {
  let arguments = (ins
-    Arg<LHLO_Buffer, "", [MemWrite]>:$output,
+    Arg<Variadic<LHLO_Buffer>, "", [MemWrite]>:$outputs,
    DefaultValuedAttr<StrAttr, "">:$config
  );
 }