diff options
Diffstat (limited to 'mlir/lib/IR/BuiltinOps.cpp')
| -rw-r--r-- | mlir/lib/IR/BuiltinOps.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/lib/IR/BuiltinOps.cpp b/mlir/lib/IR/BuiltinOps.cpp index 5d7ba237b44..dfd59c4d380 100644 --- a/mlir/lib/IR/BuiltinOps.cpp +++ b/mlir/lib/IR/BuiltinOps.cpp @@ -474,7 +474,7 @@ void ReturnOp::print(OpAsmPrinter *p) const { bool ReturnOp::verify() const { const Function *function; if (auto *stmt = dyn_cast<OperationStmt>(getOperation())) - function = cast<MLFunction>(stmt->getBlock()); + function = stmt->getBlock()->findFunction(); else function = cast<Instruction>(getOperation())->getFunction(); |
