[llvm] [SystemZ] Avoid repeated hash lookups (NFC) (PR #130392)

[Kazu Hirata via llvm-commits]

https://github.com/kazutakahirata created https://github.com/llvm/llvm-project/pull/130392

None

>From 8177889652761d1135a38b54e281bd226321b2ad Mon Sep 17 00:00:00 2001
From: Kazu Hirata <kazu at google.com>
Date: Fri, 7 Mar 2025 01:05:02 -0800
Subject: [PATCH] [SystemZ] Avoid repeated hash lookups (NFC)

---
 llvm/lib/Target/SystemZ/SystemZMachineScheduler.cpp | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/llvm/lib/Target/SystemZ/SystemZMachineScheduler.cpp b/llvm/lib/Target/SystemZ/SystemZMachineScheduler.cpp
index 4bc979de795dc..5e2365f1dc513 100644
--- a/llvm/lib/Target/SystemZ/SystemZMachineScheduler.cpp
+++ b/llvm/lib/Target/SystemZ/SystemZMachineScheduler.cpp
@@ -94,14 +94,16 @@ void SystemZPostRASchedStrategy::enterMBB(MachineBasicBlock *NextMBB) {
   // scheduled. If this is not possible, we are done.
   MachineBasicBlock *SinglePredMBB =
     getSingleSchedPred(MBB, MLI->getLoopFor(MBB));
-  if (SinglePredMBB == nullptr ||
-      SchedStates.find(SinglePredMBB) == SchedStates.end())
+  if (SinglePredMBB == nullptr)
+    return;
+  auto It = SchedStates.find(SinglePredMBB);
+  if (It == SchedStates.end())
     return;
 
   LLVM_DEBUG(dbgs() << "** Continued scheduling from "
                     << printMBBReference(*SinglePredMBB) << "\n";);
 
-  HazardRec->copyState(SchedStates[SinglePredMBB]);
+  HazardRec->copyState(It->second);
   LLVM_DEBUG(HazardRec->dumpState(););
 
   // Emit incoming terminator(s). Be optimistic and assume that branch