diff --git a/mlir/include/mlir/IR/FunctionSupport.h b/mlir/include/mlir/IR/FunctionSupport.h --- a/mlir/include/mlir/IR/FunctionSupport.h +++ b/mlir/include/mlir/IR/FunctionSupport.h @@ -206,6 +206,7 @@ //===--------------------------------------------------------------------===// // Argument Handling //===--------------------------------------------------------------------===// + using BlockArgListType = Region::BlockArgListType; unsigned getNumArguments() { return static_cast(this)->getNumFuncArguments(); @@ -224,6 +225,10 @@ args_iterator args_end() { return getBody().args_end(); } Block::BlockArgListType getArguments() { return getBody().getArguments(); } + ValueTypeRange getArgumentTypes() { + return getBody().getArgumentTypes(); + } + //===--------------------------------------------------------------------===// // Argument Attributes //===--------------------------------------------------------------------===//