Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 1 | //===-- LoopIdiomRecognize.cpp - Loop idiom recognition -------------------===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This pass implements an idiom recognizer that transforms simple loops into a |
| 11 | // non-loop form. In cases that this kicks in, it can be a significant |
| 12 | // performance win. |
| 13 | // |
| 14 | //===----------------------------------------------------------------------===// |
Chris Lattner | bdce572 | 2011-01-02 18:32:09 +0000 | [diff] [blame] | 15 | // |
| 16 | // TODO List: |
| 17 | // |
| 18 | // Future loop memory idioms to recognize: |
| 19 | // memcmp, memmove, strlen, etc. |
| 20 | // Future floating point idioms to recognize in -ffast-math mode: |
| 21 | // fpowi |
| 22 | // Future integer operation idioms to recognize: |
| 23 | // ctpop, ctlz, cttz |
| 24 | // |
| 25 | // Beware that isel's default lowering for ctpop is highly inefficient for |
| 26 | // i64 and larger types when i64 is legal and the value has few bits set. It |
| 27 | // would be good to enhance isel to emit a loop for ctpop in this case. |
| 28 | // |
| 29 | // We should enhance the memset/memcpy recognition to handle multiple stores in |
| 30 | // the loop. This would handle things like: |
| 31 | // void foo(_Complex float *P) |
| 32 | // for (i) { __real__(*P) = 0; __imag__(*P) = 0; } |
Chris Lattner | 91139cc | 2011-01-02 23:19:45 +0000 | [diff] [blame] | 33 | // |
Chris Lattner | 408b534 | 2011-02-21 02:08:54 +0000 | [diff] [blame] | 34 | // We should enhance this to handle negative strides through memory. |
| 35 | // Alternatively (and perhaps better) we could rely on an earlier pass to force |
| 36 | // forward iteration through memory, which is generally better for cache |
| 37 | // behavior. Negative strides *do* happen for memset/memcpy loops. |
| 38 | // |
Chris Lattner | d957c71 | 2011-01-03 01:10:08 +0000 | [diff] [blame] | 39 | // This could recognize common matrix multiplies and dot product idioms and |
Chris Lattner | 91139cc | 2011-01-02 23:19:45 +0000 | [diff] [blame] | 40 | // replace them with calls to BLAS (if linked in??). |
| 41 | // |
Chris Lattner | bdce572 | 2011-01-02 18:32:09 +0000 | [diff] [blame] | 42 | //===----------------------------------------------------------------------===// |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 43 | |
| 44 | #define DEBUG_TYPE "loop-idiom" |
| 45 | #include "llvm/Transforms/Scalar.h" |
Chandler Carruth | 06cb8ed | 2012-06-29 12:38:19 +0000 | [diff] [blame] | 46 | #include "llvm/IRBuilder.h" |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 47 | #include "llvm/IntrinsicInst.h" |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 48 | #include "llvm/Module.h" |
Chandler Carruth | 06cb8ed | 2012-06-29 12:38:19 +0000 | [diff] [blame] | 49 | #include "llvm/ADT/Statistic.h" |
Chris Lattner | 2e12f1a | 2010-12-27 18:39:08 +0000 | [diff] [blame] | 50 | #include "llvm/Analysis/AliasAnalysis.h" |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 51 | #include "llvm/Analysis/LoopDependenceAnalysis.h" |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 52 | #include "llvm/Analysis/LoopPass.h" |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 53 | #include "llvm/Analysis/ScalarEvolutionExpander.h" |
Chandler Carruth | 06cb8ed | 2012-06-29 12:38:19 +0000 | [diff] [blame] | 54 | #include "llvm/Analysis/ScalarEvolutionExpressions.h" |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 55 | #include "llvm/Analysis/ValueTracking.h" |
Chandler Carruth | 06cb8ed | 2012-06-29 12:38:19 +0000 | [diff] [blame] | 56 | #include "llvm/Support/Debug.h" |
| 57 | #include "llvm/Support/raw_ostream.h" |
Micah Villmow | 3574eca | 2012-10-08 16:38:25 +0000 | [diff] [blame] | 58 | #include "llvm/DataLayout.h" |
Chris Lattner | c19175c | 2011-02-18 22:22:15 +0000 | [diff] [blame] | 59 | #include "llvm/Target/TargetLibraryInfo.h" |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 60 | #include "llvm/Transforms/Utils/Local.h" |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 61 | using namespace llvm; |
| 62 | |
Chris Lattner | 4ce31fb | 2011-01-02 07:36:44 +0000 | [diff] [blame] | 63 | STATISTIC(NumMemSet, "Number of memset's formed from loop stores"); |
| 64 | STATISTIC(NumMemCpy, "Number of memcpy's formed from loop load+stores"); |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 65 | |
| 66 | namespace { |
| 67 | class LoopIdiomRecognize : public LoopPass { |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 68 | Loop *CurLoop; |
Micah Villmow | 3574eca | 2012-10-08 16:38:25 +0000 | [diff] [blame] | 69 | const DataLayout *TD; |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 70 | DominatorTree *DT; |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 71 | ScalarEvolution *SE; |
Chris Lattner | c19175c | 2011-02-18 22:22:15 +0000 | [diff] [blame] | 72 | TargetLibraryInfo *TLI; |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 73 | public: |
| 74 | static char ID; |
| 75 | explicit LoopIdiomRecognize() : LoopPass(ID) { |
| 76 | initializeLoopIdiomRecognizePass(*PassRegistry::getPassRegistry()); |
| 77 | } |
| 78 | |
| 79 | bool runOnLoop(Loop *L, LPPassManager &LPM); |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 80 | bool runOnLoopBlock(BasicBlock *BB, const SCEV *BECount, |
| 81 | SmallVectorImpl<BasicBlock*> &ExitBlocks); |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 82 | |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 83 | bool processLoopStore(StoreInst *SI, const SCEV *BECount); |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 84 | bool processLoopMemSet(MemSetInst *MSI, const SCEV *BECount); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 85 | |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 86 | bool processLoopStridedStore(Value *DestPtr, unsigned StoreSize, |
| 87 | unsigned StoreAlignment, |
| 88 | Value *SplatValue, Instruction *TheStore, |
| 89 | const SCEVAddRecExpr *Ev, |
| 90 | const SCEV *BECount); |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 91 | bool processLoopStoreOfLoopLoad(StoreInst *SI, unsigned StoreSize, |
| 92 | const SCEVAddRecExpr *StoreEv, |
| 93 | const SCEVAddRecExpr *LoadEv, |
| 94 | const SCEV *BECount); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 95 | |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 96 | /// This transformation requires natural loop information & requires that |
| 97 | /// loop preheaders be inserted into the CFG. |
| 98 | /// |
| 99 | virtual void getAnalysisUsage(AnalysisUsage &AU) const { |
| 100 | AU.addRequired<LoopInfo>(); |
| 101 | AU.addPreserved<LoopInfo>(); |
| 102 | AU.addRequiredID(LoopSimplifyID); |
| 103 | AU.addPreservedID(LoopSimplifyID); |
| 104 | AU.addRequiredID(LCSSAID); |
| 105 | AU.addPreservedID(LCSSAID); |
Chris Lattner | 2e12f1a | 2010-12-27 18:39:08 +0000 | [diff] [blame] | 106 | AU.addRequired<AliasAnalysis>(); |
| 107 | AU.addPreserved<AliasAnalysis>(); |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 108 | AU.addRequired<ScalarEvolution>(); |
| 109 | AU.addPreserved<ScalarEvolution>(); |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 110 | AU.addRequired<LoopDependenceAnalysis>(); |
| 111 | AU.addPreserved<LoopDependenceAnalysis>(); |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 112 | AU.addPreserved<DominatorTree>(); |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 113 | AU.addRequired<DominatorTree>(); |
Chris Lattner | c19175c | 2011-02-18 22:22:15 +0000 | [diff] [blame] | 114 | AU.addRequired<TargetLibraryInfo>(); |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 115 | } |
| 116 | }; |
| 117 | } |
| 118 | |
| 119 | char LoopIdiomRecognize::ID = 0; |
| 120 | INITIALIZE_PASS_BEGIN(LoopIdiomRecognize, "loop-idiom", "Recognize loop idioms", |
| 121 | false, false) |
| 122 | INITIALIZE_PASS_DEPENDENCY(LoopInfo) |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 123 | INITIALIZE_PASS_DEPENDENCY(DominatorTree) |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 124 | INITIALIZE_PASS_DEPENDENCY(LoopSimplify) |
| 125 | INITIALIZE_PASS_DEPENDENCY(LCSSA) |
| 126 | INITIALIZE_PASS_DEPENDENCY(ScalarEvolution) |
Chris Lattner | c19175c | 2011-02-18 22:22:15 +0000 | [diff] [blame] | 127 | INITIALIZE_PASS_DEPENDENCY(TargetLibraryInfo) |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 128 | INITIALIZE_PASS_DEPENDENCY(LoopDependenceAnalysis) |
Chris Lattner | 2e12f1a | 2010-12-27 18:39:08 +0000 | [diff] [blame] | 129 | INITIALIZE_AG_DEPENDENCY(AliasAnalysis) |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 130 | INITIALIZE_PASS_END(LoopIdiomRecognize, "loop-idiom", "Recognize loop idioms", |
| 131 | false, false) |
| 132 | |
| 133 | Pass *llvm::createLoopIdiomPass() { return new LoopIdiomRecognize(); } |
| 134 | |
Chris Lattner | 4f81b54 | 2011-05-22 17:39:56 +0000 | [diff] [blame] | 135 | /// deleteDeadInstruction - Delete this instruction. Before we do, go through |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 136 | /// and zero out all the operands of this instruction. If any of them become |
| 137 | /// dead, delete them and the computation tree that feeds them. |
| 138 | /// |
Benjamin Kramer | 8e0d1c0 | 2012-08-29 15:32:21 +0000 | [diff] [blame] | 139 | static void deleteDeadInstruction(Instruction *I, ScalarEvolution &SE, |
| 140 | const TargetLibraryInfo *TLI) { |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 141 | SmallVector<Instruction*, 32> NowDeadInsts; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 142 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 143 | NowDeadInsts.push_back(I); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 144 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 145 | // Before we touch this instruction, remove it from SE! |
| 146 | do { |
| 147 | Instruction *DeadInst = NowDeadInsts.pop_back_val(); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 148 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 149 | // This instruction is dead, zap it, in stages. Start by removing it from |
| 150 | // SCEV. |
| 151 | SE.forgetValue(DeadInst); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 152 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 153 | for (unsigned op = 0, e = DeadInst->getNumOperands(); op != e; ++op) { |
| 154 | Value *Op = DeadInst->getOperand(op); |
| 155 | DeadInst->setOperand(op, 0); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 156 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 157 | // If this operand just became dead, add it to the NowDeadInsts list. |
| 158 | if (!Op->use_empty()) continue; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 159 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 160 | if (Instruction *OpI = dyn_cast<Instruction>(Op)) |
Benjamin Kramer | 8e0d1c0 | 2012-08-29 15:32:21 +0000 | [diff] [blame] | 161 | if (isInstructionTriviallyDead(OpI, TLI)) |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 162 | NowDeadInsts.push_back(OpI); |
| 163 | } |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 164 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 165 | DeadInst->eraseFromParent(); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 166 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 167 | } while (!NowDeadInsts.empty()); |
| 168 | } |
| 169 | |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 170 | bool LoopIdiomRecognize::runOnLoop(Loop *L, LPPassManager &LPM) { |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 171 | CurLoop = L; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 172 | |
Benjamin Kramer | 28aff84 | 2012-09-21 17:27:23 +0000 | [diff] [blame] | 173 | // If the loop could not be converted to canonical form, it must have an |
| 174 | // indirectbr in it, just give up. |
| 175 | if (!L->getLoopPreheader()) |
| 176 | return false; |
| 177 | |
Nadav Rotem | a94d6e8 | 2012-07-24 10:51:42 +0000 | [diff] [blame] | 178 | // Disable loop idiom recognition if the function's name is a common idiom. |
Chad Rosier | 71400b6 | 2011-07-15 18:25:04 +0000 | [diff] [blame] | 179 | StringRef Name = L->getHeader()->getParent()->getName(); |
| 180 | if (Name == "memset" || Name == "memcpy") |
| 181 | return false; |
| 182 | |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 183 | // The trip count of the loop must be analyzable. |
| 184 | SE = &getAnalysis<ScalarEvolution>(); |
| 185 | if (!SE->hasLoopInvariantBackedgeTakenCount(L)) |
| 186 | return false; |
| 187 | const SCEV *BECount = SE->getBackedgeTakenCount(L); |
| 188 | if (isa<SCEVCouldNotCompute>(BECount)) return false; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 189 | |
Chris Lattner | 8e08e73 | 2011-01-02 20:24:21 +0000 | [diff] [blame] | 190 | // If this loop executes exactly one time, then it should be peeled, not |
| 191 | // optimized by this pass. |
| 192 | if (const SCEVConstant *BECst = dyn_cast<SCEVConstant>(BECount)) |
| 193 | if (BECst->getValue()->getValue() == 0) |
| 194 | return false; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 195 | |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 196 | // We require target data for now. |
Micah Villmow | 3574eca | 2012-10-08 16:38:25 +0000 | [diff] [blame] | 197 | TD = getAnalysisIfAvailable<DataLayout>(); |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 198 | if (TD == 0) return false; |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 199 | |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 200 | DT = &getAnalysis<DominatorTree>(); |
| 201 | LoopInfo &LI = getAnalysis<LoopInfo>(); |
Chris Lattner | c19175c | 2011-02-18 22:22:15 +0000 | [diff] [blame] | 202 | TLI = &getAnalysis<TargetLibraryInfo>(); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 203 | |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 204 | SmallVector<BasicBlock*, 8> ExitBlocks; |
| 205 | CurLoop->getUniqueExitBlocks(ExitBlocks); |
| 206 | |
Chris Lattner | 63f9c3c | 2011-01-02 21:14:18 +0000 | [diff] [blame] | 207 | DEBUG(dbgs() << "loop-idiom Scanning: F[" |
| 208 | << L->getHeader()->getParent()->getName() |
| 209 | << "] Loop %" << L->getHeader()->getName() << "\n"); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 210 | |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 211 | bool MadeChange = false; |
| 212 | // Scan all the blocks in the loop that are not in subloops. |
| 213 | for (Loop::block_iterator BI = L->block_begin(), E = L->block_end(); BI != E; |
| 214 | ++BI) { |
| 215 | // Ignore blocks in subloops. |
| 216 | if (LI.getLoopFor(*BI) != CurLoop) |
| 217 | continue; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 218 | |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 219 | MadeChange |= runOnLoopBlock(*BI, BECount, ExitBlocks); |
| 220 | } |
| 221 | return MadeChange; |
| 222 | } |
| 223 | |
| 224 | /// runOnLoopBlock - Process the specified block, which lives in a counted loop |
| 225 | /// with the specified backedge count. This block is known to be in the current |
| 226 | /// loop and not in any subloops. |
| 227 | bool LoopIdiomRecognize::runOnLoopBlock(BasicBlock *BB, const SCEV *BECount, |
| 228 | SmallVectorImpl<BasicBlock*> &ExitBlocks) { |
| 229 | // We can only promote stores in this block if they are unconditionally |
| 230 | // executed in the loop. For a block to be unconditionally executed, it has |
| 231 | // to dominate all the exit blocks of the loop. Verify this now. |
| 232 | for (unsigned i = 0, e = ExitBlocks.size(); i != e; ++i) |
| 233 | if (!DT->dominates(BB, ExitBlocks[i])) |
| 234 | return false; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 235 | |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 236 | bool MadeChange = false; |
| 237 | for (BasicBlock::iterator I = BB->begin(), E = BB->end(); I != E; ) { |
Chris Lattner | b7e9ef0 | 2011-01-04 07:27:30 +0000 | [diff] [blame] | 238 | Instruction *Inst = I++; |
| 239 | // Look for store instructions, which may be optimized to memset/memcpy. |
| 240 | if (StoreInst *SI = dyn_cast<StoreInst>(Inst)) { |
Chris Lattner | b7e9ef0 | 2011-01-04 07:27:30 +0000 | [diff] [blame] | 241 | WeakVH InstPtr(I); |
| 242 | if (!processLoopStore(SI, BECount)) continue; |
| 243 | MadeChange = true; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 244 | |
Chris Lattner | b7e9ef0 | 2011-01-04 07:27:30 +0000 | [diff] [blame] | 245 | // If processing the store invalidated our iterator, start over from the |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 246 | // top of the block. |
Chris Lattner | b7e9ef0 | 2011-01-04 07:27:30 +0000 | [diff] [blame] | 247 | if (InstPtr == 0) |
| 248 | I = BB->begin(); |
| 249 | continue; |
| 250 | } |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 251 | |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 252 | // Look for memset instructions, which may be optimized to a larger memset. |
| 253 | if (MemSetInst *MSI = dyn_cast<MemSetInst>(Inst)) { |
| 254 | WeakVH InstPtr(I); |
| 255 | if (!processLoopMemSet(MSI, BECount)) continue; |
| 256 | MadeChange = true; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 257 | |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 258 | // If processing the memset invalidated our iterator, start over from the |
| 259 | // top of the block. |
| 260 | if (InstPtr == 0) |
| 261 | I = BB->begin(); |
| 262 | continue; |
| 263 | } |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 264 | } |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 265 | |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 266 | return MadeChange; |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 267 | } |
| 268 | |
Chris Lattner | 62c50fd | 2011-01-02 19:01:03 +0000 | [diff] [blame] | 269 | |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 270 | /// processLoopStore - See if this store can be promoted to a memset or memcpy. |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 271 | bool LoopIdiomRecognize::processLoopStore(StoreInst *SI, const SCEV *BECount) { |
Eli Friedman | 2bc3d52 | 2011-09-12 20:23:13 +0000 | [diff] [blame] | 272 | if (!SI->isSimple()) return false; |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 273 | |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 274 | Value *StoredVal = SI->getValueOperand(); |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 275 | Value *StorePtr = SI->getPointerOperand(); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 276 | |
Chris Lattner | 95ae676 | 2010-12-28 18:53:48 +0000 | [diff] [blame] | 277 | // Reject stores that are so large that they overflow an unsigned. |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 278 | uint64_t SizeInBits = TD->getTypeSizeInBits(StoredVal->getType()); |
Chris Lattner | 95ae676 | 2010-12-28 18:53:48 +0000 | [diff] [blame] | 279 | if ((SizeInBits & 7) || (SizeInBits >> 32) != 0) |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 280 | return false; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 281 | |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 282 | // See if the pointer expression is an AddRec like {base,+,1} on the current |
| 283 | // loop, which indicates a strided store. If we have something else, it's a |
| 284 | // random store we can't handle. |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 285 | const SCEVAddRecExpr *StoreEv = |
| 286 | dyn_cast<SCEVAddRecExpr>(SE->getSCEV(StorePtr)); |
| 287 | if (StoreEv == 0 || StoreEv->getLoop() != CurLoop || !StoreEv->isAffine()) |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 288 | return false; |
| 289 | |
| 290 | // Check to see if the stride matches the size of the store. If so, then we |
| 291 | // know that every byte is touched in the loop. |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 292 | unsigned StoreSize = (unsigned)SizeInBits >> 3; |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 293 | const SCEVConstant *Stride = dyn_cast<SCEVConstant>(StoreEv->getOperand(1)); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 294 | |
Chris Lattner | 408b534 | 2011-02-21 02:08:54 +0000 | [diff] [blame] | 295 | if (Stride == 0 || StoreSize != Stride->getValue()->getValue()) { |
| 296 | // TODO: Could also handle negative stride here someday, that will require |
| 297 | // the validity check in mayLoopAccessLocation to be updated though. |
| 298 | // Enable this to print exact negative strides. |
Chris Lattner | 0e68cee | 2011-02-21 17:02:55 +0000 | [diff] [blame] | 299 | if (0 && Stride && StoreSize == -Stride->getValue()->getValue()) { |
Chris Lattner | 408b534 | 2011-02-21 02:08:54 +0000 | [diff] [blame] | 300 | dbgs() << "NEGATIVE STRIDE: " << *SI << "\n"; |
| 301 | dbgs() << "BB: " << *SI->getParent(); |
| 302 | } |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 303 | |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 304 | return false; |
Chris Lattner | 408b534 | 2011-02-21 02:08:54 +0000 | [diff] [blame] | 305 | } |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 306 | |
| 307 | // See if we can optimize just this store in isolation. |
| 308 | if (processLoopStridedStore(StorePtr, StoreSize, SI->getAlignment(), |
| 309 | StoredVal, SI, StoreEv, BECount)) |
| 310 | return true; |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 311 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 312 | // If the stored value is a strided load in the same loop with the same stride |
| 313 | // this this may be transformable into a memcpy. This kicks in for stuff like |
| 314 | // for (i) A[i] = B[i]; |
| 315 | if (LoadInst *LI = dyn_cast<LoadInst>(StoredVal)) { |
| 316 | const SCEVAddRecExpr *LoadEv = |
| 317 | dyn_cast<SCEVAddRecExpr>(SE->getSCEV(LI->getOperand(0))); |
| 318 | if (LoadEv && LoadEv->getLoop() == CurLoop && LoadEv->isAffine() && |
Eli Friedman | 2bc3d52 | 2011-09-12 20:23:13 +0000 | [diff] [blame] | 319 | StoreEv->getOperand(1) == LoadEv->getOperand(1) && LI->isSimple()) |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 320 | if (processLoopStoreOfLoopLoad(SI, StoreSize, StoreEv, LoadEv, BECount)) |
| 321 | return true; |
| 322 | } |
Chris Lattner | 4ce31fb | 2011-01-02 07:36:44 +0000 | [diff] [blame] | 323 | //errs() << "UNHANDLED strided store: " << *StoreEv << " - " << *SI << "\n"; |
Chris Lattner | 22920b5 | 2010-12-26 20:45:45 +0000 | [diff] [blame] | 324 | |
Chris Lattner | e6bb649 | 2010-12-26 19:39:38 +0000 | [diff] [blame] | 325 | return false; |
| 326 | } |
| 327 | |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 328 | /// processLoopMemSet - See if this memset can be promoted to a large memset. |
| 329 | bool LoopIdiomRecognize:: |
| 330 | processLoopMemSet(MemSetInst *MSI, const SCEV *BECount) { |
| 331 | // We can only handle non-volatile memsets with a constant size. |
| 332 | if (MSI->isVolatile() || !isa<ConstantInt>(MSI->getLength())) return false; |
| 333 | |
Chris Lattner | c19175c | 2011-02-18 22:22:15 +0000 | [diff] [blame] | 334 | // If we're not allowed to hack on memset, we fail. |
| 335 | if (!TLI->has(LibFunc::memset)) |
| 336 | return false; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 337 | |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 338 | Value *Pointer = MSI->getDest(); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 339 | |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 340 | // See if the pointer expression is an AddRec like {base,+,1} on the current |
| 341 | // loop, which indicates a strided store. If we have something else, it's a |
| 342 | // random store we can't handle. |
| 343 | const SCEVAddRecExpr *Ev = dyn_cast<SCEVAddRecExpr>(SE->getSCEV(Pointer)); |
| 344 | if (Ev == 0 || Ev->getLoop() != CurLoop || !Ev->isAffine()) |
| 345 | return false; |
| 346 | |
| 347 | // Reject memsets that are so large that they overflow an unsigned. |
| 348 | uint64_t SizeInBytes = cast<ConstantInt>(MSI->getLength())->getZExtValue(); |
| 349 | if ((SizeInBytes >> 32) != 0) |
| 350 | return false; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 351 | |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 352 | // Check to see if the stride matches the size of the memset. If so, then we |
| 353 | // know that every byte is touched in the loop. |
| 354 | const SCEVConstant *Stride = dyn_cast<SCEVConstant>(Ev->getOperand(1)); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 355 | |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 356 | // TODO: Could also handle negative stride here someday, that will require the |
| 357 | // validity check in mayLoopAccessLocation to be updated though. |
| 358 | if (Stride == 0 || MSI->getLength() != Stride->getValue()) |
| 359 | return false; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 360 | |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 361 | return processLoopStridedStore(Pointer, (unsigned)SizeInBytes, |
| 362 | MSI->getAlignment(), MSI->getValue(), |
| 363 | MSI, Ev, BECount); |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 364 | } |
| 365 | |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 366 | /// hasDependence - Uses the LoopDependenceAnalysis to determine whether 'Inst' |
| 367 | /// depends on any other value in the Loop 'L'. |
| 368 | static bool hasDependence(Instruction *Inst, Loop *L, |
| 369 | LoopDependenceAnalysis &LDA) { |
Chris Lattner | 30980b6 | 2011-01-01 19:39:01 +0000 | [diff] [blame] | 370 | for (Loop::block_iterator BI = L->block_begin(), E = L->block_end(); BI != E; |
| 371 | ++BI) |
| 372 | for (BasicBlock::iterator I = (*BI)->begin(), E = (*BI)->end(); I != E; ++I) |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 373 | if (&*I != Inst && I->mayReadOrWriteMemory() && |
| 374 | (I->mayWriteToMemory() || Inst->mayWriteToMemory()) && |
| 375 | LDA.depends(Inst, I)) |
Chris Lattner | 30980b6 | 2011-01-01 19:39:01 +0000 | [diff] [blame] | 376 | return true; |
| 377 | |
| 378 | return false; |
| 379 | } |
| 380 | |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 381 | /// getMemSetPatternValue - If a strided store of the specified value is safe to |
| 382 | /// turn into a memset_pattern16, return a ConstantArray of 16 bytes that should |
| 383 | /// be passed in. Otherwise, return null. |
| 384 | /// |
| 385 | /// Note that we don't ever attempt to use memset_pattern8 or 4, because these |
| 386 | /// just replicate their input array and then pass on to memset_pattern16. |
Micah Villmow | 3574eca | 2012-10-08 16:38:25 +0000 | [diff] [blame] | 387 | static Constant *getMemSetPatternValue(Value *V, const DataLayout &TD) { |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 388 | // If the value isn't a constant, we can't promote it to being in a constant |
| 389 | // array. We could theoretically do a store to an alloca or something, but |
| 390 | // that doesn't seem worthwhile. |
| 391 | Constant *C = dyn_cast<Constant>(V); |
| 392 | if (C == 0) return 0; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 393 | |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 394 | // Only handle simple values that are a power of two bytes in size. |
| 395 | uint64_t Size = TD.getTypeSizeInBits(V->getType()); |
| 396 | if (Size == 0 || (Size & 7) || (Size & (Size-1))) |
| 397 | return 0; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 398 | |
Chris Lattner | 80e8b50 | 2011-02-19 19:56:44 +0000 | [diff] [blame] | 399 | // Don't care enough about darwin/ppc to implement this. |
| 400 | if (TD.isBigEndian()) |
| 401 | return 0; |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 402 | |
| 403 | // Convert to size in bytes. |
| 404 | Size /= 8; |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 405 | |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 406 | // TODO: If CI is larger than 16-bytes, we can try slicing it in half to see |
Chris Lattner | 80e8b50 | 2011-02-19 19:56:44 +0000 | [diff] [blame] | 407 | // if the top and bottom are the same (e.g. for vectors and large integers). |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 408 | if (Size > 16) return 0; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 409 | |
Chris Lattner | 80e8b50 | 2011-02-19 19:56:44 +0000 | [diff] [blame] | 410 | // If the constant is exactly 16 bytes, just use it. |
| 411 | if (Size == 16) return C; |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 412 | |
Chris Lattner | 80e8b50 | 2011-02-19 19:56:44 +0000 | [diff] [blame] | 413 | // Otherwise, we'll use an array of the constants. |
| 414 | unsigned ArraySize = 16/Size; |
| 415 | ArrayType *AT = ArrayType::get(V->getType(), ArraySize); |
| 416 | return ConstantArray::get(AT, std::vector<Constant*>(ArraySize, C)); |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 417 | } |
| 418 | |
| 419 | |
| 420 | /// processLoopStridedStore - We see a strided store of some value. If we can |
| 421 | /// transform this into a memset or memset_pattern in the loop preheader, do so. |
| 422 | bool LoopIdiomRecognize:: |
| 423 | processLoopStridedStore(Value *DestPtr, unsigned StoreSize, |
| 424 | unsigned StoreAlignment, Value *StoredVal, |
| 425 | Instruction *TheStore, const SCEVAddRecExpr *Ev, |
| 426 | const SCEV *BECount) { |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 427 | |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 428 | // If the stored value is a byte-wise value (like i32 -1), then it may be |
| 429 | // turned into a memset of i8 -1, assuming that all the consecutive bytes |
| 430 | // are stored. A store of i32 0x01020304 can never be turned into a memset, |
| 431 | // but it can be turned into memset_pattern if the target supports it. |
| 432 | Value *SplatValue = isBytewiseValue(StoredVal); |
| 433 | Constant *PatternValue = 0; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 434 | |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 435 | // If we're allowed to form a memset, and the stored value would be acceptable |
| 436 | // for memset, use it. |
| 437 | if (SplatValue && TLI->has(LibFunc::memset) && |
| 438 | // Verify that the stored value is loop invariant. If not, we can't |
| 439 | // promote the memset. |
| 440 | CurLoop->isLoopInvariant(SplatValue)) { |
| 441 | // Keep and use SplatValue. |
| 442 | PatternValue = 0; |
| 443 | } else if (TLI->has(LibFunc::memset_pattern16) && |
| 444 | (PatternValue = getMemSetPatternValue(StoredVal, *TD))) { |
| 445 | // It looks like we can use PatternValue! |
| 446 | SplatValue = 0; |
| 447 | } else { |
| 448 | // Otherwise, this isn't an idiom we can transform. For example, we can't |
Eli Friedman | 5ac7c7d | 2011-09-13 00:44:16 +0000 | [diff] [blame] | 449 | // do anything with a 3-byte store. |
Chris Lattner | bafa117 | 2011-01-01 20:12:04 +0000 | [diff] [blame] | 450 | return false; |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 451 | } |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 452 | |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 453 | // Make sure the store has no dependencies (i.e. other loads and stores) in |
| 454 | // the loop. |
| 455 | if (hasDependence(TheStore, CurLoop, getAnalysis<LoopDependenceAnalysis>())) |
| 456 | return false; |
| 457 | |
Chris Lattner | 4f81b54 | 2011-05-22 17:39:56 +0000 | [diff] [blame] | 458 | // The trip count of the loop and the base pointer of the addrec SCEV is |
| 459 | // guaranteed to be loop invariant, which means that it should dominate the |
| 460 | // header. This allows us to insert code for it in the preheader. |
| 461 | BasicBlock *Preheader = CurLoop->getLoopPreheader(); |
| 462 | IRBuilder<> Builder(Preheader->getTerminator()); |
Andrew Trick | 5e7645b | 2011-06-28 05:07:32 +0000 | [diff] [blame] | 463 | SCEVExpander Expander(*SE, "loop-idiom"); |
Andrew Trick | a5d950f | 2011-06-28 05:04:16 +0000 | [diff] [blame] | 464 | |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 465 | // Okay, we have a strided store "p[i]" of a splattable value. We can turn |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 466 | // this into a memset in the loop preheader now if we want. |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 467 | unsigned AddrSpace = cast<PointerType>(DestPtr->getType())->getAddressSpace(); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 468 | Value *BasePtr = |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 469 | Expander.expandCodeFor(Ev->getStart(), Builder.getInt8PtrTy(AddrSpace), |
| 470 | Preheader->getTerminator()); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 471 | |
Chris Lattner | 4f81b54 | 2011-05-22 17:39:56 +0000 | [diff] [blame] | 472 | |
Chris Lattner | 4f81b54 | 2011-05-22 17:39:56 +0000 | [diff] [blame] | 473 | // Okay, everything looks good, insert the memset. |
| 474 | |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 475 | // The # stored bytes is (BECount+1)*Size. Expand the trip count out to |
| 476 | // pointer size if it isn't already. |
Chris Lattner | db125cf | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 477 | Type *IntPtr = TD->getIntPtrType(DestPtr->getContext()); |
Chris Lattner | 7c90b90 | 2011-01-04 00:06:55 +0000 | [diff] [blame] | 478 | BECount = SE->getTruncateOrZeroExtend(BECount, IntPtr); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 479 | |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 480 | const SCEV *NumBytesS = SE->getAddExpr(BECount, SE->getConstant(IntPtr, 1), |
Andrew Trick | 3228cc2 | 2011-03-14 16:50:06 +0000 | [diff] [blame] | 481 | SCEV::FlagNUW); |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 482 | if (StoreSize != 1) |
| 483 | NumBytesS = SE->getMulExpr(NumBytesS, SE->getConstant(IntPtr, StoreSize), |
Andrew Trick | 3228cc2 | 2011-03-14 16:50:06 +0000 | [diff] [blame] | 484 | SCEV::FlagNUW); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 485 | |
| 486 | Value *NumBytes = |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 487 | Expander.expandCodeFor(NumBytesS, IntPtr, Preheader->getTerminator()); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 488 | |
Devang Patel | cd77a50 | 2011-03-07 22:43:45 +0000 | [diff] [blame] | 489 | CallInst *NewCall; |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 490 | if (SplatValue) |
| 491 | NewCall = Builder.CreateMemSet(BasePtr, SplatValue,NumBytes,StoreAlignment); |
| 492 | else { |
| 493 | Module *M = TheStore->getParent()->getParent()->getParent(); |
| 494 | Value *MSP = M->getOrInsertFunction("memset_pattern16", |
| 495 | Builder.getVoidTy(), |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 496 | Builder.getInt8PtrTy(), |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 497 | Builder.getInt8PtrTy(), IntPtr, |
| 498 | (void*)0); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 499 | |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 500 | // Otherwise we should form a memset_pattern16. PatternValue is known to be |
| 501 | // an constant array of 16-bytes. Plop the value into a mergable global. |
| 502 | GlobalVariable *GV = new GlobalVariable(*M, PatternValue->getType(), true, |
| 503 | GlobalValue::InternalLinkage, |
| 504 | PatternValue, ".memset_pattern"); |
| 505 | GV->setUnnamedAddr(true); // Ok to merge these. |
| 506 | GV->setAlignment(16); |
Chris Lattner | 80e8b50 | 2011-02-19 19:56:44 +0000 | [diff] [blame] | 507 | Value *PatternPtr = ConstantExpr::getBitCast(GV, Builder.getInt8PtrTy()); |
Chris Lattner | 3a39372 | 2011-02-19 19:31:39 +0000 | [diff] [blame] | 508 | NewCall = Builder.CreateCall3(MSP, BasePtr, PatternPtr, NumBytes); |
| 509 | } |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 510 | |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 511 | DEBUG(dbgs() << " Formed memset: " << *NewCall << "\n" |
Chris Lattner | e41d3c0 | 2011-01-04 07:46:33 +0000 | [diff] [blame] | 512 | << " from store to: " << *Ev << " at: " << *TheStore << "\n"); |
Devang Patel | cd77a50 | 2011-03-07 22:43:45 +0000 | [diff] [blame] | 513 | NewCall->setDebugLoc(TheStore->getDebugLoc()); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 514 | |
Chris Lattner | 9f39188 | 2010-12-27 00:03:23 +0000 | [diff] [blame] | 515 | // Okay, the memset has been formed. Zap the original store and anything that |
| 516 | // feeds into it. |
Benjamin Kramer | 8e0d1c0 | 2012-08-29 15:32:21 +0000 | [diff] [blame] | 517 | deleteDeadInstruction(TheStore, *SE, TLI); |
Chris Lattner | 4ce31fb | 2011-01-02 07:36:44 +0000 | [diff] [blame] | 518 | ++NumMemSet; |
Chris Lattner | a92ff91 | 2010-12-26 23:42:51 +0000 | [diff] [blame] | 519 | return true; |
| 520 | } |
| 521 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 522 | /// processLoopStoreOfLoopLoad - We see a strided store whose value is a |
| 523 | /// same-strided load. |
| 524 | bool LoopIdiomRecognize:: |
| 525 | processLoopStoreOfLoopLoad(StoreInst *SI, unsigned StoreSize, |
| 526 | const SCEVAddRecExpr *StoreEv, |
| 527 | const SCEVAddRecExpr *LoadEv, |
| 528 | const SCEV *BECount) { |
Chris Lattner | c19175c | 2011-02-18 22:22:15 +0000 | [diff] [blame] | 529 | // If we're not allowed to form memcpy, we fail. |
| 530 | if (!TLI->has(LibFunc::memcpy)) |
| 531 | return false; |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 532 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 533 | LoadInst *LI = cast<LoadInst>(SI->getValueOperand()); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 534 | |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 535 | // Make sure the load and the store have no dependencies (i.e. other loads and |
| 536 | // stores) in the loop. |
| 537 | // FIXME: If we want to form a memmove SI and LI can be dependent but the |
| 538 | // distance must be positive. LDA doesn't provide that info currently. |
| 539 | LoopDependenceAnalysis &LDA = getAnalysis<LoopDependenceAnalysis>(); |
| 540 | if (hasDependence(SI, CurLoop, LDA) || hasDependence(LI, CurLoop, LDA)) |
| 541 | return false; |
| 542 | |
Chris Lattner | 4f81b54 | 2011-05-22 17:39:56 +0000 | [diff] [blame] | 543 | // The trip count of the loop and the base pointer of the addrec SCEV is |
| 544 | // guaranteed to be loop invariant, which means that it should dominate the |
| 545 | // header. This allows us to insert code for it in the preheader. |
| 546 | BasicBlock *Preheader = CurLoop->getLoopPreheader(); |
| 547 | IRBuilder<> Builder(Preheader->getTerminator()); |
Andrew Trick | 5e7645b | 2011-06-28 05:07:32 +0000 | [diff] [blame] | 548 | SCEVExpander Expander(*SE, "loop-idiom"); |
Andrew Trick | a5d950f | 2011-06-28 05:04:16 +0000 | [diff] [blame] | 549 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 550 | // Okay, we have a strided store "p[i]" of a loaded value. We can turn |
Benjamin Kramer | 5c6e9ae | 2012-10-21 15:03:07 +0000 | [diff] [blame] | 551 | // this into a memcpy in the loop preheader now if we want. |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 552 | Value *StoreBasePtr = |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 553 | Expander.expandCodeFor(StoreEv->getStart(), |
| 554 | Builder.getInt8PtrTy(SI->getPointerAddressSpace()), |
| 555 | Preheader->getTerminator()); |
Chris Lattner | 4f81b54 | 2011-05-22 17:39:56 +0000 | [diff] [blame] | 556 | Value *LoadBasePtr = |
| 557 | Expander.expandCodeFor(LoadEv->getStart(), |
| 558 | Builder.getInt8PtrTy(LI->getPointerAddressSpace()), |
| 559 | Preheader->getTerminator()); |
| 560 | |
Chris Lattner | 4f81b54 | 2011-05-22 17:39:56 +0000 | [diff] [blame] | 561 | // Okay, everything is safe, we can transform this! |
Andrew Trick | a5d950f | 2011-06-28 05:04:16 +0000 | [diff] [blame] | 562 | |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 563 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 564 | // The # stored bytes is (BECount+1)*Size. Expand the trip count out to |
| 565 | // pointer size if it isn't already. |
Chris Lattner | db125cf | 2011-07-18 04:54:35 +0000 | [diff] [blame] | 566 | Type *IntPtr = TD->getIntPtrType(SI->getContext()); |
Chris Lattner | 7c90b90 | 2011-01-04 00:06:55 +0000 | [diff] [blame] | 567 | BECount = SE->getTruncateOrZeroExtend(BECount, IntPtr); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 568 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 569 | const SCEV *NumBytesS = SE->getAddExpr(BECount, SE->getConstant(IntPtr, 1), |
Andrew Trick | 3228cc2 | 2011-03-14 16:50:06 +0000 | [diff] [blame] | 570 | SCEV::FlagNUW); |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 571 | if (StoreSize != 1) |
| 572 | NumBytesS = SE->getMulExpr(NumBytesS, SE->getConstant(IntPtr, StoreSize), |
Andrew Trick | 3228cc2 | 2011-03-14 16:50:06 +0000 | [diff] [blame] | 573 | SCEV::FlagNUW); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 574 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 575 | Value *NumBytes = |
| 576 | Expander.expandCodeFor(NumBytesS, IntPtr, Preheader->getTerminator()); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 577 | |
Devang Patel | af35841 | 2011-05-04 21:37:05 +0000 | [diff] [blame] | 578 | CallInst *NewCall = |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 579 | Builder.CreateMemCpy(StoreBasePtr, LoadBasePtr, NumBytes, |
| 580 | std::min(SI->getAlignment(), LI->getAlignment())); |
Devang Patel | af35841 | 2011-05-04 21:37:05 +0000 | [diff] [blame] | 581 | NewCall->setDebugLoc(SI->getDebugLoc()); |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 582 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 583 | DEBUG(dbgs() << " Formed memcpy: " << *NewCall << "\n" |
| 584 | << " from load ptr=" << *LoadEv << " at: " << *LI << "\n" |
| 585 | << " from store ptr=" << *StoreEv << " at: " << *SI << "\n"); |
Andrew Trick | a5d950f | 2011-06-28 05:04:16 +0000 | [diff] [blame] | 586 | |
Andrew Trick | d99b39e | 2011-03-14 16:48:10 +0000 | [diff] [blame] | 587 | |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 588 | // Okay, the memset has been formed. Zap the original store and anything that |
| 589 | // feeds into it. |
Benjamin Kramer | 8e0d1c0 | 2012-08-29 15:32:21 +0000 | [diff] [blame] | 590 | deleteDeadInstruction(SI, *SE, TLI); |
Chris Lattner | 4ce31fb | 2011-01-02 07:36:44 +0000 | [diff] [blame] | 591 | ++NumMemCpy; |
Chris Lattner | e2c4392 | 2011-01-02 03:37:56 +0000 | [diff] [blame] | 592 | return true; |
| 593 | } |