25 cl::desc(
"Helps find bugs by verifying the IR whenever we "
26 "emit new instructions (*very* expensive)."));
30 std::numeric_limits<unsigned long>::max();
33 cl::desc(
"Vectorize if the invocation count is < than this. 0 "
34 "disables vectorization."));
37 std::numeric_limits<unsigned long>::max();
40 cl::desc(
"Vectorize up to this many bundles."));
46 for (
Value *BndlV : Bndl) {
48 Operands.push_back(BndlI->getOperand(OpIdx));
58 "Expect Instructions!");
59 auto &Ctx = Bndl[0]->getContext();
69 case Instruction::Opcode::ZExt:
70 case Instruction::Opcode::SExt:
71 case Instruction::Opcode::FPToUI:
72 case Instruction::Opcode::FPToSI:
73 case Instruction::Opcode::FPExt:
74 case Instruction::Opcode::PtrToInt:
75 case Instruction::Opcode::IntToPtr:
76 case Instruction::Opcode::SIToFP:
77 case Instruction::Opcode::UIToFP:
78 case Instruction::Opcode::Trunc:
79 case Instruction::Opcode::FPTrunc:
80 case Instruction::Opcode::BitCast: {
85 case Instruction::Opcode::FCmp:
86 case Instruction::Opcode::ICmp: {
92 "Expected same predicate across bundle.");
96 case Instruction::Opcode::Select: {
100 case Instruction::Opcode::FNeg: {
102 auto OpC = UOp0->getOpcode();
104 WhereIt, Ctx,
"Vec");
106 case Instruction::Opcode::Add:
107 case Instruction::Opcode::FAdd:
108 case Instruction::Opcode::Sub:
109 case Instruction::Opcode::FSub:
110 case Instruction::Opcode::Mul:
111 case Instruction::Opcode::FMul:
112 case Instruction::Opcode::UDiv:
113 case Instruction::Opcode::SDiv:
114 case Instruction::Opcode::FDiv:
115 case Instruction::Opcode::URem:
116 case Instruction::Opcode::SRem:
117 case Instruction::Opcode::FRem:
118 case Instruction::Opcode::Shl:
119 case Instruction::Opcode::LShr:
120 case Instruction::Opcode::AShr:
121 case Instruction::Opcode::And:
122 case Instruction::Opcode::Or:
123 case Instruction::Opcode::Xor: {
128 BinOp0->getOpcode(),
LHS,
RHS, BinOp0, WhereIt, Ctx,
"Vec");
130 case Instruction::Opcode::Load: {
132 Value *Ptr = Ld0->getPointerOperand();
136 case Instruction::Opcode::Store: {
142 case Instruction::Opcode::UncondBr:
143 case Instruction::Opcode::CondBr:
144 case Instruction::Opcode::Ret:
145 case Instruction::Opcode::PHI:
146 case Instruction::Opcode::AddrSpaceCast:
147 case Instruction::Opcode::Call:
148 case Instruction::Opcode::GetElementPtr:
159 auto *NewI = CreateVectorInstr(Bndl,
Operands);
164void BundleVec::tryEraseDeadInstrs() {
165 DenseMap<BasicBlock *, SmallVector<Instruction *>> SortedDeadInstrCandidates;
167 for (
auto *DeadI : DeadInstrCandidates)
168 SortedDeadInstrCandidates[DeadI->getParent()].push_back(DeadI);
169 for (
auto &Pair : SortedDeadInstrCandidates)
172 for (
const auto &Pair : SortedDeadInstrCandidates) {
174 if (
I->hasNUses(0)) {
177 I->eraseFromParent();
181 DeadInstrCandidates.clear();
189 VecOp->getContext(),
"VShuf");
203 Context &Ctx = ToPack[0]->getContext();
205 unsigned InsertIdx = 0;
206 for (
Value *Elm : ToPack) {
209 if (Elm->getType()->isVectorTy()) {
212 for (
auto ExtrLane :
seq<int>(0, NumElms)) {
225 LastInsert, ExtrI, InsertLaneC, WhereIt, Ctx,
"VPack");
226 LastInsert = InsertI;
236 WhereIt, Ctx,
"Pack");
238 WhereIt = std::next(NewI->getIterator());
245 for (
Value *V : Bndl)
250 case Instruction::Opcode::Load: {
254 DeadInstrCandidates.insert(Ptr);
257 case Instruction::Opcode::Store: {
261 DeadInstrCandidates.insert(Ptr);
276 const auto &LegalityRes = StopForDebug ? Legality.getForcedPackForDebugging()
277 : Legality.canVectorize(Bndl);
287 auto ActionPtr = std::make_unique<Action>(&LegalityRes, Bndl,
289 Action *Action = ActionPtr.get();
290 IMaps->registerVector(Bndl, Action);
291 Actions.push_back(std::move(ActionPtr));
296 SmallPtrSet<Instruction *, 4> Claimed;
297 for (
const auto &NextUserBndl :
299 vectorizeRec(NextUserBndl, Bndl,
Depth + 1, Legality);
306 std::make_unique<Action>(&LegalityRes, Bndl, UserBndl,
Depth);
308 switch (LegalityRes.getSubclassID()) {
311 switch (
I->getOpcode()) {
312 case Instruction::Opcode::Load:
314 case Instruction::Opcode::Store: {
331 IMaps->registerVector(Bndl, ActionPtr.get());
341 ActionPtr->Operands = std::move(
Operands);
342 auto *Action = ActionPtr.get();
343 Actions.push_back(std::move(ActionPtr));
348void BundleVec::ActionsVector::print(raw_ostream &OS)
const {
349 for (
auto [Idx, Action] :
enumerate(Actions)) {
354void BundleVec::ActionsVector::dump()
const {
print(
dbgs()); }
362 WhereIt = std::next(VecI->getIterator());
369 "A widened Bndl should contain instrs!");
384 auto IsExternal = [
this](
const Use &
U) {
385 return !IMaps->isVectorized(
U.getUser());
388 if (
none_of(Elm->uses(), IsExternal))
391 Elm->replaceUsesWithIf(UnpackV, IsExternal);
395Value *BundleVec::emitVectors() {
396 Value *NewVec =
nullptr;
397 for (
const auto &ActionPtr : Actions) {
400 const LegalityResult &LegalityRes = *ActionPtr->LegalityRes;
401 unsigned Depth = ActionPtr->Depth;
402 auto *UserBB = !UserBndl.empty()
406 switch (LegalityRes.getSubclassID()) {
409 SmallVector<Value *, 2> VecOperands;
411 switch (
I->getOpcode()) {
412 case Instruction::Opcode::Load:
415 case Instruction::Opcode::Store:
416 VecOperands.push_back(ActionPtr->Operands[0]->Vec);
420 for (Action *OpA : ActionPtr->Operands)
421 VecOperands.push_back(OpA->Vec);
425 switch (
I->getOpcode()) {
426 case Instruction::Opcode::Load:
429 case Instruction::Opcode::Store: {
431 if (Action *OpA = IMaps->getVectorForOrig(OpBndl[0]))
432 VecOperands.push_back(OpA->Vec);
434 VecOperands.push_back(createPack(OpBndl, UserBB));
439 for (
unsigned OpIdx = 0; OpIdx <
I->getNumOperands(); ++OpIdx) {
441 if (Action *OpA = IMaps->getVectorForOrig(OpBndl[0]))
442 VecOperands.push_back(OpA->Vec);
444 VecOperands.push_back(createPack(OpBndl, UserBB));
449 NewVec = createVectorInstr(ActionPtr->Bndl, VecOperands);
452 if (NewVec !=
nullptr)
453 collectPotentiallyDeadInstrs(Bndl);
456 emitUnpacksForExternalUses(ActionPtr->Bndl, NewVec);
465 const ShuffleMask &
Mask =
467 NewVec = createShuffle(VecOp, Mask, UserBB);
468 assert(NewVec->getType() == VecOp->getType() &&
469 "Expected same type! Bad mask ?");
478 SmallVector<Value *, 4> DescrInstrs;
479 for (
const auto &ElmDescr : Descr.getDescrs()) {
480 auto *
V = ElmDescr.needsExtract() ? ElmDescr.getValue()->Vec
481 : ElmDescr.getScalar();
483 DescrInstrs.push_back(
I);
489 Context &Ctx = LastV->getContext();
491 for (
const auto &ElmDescr : Descr.getDescrs()) {
492 Value *VecOp =
nullptr;
493 Value *ValueToInsert;
494 if (ElmDescr.needsExtract()) {
495 VecOp = ElmDescr.getValue()->Vec;
499 VecOp, IdxC, WhereIt, VecOp->getContext(),
"VExt");
501 ValueToInsert = ElmDescr.getScalar();
504 if (NumLanesToInsert == 1) {
509 WhereIt, Ctx,
"VIns");
518 for (
unsigned LnCnt = 0; LnCnt != NumLanesToInsert; ++LnCnt) {
521 WhereIt, Ctx,
"VExt");
522 unsigned InsLane = Lane + LnCnt;
528 Lane += NumLanesToInsert;
537 NewVec = createPack(Bndl, UserBB);
541 if (NewVec !=
nullptr) {
543 ActionPtr->Vec = NewVec;
561 LegalityAnalysis &Legality) {
565 DeadInstrCandidates.clear();
569 vectorizeRec(Bndl, {}, 0, Legality);
571 <<
"Vec: Vectorization Actions:\n";
574 tryEraseDeadInstrs();
579 const auto &SeedSlice = Rgn.
getAux();
580 if (SeedSlice.size() < 2)
582 Function &
F = *SeedSlice[0]->getParent()->getParent();
583 IMaps = std::make_unique<InstrMaps>();
585 F.getParent()->getDataLayout(),
F.getContext(),
593 return tryVectorize(SeedSliceVals, Legality);
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
static GCRegistry::Add< ErlangGC > A("erlang", "erlang-compatible garbage collector")
#define LLVM_UNLIKELY(EXPR)
This file defines the SmallVector class.
static SymbolRef::Type getType(const Symbol *Sym)
Represent a constant reference to an array (0 or more elements consecutively in memory),...
InstListType::iterator iterator
Instruction iterators...
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
LLVM Value Representation.
static LLVM_ABI Value * createWithCopiedFlags(Instruction::Opcode Op, Value *LHS, Value *RHS, Value *CopyFrom, InsertPosition Pos, Context &Ctx, const Twine &Name="")
bool runOnRegion(Region &Rgn, const Analyses &A) final
\Returns true if it modifies R.
static LLVM_ABI Value * create(Type *DestTy, Opcode Op, Value *Operand, InsertPosition Pos, Context &Ctx, const Twine &Name="")
static LLVM_ABI Value * create(Predicate Pred, Value *S1, Value *S2, InsertPosition Pos, Context &Ctx, const Twine &Name="")
static LLVM_ABI Constant * get(Type *Ty, uint64_t V, bool IsSigned=false)
If Ty is a vector type, return a Constant with a splat of the given value.
static LLVM_ABI ConstantInt * getSigned(IntegerType *Ty, int64_t V)
Return a ConstantInt with the specified value for the specified type.
static LLVM_ABI Value * create(Value *Vec, Value *Idx, InsertPosition Pos, Context &Ctx, const Twine &Name="")
static LLVM_ABI Value * create(Value *Vec, Value *NewElt, Value *Idx, InsertPosition Pos, Context &Ctx, const Twine &Name="")
LLVM_ABI BBIterator getIterator() const
\Returns a BasicBlock::iterator for this Instruction.
Performs the legality analysis and returns a LegalityResult object.
static LLVM_ABI LoadInst * create(Type *Ty, Value *Ptr, MaybeAlign Align, InsertPosition Pos, bool IsVolatile, Context &Ctx, const Twine &Name="")
virtual void print(raw_ostream &OS) const
static LLVM_ABI PoisonValue * get(Type *T)
Static factory methods - Return an 'poison' object of the specified type.
const SmallVector< Instruction * > & getAux() const
\Returns the auxiliary vector.
static LLVM_ABI Value * create(Value *Cond, Value *True, Value *False, InsertPosition Pos, Context &Ctx, const Twine &Name="")
static LLVM_ABI Value * create(Value *V1, Value *V2, Value *Mask, InsertPosition Pos, Context &Ctx, const Twine &Name="")
static LLVM_ABI StoreInst * create(Value *V, Value *Ptr, MaybeAlign Align, InsertPosition Pos, bool IsVolatile, Context &Ctx)
static LLVM_ABI IntegerType * getInt32Ty(Context &Ctx)
static LLVM_ABI Value * createWithCopiedFlags(Instruction::Opcode Op, Value *OpV, Value *CopyFrom, InsertPosition Pos, Context &Ctx, const Twine &Name="")
static Type * getExpectedType(const Value *V)
\Returns the expected type of Value V.
static bool verifyFunction(const Function *F, raw_ostream &OS)
Equivalent to llvm::verifyFunction().
A SandboxIR Value has users. This is the base class.
static Type * getCommonScalarType(ArrayRef< Value * > Bndl)
Similar to tryGetCommonScalarType() but will assert that there is a common type.
static Instruction * getLastPHIOrSelf(Instruction *I)
If I is not a PHI it returns it.
static unsigned getNumLanes(Type *Ty)
\Returns the number of vector lanes of Ty or 1 if not a vector.
static Value * unpack(Value *FromVec, Type *ExtrTy, unsigned Lane, BasicBlock::iterator WhereIt)
Emits the necessary instruction sequence to extract element of type ExtrTy at Lane from FromVec.
static LLVM_DUMP_METHOD void dump(ArrayRef< Value * > Bndl)
Helper dump function for debugging.
static Type * getWideType(Type *ElemTy, unsigned NumElts)
\Returns <NumElts x ElemTy>.
static auto enumerateLanes(const ValueContainerT &Range)
Helper for creating LaneValueEnumerator ranges.
static Type * getElementType(Type *Ty)
Returns Ty if scalar or its element type if vector.
static BasicBlock::iterator getInsertPointAfterInstrs(ArrayRef< Value * > Vals, BasicBlock *BB)
\Returns the BB iterator after the lowest instruction in Vals (skipping instructions not in BB),...
static LLVM_ABI SmallVector< BundleTy > getNextUserBundles(ArrayRef< Value * > Bndl, const InstrMaps &IMaps, SmallPtrSet< Instruction *, 4 > &Claimed)
For each user of lane 0 in Bndl, try to form a bundle of matching users for all lanes.
#define llvm_unreachable(msg)
Marks that the current location is not supposed to be reachable.
constexpr char Align[]
Key for Kernel::Arg::Metadata::mAlign.
constexpr std::underlying_type_t< E > Mask()
Get a bitmask with 1s in all places up to the high-order bit of E's largest value.
initializer< Ty > init(const Ty &Val)
NodeAddr< UseNode * > Use
LLVM_ABI Function * getParent() const
StringLiteral schedDirectionToStr(SchedDirection Dir)
@ DiamondReuseWithShuffle
BasicBlock(llvm::BasicBlock *BB, Context &SBCtx)
friend class Instruction
Iterator for Instructions in a `BasicBlock.
SmallVector< Value *, 4 > BundleTy
static BundleTy getOperand(ArrayRef< Value * > Bndl, unsigned OpIdx)
This is an optimization pass for GlobalISel generic memory operations.
auto drop_begin(T &&RangeOrContainer, size_t N=1)
Return a range covering RangeOrContainer with the first N elements excluded.
bool all_of(R &&range, UnaryPredicate P)
Provide wrappers to std::all_of which take ranges instead of having to pass begin/end explicitly.
static cl::opt< unsigned long > StopAt("sbvec-stop-at", cl::init(StopAtDisabled), cl::Hidden, cl::desc("Vectorize if the invocation count is < than this. 0 " "disables vectorization."))
auto enumerate(FirstRange &&First, RestRanges &&...Rest)
Given two or more input ranges, returns a new range whose values are tuples (A, B,...
decltype(auto) dyn_cast(const From &Val)
dyn_cast<X> - Return the argument parameter cast to the specified type.
static constexpr unsigned long StopBundleDisabled
static cl::opt< unsigned long > StopBundle("sbvec-stop-bndl", cl::init(StopBundleDisabled), cl::Hidden, cl::desc("Vectorize up to this many bundles."))
const Value * getPointerOperand(const Value *V)
A helper function that returns the pointer operand of a load, store or GEP instruction.
RelativeUniformCounterPtr ValuesPtrExpr VTableAddr Value
auto reverse(ContainerTy &&C)
void sort(IteratorTy Start, IteratorTy End)
LLVM_ABI raw_ostream & dbgs()
dbgs() - This returns a reference to a raw_ostream for debugging messages.
bool none_of(R &&Range, UnaryPredicate P)
Provide wrappers to std::none_of which take ranges instead of having to pass begin/end explicitly.
class LLVM_GSL_OWNER SmallVector
Forward declaration of SmallVector so that calculateSmallVectorDefaultInlinedElements can reference s...
bool isa(const From &Val)
isa<X> - Return true if the parameter to the template is an instance of one of the template type argu...
ArrayRef(const T &OneElt) -> ArrayRef< T >
static constexpr unsigned long StopAtDisabled
decltype(auto) cast(const From &Val)
cast<X> - Return the argument parameter cast to the specified type.
constexpr auto seq(T Begin, T End)
Iterate over an integral type from Begin up to - but not including - End.
static cl::opt< bool > AlwaysVerify("sbvec-always-verify", cl::init(false), cl::Hidden, cl::desc("Helps find bugs by verifying the IR whenever we " "emit new instructions (*very* expensive)."))