squiid/llvm
1
0
Fork 0
Code Issues Pull requests Activity

[MLIR][NFC] Add FuncOp::getArgumentTypes()

Browse source 
Differential Revision: https://reviews.llvm.org/D85038
This commit is contained in:
Rahul Joshi 2020-07-31 11:39:49 -07:00
parent 77d5a63c19
commit eb8c72cb0d
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
mlir/include/mlir/IR/FunctionSupport.h
View file

@ -206,6 +206,7 @@ public:
//===--------------------------------------------------------------------===//
// Argument Handling
//===--------------------------------------------------------------------===//
using BlockArgListType = Region::BlockArgListType;
unsigned getNumArguments() {
return static_cast<ConcreteType *>(this)->getNumFuncArguments();
@ -224,6 +225,10 @@ public:
args_iterator args_end() { return getBody().args_end(); }
Block::BlockArgListType getArguments() { return getBody().getArguments(); }
ValueTypeRange<BlockArgListType> getArgumentTypes() {
return getBody().getArgumentTypes();
}
//===--------------------------------------------------------------------===//
// Argument Attributes
//===--------------------------------------------------------------------===//