[llvm] 506b31e - [IPO] Avoid repeated hash lookups (NFC) (#127957)

[via llvm-commits]

Author: Kazu Hirata
Date: 2025-02-20T08:55:52-08:00
New Revision: 506b31ec36746732b84fb0e9cd74af3ca885fa86

URL: https://github.com/llvm/llvm-project/commit/506b31ec36746732b84fb0e9cd74af3ca885fa86
DIFF: https://github.com/llvm/llvm-project/commit/506b31ec36746732b84fb0e9cd74af3ca885fa86.diff

LOG: [IPO] Avoid repeated hash lookups (NFC) (#127957)

Added: 
    

Modified: 
    llvm/lib/Transforms/IPO/PartialInlining.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Transforms/IPO/PartialInlining.cpp b/llvm/lib/Transforms/IPO/PartialInlining.cpp
index f2707afe195c4..56b7b8bfe1f66 100644
--- a/llvm/lib/Transforms/IPO/PartialInlining.cpp
+++ b/llvm/lib/Transforms/IPO/PartialInlining.cpp
@@ -1393,9 +1393,12 @@ bool PartialInlinerImpl::tryPartialInline(FunctionCloner &Cloner) {
     CallerORE.emit(OR);
 
     // Now update the entry count:
-    if (CalleeEntryCountV && CallSiteToProfCountMap.count(User)) {
-      uint64_t CallSiteCount = CallSiteToProfCountMap[User];
-      CalleeEntryCountV -= std::min(CalleeEntryCountV, CallSiteCount);
+    if (CalleeEntryCountV) {
+      if (auto It = CallSiteToProfCountMap.find(User);
+          It != CallSiteToProfCountMap.end()) {
+        uint64_t CallSiteCount = It->second;
+        CalleeEntryCountV -= std::min(CalleeEntryCountV, CallSiteCount);
+      }
     }
 
     AnyInline = true;