diff options
author | Mitch Phillips <mitchphillips@outlook.com> | 2017-11-08 00:20:53 +0000 |
---|---|---|
committer | Mitch Phillips <mitchphillips@outlook.com> | 2017-11-08 00:20:53 +0000 |
commit | 0222224da62299cac4b86fd9f36b82182eda6002 (patch) | |
tree | 255aaa7fcc1c0c38298019ea645db9e1e009b9f1 /llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp | |
parent | 98fe2079861fa651f83f05434951b112da6eea29 (diff) | |
download | bcm5719-llvm-0222224da62299cac4b86fd9f36b82182eda6002.tar.gz bcm5719-llvm-0222224da62299cac4b86fd9f36b82182eda6002.zip |
Revert rL317618
The implemented pass fails and is breaking a large number of unit tests.
Example:
http://lab.llvm.org:8011/builders/clang-with-lto-ubuntu/builds/5777/steps/build-stage3-compiler/logs/stdio
This reverts commit rL317618
llvm-svn: 317641
Diffstat (limited to 'llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp')
-rw-r--r-- | llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp | 459 |
1 files changed, 140 insertions, 319 deletions
diff --git a/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp b/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp index 8b8a52a040e..4232252af36 100644 --- a/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp +++ b/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp @@ -333,7 +333,7 @@ static unsigned getAltOpcode(unsigned Op) { case Instruction::Sub: return Instruction::Add; default: - return Op; + return 0; } } @@ -346,20 +346,6 @@ static bool sameOpcodeOrAlt(unsigned Opcode, unsigned AltOpcode, return Opcode == CheckedOpcode || AltOpcode == CheckedOpcode; } -/// Checks if the \p Opcode can be considered as an operand of a (possibly) -/// binary operation \p I. -/// \returns The code of the binary operation of instruction \p I if the -/// instruction with \p Opcode can be considered as an operand of \p I with the -/// default value. -static unsigned tryToRepresentAsInstArg(unsigned Opcode, Instruction *I) { - assert(!sameOpcodeOrAlt(Opcode, getAltOpcode(Opcode), I->getOpcode()) - && "Invalid Opcode"); - if (Opcode != Instruction::PHI && isa<BinaryOperator>(I) && - (I->getType()->isIntegerTy() || cast<FPMathOperator>(I)->isFast())) - return I->getOpcode(); - return 0; -} - /// Chooses the correct key for scheduling data. If \p Op has the same (or /// alternate) opcode as \p OpValue, the key is \p Op. Otherwise the key is \p /// OpValue. @@ -381,12 +367,7 @@ namespace { struct RawInstructionsData { /// Main Opcode of the instructions going to be vectorized. unsigned Opcode = 0; - /// Position of the first instruction with the \a Opcode. - unsigned OpcodePos = 0; - /// Need an additional analysis (if at least one of the instruction is not - /// same instruction kind as an instruction at OpcodePos position in the - /// list). - bool NeedAnalysis = false; + /// The list of instructions have some instructions with alternate opcodes. bool HasAltOpcodes = false; }; @@ -401,38 +382,16 @@ static RawInstructionsData getMainOpcode(ArrayRef<Value *> VL) { return {}; RawInstructionsData Res; unsigned Opcode = I0->getOpcode(); - unsigned AltOpcode = getAltOpcode(Opcode); - unsigned NewOpcodePos = 0; // Walk through the list of the vectorized instructions // in order to check its structure described by RawInstructionsData. for (unsigned Cnt = 0, E = VL.size(); Cnt != E; ++Cnt) { auto *I = dyn_cast<Instruction>(VL[Cnt]); if (!I) return {}; - if (sameOpcodeOrAlt(Opcode, AltOpcode, I->getOpcode())) { - if (Opcode != I->getOpcode()) { - Res.HasAltOpcodes = true; - if (Res.NeedAnalysis && isOdd(NewOpcodePos)) - std::swap(Opcode, AltOpcode); - } - continue; - } - if (unsigned NewOpcode = tryToRepresentAsInstArg(Opcode, I)) { - if (!Instruction::isBinaryOp(Opcode) || - !Instruction::isCommutative(Opcode)) { - NewOpcodePos = Cnt; - Opcode = NewOpcode; - AltOpcode = getAltOpcode(Opcode); - Res.NeedAnalysis = true; - } - } else if (tryToRepresentAsInstArg(I->getOpcode(), - cast<Instruction>(VL[NewOpcodePos]))) - Res.NeedAnalysis = true; - else - return {}; + if (Opcode != I->getOpcode()) + Res.HasAltOpcodes = true; } Res.Opcode = Opcode; - Res.OpcodePos = NewOpcodePos; return Res; } @@ -462,20 +421,16 @@ struct InstructionsState { static InstructionsState getSameOpcode(ArrayRef<Value *> VL) { auto Res = getMainOpcode(VL); unsigned Opcode = Res.Opcode; - if (!Res.NeedAnalysis && !Res.HasAltOpcodes) - return InstructionsState(VL[Res.OpcodePos], Opcode, false); - auto *OpInst = cast<Instruction>(VL[Res.OpcodePos]); + if (!Res.HasAltOpcodes) + return InstructionsState(VL[0], Opcode, false); + auto *OpInst = cast<Instruction>(VL[0]); unsigned AltOpcode = getAltOpcode(Opcode); // Examine each element in the list instructions VL to determine // if some operations there could be considered as an alternative - // (for example as subtraction relates to addition operation) or - // operation could be an operand of a (possibly) binary operation. + // (for example as subtraction relates to addition operation). for (int Cnt = 0, E = VL.size(); Cnt < E; Cnt++) { auto *I = cast<Instruction>(VL[Cnt]); unsigned InstOpcode = I->getOpcode(); - if (Res.NeedAnalysis && !sameOpcodeOrAlt(Opcode, AltOpcode, InstOpcode)) - if (tryToRepresentAsInstArg(InstOpcode, OpInst)) - InstOpcode = (Res.HasAltOpcodes && isOdd(Cnt)) ? AltOpcode : Opcode; if ((Res.HasAltOpcodes && InstOpcode != (isOdd(Cnt) ? AltOpcode : Opcode)) || (!Res.HasAltOpcodes && InstOpcode != Opcode)) { @@ -628,7 +583,6 @@ public: void deleteTree() { VectorizableTree.clear(); ScalarToTreeEntry.clear(); - ExtraScalarToTreeEntry.clear(); MustGather.clear(); ExternalUses.clear(); NumLoadsWantToKeepOrder = 0; @@ -768,40 +722,22 @@ private: /// The TreeEntry index containing the user of this entry. We can actually /// have multiple users so the data structure is not truly a tree. SmallVector<int, 1> UserTreeIndices; - - /// Info about instruction in this tree entry. - InstructionsState State; }; /// Create a new VectorizableTree entry. TreeEntry *newTreeEntry(ArrayRef<Value *> VL, bool Vectorized, - int &UserTreeIdx, const InstructionsState &S) { - assert((!Vectorized || S.Opcode != 0) && - "Vectorized TreeEntry without opcode"); + int &UserTreeIdx) { VectorizableTree.emplace_back(VectorizableTree); int idx = VectorizableTree.size() - 1; TreeEntry *Last = &VectorizableTree[idx]; Last->Scalars.insert(Last->Scalars.begin(), VL.begin(), VL.end()); Last->NeedToGather = !Vectorized; if (Vectorized) { - Last->State = S; - unsigned AltOpcode = getAltOpcode(S.Opcode); for (int i = 0, e = VL.size(); i != e; ++i) { - unsigned RealOpcode = - (S.IsAltShuffle && isOdd(i)) ? AltOpcode : S.Opcode; - Value *Key = (cast<Instruction>(VL[i])->getOpcode() == RealOpcode) - ? VL[i] - : S.OpValue; - assert(!getTreeEntry(VL[i], Key) && "Scalar already in tree!"); - if (VL[i] == Key) - ScalarToTreeEntry[Key] = idx; - else - ExtraScalarToTreeEntry[VL[i]][Key] = idx; + assert(!getTreeEntry(VL[i]) && "Scalar already in tree!"); + ScalarToTreeEntry[VL[i]] = idx; } } else { - Last->State.Opcode = 0; - Last->State.OpValue = VL[0]; - Last->State.IsAltShuffle = false; MustGather.insert(VL.begin(), VL.end()); } @@ -829,24 +765,8 @@ private: return nullptr; } - TreeEntry *getTreeEntry(Value *V, Value *OpValue) { - if (V == OpValue) - return getTreeEntry(V); - auto I = ExtraScalarToTreeEntry.find(V); - if (I != ExtraScalarToTreeEntry.end()) { - auto &STT = I->second; - auto STTI = STT.find(OpValue); - if (STTI != STT.end()) - return &VectorizableTree[STTI->second]; - } - return nullptr; - } - /// Maps a specific scalar to its tree entry. - SmallDenseMap<Value *, int> ScalarToTreeEntry; - - /// Maps a specific scalar to its tree entry(s) with leading scalar. - SmallDenseMap<Value *, SmallDenseMap<Value *, int>> ExtraScalarToTreeEntry; + SmallDenseMap<Value*, int> ScalarToTreeEntry; /// A list of scalars that we found that we need to keep as scalars. ValueSet MustGather; @@ -1418,15 +1338,9 @@ void BoUpSLP::buildTree(ArrayRef<Value *> Roots, continue; // For each lane: - const unsigned Opcode = Entry->State.Opcode; - const unsigned AltOpcode = getAltOpcode(Opcode); for (int Lane = 0, LE = Entry->Scalars.size(); Lane != LE; ++Lane) { Value *Scalar = Entry->Scalars[Lane]; - if (!sameOpcodeOrAlt(Opcode, AltOpcode, - cast<Instruction>(Scalar)->getOpcode())) - continue; - // Check if the scalar is externally used as an extra arg. auto ExtI = ExternallyUsedValues.find(Scalar); if (ExtI != ExternallyUsedValues.end()) { @@ -1469,38 +1383,6 @@ void BoUpSLP::buildTree(ArrayRef<Value *> Roots, } } -static Value *getDefaultConstantForOpcode(unsigned Opcode, Type *Ty) { - switch(Opcode) { - case Instruction::Add: - case Instruction::Sub: - case Instruction::Or: - case Instruction::Xor: - return ConstantInt::getNullValue(Ty); - case Instruction::Mul: - case Instruction::UDiv: - case Instruction::SDiv: - case Instruction::URem: - case Instruction::SRem: - return ConstantInt::get(Ty, /*V=*/1); - case Instruction::FAdd: - case Instruction::FSub: - return ConstantFP::get(Ty, /*V=*/0.0); - case Instruction::FMul: - case Instruction::FDiv: - case Instruction::FRem: - return ConstantFP::get(Ty, /*V=*/1.0); - case Instruction::And: - return ConstantInt::getAllOnesValue(Ty); - case Instruction::Shl: - case Instruction::LShr: - case Instruction::AShr: - return ConstantInt::getNullValue(Type::getInt32Ty(Ty->getContext())); - default: - break; - } - llvm_unreachable("unknown binop for default constant value"); -} - void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, int UserTreeIdx) { assert((allConstant(VL) || allSameType(VL)) && "Invalid types!"); @@ -1508,46 +1390,31 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, InstructionsState S = getSameOpcode(VL); if (Depth == RecursionMaxDepth) { DEBUG(dbgs() << "SLP: Gathering due to max recursion depth.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } // Don't handle vectors. if (S.OpValue->getType()->isVectorTy()) { DEBUG(dbgs() << "SLP: Gathering due to vector type.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } if (StoreInst *SI = dyn_cast<StoreInst>(S.OpValue)) if (SI->getValueOperand()->getType()->isVectorTy()) { DEBUG(dbgs() << "SLP: Gathering due to store vector type.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } // If all of the operands are identical or constant we have a simple solution. if (allConstant(VL) || isSplat(VL) || !allSameBlock(VL) || !S.Opcode) { DEBUG(dbgs() << "SLP: Gathering due to C,S,B,O. \n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } - // Avoid any vectors that are wider than two elements and - // with real operations less than or equal to half of vector - // to others members are operands to that operations. - unsigned AltOpcode = getAltOpcode(S.Opcode); - unsigned SameOrAlt = 0; - if (VL.size() > 2) { - for (Value *V : VL) { - auto *Instr = cast<Instruction>(V); - if (sameOpcodeOrAlt(S.Opcode, AltOpcode, Instr->getOpcode())) - SameOrAlt++; - } - if (SameOrAlt <= (VL.size() / 2)) - return; - } - // We now know that this is a vector of instructions of the same type from // the same block. @@ -1556,7 +1423,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, if (EphValues.count(VL[i])) { DEBUG(dbgs() << "SLP: The instruction (" << *VL[i] << ") is ephemeral.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } } @@ -1567,7 +1434,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, DEBUG(dbgs() << "SLP: \tChecking bundle: " << *VL[i] << ".\n"); if (E->Scalars[i] != VL[i]) { DEBUG(dbgs() << "SLP: Gathering due to partial overlap.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } } @@ -1586,7 +1453,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, if (getTreeEntry(I)) { DEBUG(dbgs() << "SLP: The instruction (" << *VL[i] << ") is already in tree.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } } @@ -1596,7 +1463,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, for (unsigned i = 0, e = VL.size(); i != e; ++i) { if (MustGather.count(VL[i])) { DEBUG(dbgs() << "SLP: Gathering due to gathered scalar.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } } @@ -1610,7 +1477,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, // Don't go into unreachable blocks. They may contain instructions with // dependency cycles which confuse the final scheduling. DEBUG(dbgs() << "SLP: bundle in unreachable block.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } @@ -1619,7 +1486,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, for (unsigned j = i + 1; j < e; ++j) if (VL[i] == VL[j]) { DEBUG(dbgs() << "SLP: Scalar used twice in bundle.\n"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } @@ -1634,7 +1501,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, assert((!BS.getScheduleData(VL0) || !BS.getScheduleData(VL0)->isPartOfBundle()) && "tryScheduleBundle should cancelScheduling on failure"); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } DEBUG(dbgs() << "SLP: We are able to schedule this bundle.\n"); @@ -1653,12 +1520,12 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, if (Term) { DEBUG(dbgs() << "SLP: Need to swizzle PHINodes (TerminatorInst use).\n"); BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } } - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); DEBUG(dbgs() << "SLP: added a vector of PHINodes.\n"); for (unsigned i = 0, e = PH->getNumIncomingValues(); i < e; ++i) { @@ -1680,7 +1547,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, } else { BS.cancelScheduling(VL, VL0); } - newTreeEntry(VL, Reuse, UserTreeIdx, S); + newTreeEntry(VL, Reuse, UserTreeIdx); return; } case Instruction::Load: { @@ -1695,7 +1562,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, if (DL->getTypeSizeInBits(ScalarTy) != DL->getTypeAllocSizeInBits(ScalarTy)) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: Gathering loads of non-packed type.\n"); return; } @@ -1706,7 +1573,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, LoadInst *L = cast<LoadInst>(VL[i]); if (!L->isSimple()) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: Gathering non-simple loads.\n"); return; } @@ -1728,7 +1595,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, if (Consecutive) { ++NumLoadsWantToKeepOrder; - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); DEBUG(dbgs() << "SLP: added a vector of loads.\n"); return; } @@ -1743,7 +1610,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, } BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); if (ReverseConsecutive) { ++NumLoadsWantToChangeOrder; @@ -1770,12 +1637,12 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, Type *Ty = cast<Instruction>(VL[i])->getOperand(0)->getType(); if (Ty != SrcTy || !isValidElementType(Ty)) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: Gathering casts with different src types.\n"); return; } } - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); DEBUG(dbgs() << "SLP: added a vector of casts.\n"); for (unsigned i = 0, e = VL0->getNumOperands(); i < e; ++i) { @@ -1798,13 +1665,13 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, if (Cmp->getPredicate() != P0 || Cmp->getOperand(0)->getType() != ComparedTy) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: Gathering cmp with different predicate.\n"); return; } } - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); DEBUG(dbgs() << "SLP: added a vector of compares.\n"); for (unsigned i = 0, e = VL0->getNumOperands(); i < e; ++i) { @@ -1836,7 +1703,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, case Instruction::And: case Instruction::Or: case Instruction::Xor: - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); DEBUG(dbgs() << "SLP: added a vector of bin op.\n"); // Sort operands of the instructions so that each side is more likely to @@ -1852,21 +1719,10 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, for (unsigned i = 0, e = VL0->getNumOperands(); i < e; ++i) { ValueList Operands; // Prepare the operand vector. - for (Value *VecOp : VL) { - auto *I = cast<Instruction>(VecOp); - if (I->getOpcode() == S.Opcode) { - Operands.push_back(I->getOperand(i)); - continue; - } - assert(Instruction::isBinaryOp(S.Opcode) && - "Expected a binary operation."); - Value *Operand = isOdd(i) - ? getDefaultConstantForOpcode(S.Opcode, I->getType()) - : VecOp; - Operands.push_back(Operand); - } - if (allSameType(Operands)) - buildTree_rec(Operands, Depth + 1, UserTreeIdx); + for (Value *j : VL) + Operands.push_back(cast<Instruction>(j)->getOperand(i)); + + buildTree_rec(Operands, Depth + 1, UserTreeIdx); } return; @@ -1876,7 +1732,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, if (cast<Instruction>(VL[j])->getNumOperands() != 2) { DEBUG(dbgs() << "SLP: not-vectorizable GEP (nested indexes).\n"); BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } } @@ -1889,7 +1745,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, if (Ty0 != CurTy) { DEBUG(dbgs() << "SLP: not-vectorizable GEP (different types).\n"); BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } } @@ -1901,12 +1757,12 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, DEBUG( dbgs() << "SLP: not-vectorizable GEP (non-constant indexes).\n"); BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); return; } } - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); DEBUG(dbgs() << "SLP: added a vector of GEPs.\n"); for (unsigned i = 0, e = 2; i < e; ++i) { ValueList Operands; @@ -1923,12 +1779,12 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, for (unsigned i = 0, e = VL.size() - 1; i < e; ++i) if (!isConsecutiveAccess(VL[i], VL[i + 1], *DL, *SE)) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: Non-consecutive store.\n"); return; } - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); DEBUG(dbgs() << "SLP: added a vector of stores.\n"); ValueList Operands; @@ -1946,7 +1802,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, Intrinsic::ID ID = getVectorIntrinsicIDForCall(CI, TLI); if (!isTriviallyVectorizable(ID)) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: Non-vectorizable call.\n"); return; } @@ -1960,7 +1816,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, getVectorIntrinsicIDForCall(CI2, TLI) != ID || !CI->hasIdenticalOperandBundleSchema(*CI2)) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: mismatched calls:" << *CI << "!=" << *VL[i] << "\n"); return; @@ -1971,7 +1827,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, Value *A1J = CI2->getArgOperand(1); if (A1I != A1J) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: mismatched arguments in call:" << *CI << " argument "<< A1I<<"!=" << A1J << "\n"); @@ -1984,14 +1840,14 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, CI->op_begin() + CI->getBundleOperandsEndIndex(), CI2->op_begin() + CI2->getBundleOperandsStartIndex())) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: mismatched bundle operands in calls:" << *CI << "!=" << *VL[i] << '\n'); return; } } - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); for (unsigned i = 0, e = CI->getNumArgOperands(); i != e; ++i) { ValueList Operands; // Prepare the operand vector. @@ -2008,11 +1864,11 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, // then do not vectorize this instruction. if (!S.IsAltShuffle) { BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: ShuffleVector are not vectorized.\n"); return; } - newTreeEntry(VL, true, UserTreeIdx, S); + newTreeEntry(VL, true, UserTreeIdx); DEBUG(dbgs() << "SLP: added a ShuffleVector op.\n"); // Reorder operands if reordering would enable vectorization. @@ -2027,19 +1883,8 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, for (unsigned i = 0, e = VL0->getNumOperands(); i < e; ++i) { ValueList Operands; // Prepare the operand vector. - for (Value *VecOp : VL) { - auto *I = cast<Instruction>(VecOp); - if (sameOpcodeOrAlt(S.Opcode, AltOpcode, I->getOpcode())) { - Operands.push_back(I->getOperand(i)); - continue; - } - assert(Instruction::isBinaryOp(S.Opcode) && - "Expected a binary operation."); - Value *Operand = isOdd(i) - ? getDefaultConstantForOpcode(S.Opcode, I->getType()) - : VecOp; - Operands.push_back(Operand); - } + for (Value *j : VL) + Operands.push_back(cast<Instruction>(j)->getOperand(i)); buildTree_rec(Operands, Depth + 1, UserTreeIdx); } @@ -2047,7 +1892,7 @@ void BoUpSLP::buildTree_rec(ArrayRef<Value *> VL, unsigned Depth, default: BS.cancelScheduling(VL, VL0); - newTreeEntry(VL, false, UserTreeIdx, S); + newTreeEntry(VL, false, UserTreeIdx); DEBUG(dbgs() << "SLP: Gathering unknown instruction.\n"); return; } @@ -2168,17 +2013,18 @@ int BoUpSLP::getEntryCost(TreeEntry *E) { } return getGatherCost(E->Scalars); } - assert(E->State.Opcode && allSameType(VL) && allSameBlock(VL) && "Invalid VL"); - auto *VL0 = cast<Instruction>(E->State.OpValue); - unsigned ShuffleOrOp = E->State.IsAltShuffle ? - (unsigned) Instruction::ShuffleVector : E->State.Opcode; + InstructionsState S = getSameOpcode(VL); + assert(S.Opcode && allSameType(VL) && allSameBlock(VL) && "Invalid VL"); + Instruction *VL0 = cast<Instruction>(S.OpValue); + unsigned ShuffleOrOp = S.IsAltShuffle ? + (unsigned) Instruction::ShuffleVector : S.Opcode; switch (ShuffleOrOp) { case Instruction::PHI: return 0; case Instruction::ExtractValue: case Instruction::ExtractElement: - if (canReuseExtract(VL, E->State.OpValue)) { + if (canReuseExtract(VL, S.OpValue)) { int DeadCost = 0; for (unsigned i = 0, e = VL.size(); i < e; ++i) { Instruction *E = cast<Instruction>(VL[i]); @@ -2222,8 +2068,8 @@ int BoUpSLP::getEntryCost(TreeEntry *E) { // Calculate the cost of this instruction. VectorType *MaskTy = VectorType::get(Builder.getInt1Ty(), VL.size()); int ScalarCost = VecTy->getNumElements() * - TTI->getCmpSelInstrCost(ShuffleOrOp, ScalarTy, Builder.getInt1Ty(), VL0); - int VecCost = TTI->getCmpSelInstrCost(ShuffleOrOp, VecTy, MaskTy, VL0); + TTI->getCmpSelInstrCost(S.Opcode, ScalarTy, Builder.getInt1Ty(), VL0); + int VecCost = TTI->getCmpSelInstrCost(S.Opcode, VecTy, MaskTy, VL0); return VecCost - ScalarCost; } case Instruction::Add: @@ -2249,7 +2095,7 @@ int BoUpSLP::getEntryCost(TreeEntry *E) { TargetTransformInfo::OperandValueKind Op1VK = TargetTransformInfo::OK_AnyValue; TargetTransformInfo::OperandValueKind Op2VK = - TargetTransformInfo::OK_AnyValue; + TargetTransformInfo::OK_UniformConstantValue; TargetTransformInfo::OperandValueProperties Op1VP = TargetTransformInfo::OP_None; TargetTransformInfo::OperandValueProperties Op2VP = @@ -2260,33 +2106,34 @@ int BoUpSLP::getEntryCost(TreeEntry *E) { // If instead not all operands are constants, then set the operand kind // to OK_AnyValue. If all operands are constants but not the same, // then set the operand kind to OK_NonUniformConstantValue. - if (auto *CInt = dyn_cast<ConstantInt>(VL0->getOperand(1))) { - Op2VK = TargetTransformInfo::OK_UniformConstantValue; - const unsigned Opcode = E->State.Opcode; - for (auto *V : VL) { - auto *I = cast<Instruction>(V); - if (I == VL0 || Opcode != I->getOpcode()) - continue; - if (!isa<ConstantInt>(I->getOperand(1))) { - Op2VK = TargetTransformInfo::OK_AnyValue; - break; - } - if (Op2VK == TargetTransformInfo::OK_UniformConstantValue && - CInt != cast<ConstantInt>(I->getOperand(1))) - Op2VK = TargetTransformInfo::OK_NonUniformConstantValue; + ConstantInt *CInt = nullptr; + for (unsigned i = 0; i < VL.size(); ++i) { + const Instruction *I = cast<Instruction>(VL[i]); + if (!isa<ConstantInt>(I->getOperand(1))) { + Op2VK = TargetTransformInfo::OK_AnyValue; + break; + } + if (i == 0) { + CInt = cast<ConstantInt>(I->getOperand(1)); + continue; } - // FIXME: Currently cost of model modification for division by power of - // 2 is handled for X86 and AArch64. Add support for other targets. if (Op2VK == TargetTransformInfo::OK_UniformConstantValue && - CInt->getValue().isPowerOf2()) - Op2VP = TargetTransformInfo::OP_PowerOf2; + CInt != cast<ConstantInt>(I->getOperand(1))) + Op2VK = TargetTransformInfo::OK_NonUniformConstantValue; } + // FIXME: Currently cost of model modification for division by power of + // 2 is handled for X86 and AArch64. Add support for other targets. + if (Op2VK == TargetTransformInfo::OK_UniformConstantValue && CInt && + CInt->getValue().isPowerOf2()) + Op2VP = TargetTransformInfo::OP_PowerOf2; - int ScalarCost = VecTy->getNumElements() * - TTI->getArithmeticInstrCost(E->State.Opcode, ScalarTy, - Op1VK, Op2VK, Op1VP, Op2VP); - int VecCost = TTI->getArithmeticInstrCost(E->State.Opcode, VecTy, Op1VK, - Op2VK, Op1VP, Op2VP); + SmallVector<const Value *, 4> Operands(VL0->operand_values()); + int ScalarCost = + VecTy->getNumElements() * + TTI->getArithmeticInstrCost(S.Opcode, ScalarTy, Op1VK, Op2VK, Op1VP, + Op2VP, Operands); + int VecCost = TTI->getArithmeticInstrCost(S.Opcode, VecTy, Op1VK, Op2VK, + Op1VP, Op2VP, Operands); return VecCost - ScalarCost; } case Instruction::GetElementPtr: { @@ -2352,18 +2199,23 @@ int BoUpSLP::getEntryCost(TreeEntry *E) { TargetTransformInfo::OK_AnyValue; TargetTransformInfo::OperandValueKind Op2VK = TargetTransformInfo::OK_AnyValue; - unsigned AltOpcode = getAltOpcode(E->State.Opcode); - int ScalarCost = - TTI->getArithmeticInstrCost(E->State.Opcode, ScalarTy, Op1VK, Op2VK) * - VL.size() / 2; - ScalarCost += - TTI->getArithmeticInstrCost(AltOpcode, ScalarTy, Op1VK, Op2VK) * - VL.size() / 2; + int ScalarCost = 0; + int VecCost = 0; + for (Value *i : VL) { + Instruction *I = cast<Instruction>(i); + if (!I) + break; + ScalarCost += + TTI->getArithmeticInstrCost(I->getOpcode(), ScalarTy, Op1VK, Op2VK); + } // VecCost is equal to sum of the cost of creating 2 vectors // and the cost of creating shuffle. - int VecCost = - TTI->getArithmeticInstrCost(E->State.Opcode, VecTy, Op1VK, Op2VK); - VecCost += TTI->getArithmeticInstrCost(AltOpcode, VecTy, Op1VK, Op2VK); + Instruction *I0 = cast<Instruction>(VL[0]); + VecCost = + TTI->getArithmeticInstrCost(I0->getOpcode(), VecTy, Op1VK, Op2VK); + Instruction *I1 = cast<Instruction>(VL[1]); + VecCost += + TTI->getArithmeticInstrCost(I1->getOpcode(), VecTy, Op1VK, Op2VK); VecCost += TTI->getShuffleCost(TargetTransformInfo::SK_Alternate, VecTy, 0); return VecCost - ScalarCost; @@ -2429,7 +2281,7 @@ int BoUpSLP::getSpillCost() { Instruction *PrevInst = nullptr; for (const auto &N : VectorizableTree) { - Instruction *Inst = dyn_cast<Instruction>(N.State.OpValue); + Instruction *Inst = dyn_cast<Instruction>(N.Scalars[0]); if (!Inst) continue; @@ -2489,7 +2341,7 @@ int BoUpSLP::getTreeCost() { for (TreeEntry &TE : VectorizableTree) { int C = getEntryCost(&TE); DEBUG(dbgs() << "SLP: Adding cost " << C << " for bundle that starts with " - << *TE.State.OpValue << ".\n"); + << *TE.Scalars[0] << ".\n"); Cost += C; } @@ -2510,7 +2362,7 @@ int BoUpSLP::getTreeCost() { // extend the extracted value back to the original type. Here, we account // for the extract and the added cost of the sign extend if needed. auto *VecTy = VectorType::get(EU.Scalar->getType(), BundleWidth); - auto *ScalarRoot = VectorizableTree[0].State.OpValue; + auto *ScalarRoot = VectorizableTree[0].Scalars[0]; if (MinBWs.count(ScalarRoot)) { auto *MinTy = IntegerType::get(F->getContext(), MinBWs[ScalarRoot].first); auto Extend = @@ -2573,15 +2425,13 @@ void BoUpSLP::reorderAltShuffleOperands(unsigned Opcode, ArrayRef<Value *> VL, SmallVectorImpl<Value *> &Right) { // Push left and right operands of binary operation into Left and Right unsigned AltOpcode = getAltOpcode(Opcode); + (void)AltOpcode; for (Value *V : VL) { auto *I = cast<Instruction>(V); - if (sameOpcodeOrAlt(Opcode, AltOpcode, I->getOpcode())) { - Left.push_back(I->getOperand(0)); - Right.push_back(I->getOperand(1)); - } else { - Left.push_back(I); - Right.push_back(getDefaultConstantForOpcode(Opcode, I->getType())); - } + assert(sameOpcodeOrAlt(Opcode, AltOpcode, I->getOpcode()) && + "Incorrect instruction in vector"); + Left.push_back(I->getOperand(0)); + Right.push_back(I->getOperand(1)); } // Reorder if we have a commutative operation and consecutive access @@ -2630,13 +2480,8 @@ static bool shouldReorderOperands( int i, unsigned Opcode, Instruction &I, ArrayRef<Value *> Left, ArrayRef<Value *> Right, bool AllSameOpcodeLeft, bool AllSameOpcodeRight, bool SplatLeft, bool SplatRight, Value *&VLeft, Value *&VRight) { - if (I.getOpcode() == Opcode) { - VLeft = I.getOperand(0); - VRight = I.getOperand(1); - } else { - VLeft = &I; - VRight = getDefaultConstantForOpcode(Opcode, I.getType()); - } + VLeft = I.getOperand(0); + VRight = I.getOperand(1); // If we have "SplatRight", try to see if commuting is needed to preserve it. if (SplatRight) { if (VRight == Right[i - 1]) @@ -2700,15 +2545,8 @@ void BoUpSLP::reorderInputsAccordingToOpcode(unsigned Opcode, // Peel the first iteration out of the loop since there's nothing // interesting to do anyway and it simplifies the checks in the loop. auto *I = cast<Instruction>(VL[0]); - Value *VLeft; - Value *VRight; - if (I->getOpcode() == Opcode) { - VLeft = I->getOperand(0); - VRight = I->getOperand(1); - } else { - VLeft = I; - VRight = getDefaultConstantForOpcode(Opcode, I->getType()); - } + Value *VLeft = I->getOperand(0); + Value *VRight = I->getOperand(1); if (!isa<Instruction>(VRight) && isa<Instruction>(VLeft)) // Favor having instruction to the right. FIXME: why? std::swap(VLeft, VRight); @@ -2913,11 +2751,12 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E) { IRBuilder<>::InsertPointGuard Guard(Builder); if (E->VectorizedValue) { - DEBUG(dbgs() << "SLP: Diamond merged for " << *E->State.OpValue << ".\n"); + DEBUG(dbgs() << "SLP: Diamond merged for " << *E->Scalars[0] << ".\n"); return E->VectorizedValue; } - Instruction *VL0 = cast<Instruction>(E->State.OpValue); + InstructionsState S = getSameOpcode(E->Scalars); + Instruction *VL0 = cast<Instruction>(E->Scalars[0]); Type *ScalarTy = VL0->getType(); if (StoreInst *SI = dyn_cast<StoreInst>(VL0)) ScalarTy = SI->getValueOperand()->getType(); @@ -2930,8 +2769,8 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E) { return V; } - unsigned ShuffleOrOp = E->State.IsAltShuffle ? - (unsigned) Instruction::ShuffleVector : E->State.Opcode; + unsigned ShuffleOrOp = S.IsAltShuffle ? + (unsigned) Instruction::ShuffleVector : S.Opcode; switch (ShuffleOrOp) { case Instruction::PHI: { PHINode *PH = dyn_cast<PHINode>(VL0); @@ -3041,7 +2880,7 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E) { CmpInst::Predicate P0 = cast<CmpInst>(VL0)->getPredicate(); Value *V; - if (E->State.Opcode == Instruction::FCmp) + if (S.Opcode == Instruction::FCmp) V = Builder.CreateFCmp(P0, L, R); else V = Builder.CreateICmp(P0, L, R); @@ -3093,19 +2932,13 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E) { case Instruction::Xor: { ValueList LHSVL, RHSVL; if (isa<BinaryOperator>(VL0) && VL0->isCommutative()) - reorderInputsAccordingToOpcode(E->State.Opcode, E->Scalars, LHSVL, + reorderInputsAccordingToOpcode(S.Opcode, E->Scalars, LHSVL, RHSVL); else for (Value *V : E->Scalars) { auto *I = cast<Instruction>(V); - if (I->getOpcode() == E->State.Opcode) { - LHSVL.push_back(I->getOperand(0)); - RHSVL.push_back(I->getOperand(1)); - } else { - LHSVL.push_back(V); - RHSVL.push_back( - getDefaultConstantForOpcode(E->State.Opcode, I->getType())); - } + LHSVL.push_back(I->getOperand(0)); + RHSVL.push_back(I->getOperand(1)); } setInsertPointAfterBundle(E->Scalars, VL0); @@ -3117,7 +2950,7 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E) { return V; Value *V = Builder.CreateBinOp( - static_cast<Instruction::BinaryOps>(E->State.Opcode), LHS, RHS); + static_cast<Instruction::BinaryOps>(S.Opcode), LHS, RHS); E->VectorizedValue = V; propagateIRFlags(E->VectorizedValue, E->Scalars, VL0); ++NumVectorInstructions; @@ -3267,9 +3100,9 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E) { } case Instruction::ShuffleVector: { ValueList LHSVL, RHSVL; - assert(Instruction::isBinaryOp(E->State.Opcode) && + assert(Instruction::isBinaryOp(S.Opcode) && "Invalid Shuffle Vector Operand"); - reorderAltShuffleOperands(E->State.Opcode, E->Scalars, LHSVL, RHSVL); + reorderAltShuffleOperands(S.Opcode, E->Scalars, LHSVL, RHSVL); setInsertPointAfterBundle(E->Scalars, VL0); Value *LHS = vectorizeTree(LHSVL); @@ -3280,9 +3113,9 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E) { // Create a vector of LHS op1 RHS Value *V0 = Builder.CreateBinOp( - static_cast<Instruction::BinaryOps>(E->State.Opcode), LHS, RHS); + static_cast<Instruction::BinaryOps>(S.Opcode), LHS, RHS); - unsigned AltOpcode = getAltOpcode(E->State.Opcode); + unsigned AltOpcode = getAltOpcode(S.Opcode); // Create a vector of LHS op2 RHS Value *V1 = Builder.CreateBinOp( static_cast<Instruction::BinaryOps>(AltOpcode), LHS, RHS); @@ -3304,13 +3137,8 @@ Value *BoUpSLP::vectorizeTree(TreeEntry *E) { } Value *ShuffleMask = ConstantVector::get(Mask); - InstructionsState S = getSameOpcode(EvenScalars); - assert(!S.IsAltShuffle && "Unexpected alternate opcode"); - propagateIRFlags(V0, EvenScalars, S.OpValue); - - S = getSameOpcode(OddScalars); - assert(!S.IsAltShuffle && "Unexpected alternate opcode"); - propagateIRFlags(V1, OddScalars, S.OpValue); + propagateIRFlags(V0, EvenScalars); + propagateIRFlags(V1, OddScalars); Value *V = Builder.CreateShuffleVector(V0, V1, ShuffleMask); E->VectorizedValue = V; @@ -3344,7 +3172,7 @@ BoUpSLP::vectorizeTree(ExtraValueToDebugLocsMap &ExternallyUsedValues) { // If the vectorized tree can be rewritten in a smaller type, we truncate the // vectorized root. InstCombine will then rewrite the entire expression. We // sign extend the extracted values below. - auto *ScalarRoot = VectorizableTree[0].State.OpValue; + auto *ScalarRoot = VectorizableTree[0].Scalars[0]; if (MinBWs.count(ScalarRoot)) { if (auto *I = dyn_cast<Instruction>(VectorRoot)) Builder.SetInsertPoint(&*++BasicBlock::iterator(I)); @@ -3455,15 +3283,9 @@ BoUpSLP::vectorizeTree(ExtraValueToDebugLocsMap &ExternallyUsedValues) { assert(Entry->VectorizedValue && "Can't find vectorizable value"); // For each lane: - const unsigned Opcode = Entry->State.Opcode; - const unsigned AltOpcode = getAltOpcode(Opcode); for (int Lane = 0, LE = Entry->Scalars.size(); Lane != LE; ++Lane) { Value *Scalar = Entry->Scalars[Lane]; - if (!sameOpcodeOrAlt(Opcode, AltOpcode, - cast<Instruction>(Scalar)->getOpcode())) - continue; - Type *Ty = Scalar->getType(); if (!Ty->isVoidTy()) { #ifndef NDEBUG @@ -3595,7 +3417,7 @@ bool BoUpSLP::BlockScheduling::tryScheduleBundle(ArrayRef<Value *> VL, } for (Value *V : VL) { - ScheduleData *BundleMember = getScheduleData(V, isOneOf(OpValue, V)); + ScheduleData *BundleMember = getScheduleData(V); assert(BundleMember && "no ScheduleData for bundle member (maybe not in same basic block)"); if (BundleMember->IsScheduled) { @@ -3668,7 +3490,7 @@ void BoUpSLP::BlockScheduling::cancelScheduling(ArrayRef<Value *> VL, if (isa<PHINode>(OpValue)) return; - ScheduleData *Bundle = getScheduleData(OpValue)->FirstInBundle; + ScheduleData *Bundle = getScheduleData(OpValue); DEBUG(dbgs() << "SLP: cancel scheduling of " << *Bundle << "\n"); assert(!Bundle->IsScheduled && "Can't cancel bundle which is already scheduled"); @@ -3971,7 +3793,7 @@ void BoUpSLP::scheduleBlock(BlockScheduling *BS) { I = I->getNextNode()) { BS->doForAllOpcodes(I, [this, &Idx, &NumToSchedule, BS](ScheduleData *SD) { assert(SD->isPartOfBundle() == - (getTreeEntry(SD->Inst, SD->OpValue) != nullptr) && + (getTreeEntry(SD->Inst) != nullptr) && "scheduler and vectorizer bundle mismatch"); SD->FirstInBundle->SchedulingPriority = Idx++; if (SD->isSchedulingEntity()) { @@ -3994,13 +3816,12 @@ void BoUpSLP::scheduleBlock(BlockScheduling *BS) { ScheduleData *BundleMember = picked; while (BundleMember) { Instruction *pickedInst = BundleMember->Inst; - if (pickedInst == BundleMember->OpValue) { - if (LastScheduledInst->getNextNode() != pickedInst) { - BS->BB->getInstList().remove(pickedInst); - BS->BB->getInstList().insert(LastScheduledInst->getIterator(), pickedInst); - } - LastScheduledInst = pickedInst; + if (LastScheduledInst->getNextNode() != pickedInst) { + BS->BB->getInstList().remove(pickedInst); + BS->BB->getInstList().insert(LastScheduledInst->getIterator(), + pickedInst); } + LastScheduledInst = pickedInst; BundleMember = BundleMember->NextInBundle; } |