diff options
| author | Vitaly Buka <vitalybuka@google.com> | 2020-08-27 02:01:26 -0700 |
|---|---|---|
| committer | Vitaly Buka <vitalybuka@google.com> | 2020-08-27 02:13:52 -0700 |
| commit | 469debe0275b5800a2231ae670d235a83e6fbb27 (patch) | |
| tree | f2ed1d64b3a0b6c34deb0261875be41c82196819 /llvm/lib/Analysis/ValueTracking.cpp | |
| parent | 154901c287c370ef7a2bbe57b034e60c25682bfe (diff) | |
| download | llvm-469debe0275b5800a2231ae670d235a83e6fbb27.zip llvm-469debe0275b5800a2231ae670d235a83e6fbb27.tar.gz llvm-469debe0275b5800a2231ae670d235a83e6fbb27.tar.bz2 | |
[ValueTracking] Support select in findAllocaForValue
Diffstat (limited to 'llvm/lib/Analysis/ValueTracking.cpp')
| -rw-r--r-- | llvm/lib/Analysis/ValueTracking.cpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/llvm/lib/Analysis/ValueTracking.cpp b/llvm/lib/Analysis/ValueTracking.cpp index 36998cd..5be566b 100644 --- a/llvm/lib/Analysis/ValueTracking.cpp +++ b/llvm/lib/Analysis/ValueTracking.cpp @@ -4345,7 +4345,14 @@ findAllocaForValue(Value *V, DenseMap<Value *, AllocaInst *> &AllocaForValue) { AllocaInst *Res = nullptr; if (CastInst *CI = dyn_cast<CastInst>(V)) Res = findAllocaForValue(CI->getOperand(0), AllocaForValue); - else if (PHINode *PN = dyn_cast<PHINode>(V)) { + else if (auto *SI = dyn_cast<SelectInst>(V)) { + Res = findAllocaForValue(SI->getTrueValue(), AllocaForValue); + if (!Res) + return nullptr; + AllocaInst *F = findAllocaForValue(SI->getFalseValue(), AllocaForValue); + if (F != Res) + return nullptr; + } else if (PHINode *PN = dyn_cast<PHINode>(V)) { for (Value *IncValue : PN->incoming_values()) { // Allow self-referencing phi-nodes. if (IncValue == PN) |