wangtao
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[InstCombine] reduce indent; NFC 

llvm-svn: 272193
This commit is contained in:
Sanjay Patel 2016-06-08 20:09:04 +00:00
parent 7aa0ad65ce
commit 10a2c38d83
1 changed files with 65 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

129
llvm/lib/Transforms/InstCombine/InstCombineSelect.cpp
View File

@ -984,75 +984,76 @@ Instruction *InstCombiner::visitSelectInst(SelectInst &SI) {
if (Instruction *Result = visitSelectInstWithICmp(SI, ICI)) if (Instruction *Result = visitSelectInstWithICmp(SI, ICI))
return Result; return Result;
if (Instruction *TI = dyn_cast<Instruction>(TrueVal)) auto *TI = dyn_cast<Instruction>(TrueVal);
if (Instruction *FI = dyn_cast<Instruction>(FalseVal)) auto *FI = dyn_cast<Instruction>(FalseVal);
if (TI->hasOneUse() && FI->hasOneUse()) { if (TI && FI && TI->hasOneUse() && FI->hasOneUse()) {
Instruction *AddOp = nullptr, *SubOp = nullptr; Instruction *AddOp = nullptr, *SubOp = nullptr;
// Turn (select C, (op X, Y), (op X, Z)) -> (op X, (select C, Y, Z)) // Turn (select C, (op X, Y), (op X, Z)) -> (op X, (select C, Y, Z))
if (TI->getOpcode() == FI->getOpcode()) if (TI->getOpcode() == FI->getOpcode())
if (Instruction *IV = FoldSelectOpOp(SI, TI, FI)) if (Instruction *IV = FoldSelectOpOp(SI, TI, FI))
return IV; return IV;
// Turn select C, (X+Y), (X-Y) --> (X+(select C, Y, (-Y))). This is // Turn select C, (X+Y), (X-Y) --> (X+(select C, Y, (-Y))). This is
// even legal for FP. // even legal for FP.
if ((TI->getOpcode() == Instruction::Sub && if ((TI->getOpcode() == Instruction::Sub &&
FI->getOpcode() == Instruction::Add) || FI->getOpcode() == Instruction::Add) ||
(TI->getOpcode() == Instruction::FSub && (TI->getOpcode() == Instruction::FSub &&
FI->getOpcode() == Instruction::FAdd)) { FI->getOpcode() == Instruction::FAdd)) {
AddOp = FI; SubOp = TI; AddOp = FI;
} else if ((FI->getOpcode() == Instruction::Sub && SubOp = TI;
TI->getOpcode() == Instruction::Add) || } else if ((FI->getOpcode() == Instruction::Sub &&
(FI->getOpcode() == Instruction::FSub && TI->getOpcode() == Instruction::Add) ||
TI->getOpcode() == Instruction::FAdd)) { (FI->getOpcode() == Instruction::FSub &&
AddOp = TI; SubOp = FI; TI->getOpcode() == Instruction::FAdd)) {
} AddOp = TI;
SubOp = FI;
}
if (AddOp) { if (AddOp) {
Value *OtherAddOp = nullptr; Value *OtherAddOp = nullptr;
if (SubOp->getOperand(0) == AddOp->getOperand(0)) { if (SubOp->getOperand(0) == AddOp->getOperand(0)) {
OtherAddOp = AddOp->getOperand(1); OtherAddOp = AddOp->getOperand(1);
} else if (SubOp->getOperand(0) == AddOp->getOperand(1)) { } else if (SubOp->getOperand(0) == AddOp->getOperand(1)) {
OtherAddOp = AddOp->getOperand(0); OtherAddOp = AddOp->getOperand(0);
}
if (OtherAddOp) {
// So at this point we know we have (Y -> OtherAddOp):
// select C, (add X, Y), (sub X, Z)
Value *NegVal; // Compute -Z
if (SI.getType()->isFPOrFPVectorTy()) {
NegVal = Builder->CreateFNeg(SubOp->getOperand(1));
if (Instruction *NegInst = dyn_cast<Instruction>(NegVal)) {
FastMathFlags Flags = AddOp->getFastMathFlags();
Flags &= SubOp->getFastMathFlags();
NegInst->setFastMathFlags(Flags);
}
} else {
NegVal = Builder->CreateNeg(SubOp->getOperand(1));
}
Value *NewTrueOp = OtherAddOp;
Value *NewFalseOp = NegVal;
if (AddOp != TI)
std::swap(NewTrueOp, NewFalseOp);
Value *NewSel =
Builder->CreateSelect(CondVal, NewTrueOp,
NewFalseOp, SI.getName() + ".p");
if (SI.getType()->isFPOrFPVectorTy()) {
Instruction *RI =
BinaryOperator::CreateFAdd(SubOp->getOperand(0), NewSel);
FastMathFlags Flags = AddOp->getFastMathFlags();
Flags &= SubOp->getFastMathFlags();
RI->setFastMathFlags(Flags);
return RI;
} else
return BinaryOperator::CreateAdd(SubOp->getOperand(0), NewSel);
}
}
} }
if (OtherAddOp) {
// So at this point we know we have (Y -> OtherAddOp):
// select C, (add X, Y), (sub X, Z)
Value *NegVal; // Compute -Z
if (SI.getType()->isFPOrFPVectorTy()) {
NegVal = Builder->CreateFNeg(SubOp->getOperand(1));
if (Instruction *NegInst = dyn_cast<Instruction>(NegVal)) {
FastMathFlags Flags = AddOp->getFastMathFlags();
Flags &= SubOp->getFastMathFlags();
NegInst->setFastMathFlags(Flags);
}
} else {
NegVal = Builder->CreateNeg(SubOp->getOperand(1));
}
Value *NewTrueOp = OtherAddOp;
Value *NewFalseOp = NegVal;
if (AddOp != TI)
std::swap(NewTrueOp, NewFalseOp);
Value *NewSel = Builder->CreateSelect(CondVal, NewTrueOp, NewFalseOp,
SI.getName() + ".p");
if (SI.getType()->isFPOrFPVectorTy()) {
Instruction *RI =
BinaryOperator::CreateFAdd(SubOp->getOperand(0), NewSel);
FastMathFlags Flags = AddOp->getFastMathFlags();
Flags &= SubOp->getFastMathFlags();
RI->setFastMathFlags(Flags);
return RI;
} else
return BinaryOperator::CreateAdd(SubOp->getOperand(0), NewSel);
}
}
}
// See if we can fold the select into one of our operands. // See if we can fold the select into one of our operands.
if (SI.getType()->isIntOrIntVectorTy() || SI.getType()->isFPOrFPVectorTy()) { if (SI.getType()->isIntOrIntVectorTy() || SI.getType()->isFPOrFPVectorTy()) {
if (Instruction *FoldI = FoldSelectIntoOp(SI, TrueVal, FalseVal)) if (Instruction *FoldI = FoldSelectIntoOp(SI, TrueVal, FalseVal))