ScopStmt: Provide a function to allow users look up the corresponding

  memory access of a particular instruction, the function will return
  null if no such memory access.

llvm-svn: 155544

1 files changed, 6 insertions, 0 deletions

diff --git a/polly/include/polly/ScopInfo.h b/polly/include/polly/ScopInfo.h
index 3d7ff725e0e..4eb1ece2d55 100755
--- a/polly/include/polly/ScopInfo.h
+++ b/polly/include/polly/ScopInfo.h
@@ -321,6 +321,12 @@ public:
     return *InstructionToAccess[Inst];
   }
 
+  MemoryAccess *lookupAccessFor(const Instruction *Inst) const {
+    std::map<const Instruction*, MemoryAccess*>::const_iterator at
+      = InstructionToAccess.find(Inst);
+    return at == InstructionToAccess.end() ? NULL : at->second;
+  }
+
   void setBasicBlock(BasicBlock *Block) { BB = Block; }
 
   typedef MemoryAccessVec::iterator memacc_iterator;