RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-11-25 20:23:11 +01:00
Code Issues Projects Releases Wiki Activity

[ConstantFolding] refactor helper for vector reductions; NFC

Browse Source 
We should handle other cases (undef/poison), so reduce
the duplication of repeated switches.
This commit is contained in:
Sanjay Patel 2021-04-29 12:07:51 -04:00
parent 7ddc316273
commit b8ab3f33f1
1 changed files with 32 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
lib/Analysis/ConstantFolding.cpp
View File

@ -1702,19 +1702,29 @@ Constant *ConstantFoldBinaryFP(double (*NativeFP)(double, double), double V,
return GetConstantFoldFPValue(V, Ty);
}
Constant *ConstantFoldVectorReduce(Intrinsic::ID IID, Constant *Op) {
Constant *constantFoldVectorReduce(Intrinsic::ID IID, Constant *Op) {
FixedVectorType *VT = dyn_cast<FixedVectorType>(Op->getType());
if (!VT)
return nullptr;
ConstantInt *CI = dyn_cast<ConstantInt>(Op->getAggregateElement(0U));
if (!CI)
return nullptr;
APInt Acc = CI->getValue();
for (unsigned I = 1; I < VT->getNumElements(); I++) {
if (!(CI = dyn_cast<ConstantInt>(Op->getAggregateElement(I))))
// This isn't strictly necessary, but handle the special/common case of zero:
// all integer reductions of a zero input produce zero.
if (isa<ConstantAggregateZero>(Op))
return ConstantInt::get(VT->getElementType(), 0);
// TODO: Handle undef and poison.
if (!isa<ConstantVector>(Op) && !isa<ConstantDataVector>(Op))
return nullptr;
const APInt &X = CI->getValue();
auto *EltC = dyn_cast<ConstantInt>(Op->getAggregateElement(0U));
if (!EltC)
return nullptr;
APInt Acc = EltC->getValue();
for (unsigned I = 1, E = VT->getNumElements(); I != E; I++) {
if (!(EltC = dyn_cast<ConstantInt>(Op->getAggregateElement(I))))
return nullptr;
const APInt &X = EltC->getValue();
switch (IID) {
case Intrinsic::vector_reduce_add:
Acc = Acc + X;
@ -2241,7 +2251,6 @@ static Constant *ConstantFoldScalarCall1(StringRef Name,
}
}
if (isa<ConstantAggregateZero>(Operands[0])) {
switch (IntrinsicID) {
default: break;
case Intrinsic::vector_reduce_add:
@ -2253,8 +2262,9 @@ static Constant *ConstantFoldScalarCall1(StringRef Name,
case Intrinsic::vector_reduce_smax:
case Intrinsic::vector_reduce_umin:
case Intrinsic::vector_reduce_umax:
return ConstantInt::get(Ty, 0);
}
if (Constant *C = constantFoldVectorReduce(IntrinsicID, Operands[0]))
return C;
break;
}
// Support ConstantVector in case we have an Undef in the top.
@ -2263,18 +2273,6 @@ static Constant *ConstantFoldScalarCall1(StringRef Name,
auto *Op = cast<Constant>(Operands[0]);
switch (IntrinsicID) {
default: break;
case Intrinsic::vector_reduce_add:
case Intrinsic::vector_reduce_mul:
case Intrinsic::vector_reduce_and:
case Intrinsic::vector_reduce_or:
case Intrinsic::vector_reduce_xor:
case Intrinsic::vector_reduce_smin:
case Intrinsic::vector_reduce_smax:
case Intrinsic::vector_reduce_umin:
case Intrinsic::vector_reduce_umax:
if (Constant *C = ConstantFoldVectorReduce(IntrinsicID, Op))
return C;
break;
case Intrinsic::x86_sse_cvtss2si:
case Intrinsic::x86_sse_cvtss2si64:
case Intrinsic::x86_sse2_cvtsd2si: