diff options
| -rw-r--r-- | llvm/lib/Transforms/Scalar/LoopIdiomRecognize.cpp | 18 |
1 files changed, 9 insertions, 9 deletions
diff --git a/llvm/lib/Transforms/Scalar/LoopIdiomRecognize.cpp b/llvm/lib/Transforms/Scalar/LoopIdiomRecognize.cpp index 880cbc3db88..6d73091592e 100644 --- a/llvm/lib/Transforms/Scalar/LoopIdiomRecognize.cpp +++ b/llvm/lib/Transforms/Scalar/LoopIdiomRecognize.cpp @@ -163,8 +163,9 @@ private: void collectStores(BasicBlock *BB); LegalStoreKind isLegalStore(StoreInst *SI); + enum class ForMemset { No, Yes }; bool processLoopStores(SmallVectorImpl<StoreInst *> &SL, const SCEV *BECount, - bool ForMemset); + ForMemset For); bool processLoopMemSet(MemSetInst *MSI, const SCEV *BECount); bool processLoopStridedStore(Value *DestPtr, unsigned StoreSize, @@ -543,10 +544,10 @@ bool LoopIdiomRecognize::runOnLoopBlock( // optimized into a memset (memset_pattern). The latter most commonly happens // with structs and handunrolled loops. for (auto &SL : StoreRefsForMemset) - MadeChange |= processLoopStores(SL.second, BECount, true); + MadeChange |= processLoopStores(SL.second, BECount, ForMemset::Yes); for (auto &SL : StoreRefsForMemsetPattern) - MadeChange |= processLoopStores(SL.second, BECount, false); + MadeChange |= processLoopStores(SL.second, BECount, ForMemset::No); // Optimize the store into a memcpy, if it feeds an similarly strided load. for (auto &SI : StoreRefsForMemcpy) @@ -572,10 +573,9 @@ bool LoopIdiomRecognize::runOnLoopBlock( return MadeChange; } -/// processLoopStores - See if this store(s) can be promoted to a memset. +/// See if this store(s) can be promoted to a memset. bool LoopIdiomRecognize::processLoopStores(SmallVectorImpl<StoreInst *> &SL, - const SCEV *BECount, - bool ForMemset) { + const SCEV *BECount, ForMemset For) { // Try to find consecutive stores that can be transformed into memsets. SetVector<StoreInst *> Heads, Tails; SmallDenseMap<StoreInst *, StoreInst *> ConsecutiveChain; @@ -602,7 +602,7 @@ bool LoopIdiomRecognize::processLoopStores(SmallVectorImpl<StoreInst *> &SL, Value *FirstSplatValue = nullptr; Constant *FirstPatternValue = nullptr; - if (ForMemset) + if (For == ForMemset::Yes) FirstSplatValue = isBytewiseValue(FirstStoredVal); else FirstPatternValue = getMemSetPatternValue(FirstStoredVal, DL); @@ -635,7 +635,7 @@ bool LoopIdiomRecognize::processLoopStores(SmallVectorImpl<StoreInst *> &SL, Value *SecondSplatValue = nullptr; Constant *SecondPatternValue = nullptr; - if (ForMemset) + if (For == ForMemset::Yes) SecondSplatValue = isBytewiseValue(SecondStoredVal); else SecondPatternValue = getMemSetPatternValue(SecondStoredVal, DL); @@ -644,7 +644,7 @@ bool LoopIdiomRecognize::processLoopStores(SmallVectorImpl<StoreInst *> &SL, "Expected either splat value or pattern value."); if (isConsecutiveAccess(SL[i], SL[k], *DL, *SE, false)) { - if (ForMemset) { + if (For == ForMemset::Yes) { if (FirstSplatValue != SecondSplatValue) continue; } else { |
