Make fix for r60829 less conservative to allow the proper optimization for

vec_extract-sse4.ll.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@60865 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Mon P Wang 2008-12-11 00:26:16 +00:00
parent 0ffa3de9e9
commit a60b5236c1

View File

@ -4884,14 +4884,16 @@ SDValue DAGCombiner::visitEXTRACT_VECTOR_ELT(SDNode *N) {
if (isa<ConstantSDNode>(EltNo)) { if (isa<ConstantSDNode>(EltNo)) {
unsigned Elt = cast<ConstantSDNode>(EltNo)->getZExtValue(); unsigned Elt = cast<ConstantSDNode>(EltNo)->getZExtValue();
bool NewLoad = false; bool NewLoad = false;
bool BCNumEltsChanged = false;
MVT VT = InVec.getValueType(); MVT VT = InVec.getValueType();
MVT EVT = VT.getVectorElementType(); MVT EVT = VT.getVectorElementType();
MVT LVT = EVT; MVT LVT = EVT;
if (InVec.getOpcode() == ISD::BIT_CONVERT) { if (InVec.getOpcode() == ISD::BIT_CONVERT) {
MVT BCVT = InVec.getOperand(0).getValueType(); MVT BCVT = InVec.getOperand(0).getValueType();
if (!BCVT.isVector() || EVT.bitsGT(BCVT.getVectorElementType()) || if (!BCVT.isVector() || EVT.bitsGT(BCVT.getVectorElementType()))
VT.getVectorNumElements() != BCVT.getVectorNumElements())
return SDValue(); return SDValue();
if (VT.getVectorNumElements() != BCVT.getVectorNumElements())
BCNumEltsChanged = true;
InVec = InVec.getOperand(0); InVec = InVec.getOperand(0);
EVT = BCVT.getVectorElementType(); EVT = BCVT.getVectorElementType();
NewLoad = true; NewLoad = true;
@ -4908,6 +4910,11 @@ SDValue DAGCombiner::visitEXTRACT_VECTOR_ELT(SDNode *N) {
// (vextract (vector_shuffle (load $addr), v2, <1, u, u, u>), 1) // (vextract (vector_shuffle (load $addr), v2, <1, u, u, u>), 1)
// => // =>
// (load $addr+1*size) // (load $addr+1*size)
// If the bit convert changed the number of elements, it is unsafe
// to examine the mask.
if (BCNumEltsChanged)
return SDValue();
unsigned Idx = cast<ConstantSDNode>(InVec.getOperand(2). unsigned Idx = cast<ConstantSDNode>(InVec.getOperand(2).
getOperand(Elt))->getZExtValue(); getOperand(Elt))->getZExtValue();
unsigned NumElems = InVec.getOperand(2).getNumOperands(); unsigned NumElems = InVec.getOperand(2).getNumOperands();