1239310Sdim//===-- PPCCTRLoops.cpp - Identify and generate CTR loops -----------------===// 2239310Sdim// 3239310Sdim// The LLVM Compiler Infrastructure 4239310Sdim// 5239310Sdim// This file is distributed under the University of Illinois Open Source 6239310Sdim// License. See LICENSE.TXT for details. 7239310Sdim// 8239310Sdim//===----------------------------------------------------------------------===// 9239310Sdim// 10239310Sdim// This pass identifies loops where we can generate the PPC branch instructions 11239310Sdim// that decrement and test the count register (CTR) (bdnz and friends). 12239310Sdim// 13239310Sdim// The pattern that defines the induction variable can changed depending on 14239310Sdim// prior optimizations. For example, the IndVarSimplify phase run by 'opt' 15239310Sdim// normalizes induction variables, and the Loop Strength Reduction pass 16239310Sdim// run by 'llc' may also make changes to the induction variable. 17239310Sdim// 18239310Sdim// Criteria for CTR loops: 19239310Sdim// - Countable loops (w/ ind. var for a trip count) 20239310Sdim// - Try inner-most loops first 21239310Sdim// - No nested CTR loops. 22239310Sdim// - No function calls in loops. 23239310Sdim// 24239310Sdim//===----------------------------------------------------------------------===// 25239310Sdim 26239310Sdim#define DEBUG_TYPE "ctrloops" 27263509Sdim 28263509Sdim#include "llvm/Transforms/Scalar.h" 29263509Sdim#include "llvm/ADT/Statistic.h" 30263509Sdim#include "llvm/ADT/STLExtras.h" 31263509Sdim#include "llvm/Analysis/Dominators.h" 32263509Sdim#include "llvm/Analysis/LoopInfo.h" 33263509Sdim#include "llvm/Analysis/ScalarEvolutionExpander.h" 34263509Sdim#include "llvm/IR/Constants.h" 35263509Sdim#include "llvm/IR/DerivedTypes.h" 36263509Sdim#include "llvm/IR/InlineAsm.h" 37263509Sdim#include "llvm/IR/Instructions.h" 38263509Sdim#include "llvm/IR/IntrinsicInst.h" 39263509Sdim#include "llvm/IR/Module.h" 40263509Sdim#include "llvm/PassSupport.h" 41263509Sdim#include "llvm/Support/CommandLine.h" 42263509Sdim#include "llvm/Support/Debug.h" 43263509Sdim#include "llvm/Support/ValueHandle.h" 44263509Sdim#include "llvm/Support/raw_ostream.h" 45263509Sdim#include "llvm/Transforms/Utils/BasicBlockUtils.h" 46263509Sdim#include "llvm/Transforms/Utils/Local.h" 47263509Sdim#include "llvm/Transforms/Utils/LoopUtils.h" 48263509Sdim#include "llvm/Target/TargetLibraryInfo.h" 49263509Sdim#include "PPCTargetMachine.h" 50239310Sdim#include "PPC.h" 51263509Sdim 52263509Sdim#ifndef NDEBUG 53239310Sdim#include "llvm/CodeGen/MachineDominators.h" 54239310Sdim#include "llvm/CodeGen/MachineFunction.h" 55239310Sdim#include "llvm/CodeGen/MachineFunctionPass.h" 56239310Sdim#include "llvm/CodeGen/MachineRegisterInfo.h" 57263509Sdim#endif 58263509Sdim 59239310Sdim#include <algorithm> 60263509Sdim#include <vector> 61239310Sdim 62239310Sdimusing namespace llvm; 63239310Sdim 64263509Sdim#ifndef NDEBUG 65263509Sdimstatic cl::opt<int> CTRLoopLimit("ppc-max-ctrloop", cl::Hidden, cl::init(-1)); 66263509Sdim#endif 67263509Sdim 68239310SdimSTATISTIC(NumCTRLoops, "Number of loops converted to CTR loops"); 69239310Sdim 70252723Sdimnamespace llvm { 71252723Sdim void initializePPCCTRLoopsPass(PassRegistry&); 72263509Sdim#ifndef NDEBUG 73263509Sdim void initializePPCCTRLoopsVerifyPass(PassRegistry&); 74263509Sdim#endif 75252723Sdim} 76252723Sdim 77239310Sdimnamespace { 78263509Sdim struct PPCCTRLoops : public FunctionPass { 79239310Sdim 80263509Sdim#ifndef NDEBUG 81263509Sdim static int Counter; 82263509Sdim#endif 83263509Sdim 84239310Sdim public: 85263509Sdim static char ID; 86239310Sdim 87263509Sdim PPCCTRLoops() : FunctionPass(ID), TM(0) { 88252723Sdim initializePPCCTRLoopsPass(*PassRegistry::getPassRegistry()); 89252723Sdim } 90263509Sdim PPCCTRLoops(PPCTargetMachine &TM) : FunctionPass(ID), TM(&TM) { 91263509Sdim initializePPCCTRLoopsPass(*PassRegistry::getPassRegistry()); 92263509Sdim } 93239310Sdim 94263509Sdim virtual bool runOnFunction(Function &F); 95239310Sdim 96239310Sdim virtual void getAnalysisUsage(AnalysisUsage &AU) const { 97263509Sdim AU.addRequired<LoopInfo>(); 98263509Sdim AU.addPreserved<LoopInfo>(); 99263509Sdim AU.addRequired<DominatorTree>(); 100263509Sdim AU.addPreserved<DominatorTree>(); 101263509Sdim AU.addRequired<ScalarEvolution>(); 102239310Sdim } 103239310Sdim 104239310Sdim private: 105263509Sdim bool mightUseCTR(const Triple &TT, BasicBlock *BB); 106263509Sdim bool convertToCTRLoop(Loop *L); 107239310Sdim 108263509Sdim private: 109263509Sdim PPCTargetMachine *TM; 110263509Sdim LoopInfo *LI; 111263509Sdim ScalarEvolution *SE; 112263509Sdim DataLayout *TD; 113263509Sdim DominatorTree *DT; 114263509Sdim const TargetLibraryInfo *LibInfo; 115239310Sdim }; 116239310Sdim 117239310Sdim char PPCCTRLoops::ID = 0; 118263509Sdim#ifndef NDEBUG 119263509Sdim int PPCCTRLoops::Counter = 0; 120263509Sdim#endif 121239310Sdim 122263509Sdim#ifndef NDEBUG 123263509Sdim struct PPCCTRLoopsVerify : public MachineFunctionPass { 124239310Sdim public: 125263509Sdim static char ID; 126239310Sdim 127263509Sdim PPCCTRLoopsVerify() : MachineFunctionPass(ID) { 128263509Sdim initializePPCCTRLoopsVerifyPass(*PassRegistry::getPassRegistry()); 129263509Sdim } 130239310Sdim 131263509Sdim virtual void getAnalysisUsage(AnalysisUsage &AU) const { 132263509Sdim AU.addRequired<MachineDominatorTree>(); 133263509Sdim MachineFunctionPass::getAnalysisUsage(AU); 134239310Sdim } 135239310Sdim 136263509Sdim virtual bool runOnMachineFunction(MachineFunction &MF); 137263509Sdim 138263509Sdim private: 139263509Sdim MachineDominatorTree *MDT; 140239310Sdim }; 141263509Sdim 142263509Sdim char PPCCTRLoopsVerify::ID = 0; 143263509Sdim#endif // NDEBUG 144239310Sdim} // end anonymous namespace 145239310Sdim 146252723SdimINITIALIZE_PASS_BEGIN(PPCCTRLoops, "ppc-ctr-loops", "PowerPC CTR Loops", 147252723Sdim false, false) 148263509SdimINITIALIZE_PASS_DEPENDENCY(DominatorTree) 149263509SdimINITIALIZE_PASS_DEPENDENCY(LoopInfo) 150263509SdimINITIALIZE_PASS_DEPENDENCY(ScalarEvolution) 151252723SdimINITIALIZE_PASS_END(PPCCTRLoops, "ppc-ctr-loops", "PowerPC CTR Loops", 152252723Sdim false, false) 153239310Sdim 154263509SdimFunctionPass *llvm::createPPCCTRLoops(PPCTargetMachine &TM) { 155263509Sdim return new PPCCTRLoops(TM); 156239310Sdim} 157239310Sdim 158263509Sdim#ifndef NDEBUG 159263509SdimINITIALIZE_PASS_BEGIN(PPCCTRLoopsVerify, "ppc-ctr-loops-verify", 160263509Sdim "PowerPC CTR Loops Verify", false, false) 161263509SdimINITIALIZE_PASS_DEPENDENCY(MachineDominatorTree) 162263509SdimINITIALIZE_PASS_END(PPCCTRLoopsVerify, "ppc-ctr-loops-verify", 163263509Sdim "PowerPC CTR Loops Verify", false, false) 164239310Sdim 165263509SdimFunctionPass *llvm::createPPCCTRLoopsVerify() { 166263509Sdim return new PPCCTRLoopsVerify(); 167239310Sdim} 168263509Sdim#endif // NDEBUG 169239310Sdim 170263509Sdimbool PPCCTRLoops::runOnFunction(Function &F) { 171263509Sdim LI = &getAnalysis<LoopInfo>(); 172263509Sdim SE = &getAnalysis<ScalarEvolution>(); 173263509Sdim DT = &getAnalysis<DominatorTree>(); 174263509Sdim TD = getAnalysisIfAvailable<DataLayout>(); 175263509Sdim LibInfo = getAnalysisIfAvailable<TargetLibraryInfo>(); 176239310Sdim 177263509Sdim bool MadeChange = false; 178239310Sdim 179263509Sdim for (LoopInfo::iterator I = LI->begin(), E = LI->end(); 180239310Sdim I != E; ++I) { 181263509Sdim Loop *L = *I; 182263509Sdim if (!L->getParentLoop()) 183263509Sdim MadeChange |= convertToCTRLoop(L); 184239310Sdim } 185239310Sdim 186263509Sdim return MadeChange; 187239310Sdim} 188239310Sdim 189266759Sdimstatic bool isLargeIntegerTy(bool Is32Bit, Type *Ty) { 190266759Sdim if (IntegerType *ITy = dyn_cast<IntegerType>(Ty)) 191266759Sdim return ITy->getBitWidth() > (Is32Bit ? 32 : 64); 192266759Sdim 193266759Sdim return false; 194266759Sdim} 195266759Sdim 196263509Sdimbool PPCCTRLoops::mightUseCTR(const Triple &TT, BasicBlock *BB) { 197263509Sdim for (BasicBlock::iterator J = BB->begin(), JE = BB->end(); 198263509Sdim J != JE; ++J) { 199263509Sdim if (CallInst *CI = dyn_cast<CallInst>(J)) { 200263509Sdim if (InlineAsm *IA = dyn_cast<InlineAsm>(CI->getCalledValue())) { 201263509Sdim // Inline ASM is okay, unless it clobbers the ctr register. 202263509Sdim InlineAsm::ConstraintInfoVector CIV = IA->ParseConstraints(); 203263509Sdim for (unsigned i = 0, ie = CIV.size(); i < ie; ++i) { 204263509Sdim InlineAsm::ConstraintInfo &C = CIV[i]; 205263509Sdim if (C.Type != InlineAsm::isInput) 206263509Sdim for (unsigned j = 0, je = C.Codes.size(); j < je; ++j) 207263509Sdim if (StringRef(C.Codes[j]).equals_lower("{ctr}")) 208263509Sdim return true; 209263509Sdim } 210239310Sdim 211263509Sdim continue; 212239310Sdim } 213239310Sdim 214263509Sdim if (!TM) 215263509Sdim return true; 216263509Sdim const TargetLowering *TLI = TM->getTargetLowering(); 217239310Sdim 218263509Sdim if (Function *F = CI->getCalledFunction()) { 219263509Sdim // Most intrinsics don't become function calls, but some might. 220263509Sdim // sin, cos, exp and log are always calls. 221263509Sdim unsigned Opcode; 222263509Sdim if (F->getIntrinsicID() != Intrinsic::not_intrinsic) { 223263509Sdim switch (F->getIntrinsicID()) { 224263509Sdim default: continue; 225239310Sdim 226263509Sdim// VisualStudio defines setjmp as _setjmp 227263509Sdim#if defined(_MSC_VER) && defined(setjmp) && \ 228263509Sdim !defined(setjmp_undefined_for_msvc) 229263509Sdim# pragma push_macro("setjmp") 230263509Sdim# undef setjmp 231263509Sdim# define setjmp_undefined_for_msvc 232263509Sdim#endif 233239310Sdim 234263509Sdim case Intrinsic::setjmp: 235239310Sdim 236263509Sdim#if defined(_MSC_VER) && defined(setjmp_undefined_for_msvc) 237263509Sdim // let's return it to _setjmp state 238263509Sdim# pragma pop_macro("setjmp") 239263509Sdim# undef setjmp_undefined_for_msvc 240263509Sdim#endif 241239310Sdim 242263509Sdim case Intrinsic::longjmp: 243239310Sdim 244263509Sdim // Exclude eh_sjlj_setjmp; we don't need to exclude eh_sjlj_longjmp 245263509Sdim // because, although it does clobber the counter register, the 246263509Sdim // control can't then return to inside the loop unless there is also 247263509Sdim // an eh_sjlj_setjmp. 248263509Sdim case Intrinsic::eh_sjlj_setjmp: 249239310Sdim 250263509Sdim case Intrinsic::memcpy: 251263509Sdim case Intrinsic::memmove: 252263509Sdim case Intrinsic::memset: 253263509Sdim case Intrinsic::powi: 254263509Sdim case Intrinsic::log: 255263509Sdim case Intrinsic::log2: 256263509Sdim case Intrinsic::log10: 257263509Sdim case Intrinsic::exp: 258263509Sdim case Intrinsic::exp2: 259263509Sdim case Intrinsic::pow: 260263509Sdim case Intrinsic::sin: 261263509Sdim case Intrinsic::cos: 262263509Sdim return true; 263263509Sdim case Intrinsic::copysign: 264263509Sdim if (CI->getArgOperand(0)->getType()->getScalarType()-> 265263509Sdim isPPC_FP128Ty()) 266263509Sdim return true; 267263509Sdim else 268263509Sdim continue; // ISD::FCOPYSIGN is never a library call. 269263509Sdim case Intrinsic::sqrt: Opcode = ISD::FSQRT; break; 270263509Sdim case Intrinsic::floor: Opcode = ISD::FFLOOR; break; 271263509Sdim case Intrinsic::ceil: Opcode = ISD::FCEIL; break; 272263509Sdim case Intrinsic::trunc: Opcode = ISD::FTRUNC; break; 273263509Sdim case Intrinsic::rint: Opcode = ISD::FRINT; break; 274263509Sdim case Intrinsic::nearbyint: Opcode = ISD::FNEARBYINT; break; 275263509Sdim case Intrinsic::round: Opcode = ISD::FROUND; break; 276263509Sdim } 277263509Sdim } 278239310Sdim 279263509Sdim // PowerPC does not use [US]DIVREM or other library calls for 280263509Sdim // operations on regular types which are not otherwise library calls 281263509Sdim // (i.e. soft float or atomics). If adapting for targets that do, 282263509Sdim // additional care is required here. 283239310Sdim 284263509Sdim LibFunc::Func Func; 285263509Sdim if (!F->hasLocalLinkage() && F->hasName() && LibInfo && 286263509Sdim LibInfo->getLibFunc(F->getName(), Func) && 287263509Sdim LibInfo->hasOptimizedCodeGen(Func)) { 288263509Sdim // Non-read-only functions are never treated as intrinsics. 289263509Sdim if (!CI->onlyReadsMemory()) 290263509Sdim return true; 291239310Sdim 292263509Sdim // Conversion happens only for FP calls. 293263509Sdim if (!CI->getArgOperand(0)->getType()->isFloatingPointTy()) 294263509Sdim return true; 295252723Sdim 296263509Sdim switch (Func) { 297263509Sdim default: return true; 298263509Sdim case LibFunc::copysign: 299263509Sdim case LibFunc::copysignf: 300263509Sdim continue; // ISD::FCOPYSIGN is never a library call. 301263509Sdim case LibFunc::copysignl: 302263509Sdim return true; 303263509Sdim case LibFunc::fabs: 304263509Sdim case LibFunc::fabsf: 305263509Sdim case LibFunc::fabsl: 306263509Sdim continue; // ISD::FABS is never a library call. 307263509Sdim case LibFunc::sqrt: 308263509Sdim case LibFunc::sqrtf: 309263509Sdim case LibFunc::sqrtl: 310263509Sdim Opcode = ISD::FSQRT; break; 311263509Sdim case LibFunc::floor: 312263509Sdim case LibFunc::floorf: 313263509Sdim case LibFunc::floorl: 314263509Sdim Opcode = ISD::FFLOOR; break; 315263509Sdim case LibFunc::nearbyint: 316263509Sdim case LibFunc::nearbyintf: 317263509Sdim case LibFunc::nearbyintl: 318263509Sdim Opcode = ISD::FNEARBYINT; break; 319263509Sdim case LibFunc::ceil: 320263509Sdim case LibFunc::ceilf: 321263509Sdim case LibFunc::ceill: 322263509Sdim Opcode = ISD::FCEIL; break; 323263509Sdim case LibFunc::rint: 324263509Sdim case LibFunc::rintf: 325263509Sdim case LibFunc::rintl: 326263509Sdim Opcode = ISD::FRINT; break; 327263509Sdim case LibFunc::round: 328263509Sdim case LibFunc::roundf: 329263509Sdim case LibFunc::roundl: 330263509Sdim Opcode = ISD::FROUND; break; 331263509Sdim case LibFunc::trunc: 332263509Sdim case LibFunc::truncf: 333263509Sdim case LibFunc::truncl: 334263509Sdim Opcode = ISD::FTRUNC; break; 335239310Sdim } 336239310Sdim 337263509Sdim MVT VTy = 338263509Sdim TLI->getSimpleValueType(CI->getArgOperand(0)->getType(), true); 339263509Sdim if (VTy == MVT::Other) 340263509Sdim return true; 341263509Sdim 342263509Sdim if (TLI->isOperationLegalOrCustom(Opcode, VTy)) 343263509Sdim continue; 344263509Sdim else if (VTy.isVector() && 345263509Sdim TLI->isOperationLegalOrCustom(Opcode, VTy.getScalarType())) 346263509Sdim continue; 347252723Sdim 348263509Sdim return true; 349239310Sdim } 350239310Sdim } 351239310Sdim 352239310Sdim return true; 353263509Sdim } else if (isa<BinaryOperator>(J) && 354263509Sdim J->getType()->getScalarType()->isPPC_FP128Ty()) { 355263509Sdim // Most operations on ppc_f128 values become calls. 356263509Sdim return true; 357263509Sdim } else if (isa<UIToFPInst>(J) || isa<SIToFPInst>(J) || 358263509Sdim isa<FPToUIInst>(J) || isa<FPToSIInst>(J)) { 359263509Sdim CastInst *CI = cast<CastInst>(J); 360263509Sdim if (CI->getSrcTy()->getScalarType()->isPPC_FP128Ty() || 361263509Sdim CI->getDestTy()->getScalarType()->isPPC_FP128Ty() || 362266759Sdim isLargeIntegerTy(TT.isArch32Bit(), CI->getSrcTy()->getScalarType()) || 363266759Sdim isLargeIntegerTy(TT.isArch32Bit(), CI->getDestTy()->getScalarType())) 364263509Sdim return true; 365266759Sdim } else if (isLargeIntegerTy(TT.isArch32Bit(), 366266759Sdim J->getType()->getScalarType()) && 367263509Sdim (J->getOpcode() == Instruction::UDiv || 368263509Sdim J->getOpcode() == Instruction::SDiv || 369263509Sdim J->getOpcode() == Instruction::URem || 370263509Sdim J->getOpcode() == Instruction::SRem)) { 371263509Sdim return true; 372263509Sdim } else if (isa<IndirectBrInst>(J) || isa<InvokeInst>(J)) { 373263509Sdim // On PowerPC, indirect jumps use the counter register. 374263509Sdim return true; 375263509Sdim } else if (SwitchInst *SI = dyn_cast<SwitchInst>(J)) { 376263509Sdim if (!TM) 377263509Sdim return true; 378263509Sdim const TargetLowering *TLI = TM->getTargetLowering(); 379239310Sdim 380263509Sdim if (TLI->supportJumpTables() && 381263509Sdim SI->getNumCases()+1 >= (unsigned) TLI->getMinimumJumpTableEntries()) 382239310Sdim return true; 383239310Sdim } 384239310Sdim } 385263509Sdim 386239310Sdim return false; 387239310Sdim} 388239310Sdim 389263509Sdimbool PPCCTRLoops::convertToCTRLoop(Loop *L) { 390263509Sdim bool MadeChange = false; 391239310Sdim 392263509Sdim Triple TT = Triple(L->getHeader()->getParent()->getParent()-> 393263509Sdim getTargetTriple()); 394263509Sdim if (!TT.isArch32Bit() && !TT.isArch64Bit()) 395263509Sdim return MadeChange; // Unknown arch. type. 396239310Sdim 397263509Sdim // Process nested loops first. 398263509Sdim for (Loop::iterator I = L->begin(), E = L->end(); I != E; ++I) { 399263509Sdim MadeChange |= convertToCTRLoop(*I); 400263509Sdim } 401239310Sdim 402263509Sdim // If a nested loop has been converted, then we can't convert this loop. 403263509Sdim if (MadeChange) 404263509Sdim return MadeChange; 405239310Sdim 406263509Sdim#ifndef NDEBUG 407263509Sdim // Stop trying after reaching the limit (if any). 408263509Sdim int Limit = CTRLoopLimit; 409263509Sdim if (Limit >= 0) { 410263509Sdim if (Counter >= CTRLoopLimit) 411263509Sdim return false; 412263509Sdim Counter++; 413239310Sdim } 414263509Sdim#endif 415239310Sdim 416263509Sdim // We don't want to spill/restore the counter register, and so we don't 417263509Sdim // want to use the counter register if the loop contains calls. 418263509Sdim for (Loop::block_iterator I = L->block_begin(), IE = L->block_end(); 419263509Sdim I != IE; ++I) 420263509Sdim if (mightUseCTR(TT, *I)) 421263509Sdim return MadeChange; 422239310Sdim 423263509Sdim SmallVector<BasicBlock*, 4> ExitingBlocks; 424263509Sdim L->getExitingBlocks(ExitingBlocks); 425239310Sdim 426263509Sdim BasicBlock *CountedExitBlock = 0; 427263509Sdim const SCEV *ExitCount = 0; 428263509Sdim BranchInst *CountedExitBranch = 0; 429263509Sdim for (SmallVectorImpl<BasicBlock *>::iterator I = ExitingBlocks.begin(), 430263509Sdim IE = ExitingBlocks.end(); I != IE; ++I) { 431263509Sdim const SCEV *EC = SE->getExitCount(L, *I); 432263509Sdim DEBUG(dbgs() << "Exit Count for " << *L << " from block " << 433263509Sdim (*I)->getName() << ": " << *EC << "\n"); 434263509Sdim if (isa<SCEVCouldNotCompute>(EC)) 435263509Sdim continue; 436263509Sdim if (const SCEVConstant *ConstEC = dyn_cast<SCEVConstant>(EC)) { 437263509Sdim if (ConstEC->getValue()->isZero()) 438263509Sdim continue; 439263509Sdim } else if (!SE->isLoopInvariant(EC, L)) 440263509Sdim continue; 441239310Sdim 442263509Sdim if (SE->getTypeSizeInBits(EC->getType()) > (TT.isArch64Bit() ? 64 : 32)) 443263509Sdim continue; 444263509Sdim 445263509Sdim // We now have a loop-invariant count of loop iterations (which is not the 446263509Sdim // constant zero) for which we know that this loop will not exit via this 447263509Sdim // exisiting block. 448263509Sdim 449263509Sdim // We need to make sure that this block will run on every loop iteration. 450263509Sdim // For this to be true, we must dominate all blocks with backedges. Such 451263509Sdim // blocks are in-loop predecessors to the header block. 452263509Sdim bool NotAlways = false; 453263509Sdim for (pred_iterator PI = pred_begin(L->getHeader()), 454263509Sdim PIE = pred_end(L->getHeader()); PI != PIE; ++PI) { 455263509Sdim if (!L->contains(*PI)) 456239310Sdim continue; 457263509Sdim 458263509Sdim if (!DT->dominates(*I, *PI)) { 459263509Sdim NotAlways = true; 460263509Sdim break; 461239310Sdim } 462239310Sdim } 463239310Sdim 464263509Sdim if (NotAlways) 465263509Sdim continue; 466239310Sdim 467263509Sdim // Make sure this blocks ends with a conditional branch. 468263509Sdim Instruction *TI = (*I)->getTerminator(); 469263509Sdim if (!TI) 470263509Sdim continue; 471239310Sdim 472263509Sdim if (BranchInst *BI = dyn_cast<BranchInst>(TI)) { 473263509Sdim if (!BI->isConditional()) 474263509Sdim continue; 475252723Sdim 476263509Sdim CountedExitBranch = BI; 477263509Sdim } else 478263509Sdim continue; 479252723Sdim 480263509Sdim // Note that this block may not be the loop latch block, even if the loop 481263509Sdim // has a latch block. 482263509Sdim CountedExitBlock = *I; 483263509Sdim ExitCount = EC; 484263509Sdim break; 485252723Sdim } 486252723Sdim 487263509Sdim if (!CountedExitBlock) 488263509Sdim return MadeChange; 489239310Sdim 490263509Sdim BasicBlock *Preheader = L->getLoopPreheader(); 491239310Sdim 492263509Sdim // If we don't have a preheader, then insert one. If we already have a 493263509Sdim // preheader, then we can use it (except if the preheader contains a use of 494263509Sdim // the CTR register because some such uses might be reordered by the 495263509Sdim // selection DAG after the mtctr instruction). 496263509Sdim if (!Preheader || mightUseCTR(TT, Preheader)) 497263509Sdim Preheader = InsertPreheaderForLoop(L, this); 498263509Sdim if (!Preheader) 499263509Sdim return MadeChange; 500239310Sdim 501263509Sdim DEBUG(dbgs() << "Preheader for exit count: " << Preheader->getName() << "\n"); 502263509Sdim 503263509Sdim // Insert the count into the preheader and replace the condition used by the 504263509Sdim // selected branch. 505263509Sdim MadeChange = true; 506263509Sdim 507263509Sdim SCEVExpander SCEVE(*SE, "loopcnt"); 508263509Sdim LLVMContext &C = SE->getContext(); 509263509Sdim Type *CountType = TT.isArch64Bit() ? Type::getInt64Ty(C) : 510263509Sdim Type::getInt32Ty(C); 511263509Sdim if (!ExitCount->getType()->isPointerTy() && 512263509Sdim ExitCount->getType() != CountType) 513263509Sdim ExitCount = SE->getZeroExtendExpr(ExitCount, CountType); 514263509Sdim ExitCount = SE->getAddExpr(ExitCount, 515263509Sdim SE->getConstant(CountType, 1)); 516263509Sdim Value *ECValue = SCEVE.expandCodeFor(ExitCount, CountType, 517263509Sdim Preheader->getTerminator()); 518263509Sdim 519263509Sdim IRBuilder<> CountBuilder(Preheader->getTerminator()); 520263509Sdim Module *M = Preheader->getParent()->getParent(); 521263509Sdim Value *MTCTRFunc = Intrinsic::getDeclaration(M, Intrinsic::ppc_mtctr, 522263509Sdim CountType); 523263509Sdim CountBuilder.CreateCall(MTCTRFunc, ECValue); 524263509Sdim 525263509Sdim IRBuilder<> CondBuilder(CountedExitBranch); 526263509Sdim Value *DecFunc = 527263509Sdim Intrinsic::getDeclaration(M, Intrinsic::ppc_is_decremented_ctr_nonzero); 528263509Sdim Value *NewCond = CondBuilder.CreateCall(DecFunc); 529263509Sdim Value *OldCond = CountedExitBranch->getCondition(); 530263509Sdim CountedExitBranch->setCondition(NewCond); 531263509Sdim 532263509Sdim // The false branch must exit the loop. 533263509Sdim if (!L->contains(CountedExitBranch->getSuccessor(0))) 534263509Sdim CountedExitBranch->swapSuccessors(); 535263509Sdim 536263509Sdim // The old condition may be dead now, and may have even created a dead PHI 537263509Sdim // (the original induction variable). 538263509Sdim RecursivelyDeleteTriviallyDeadInstructions(OldCond); 539263509Sdim DeleteDeadPHIs(CountedExitBlock); 540263509Sdim 541263509Sdim ++NumCTRLoops; 542263509Sdim return MadeChange; 543263509Sdim} 544263509Sdim 545263509Sdim#ifndef NDEBUG 546263509Sdimstatic bool clobbersCTR(const MachineInstr *MI) { 547263509Sdim for (unsigned i = 0, e = MI->getNumOperands(); i != e; ++i) { 548263509Sdim const MachineOperand &MO = MI->getOperand(i); 549263509Sdim if (MO.isReg()) { 550263509Sdim if (MO.isDef() && (MO.getReg() == PPC::CTR || MO.getReg() == PPC::CTR8)) 551263509Sdim return true; 552263509Sdim } else if (MO.isRegMask()) { 553263509Sdim if (MO.clobbersPhysReg(PPC::CTR) || MO.clobbersPhysReg(PPC::CTR8)) 554263509Sdim return true; 555239310Sdim } 556239310Sdim } 557239310Sdim 558263509Sdim return false; 559263509Sdim} 560239310Sdim 561263509Sdimstatic bool verifyCTRBranch(MachineBasicBlock *MBB, 562263509Sdim MachineBasicBlock::iterator I) { 563263509Sdim MachineBasicBlock::iterator BI = I; 564263509Sdim SmallSet<MachineBasicBlock *, 16> Visited; 565263509Sdim SmallVector<MachineBasicBlock *, 8> Preds; 566263509Sdim bool CheckPreds; 567239310Sdim 568263509Sdim if (I == MBB->begin()) { 569263509Sdim Visited.insert(MBB); 570263509Sdim goto queue_preds; 571263509Sdim } else 572263509Sdim --I; 573239310Sdim 574263509Sdimcheck_block: 575263509Sdim Visited.insert(MBB); 576263509Sdim if (I == MBB->end()) 577263509Sdim goto queue_preds; 578263509Sdim 579263509Sdim CheckPreds = true; 580263509Sdim for (MachineBasicBlock::iterator IE = MBB->begin();; --I) { 581263509Sdim unsigned Opc = I->getOpcode(); 582263509Sdim if (Opc == PPC::MTCTRloop || Opc == PPC::MTCTR8loop) { 583263509Sdim CheckPreds = false; 584263509Sdim break; 585239310Sdim } 586239310Sdim 587263509Sdim if (I != BI && clobbersCTR(I)) { 588263509Sdim DEBUG(dbgs() << "BB#" << MBB->getNumber() << " (" << 589263509Sdim MBB->getFullName() << ") instruction " << *I << 590263509Sdim " clobbers CTR, invalidating " << "BB#" << 591263509Sdim BI->getParent()->getNumber() << " (" << 592263509Sdim BI->getParent()->getFullName() << ") instruction " << 593263509Sdim *BI << "\n"); 594263509Sdim return false; 595263509Sdim } 596239310Sdim 597263509Sdim if (I == IE) 598263509Sdim break; 599239310Sdim } 600239310Sdim 601263509Sdim if (!CheckPreds && Preds.empty()) 602263509Sdim return true; 603239310Sdim 604263509Sdim if (CheckPreds) { 605263509Sdimqueue_preds: 606263509Sdim if (MachineFunction::iterator(MBB) == MBB->getParent()->begin()) { 607263509Sdim DEBUG(dbgs() << "Unable to find a MTCTR instruction for BB#" << 608263509Sdim BI->getParent()->getNumber() << " (" << 609263509Sdim BI->getParent()->getFullName() << ") instruction " << 610263509Sdim *BI << "\n"); 611263509Sdim return false; 612263509Sdim } 613239310Sdim 614263509Sdim for (MachineBasicBlock::pred_iterator PI = MBB->pred_begin(), 615263509Sdim PIE = MBB->pred_end(); PI != PIE; ++PI) 616263509Sdim Preds.push_back(*PI); 617239310Sdim } 618239310Sdim 619263509Sdim do { 620263509Sdim MBB = Preds.pop_back_val(); 621263509Sdim if (!Visited.count(MBB)) { 622263509Sdim I = MBB->getLastNonDebugInstr(); 623263509Sdim goto check_block; 624263509Sdim } 625263509Sdim } while (!Preds.empty()); 626239310Sdim 627263509Sdim return true; 628263509Sdim} 629239310Sdim 630263509Sdimbool PPCCTRLoopsVerify::runOnMachineFunction(MachineFunction &MF) { 631263509Sdim MDT = &getAnalysis<MachineDominatorTree>(); 632239310Sdim 633263509Sdim // Verify that all bdnz/bdz instructions are dominated by a loop mtctr before 634263509Sdim // any other instructions that might clobber the ctr register. 635263509Sdim for (MachineFunction::iterator I = MF.begin(), IE = MF.end(); 636263509Sdim I != IE; ++I) { 637263509Sdim MachineBasicBlock *MBB = I; 638263509Sdim if (!MDT->isReachableFromEntry(MBB)) 639263509Sdim continue; 640239310Sdim 641263509Sdim for (MachineBasicBlock::iterator MII = MBB->getFirstTerminator(), 642263509Sdim MIIE = MBB->end(); MII != MIIE; ++MII) { 643263509Sdim unsigned Opc = MII->getOpcode(); 644263509Sdim if (Opc == PPC::BDNZ8 || Opc == PPC::BDNZ || 645263509Sdim Opc == PPC::BDZ8 || Opc == PPC::BDZ) 646263509Sdim if (!verifyCTRBranch(MBB, MII)) 647263509Sdim llvm_unreachable("Invalid PPC CTR loop!"); 648263509Sdim } 649263509Sdim } 650263509Sdim 651263509Sdim return false; 652239310Sdim} 653263509Sdim#endif // NDEBUG 654239310Sdim 655