colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Restore SingleBlockImplicitTerminator verification to mhlo.while 

The internal users have been cleaned up, so we can roll this forward again.

PiperOrigin-RevId: 366313960
This commit is contained in:
Geoffrey Martin-Noble 2021-04-01 13:03:17 -07:00 committed by TensorFlow MLIR Team
parent f1bed7b340
commit 763ff55970
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
View File

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