Sign-extend integer constants from original type size to 64 bits!

llvm-svn: 3958

1 files changed, 9 insertions, 5 deletions

diff --git a/llvm/lib/CodeGen/InstrSelection/InstrSelectionSupport.cpp b/llvm/lib/CodeGen/InstrSelection/InstrSelectionSupport.cpp
index 508c0b6e5e0..a8f14e20bca 100644
--- a/llvm/lib/CodeGen/InstrSelection/InstrSelectionSupport.cpp
+++ b/llvm/lib/CodeGen/InstrSelection/InstrSelectionSupport.cpp
@@ -373,11 +373,15 @@ ChooseRegOrImmed(Value* val,
   else if (CPV->getType()->isSigned())
     intValue = cast<ConstantSInt>(CPV)->getValue();
   else
-    {
-      assert(CPV->getType()->isUnsigned() && "Not pointer, bool, or integer?");
-      uint64_t V = cast<ConstantUInt>(CPV)->getValue();
-      if (V >= INT64_MAX) return MachineOperand::MO_VirtualRegister;
-      intValue = (int64_t) V;
+    { // get the int value and sign-extend if original was less than 64 bits
+      intValue = (int64_t) cast<ConstantUInt>(CPV)->getValue();
+      switch(CPV->getType()->getPrimitiveID())
+        {
+        case Type::UByteTyID:  intValue = (int64_t) (int8_t) intValue; break;
+        case Type::UShortTyID: intValue = (int64_t) (short)  intValue; break;
+        case Type::UIntTyID:   intValue = (int64_t) (int)    intValue; break;
+        default: break;
+        }
     }
 
   return ChooseRegOrImmed(intValue, CPV->getType()->isSigned(),