Add method to assign stack slot to virtual register without creating a

new one.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@13895 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/CodeGen/VirtRegMap.cpp b/lib/CodeGen/VirtRegMap.cpp
index 7e8e416034c8d6a9bdb6d39be235721f2e702ac2..5cec9e97c6e250b1484e28a56d9cd8983cbd4b21 100644 (file)
--- a/lib/CodeGen/VirtRegMap.cpp
+++ b/lib/CodeGen/VirtRegMap.cpp
@@ -61,6 +61,14 @@ int VirtRegMap::assignVirt2StackSlot(unsigned virtReg)
     return frameIndex;
 }
 
+void VirtRegMap::assignVirt2StackSlot(unsigned virtReg, int frameIndex)
+{
+    assert(MRegisterInfo::isVirtualRegister(virtReg));
+    assert(v2ssMap_[virtReg] == NO_STACK_SLOT &&
+           "attempt to assign stack slot to already spilled register");
+     v2ssMap_[virtReg] = frameIndex;
+}
+
 void VirtRegMap::virtFolded(unsigned virtReg,
                             MachineInstr* oldMI,
                             MachineInstr* newMI)

diff --git a/lib/CodeGen/VirtRegMap.h b/lib/CodeGen/VirtRegMap.h
index c4188bcfbcd3eb52ddfa75abfac6601518de5cc8..e9bfe76d4bf7785c8dc270997eb34cbc626a9a10 100644 (file)
--- a/lib/CodeGen/VirtRegMap.h
+++ b/lib/CodeGen/VirtRegMap.h
@@ -97,6 +97,7 @@ namespace llvm {
         }
 
         int assignVirt2StackSlot(unsigned virtReg);
+        void assignVirt2StackSlot(unsigned virtReg, int frameIndex);
 
         void virtFolded(unsigned virtReg,
                         MachineInstr* oldMI,