intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-02-03 02:26:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add MLFunction::getReturnStmt.

Browse Source 
PiperOrigin-RevId: 208514441
This commit is contained in:
Uday Bondhugula
2018-08-13 11:30:36 -07:00
committed by jpienaar
parent 067d70f20d
commit 95c1bf445a
2 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/include/mlir/IR/MLFunction.h
View File

@@ -80,6 +80,10 @@ public:
~MLFunction();
// Return the 'return' statement of this MLFunction.
const OperationStmt *getReturnStmt() const;
OperationStmt *getReturnStmt();
/// Methods for support type inquiry through isa, cast, and dyn_cast.
static bool classof(const Function *func) {
return func->getKind() == Function::Kind::MLFunc;

8
mlir/lib/IR/Function.cpp
View File

@@ -152,3 +152,11 @@ void MLFunction::destroy() {
this->~MLFunction();
free(this);
}
const OperationStmt *MLFunction::getReturnStmt() const {
return cast<OperationStmt>(&back());
}
OperationStmt *MLFunction::getReturnStmt() {
return cast<OperationStmt>(&back());
}