[llvm-commits] CVS: llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp

[Chris Lattner]

Changes in directory llvm/lib/CodeGen/SelectionDAG:

LegalizeDAG.cpp updated: 1.321 -> 1.322
---
Log message:

Implement expand of BUILD_VECTOR containing variable elements.
This implements CodeGen/Generic/vector.ll:test_variable_buildvector


---
Diffs of the changes:  (+32 -2)

 LegalizeDAG.cpp |   34 ++++++++++++++++++++++++++++++++--
 1 files changed, 32 insertions(+), 2 deletions(-)


Index: llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp
diff -u llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp:1.321 llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp:1.322
--- llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp:1.321	Sat Mar 18 19:17:20 2006
+++ llvm/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp	Sat Mar 18 22:18:56 2006
@@ -779,8 +779,38 @@
       // Otherwise, this isn't a constant entry.  Allocate a sufficiently
       // aligned object on the stack, store each element into it, then load
       // the result as a vector.
-      assert(0 && "Cannot lower variable BUILD_VECTOR yet!");
-      abort();
+      MVT::ValueType VT = Node->getValueType(0);
+      // Create the stack frame object.
+      MachineFrameInfo *FrameInfo = DAG.getMachineFunction().getFrameInfo();
+      unsigned ByteSize = MVT::getSizeInBits(VT)/8;
+      int FrameIdx = FrameInfo->CreateStackObject(ByteSize, ByteSize);
+      SDOperand FIPtr = DAG.getFrameIndex(FrameIdx, TLI.getPointerTy());
+      
+      // Emit a store of each element to the stack slot.
+      std::vector<SDOperand> Stores;
+      bool isLittleEndian = TLI.isLittleEndian();
+      unsigned TypeByteSize = 
+        MVT::getSizeInBits(Node->getOperand(0).getValueType())/8;
+      unsigned VectorSize = MVT::getSizeInBits(VT)/8;
+      // Store (in the right endianness) the elements to memory.
+      for (unsigned i = 0, e = Node->getNumOperands(); i != e; ++i) {
+        unsigned Offset;
+        if (isLittleEndian) 
+          Offset = TypeByteSize*i;
+        else
+          Offset = TypeByteSize*(e-i-1);
+
+        SDOperand Idx = DAG.getConstant(Offset, FIPtr.getValueType());
+        Idx = DAG.getNode(ISD::ADD, FIPtr.getValueType(), FIPtr, Idx);
+
+        Stores.push_back(DAG.getNode(ISD::STORE, MVT::Other, DAG.getEntryNode(),
+                                     Node->getOperand(i), Idx, 
+                                     DAG.getSrcValue(NULL)));
+      }
+      SDOperand StoreChain = DAG.getNode(ISD::TokenFactor, MVT::Other, Stores);
+      
+      // Result is a load from the stack slot.
+      Result = DAG.getLoad(VT, StoreChain, FIPtr, DAG.getSrcValue(0));
       break;
     }
     }