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

Temporarily relax restriction on mhlo.while terminator 

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

PiperOrigin-RevId: 365949611
This commit is contained in:
Geoffrey Martin-Noble 2021-03-30 19:40:48 -07:00 committed by TensorFlow MLIR Team
parent 5d65758e8c
commit 5ec66775d4
1 changed files with 1 additions and 2 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,8 +578,7 @@ 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);