diff --git a/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td b/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td --- a/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td +++ b/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td @@ -790,7 +790,7 @@ // ReturnOp //===----------------------------------------------------------------------===// -def LLVM_ReturnOp : LLVM_TerminatorOp<"return", [Pure]> { +def LLVM_ReturnOp : LLVM_TerminatorOp<"return", [Pure, ReturnLike]> { let arguments = (ins Optional:$arg); let assemblyFormat = "attr-dict ($arg^ `:` type($arg))?";