commit 2ce3b577fb44a28c8021bda739fc1f65e8797fbf
author Uday Bondhugula <bondhugula@google.com> Mon Aug 13 11:30:36 2018 -0700
committer jpienaar <jpienaar@google.com> Fri Mar 29 12:58:45 2019 -0700
tree 223e3c7ed0a02645f89f0c0e0a03ea31bf9e1075
parent c03c695772f409f9ee0a55229983c98cb77d5403

Add MLFunction::getReturnStmt.

PiperOrigin-RevId: 208514441

lib/IR/Function.cpp

diff --git a/lib/IR/Function.cpp b/lib/IR/Function.cpp
index 4719b35..1c89de8 100644
--- a/lib/IR/Function.cpp
+++ b/lib/IR/Function.cpp
@@ -152,3 +152,11 @@
   this->~MLFunction();
   free(this);
 }
+
+const OperationStmt *MLFunction::getReturnStmt() const {
+  return cast<OperationStmt>(&back());
+}
+
+OperationStmt *MLFunction::getReturnStmt() {
+  return cast<OperationStmt>(&back());
+}