From 8168a54b9284d49b2247a68cb261bf4c3c2cdf47 Mon Sep 17 00:00:00 2001
From: Kazu Hirata <kazu@google.com>
Date: Tue, 18 Mar 2025 12:30:06 -0700
Subject: [PATCH] [AMDGPU] Avoid repeated hash lookups (NFC)

---
 llvm/lib/Target/AMDGPU/AMDGPUSetWavePriority.cpp | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/llvm/lib/Target/AMDGPU/AMDGPUSetWavePriority.cpp b/llvm/lib/Target/AMDGPU/AMDGPUSetWavePriority.cpp
index c16d33f1453c0..d760cd3115eaa 100644
--- a/llvm/lib/Target/AMDGPU/AMDGPUSetWavePriority.cpp
+++ b/llvm/lib/Target/AMDGPU/AMDGPUSetWavePriority.cpp
@@ -200,12 +200,12 @@ bool AMDGPUSetWavePriority::runOnMachineFunction(MachineFunction &MF) {
   }
 
   for (MachineBasicBlock *MBB : PriorityLoweringBlocks) {
-    BuildSetprioMI(
-        *MBB,
-        MBBInfos[MBB].LastVMEMLoad
-            ? std::next(MachineBasicBlock::iterator(MBBInfos[MBB].LastVMEMLoad))
-            : MBB->begin(),
-        LowPriority);
+    MachineInstr *LastVMEMLoad = MBBInfos[MBB].LastVMEMLoad;
+    BuildSetprioMI(*MBB,
+                   LastVMEMLoad
+                       ? std::next(MachineBasicBlock::iterator(LastVMEMLoad))
+                       : MBB->begin(),
+                   LowPriority);
   }
 
   return true;