From 5b2316e2b06e0fb6aea31d9d3eb61376b40c23a6 Mon Sep 17 00:00:00 2001 From: Chris Lattner Date: Tue, 28 Mar 2006 20:24:43 +0000 Subject: [PATCH] Teach Legalize how to pack VVECTOR_SHUFFLE nodes into VECTOR_SHUFFLE nodes. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@27232 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/CodeGen/SelectionDAG/LegalizeDAG.cpp | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp b/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp index 8bcb37ece99..d6c3fe0920f 100644 --- a/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp +++ b/lib/CodeGen/SelectionDAG/LegalizeDAG.cpp @@ -4532,6 +4532,27 @@ SDOperand SelectionDAGLegalize::PackVectorOp(SDOperand Op, Node->getOperand(1), Node->getOperand(2)); } break; + case ISD::VVECTOR_SHUFFLE: + if (!MVT::isVector(NewVT)) { + // Returning a scalar? Figure out if it is the LHS or RHS and return it. + SDOperand EltNum = Node->getOperand(2).getOperand(0); + if (cast(EltNum)->getValue()) + Result = PackVectorOp(Node->getOperand(1), NewVT); + else + Result = PackVectorOp(Node->getOperand(0), NewVT); + } else { + // Otherwise, return a VECTOR_SHUFFLE node. First convert the index + // vector from a VBUILD_VECTOR to a BUILD_VECTOR. + std::vector BuildVecIdx(Node->getOperand(2).Val->op_begin(), + Node->getOperand(2).Val->op_end()-2); + MVT::ValueType BVT = MVT::getIntVectorWithNumElements(BuildVecIdx.size()); + SDOperand BV = DAG.getNode(ISD::BUILD_VECTOR, BVT, BuildVecIdx); + + Result = DAG.getNode(ISD::VECTOR_SHUFFLE, NewVT, + PackVectorOp(Node->getOperand(0), NewVT), + PackVectorOp(Node->getOperand(1), NewVT), BV); + } + break; case ISD::VBIT_CONVERT: if (Op.getOperand(0).getValueType() != MVT::Vector) Result = DAG.getNode(ISD::BIT_CONVERT, NewVT, Op.getOperand(0));