[llvm] 8ad4f1a - [Analysis] Avoid repeated hash lookups (NFC) (#123893)

[via llvm-commits]

Author: Kazu Hirata
Date: 2025-01-22T00:17:29-08:00
New Revision: 8ad4f1a9c110632e419cc64931d432e91559407a

URL: https://github.com/llvm/llvm-project/commit/8ad4f1a9c110632e419cc64931d432e91559407a
DIFF: https://github.com/llvm/llvm-project/commit/8ad4f1a9c110632e419cc64931d432e91559407a.diff

LOG: [Analysis] Avoid repeated hash lookups (NFC) (#123893)

Added: 
    

Modified: 
    llvm/lib/Analysis/BranchProbabilityInfo.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Analysis/BranchProbabilityInfo.cpp b/llvm/lib/Analysis/BranchProbabilityInfo.cpp
index 8906dc4e5cacf8..8077c28f79a390 100644
--- a/llvm/lib/Analysis/BranchProbabilityInfo.cpp
+++ b/llvm/lib/Analysis/BranchProbabilityInfo.cpp
@@ -1176,10 +1176,12 @@ void BranchProbabilityInfo::copyEdgeProbabilities(BasicBlock *Src,
 
 void BranchProbabilityInfo::swapSuccEdgesProbabilities(const BasicBlock *Src) {
   assert(Src->getTerminator()->getNumSuccessors() == 2);
-  if (!Probs.contains(std::make_pair(Src, 0)))
+  auto It0 = Probs.find(std::make_pair(Src, 0));
+  if (It0 == Probs.end())
     return; // No probability is set for edges from Src
-  assert(Probs.contains(std::make_pair(Src, 1)));
-  std::swap(Probs[std::make_pair(Src, 0)], Probs[std::make_pair(Src, 1)]);
+  auto It1 = Probs.find(std::make_pair(Src, 1));
+  assert(It1 != Probs.end());
+  std::swap(It0->second, It1->second);
 }
 
 raw_ostream &