shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

[SLP] Migrate away from PointerUnion::get

Browse Source
This commit is contained in:
Fangrui Song 2024-12-27 21:01:08 -08:00
parent 092966a44d
commit edc42b2dc1
1 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp
View File

@ -10776,23 +10776,21 @@ public:
}
if (ForExtracts) {
// No need to add vectors here, already handled them in adjustExtracts.
assert(
InVectors.size() == 1 && isa<const TreeEntry *>(InVectors.front()) &&
!CommonMask.empty() &&
all_of(enumerate(CommonMask),
[&](auto P) {
Value *Scalar =
InVectors.front().get<const TreeEntry *>()->getOrdered(
P.index());
if (P.value() == PoisonMaskElem)
return P.value() == Mask[P.index()] ||
isa<UndefValue>(Scalar);
if (isa<Constant>(V1))
return true;
auto *EI = cast<ExtractElementInst>(Scalar);
return EI->getVectorOperand() == V1;
}) &&
"Expected only tree entry for extractelement vectors.");
assert(InVectors.size() == 1 && isa<const TreeEntry *>(InVectors[0]) &&
!CommonMask.empty() &&
all_of(enumerate(CommonMask),
[&](auto P) {
Value *Scalar = cast<const TreeEntry *>(InVectors[0])
->getOrdered(P.index());
if (P.value() == PoisonMaskElem)
return P.value() == Mask[P.index()] ||
isa<UndefValue>(Scalar);
if (isa<Constant>(V1))
return true;
auto *EI = cast<ExtractElementInst>(Scalar);
return EI->getVectorOperand() == V1;
}) &&
"Expected only tree entry for extractelement vectors.");
return;
}
assert(!InVectors.empty() && !CommonMask.empty() &&