AssumptionTracker: Don't create temporary CallbackVHs.

Those are expensive to create in cold cache scenarios. NFC.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@219575 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/include/llvm/Analysis/AssumptionTracker.h b/include/llvm/Analysis/AssumptionTracker.h
index 80ae513f4f93831f014a2887422f592ea35ff03f..5a050a8abf552d94bec88a242d32e20637c0432a 100644 (file)
--- a/include/llvm/Analysis/AssumptionTracker.h
+++ b/include/llvm/Analysis/AssumptionTracker.h
@@ -99,7 +99,7 @@ public:
   typedef iterator_range<assumption_iterator> assumption_range;
 
   inline assumption_range assumptions(Function *F) {
-    FunctionCallsMap::iterator I = CachedAssumeCalls.find(F);
+    FunctionCallsMap::iterator I = CachedAssumeCalls.find_as(F);
     if (I == CachedAssumeCalls.end()) {
       I = scanFunction(F);
     }

diff --git a/lib/Analysis/AssumptionTracker.cpp b/lib/Analysis/AssumptionTracker.cpp
index 3441030e49e96e2247a83229b63f06019ca8e8fb..b62f17b015ccd63e86a3d1a32f2b6684548d09d3 100644 (file)
--- a/lib/Analysis/AssumptionTracker.cpp
+++ b/lib/Analysis/AssumptionTracker.cpp
@@ -29,12 +29,14 @@ void AssumptionTracker::FunctionCallbackVH::deleted() {
 }
 
 void AssumptionTracker::forgetCachedAssumptions(Function *F) {
-  CachedAssumeCalls.erase(F);
+  auto I = CachedAssumeCalls.find_as(F);
+  if (I != CachedAssumeCalls.end())
+    CachedAssumeCalls.erase(I);
 }
 
 void AssumptionTracker::CallCallbackVH::deleted() {
   assert(F && "delete callback called on dummy handle");
-  FunctionCallsMap::iterator I = AT->CachedAssumeCalls.find(F);
+  FunctionCallsMap::iterator I = AT->CachedAssumeCalls.find_as(F);
   assert(I != AT->CachedAssumeCalls.end() &&
          "Function cleared from the map without removing the values?");
 
@@ -88,7 +90,7 @@ void AssumptionTracker::registerAssumption(CallInst *CI) {
   Function *F = CI->getParent()->getParent();
   assert(F && "Cannot register @llvm.assume call not in a function");
 
-  FunctionCallsMap::iterator I = CachedAssumeCalls.find(F);
+  FunctionCallsMap::iterator I = CachedAssumeCalls.find_as(F);
   if (I == CachedAssumeCalls.end()) {
     // If this function has not already been scanned, then don't do anything
     // here. This intrinsic will be found, if it still exists, if the list of