Restore SingleBlockImplicitTerminator verification to mhlo.while

The internal users have been cleaned up, so we can roll this forward again. PiperOrigin-RevId: 366313960
2021-04-01 13:03:17 -07:00 · 2021-04-01 13:03:17 -07:00 · 763ff55970
parent f1bed7b340
commit 763ff55970
1 changed files with 2 additions and 1 deletions
--- a/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
+++ b/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
@ -578,7 +578,8 @@ def HLO_CaseOp: HLO_Op<"case", [

 def HLO_WhileOp: HLO_Op<"while", [
      RecursiveSideEffects,
-      SameOperandsAndResultType
+      SameOperandsAndResultType,
+      SingleBlockImplicitTerminator<"ReturnOp">
    ]>, BASE_HLO_WhileOp {
  let arguments = (ins HLO_TensorOrTuple:$val);