From 46ade282c8d98558d0d1b8e79d2eee3ae00086f1 Mon Sep 17 00:00:00 2001
From: Chris Lattner <clattner@google.com>
Date: Mon, 25 Mar 2019 18:02:49 -0700
Subject: Make FunctionPass::getFunction() return a reference to the function,
 instead of a pointer.  This makes it consistent with all the other methods in
 FunctionPass, as well as with ModulePass::getModule().  NFC.

PiperOrigin-RevId: 240257910
---
 mlir/lib/Transforms/LowerVectorTransfers.cpp | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

(limited to 'mlir/lib/Transforms/LowerVectorTransfers.cpp')

diff --git a/mlir/lib/Transforms/LowerVectorTransfers.cpp b/mlir/lib/Transforms/LowerVectorTransfers.cpp
index 860d4f3c2de..e6b1950c222 100644
--- a/mlir/lib/Transforms/LowerVectorTransfers.cpp
+++ b/mlir/lib/Transforms/LowerVectorTransfers.cpp
@@ -43,7 +43,6 @@
 #include "mlir/Transforms/MLPatternLoweringPass.h"
 #include "mlir/Transforms/Passes.h"
 
-///
 /// Implements lowering of VectorTransferReadOp and VectorTransferWriteOp to a
 /// proper abstraction for the hardware.
 ///
@@ -376,9 +375,9 @@ public:
 struct LowerVectorTransfersPass
     : public FunctionPass<LowerVectorTransfersPass> {
   void runOnFunction() {
-    Function *f = getFunction();
+    auto &f = getFunction();
     applyMLPatternsGreedily<VectorTransferExpander<VectorTransferReadOp>,
-                            VectorTransferExpander<VectorTransferWriteOp>>(f);
+                            VectorTransferExpander<VectorTransferWriteOp>>(&f);
   }
 };
 
-- 
cgit v1.2.3