[llvm] 4230858 - [ARM] Avoid repeated map lookups (NFC) (#131420)

[via llvm-commits]

Author: Kazu Hirata
Date: 2025-03-15T09:11:18-07:00
New Revision: 4230858a52abcd165dc52edd194f18cef436cbf3

URL: https://github.com/llvm/llvm-project/commit/4230858a52abcd165dc52edd194f18cef436cbf3
DIFF: https://github.com/llvm/llvm-project/commit/4230858a52abcd165dc52edd194f18cef436cbf3.diff

LOG: [ARM] Avoid repeated map lookups (NFC) (#131420)

Added: 
    

Modified: 
    llvm/lib/Target/ARM/ARMParallelDSP.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Target/ARM/ARMParallelDSP.cpp b/llvm/lib/Target/ARM/ARMParallelDSP.cpp
index 4e92720de4755..e8bbd15c1ca03 100644
--- a/llvm/lib/Target/ARM/ARMParallelDSP.cpp
+++ b/llvm/lib/Target/ARM/ARMParallelDSP.cpp
@@ -715,10 +715,14 @@ void ARMParallelDSP::InsertParallelMACs(Reduction &R) {
     MulCandidate *RHSMul = Pair.second;
     LoadInst *BaseLHS = LHSMul->getBaseLoad();
     LoadInst *BaseRHS = RHSMul->getBaseLoad();
-    LoadInst *WideLHS = WideLoads.count(BaseLHS) ?
-      WideLoads[BaseLHS]->getLoad() : CreateWideLoad(LHSMul->VecLd, Ty);
-    LoadInst *WideRHS = WideLoads.count(BaseRHS) ?
-      WideLoads[BaseRHS]->getLoad() : CreateWideLoad(RHSMul->VecLd, Ty);
+    auto LIt = WideLoads.find(BaseLHS);
+    LoadInst *WideLHS = LIt != WideLoads.end()
+                            ? LIt->second->getLoad()
+                            : CreateWideLoad(LHSMul->VecLd, Ty);
+    auto RIt = WideLoads.find(BaseRHS);
+    LoadInst *WideRHS = RIt != WideLoads.end()
+                            ? RIt->second->getLoad()
+                            : CreateWideLoad(RHSMul->VecLd, Ty);
 
     Instruction *InsertAfter = GetInsertPoint(WideLHS, WideRHS);
     InsertAfter = GetInsertPoint(InsertAfter, Acc);