From 5ec66775d41d5a3dedb58922259deddf41c679af Mon Sep 17 00:00:00 2001
From: Geoffrey Martin-Noble <gcmn@google.com>
Date: Tue, 30 Mar 2021 19:40:48 -0700
Subject: [PATCH] Temporarily relax restriction on mhlo.while terminator

Some internal tests are failing, so relaxing this restriction
temporarily while we investigate.

PiperOrigin-RevId: 365949611
---
 include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td | 3 +--
 1 file changed, 1 insertion(+), 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 efb041a..44ab6d2 100644
--- a/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
+++ b/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
@@ -578,8 +578,7 @@ def HLO_CaseOp: HLO_Op<"case", [
 
 def HLO_WhileOp: HLO_Op<"while", [
       RecursiveSideEffects,
-      SameOperandsAndResultType,
-      SingleBlockImplicitTerminator<"ReturnOp">
+      SameOperandsAndResultType
     ]>, BASE_HLO_WhileOp {
   let arguments = (ins HLO_TensorOrTuple:$val);