From 8941071ae120397676a95202462c05c76288d9c4 Mon Sep 17 00:00:00 2001
From: Evan Cheng <evan.cheng@apple.com>
Date: Wed, 18 Jul 2007 21:26:06 +0000
Subject: [PATCH] Use MOV instead of LEA to restore ESP if callee-saved frame
 size is 0; if previous instruction updates esp, fold it in.

llvm-svn: 40018
---
 llvm/lib/Target/X86/X86RegisterInfo.cpp | 34 +++++++++++++++----------
 1 file changed, 20 insertions(+), 14 deletions(-)

diff --git a/llvm/lib/Target/X86/X86RegisterInfo.cpp b/llvm/lib/Target/X86/X86RegisterInfo.cpp
index da65db03e4b6..b64d864d65f2 100644
--- a/llvm/lib/Target/X86/X86RegisterInfo.cpp
+++ b/llvm/lib/Target/X86/X86RegisterInfo.cpp
@@ -1361,17 +1361,7 @@ void X86RegisterInfo::emitEpilogue(MachineFunction &MF,
     --MBBI;
   }
 
-  // If dynamic alloca is used, then reset esp to point to the last
-  // callee-saved slot before popping them off!
-  if (MFI->hasVarSizedObjects()) {
-    unsigned Opc = Is64Bit ? X86::LEA64r : X86::LEA32r;
-    MachineInstr *MI = addRegOffset(BuildMI(TII.get(Opc), StackPtr),
-                                    FramePtr, -CSSize);
-    MBB.insert(MBBI, MI);
-    NumBytes = 0;
-  }
-
-  if (NumBytes) {    // adjust stack pointer back: ESP += numbytes
+  if (NumBytes || MFI->hasVarSizedObjects()) {
     // If there is an ADD32ri or SUB32ri of ESP immediately before this
     // instruction, merge the two instructions.
     if (MBBI != MBB.begin()) {
@@ -1389,11 +1379,27 @@ void X86RegisterInfo::emitEpilogue(MachineFunction &MF,
         MBB.erase(PI);
       }
     }
-
-    if (NumBytes)
-      emitSPUpdate(MBB, MBBI, StackPtr, NumBytes, Is64Bit, TII);
   }
 
+  // If dynamic alloca is used, then reset esp to point to the last
+  // callee-saved slot before popping them off!
+  if (MFI->hasVarSizedObjects()) {
+    unsigned Opc = Is64Bit ? X86::LEA64r : X86::LEA32r;
+    if (CSSize) {
+      MachineInstr *MI = addRegOffset(BuildMI(TII.get(Opc), StackPtr),
+                                      FramePtr, -CSSize);
+      MBB.insert(MBBI, MI);
+    } else
+      BuildMI(MBB, MBBI, TII.get(Is64Bit ? X86::MOV64rr : X86::MOV32rr),StackPtr).
+        addReg(FramePtr);
+
+    NumBytes = 0;
+  }
+
+  // adjust stack pointer back: ESP += numbytes
+  if (NumBytes)
+    emitSPUpdate(MBB, MBBI, StackPtr, NumBytes, Is64Bit, TII);
+
   // We're returning from function via eh_return.
   if (RetOpcode == X86::EH_RETURN) {
     MBBI = prior(MBB.end());