Fix a memory leak.

llvm-svn: 262682

diff --git a/llvm/lib/Analysis/InlineCost.cpp b/llvm/lib/Analysis/InlineCost.cpp
index 4f69d1f..355c326 100644 (file)
--- a/llvm/lib/Analysis/InlineCost.cpp
+++ b/llvm/lib/Analysis/InlineCost.cpp
@@ -1600,5 +1600,8 @@ BlockFrequencyInfo *BlockFrequencyAnalysis::getBlockFrequencyInfo(Function *F) {
 
 /// \brief Invalidate BlockFrequencyInfo for a function.
 void BlockFrequencyAnalysis::invalidateBlockFrequencyInfo(Function *F) {
-  BFM.erase(F);
+  if (BFM.count(F)) {
+    delete BFM[F];
+    BFM.erase(F);
+  }
 }