From af01c08ce6920c2e360b3d8bf1cf4225f5db0ed3 Mon Sep 17 00:00:00 2001
From: Mehdi Amini <aminim@google.com>
Date: Wed, 26 May 2021 12:55:03 -0700
Subject: [PATCH] Allow tuple as results of mhlo.custom_call op (NFC)

PiperOrigin-RevId: 376009518
---
 include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td b/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
index 80fd274..33c8ecd 100644
--- a/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
+++ b/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
@@ -1471,12 +1471,12 @@ def HLO_CustomCallOp: HLO_Op<"custom_call", []> {
     See https://www.tensorflow.org/xla/operation_semantics#customcall.
   }];
   let arguments = (ins
-    Variadic<HLO_Tensor>:$args,
+    Variadic<HLO_TensorOrTuple>:$args,
     StrAttr:$call_target_name,
     DefaultValuedAttr<BoolAttr, "false">:$has_side_effect,
     DefaultValuedAttr<StrAttr, "">:$backend_config
   );
-  let results = (outs Variadic<HLO_Tensor>);
+  let results = (outs Variadic<HLO_TensorOrTuple>);
   let hasCustomHLOConverter = 1;
 }