[AMDGPU] Fix

[shiltian]

No description provided.

[shiltian]

Warning

This pull request is not mergeable via GitHub because a downstack PR is open. Once all requirements are satisfied, merge this PR as a stack on Graphite.
Learn more

• [AMDGPU] Fix #115505 👈
• [AMDGPU] Reorganize tests to unblock #112403 #115503 : 1 other dependent PR (#112403 )
• main

This stack of pull requests is managed by Graphite. Learn more about stacking.

Join @shiltian and the rest of your teammates on Graphite

[llvmbot]

@llvm/pr-subscribers-backend-amdgpu

Author: Shilei Tian (shiltian)

Changes

---

Full diff: https://github.com/llvm/llvm-project/pull/115505.diff

1 Files Affected:

• (modified) llvm/lib/Target/AMDGPU/SIFrameLowering.cpp (+10-1)

diff --git a/llvm/lib/Target/AMDGPU/SIFrameLowering.cpp b/llvm/lib/Target/AMDGPU/SIFrameLowering.cpp
index 13a2db7a87b437..8e3208abaec9a5 100644
--- a/llvm/lib/Target/AMDGPU/SIFrameLowering.cpp
+++ b/llvm/lib/Target/AMDGPU/SIFrameLowering.cpp
@@ -59,9 +59,18 @@ static MCRegister findScratchNonCalleeSaveRegister(
   if (Unused)
     return findUnusedRegister(MRI, LiveUnits, RC);
 
+  dbgs() << "--------\n";
+
   for (MCRegister Reg : RC) {
-    if (LiveUnits.available(Reg) && !MRI.isReserved(Reg))
+    dbgs() << "reg " << Reg << " avail? " << LiveUnits.available(Reg)
+           << ", reserved? " << MRI.isReserved(Reg) << "\n";
+  }
+
+  for (MCRegister Reg : RC) {
+    if (LiveUnits.available(Reg) && !MRI.isReserved(Reg)) {
+      dbgs() << "--------choose reg " << Reg << '\n';
       return Reg;
+    }
   }
 
   return MCRegister();