[llvm-commits] [llvm] r53541 - /llvm/trunk/lib/VMCore/Constants.cpp

[Chris Lattner]

Author: lattner
Date: Mon Jul 14 00:10:41 2008
New Revision: 53541

URL: http://llvm.org/viewvc/llvm-project?rev=53541&view=rev
Log:
Document and fix Constant::getVectorElements to return an empty vector
when presented with a constant expr.

If ConstantExpr::getV[IF]Cmp to work when ConstantFoldCompareInstruction
returns an undef or constant expr.  


Modified:
    llvm/trunk/lib/VMCore/Constants.cpp

Modified: llvm/trunk/lib/VMCore/Constants.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/VMCore/Constants.cpp?rev=53541&r1=53540&r2=53541&view=diff

==============================================================================
--- llvm/trunk/lib/VMCore/Constants.cpp (original)
+++ llvm/trunk/lib/VMCore/Constants.cpp Mon Jul 14 00:10:41 2008
@@ -157,7 +157,8 @@
 
 /// getVectorElements - This method, which is only valid on constant of vector
 /// type, returns the elements of the vector in the specified smallvector.
-/// This handles breaking down a vector undef into undef elements, etc.
+/// This handles breaking down a vector undef into undef elements, etc.  For
+/// constant exprs and other cases we can't handle, we return an empty vector.
 void Constant::getVectorElements(SmallVectorImpl<Constant*> &Elts) const {
   assert(isa<VectorType>(getType()) && "Not a vector constant!");
   
@@ -174,8 +175,12 @@
     return;
   }
   
-  assert(isa<UndefValue>(this) && "Unknown vector constant type!");
-  Elts.assign(VT->getNumElements(), UndefValue::get(VT->getElementType()));
+  if (isa<UndefValue>(this)) {
+    Elts.assign(VT->getNumElements(), UndefValue::get(VT->getElementType()));
+    return;
+  }
+  
+  // Unknown type, must be constant expr etc.
 }
 
 
@@ -2206,16 +2211,19 @@
   const Type *EltTy = VTy->getElementType();
   unsigned NumElts = VTy->getNumElements();
 
-  SmallVector<Constant *, 8> Elts;
+  SmallVector<Constant *, 16> Elts;
   for (unsigned i = 0; i != NumElts; ++i) {
     Constant *FC = ConstantFoldCompareInstruction(pred, LHS->getOperand(i),
-                                                        RHS->getOperand(i));
-    if (FC) {
-      uint64_t Val = cast<ConstantInt>(FC)->getZExtValue();
-      if (Val != 0ULL)
+                                                  RHS->getOperand(i));
+    if (ConstantInt *FCI = dyn_cast_or_null<ConstantInt>(FC)) {
+      if (FCI->getZExtValue())
         Elts.push_back(ConstantInt::getAllOnesValue(EltTy));
       else
         Elts.push_back(ConstantInt::get(EltTy, 0ULL));
+    } else if (FC && isa<UndefValue>(FC)) {
+      Elts.push_back(UndefValue::get(EltTy));
+    } else {
+      break;
     }
   }
   if (Elts.size() == NumElts)
@@ -2243,16 +2251,19 @@
   const Type *REltTy = IntegerType::get(EltTy->getPrimitiveSizeInBits());
   const Type *ResultTy = VectorType::get(REltTy, NumElts);
 
-  SmallVector<Constant *, 8> Elts;
+  SmallVector<Constant *, 16> Elts;
   for (unsigned i = 0; i != NumElts; ++i) {
     Constant *FC = ConstantFoldCompareInstruction(pred, LHS->getOperand(i),
                                                         RHS->getOperand(i));
-    if (FC) {
-      uint64_t Val = cast<ConstantInt>(FC)->getZExtValue();
-      if (Val != 0ULL)
+    if (ConstantInt *FCI = dyn_cast_or_null<ConstantInt>(FC)) {
+      if (FCI->getZExtValue())
         Elts.push_back(ConstantInt::getAllOnesValue(REltTy));
       else
         Elts.push_back(ConstantInt::get(REltTy, 0ULL));
+    } else if (FC && isa<UndefValue>(FC)) {
+      Elts.push_back(UndefValue::get(REltTy));
+    } else {
+      break;
     }
   }
   if (Elts.size() == NumElts)