[TSAR, Memory] Calculate trip count using nodes.

Author: yukatan1701

include/tsar/Analysis/Memory/DefinedMemory.h

@@ -52,6 +52,7 @@
 #include <llvm/Pass.h>
 
 namespace llvm {
+class AssumptionCache;
 class DominatorTree;
 class Value;
 class Instruction;
@@ -92,6 +93,11 @@ class DefUseSet {
   /// Set of memory locations.
   typedef MemorySet<MemoryLocationRange> LocationSet;
 
+  /// Set of data flow nodes with which memory locations from the 
+  /// ExplicitAccesses set are associated.
+  typedef llvm::DenseMap<const llvm::Value *, llvm::DenseSet<const DFNode *>>
+      ExplicitNodeMap;
+
   /// Returns set of the must defined locations.
   const LocationSet & getDefs() const { return mDefs; }
 
@@ -386,6 +392,33 @@ class DefUseSet {
         .second;
   }
 
+  /// TODO: description
+  void addExplicitAccessAndDFNode(const llvm::Value *Ptr, const DFNode *N) {
+    assert(Ptr && "Location pointer must not be null!");
+    mCollapsableExplicitAccesses.try_emplace(Ptr).first->second.insert(N);
+  }
+
+  /// TODO: description
+  const ExplicitNodeMap & getCollapsableExplicitAccesses() const {
+    return mCollapsableExplicitAccesses;
+  }
+
+  /// TODO: description
+  void addCollapsableExplicitAccesses(const ExplicitNodeMap &ENM) {
+    for (auto Itr = ENM.begin(); Itr != ENM.end(); ++Itr) {
+      auto &NodeList = Itr->second;
+      mCollapsableExplicitAccesses.try_emplace(Itr->first).
+          first->second.insert(NodeList.begin(), NodeList.end());
+    }
+  }
+
+  /// TODO: description
+  const llvm::DenseSet<const DFNode *> *
+  getCollapExpAccessNodes(const llvm::Value *Ptr) const {
+    auto Itr = mCollapsableExplicitAccesses.find(Ptr);
+    return Itr == mCollapsableExplicitAccesses.end() ? nullptr : &Itr->second;
+  }
+
 private:
   LocationSet mDefs;
   LocationSet mMayDefs;
@@ -396,6 +429,7 @@ class DefUseSet {
   InstructionSet mExplicitUnknowns;
   InstructionSet mAddressUnknowns;
   TransitiveMap mAddressTransitives;
+  ExplicitNodeMap mCollapsableExplicitAccesses;
 };
 
 /// This presents information whether a location has definition after a node
@@ -449,29 +483,32 @@ class ReachDFFwk : private bcl::Uncopyable {
   ReachDFFwk(AliasTree &AT, llvm::TargetLibraryInfo &TLI,
       const DFRegionInfo &DFI, const llvm::DominatorTree &DT,
       const DelinearizeInfo &DI, llvm::ScalarEvolution &SE,
-      const llvm::DataLayout &DL, const GlobalOptions &GO,
-      DefinedMemoryInfo &DefInfo) :
+      const llvm::DataLayout &DL, llvm::AssumptionCache &AC,
+      const GlobalOptions &GO, DefinedMemoryInfo &DefInfo) :
     mAliasTree(&AT), mTLI(&TLI), mRegionInfo(&DFI),
-    mDT(&DT), mDI(&DI), mSE(&SE), mDL(&DL), mGO(&GO), mDefInfo(&DefInfo) {}
+    mDT(&DT), mDI(&DI), mSE(&SE), mDL(&DL), mAC(&AC), mGO(&GO),
+    mDefInfo(&DefInfo) {}
 
   /// Creates data-flow framework.
   ReachDFFwk(AliasTree &AT, llvm::TargetLibraryInfo &TLI,
       const DFRegionInfo &DFI, const llvm::DominatorTree &DT,
       const DelinearizeInfo &DI, llvm::ScalarEvolution &SE,
-      const llvm::DataLayout &DL, const GlobalOptions &GO,
-      DefinedMemoryInfo &DefInfo, InterprocDefUseInfo &InterprocDUInfo) :
+      const llvm::DataLayout &DL, llvm::AssumptionCache &AC,
+      const GlobalOptions &GO, DefinedMemoryInfo &DefInfo,
+      InterprocDefUseInfo &InterprocDUInfo) :
     mAliasTree(&AT), mTLI(&TLI), mRegionInfo(&DFI), mDT(&DT),
-    mDI(&DI), mSE(&SE), mDL(&DL), mGO(&GO), mDefInfo(&DefInfo),
+    mDI(&DI), mSE(&SE), mDL(&DL), mAC(&AC), mGO(&GO), mDefInfo(&DefInfo),
     mInterprocDUInfo(&InterprocDUInfo) {}
 
   /// Creates data-flow framework.
   ReachDFFwk(AliasTree &AT, llvm::TargetLibraryInfo &TLI,
       const llvm::DominatorTree &DT, const DelinearizeInfo &DI,
       llvm::ScalarEvolution &SE, const llvm::DataLayout &DL,
-      const GlobalOptions &GO, DefinedMemoryInfo &DefInfo,
-      InterprocDefUseInfo &InterprocDUInfo) :
+      llvm::AssumptionCache &AC, const GlobalOptions &GO,
+      DefinedMemoryInfo &DefInfo, InterprocDefUseInfo &InterprocDUInfo) :
     mAliasTree(&AT), mTLI(&TLI), mDT(&DT), mDI(&DI), mSE(&SE), mDL(&DL),
-    mGO(&GO), mDefInfo(&DefInfo), mInterprocDUInfo(&InterprocDUInfo) {}
+    mAC(&AC), mGO(&GO), mDefInfo(&DefInfo),
+    mInterprocDUInfo(&InterprocDUInfo) {}
 
   /// Return results of interprocedural analysis or nullptr.
   InterprocDefUseInfo * getInterprocDefUseInfo() noexcept {
@@ -510,6 +547,11 @@ class ReachDFFwk : private bcl::Uncopyable {
   /// Returns data layout.
   const llvm::DataLayout & getDataLayout() const noexcept { return *mDL; }
 
+  /// Returns assumption cache.
+  llvm::AssumptionCache & getAssumptionCache() const noexcept {
+    return *mAC;
+  }
+
   /// Returns global options.
   const GlobalOptions & getGlobalOptions() const noexcept { return *mGO; }
 
@@ -526,6 +568,7 @@ class ReachDFFwk : private bcl::Uncopyable {
   const DelinearizeInfo *mDI;
   llvm::ScalarEvolution *mSE;
   const llvm::DataLayout *mDL;
+  llvm::AssumptionCache *mAC;
   const GlobalOptions *mGO;
 };