1 //===- GVN.cpp - Eliminate redundant values and loads ---------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // This pass performs global value numbering to eliminate fully redundant 10 // instructions. It also performs simple dead load elimination. 11 // 12 // Note that this pass does the value numbering itself; it does not use the 13 // ValueNumbering analysis passes. 14 // 15 //===----------------------------------------------------------------------===// 16 17 #include "llvm/Transforms/Scalar/GVN.h" 18 #include "llvm/ADT/DenseMap.h" 19 #include "llvm/ADT/DepthFirstIterator.h" 20 #include "llvm/ADT/Hashing.h" 21 #include "llvm/ADT/MapVector.h" 22 #include "llvm/ADT/PointerIntPair.h" 23 #include "llvm/ADT/PostOrderIterator.h" 24 #include "llvm/ADT/STLExtras.h" 25 #include "llvm/ADT/SetVector.h" 26 #include "llvm/ADT/SmallPtrSet.h" 27 #include "llvm/ADT/SmallVector.h" 28 #include "llvm/ADT/Statistic.h" 29 #include "llvm/Analysis/AliasAnalysis.h" 30 #include "llvm/Analysis/AssumeBundleQueries.h" 31 #include "llvm/Analysis/AssumptionCache.h" 32 #include "llvm/Analysis/CFG.h" 33 #include "llvm/Analysis/DomTreeUpdater.h" 34 #include "llvm/Analysis/GlobalsModRef.h" 35 #include "llvm/Analysis/InstructionSimplify.h" 36 #include "llvm/Analysis/LoopInfo.h" 37 #include "llvm/Analysis/MemoryBuiltins.h" 38 #include "llvm/Analysis/MemoryDependenceAnalysis.h" 39 #include "llvm/Analysis/MemorySSA.h" 40 #include "llvm/Analysis/MemorySSAUpdater.h" 41 #include "llvm/Analysis/OptimizationRemarkEmitter.h" 42 #include "llvm/Analysis/PHITransAddr.h" 43 #include "llvm/Analysis/TargetLibraryInfo.h" 44 #include "llvm/Analysis/ValueTracking.h" 45 #include "llvm/Config/llvm-config.h" 46 #include "llvm/IR/Attributes.h" 47 #include "llvm/IR/BasicBlock.h" 48 #include "llvm/IR/Constant.h" 49 #include "llvm/IR/Constants.h" 50 #include "llvm/IR/DataLayout.h" 51 #include "llvm/IR/DebugLoc.h" 52 #include "llvm/IR/Dominators.h" 53 #include "llvm/IR/Function.h" 54 #include "llvm/IR/InstrTypes.h" 55 #include "llvm/IR/Instruction.h" 56 #include "llvm/IR/Instructions.h" 57 #include "llvm/IR/IntrinsicInst.h" 58 #include "llvm/IR/Intrinsics.h" 59 #include "llvm/IR/LLVMContext.h" 60 #include "llvm/IR/Metadata.h" 61 #include "llvm/IR/Module.h" 62 #include "llvm/IR/Operator.h" 63 #include "llvm/IR/PassManager.h" 64 #include "llvm/IR/PatternMatch.h" 65 #include "llvm/IR/Type.h" 66 #include "llvm/IR/Use.h" 67 #include "llvm/IR/Value.h" 68 #include "llvm/InitializePasses.h" 69 #include "llvm/Pass.h" 70 #include "llvm/Support/Casting.h" 71 #include "llvm/Support/CommandLine.h" 72 #include "llvm/Support/Compiler.h" 73 #include "llvm/Support/Debug.h" 74 #include "llvm/Support/raw_ostream.h" 75 #include "llvm/Transforms/Utils.h" 76 #include "llvm/Transforms/Utils/AssumeBundleBuilder.h" 77 #include "llvm/Transforms/Utils/BasicBlockUtils.h" 78 #include "llvm/Transforms/Utils/Local.h" 79 #include "llvm/Transforms/Utils/SSAUpdater.h" 80 #include "llvm/Transforms/Utils/VNCoercion.h" 81 #include <algorithm> 82 #include <cassert> 83 #include <cstdint> 84 #include <utility> 85 #include <vector> 86 87 using namespace llvm; 88 using namespace llvm::gvn; 89 using namespace llvm::VNCoercion; 90 using namespace PatternMatch; 91 92 #define DEBUG_TYPE "gvn" 93 94 STATISTIC(NumGVNInstr, "Number of instructions deleted"); 95 STATISTIC(NumGVNLoad, "Number of loads deleted"); 96 STATISTIC(NumGVNPRE, "Number of instructions PRE'd"); 97 STATISTIC(NumGVNBlocks, "Number of blocks merged"); 98 STATISTIC(NumGVNSimpl, "Number of instructions simplified"); 99 STATISTIC(NumGVNEqProp, "Number of equalities propagated"); 100 STATISTIC(NumPRELoad, "Number of loads PRE'd"); 101 STATISTIC(NumPRELoopLoad, "Number of loop loads PRE'd"); 102 103 STATISTIC(IsValueFullyAvailableInBlockNumSpeculationsMax, 104 "Number of blocks speculated as available in " 105 "IsValueFullyAvailableInBlock(), max"); 106 STATISTIC(MaxBBSpeculationCutoffReachedTimes, 107 "Number of times we we reached gvn-max-block-speculations cut-off " 108 "preventing further exploration"); 109 110 static cl::opt<bool> GVNEnablePRE("enable-pre", cl::init(true), cl::Hidden); 111 static cl::opt<bool> GVNEnableLoadPRE("enable-load-pre", cl::init(true)); 112 static cl::opt<bool> GVNEnableLoadInLoopPRE("enable-load-in-loop-pre", 113 cl::init(true)); 114 static cl::opt<bool> 115 GVNEnableSplitBackedgeInLoadPRE("enable-split-backedge-in-load-pre", 116 cl::init(true)); 117 static cl::opt<bool> GVNEnableMemDep("enable-gvn-memdep", cl::init(true)); 118 119 static cl::opt<uint32_t> MaxNumDeps( 120 "gvn-max-num-deps", cl::Hidden, cl::init(100), cl::ZeroOrMore, 121 cl::desc("Max number of dependences to attempt Load PRE (default = 100)")); 122 123 // This is based on IsValueFullyAvailableInBlockNumSpeculationsMax stat. 124 static cl::opt<uint32_t> MaxBBSpeculations( 125 "gvn-max-block-speculations", cl::Hidden, cl::init(600), cl::ZeroOrMore, 126 cl::desc("Max number of blocks we're willing to speculate on (and recurse " 127 "into) when deducing if a value is fully available or not in GVN " 128 "(default = 600)")); 129 130 struct llvm::GVN::Expression { 131 uint32_t opcode; 132 bool commutative = false; 133 Type *type = nullptr; 134 SmallVector<uint32_t, 4> varargs; 135 136 Expression(uint32_t o = ~2U) : opcode(o) {} 137 138 bool operator==(const Expression &other) const { 139 if (opcode != other.opcode) 140 return false; 141 if (opcode == ~0U || opcode == ~1U) 142 return true; 143 if (type != other.type) 144 return false; 145 if (varargs != other.varargs) 146 return false; 147 return true; 148 } 149 150 friend hash_code hash_value(const Expression &Value) { 151 return hash_combine( 152 Value.opcode, Value.type, 153 hash_combine_range(Value.varargs.begin(), Value.varargs.end())); 154 } 155 }; 156 157 namespace llvm { 158 159 template <> struct DenseMapInfo<GVN::Expression> { 160 static inline GVN::Expression getEmptyKey() { return ~0U; } 161 static inline GVN::Expression getTombstoneKey() { return ~1U; } 162 163 static unsigned getHashValue(const GVN::Expression &e) { 164 using llvm::hash_value; 165 166 return static_cast<unsigned>(hash_value(e)); 167 } 168 169 static bool isEqual(const GVN::Expression &LHS, const GVN::Expression &RHS) { 170 return LHS == RHS; 171 } 172 }; 173 174 } // end namespace llvm 175 176 /// Represents a particular available value that we know how to materialize. 177 /// Materialization of an AvailableValue never fails. An AvailableValue is 178 /// implicitly associated with a rematerialization point which is the 179 /// location of the instruction from which it was formed. 180 struct llvm::gvn::AvailableValue { 181 enum ValType { 182 SimpleVal, // A simple offsetted value that is accessed. 183 LoadVal, // A value produced by a load. 184 MemIntrin, // A memory intrinsic which is loaded from. 185 UndefVal // A UndefValue representing a value from dead block (which 186 // is not yet physically removed from the CFG). 187 }; 188 189 /// V - The value that is live out of the block. 190 PointerIntPair<Value *, 2, ValType> Val; 191 192 /// Offset - The byte offset in Val that is interesting for the load query. 193 unsigned Offset = 0; 194 195 static AvailableValue get(Value *V, unsigned Offset = 0) { 196 AvailableValue Res; 197 Res.Val.setPointer(V); 198 Res.Val.setInt(SimpleVal); 199 Res.Offset = Offset; 200 return Res; 201 } 202 203 static AvailableValue getMI(MemIntrinsic *MI, unsigned Offset = 0) { 204 AvailableValue Res; 205 Res.Val.setPointer(MI); 206 Res.Val.setInt(MemIntrin); 207 Res.Offset = Offset; 208 return Res; 209 } 210 211 static AvailableValue getLoad(LoadInst *Load, unsigned Offset = 0) { 212 AvailableValue Res; 213 Res.Val.setPointer(Load); 214 Res.Val.setInt(LoadVal); 215 Res.Offset = Offset; 216 return Res; 217 } 218 219 static AvailableValue getUndef() { 220 AvailableValue Res; 221 Res.Val.setPointer(nullptr); 222 Res.Val.setInt(UndefVal); 223 Res.Offset = 0; 224 return Res; 225 } 226 227 bool isSimpleValue() const { return Val.getInt() == SimpleVal; } 228 bool isCoercedLoadValue() const { return Val.getInt() == LoadVal; } 229 bool isMemIntrinValue() const { return Val.getInt() == MemIntrin; } 230 bool isUndefValue() const { return Val.getInt() == UndefVal; } 231 232 Value *getSimpleValue() const { 233 assert(isSimpleValue() && "Wrong accessor"); 234 return Val.getPointer(); 235 } 236 237 LoadInst *getCoercedLoadValue() const { 238 assert(isCoercedLoadValue() && "Wrong accessor"); 239 return cast<LoadInst>(Val.getPointer()); 240 } 241 242 MemIntrinsic *getMemIntrinValue() const { 243 assert(isMemIntrinValue() && "Wrong accessor"); 244 return cast<MemIntrinsic>(Val.getPointer()); 245 } 246 247 /// Emit code at the specified insertion point to adjust the value defined 248 /// here to the specified type. This handles various coercion cases. 249 Value *MaterializeAdjustedValue(LoadInst *Load, Instruction *InsertPt, 250 GVN &gvn) const; 251 }; 252 253 /// Represents an AvailableValue which can be rematerialized at the end of 254 /// the associated BasicBlock. 255 struct llvm::gvn::AvailableValueInBlock { 256 /// BB - The basic block in question. 257 BasicBlock *BB = nullptr; 258 259 /// AV - The actual available value 260 AvailableValue AV; 261 262 static AvailableValueInBlock get(BasicBlock *BB, AvailableValue &&AV) { 263 AvailableValueInBlock Res; 264 Res.BB = BB; 265 Res.AV = std::move(AV); 266 return Res; 267 } 268 269 static AvailableValueInBlock get(BasicBlock *BB, Value *V, 270 unsigned Offset = 0) { 271 return get(BB, AvailableValue::get(V, Offset)); 272 } 273 274 static AvailableValueInBlock getUndef(BasicBlock *BB) { 275 return get(BB, AvailableValue::getUndef()); 276 } 277 278 /// Emit code at the end of this block to adjust the value defined here to 279 /// the specified type. This handles various coercion cases. 280 Value *MaterializeAdjustedValue(LoadInst *Load, GVN &gvn) const { 281 return AV.MaterializeAdjustedValue(Load, BB->getTerminator(), gvn); 282 } 283 }; 284 285 //===----------------------------------------------------------------------===// 286 // ValueTable Internal Functions 287 //===----------------------------------------------------------------------===// 288 289 GVN::Expression GVN::ValueTable::createExpr(Instruction *I) { 290 Expression e; 291 e.type = I->getType(); 292 e.opcode = I->getOpcode(); 293 if (const GCRelocateInst *GCR = dyn_cast<GCRelocateInst>(I)) { 294 // gc.relocate is 'special' call: its second and third operands are 295 // not real values, but indices into statepoint's argument list. 296 // Use the refered to values for purposes of identity. 297 e.varargs.push_back(lookupOrAdd(GCR->getOperand(0))); 298 e.varargs.push_back(lookupOrAdd(GCR->getBasePtr())); 299 e.varargs.push_back(lookupOrAdd(GCR->getDerivedPtr())); 300 } else { 301 for (Use &Op : I->operands()) 302 e.varargs.push_back(lookupOrAdd(Op)); 303 } 304 if (I->isCommutative()) { 305 // Ensure that commutative instructions that only differ by a permutation 306 // of their operands get the same value number by sorting the operand value 307 // numbers. Since commutative operands are the 1st two operands it is more 308 // efficient to sort by hand rather than using, say, std::sort. 309 assert(I->getNumOperands() >= 2 && "Unsupported commutative instruction!"); 310 if (e.varargs[0] > e.varargs[1]) 311 std::swap(e.varargs[0], e.varargs[1]); 312 e.commutative = true; 313 } 314 315 if (auto *C = dyn_cast<CmpInst>(I)) { 316 // Sort the operand value numbers so x<y and y>x get the same value number. 317 CmpInst::Predicate Predicate = C->getPredicate(); 318 if (e.varargs[0] > e.varargs[1]) { 319 std::swap(e.varargs[0], e.varargs[1]); 320 Predicate = CmpInst::getSwappedPredicate(Predicate); 321 } 322 e.opcode = (C->getOpcode() << 8) | Predicate; 323 e.commutative = true; 324 } else if (auto *E = dyn_cast<InsertValueInst>(I)) { 325 e.varargs.append(E->idx_begin(), E->idx_end()); 326 } else if (auto *SVI = dyn_cast<ShuffleVectorInst>(I)) { 327 ArrayRef<int> ShuffleMask = SVI->getShuffleMask(); 328 e.varargs.append(ShuffleMask.begin(), ShuffleMask.end()); 329 } 330 331 return e; 332 } 333 334 GVN::Expression GVN::ValueTable::createCmpExpr(unsigned Opcode, 335 CmpInst::Predicate Predicate, 336 Value *LHS, Value *RHS) { 337 assert((Opcode == Instruction::ICmp || Opcode == Instruction::FCmp) && 338 "Not a comparison!"); 339 Expression e; 340 e.type = CmpInst::makeCmpResultType(LHS->getType()); 341 e.varargs.push_back(lookupOrAdd(LHS)); 342 e.varargs.push_back(lookupOrAdd(RHS)); 343 344 // Sort the operand value numbers so x<y and y>x get the same value number. 345 if (e.varargs[0] > e.varargs[1]) { 346 std::swap(e.varargs[0], e.varargs[1]); 347 Predicate = CmpInst::getSwappedPredicate(Predicate); 348 } 349 e.opcode = (Opcode << 8) | Predicate; 350 e.commutative = true; 351 return e; 352 } 353 354 GVN::Expression GVN::ValueTable::createExtractvalueExpr(ExtractValueInst *EI) { 355 assert(EI && "Not an ExtractValueInst?"); 356 Expression e; 357 e.type = EI->getType(); 358 e.opcode = 0; 359 360 WithOverflowInst *WO = dyn_cast<WithOverflowInst>(EI->getAggregateOperand()); 361 if (WO != nullptr && EI->getNumIndices() == 1 && *EI->idx_begin() == 0) { 362 // EI is an extract from one of our with.overflow intrinsics. Synthesize 363 // a semantically equivalent expression instead of an extract value 364 // expression. 365 e.opcode = WO->getBinaryOp(); 366 e.varargs.push_back(lookupOrAdd(WO->getLHS())); 367 e.varargs.push_back(lookupOrAdd(WO->getRHS())); 368 return e; 369 } 370 371 // Not a recognised intrinsic. Fall back to producing an extract value 372 // expression. 373 e.opcode = EI->getOpcode(); 374 for (Use &Op : EI->operands()) 375 e.varargs.push_back(lookupOrAdd(Op)); 376 377 append_range(e.varargs, EI->indices()); 378 379 return e; 380 } 381 382 //===----------------------------------------------------------------------===// 383 // ValueTable External Functions 384 //===----------------------------------------------------------------------===// 385 386 GVN::ValueTable::ValueTable() = default; 387 GVN::ValueTable::ValueTable(const ValueTable &) = default; 388 GVN::ValueTable::ValueTable(ValueTable &&) = default; 389 GVN::ValueTable::~ValueTable() = default; 390 GVN::ValueTable &GVN::ValueTable::operator=(const GVN::ValueTable &Arg) = default; 391 392 /// add - Insert a value into the table with a specified value number. 393 void GVN::ValueTable::add(Value *V, uint32_t num) { 394 valueNumbering.insert(std::make_pair(V, num)); 395 if (PHINode *PN = dyn_cast<PHINode>(V)) 396 NumberingPhi[num] = PN; 397 } 398 399 uint32_t GVN::ValueTable::lookupOrAddCall(CallInst *C) { 400 if (AA->doesNotAccessMemory(C)) { 401 Expression exp = createExpr(C); 402 uint32_t e = assignExpNewValueNum(exp).first; 403 valueNumbering[C] = e; 404 return e; 405 } else if (MD && AA->onlyReadsMemory(C)) { 406 Expression exp = createExpr(C); 407 auto ValNum = assignExpNewValueNum(exp); 408 if (ValNum.second) { 409 valueNumbering[C] = ValNum.first; 410 return ValNum.first; 411 } 412 413 MemDepResult local_dep = MD->getDependency(C); 414 415 if (!local_dep.isDef() && !local_dep.isNonLocal()) { 416 valueNumbering[C] = nextValueNumber; 417 return nextValueNumber++; 418 } 419 420 if (local_dep.isDef()) { 421 // For masked load/store intrinsics, the local_dep may actully be 422 // a normal load or store instruction. 423 CallInst *local_cdep = dyn_cast<CallInst>(local_dep.getInst()); 424 425 if (!local_cdep || 426 local_cdep->getNumArgOperands() != C->getNumArgOperands()) { 427 valueNumbering[C] = nextValueNumber; 428 return nextValueNumber++; 429 } 430 431 for (unsigned i = 0, e = C->getNumArgOperands(); i < e; ++i) { 432 uint32_t c_vn = lookupOrAdd(C->getArgOperand(i)); 433 uint32_t cd_vn = lookupOrAdd(local_cdep->getArgOperand(i)); 434 if (c_vn != cd_vn) { 435 valueNumbering[C] = nextValueNumber; 436 return nextValueNumber++; 437 } 438 } 439 440 uint32_t v = lookupOrAdd(local_cdep); 441 valueNumbering[C] = v; 442 return v; 443 } 444 445 // Non-local case. 446 const MemoryDependenceResults::NonLocalDepInfo &deps = 447 MD->getNonLocalCallDependency(C); 448 // FIXME: Move the checking logic to MemDep! 449 CallInst* cdep = nullptr; 450 451 // Check to see if we have a single dominating call instruction that is 452 // identical to C. 453 for (unsigned i = 0, e = deps.size(); i != e; ++i) { 454 const NonLocalDepEntry *I = &deps[i]; 455 if (I->getResult().isNonLocal()) 456 continue; 457 458 // We don't handle non-definitions. If we already have a call, reject 459 // instruction dependencies. 460 if (!I->getResult().isDef() || cdep != nullptr) { 461 cdep = nullptr; 462 break; 463 } 464 465 CallInst *NonLocalDepCall = dyn_cast<CallInst>(I->getResult().getInst()); 466 // FIXME: All duplicated with non-local case. 467 if (NonLocalDepCall && DT->properlyDominates(I->getBB(), C->getParent())){ 468 cdep = NonLocalDepCall; 469 continue; 470 } 471 472 cdep = nullptr; 473 break; 474 } 475 476 if (!cdep) { 477 valueNumbering[C] = nextValueNumber; 478 return nextValueNumber++; 479 } 480 481 if (cdep->getNumArgOperands() != C->getNumArgOperands()) { 482 valueNumbering[C] = nextValueNumber; 483 return nextValueNumber++; 484 } 485 for (unsigned i = 0, e = C->getNumArgOperands(); i < e; ++i) { 486 uint32_t c_vn = lookupOrAdd(C->getArgOperand(i)); 487 uint32_t cd_vn = lookupOrAdd(cdep->getArgOperand(i)); 488 if (c_vn != cd_vn) { 489 valueNumbering[C] = nextValueNumber; 490 return nextValueNumber++; 491 } 492 } 493 494 uint32_t v = lookupOrAdd(cdep); 495 valueNumbering[C] = v; 496 return v; 497 } else { 498 valueNumbering[C] = nextValueNumber; 499 return nextValueNumber++; 500 } 501 } 502 503 /// Returns true if a value number exists for the specified value. 504 bool GVN::ValueTable::exists(Value *V) const { return valueNumbering.count(V) != 0; } 505 506 /// lookup_or_add - Returns the value number for the specified value, assigning 507 /// it a new number if it did not have one before. 508 uint32_t GVN::ValueTable::lookupOrAdd(Value *V) { 509 DenseMap<Value*, uint32_t>::iterator VI = valueNumbering.find(V); 510 if (VI != valueNumbering.end()) 511 return VI->second; 512 513 if (!isa<Instruction>(V)) { 514 valueNumbering[V] = nextValueNumber; 515 return nextValueNumber++; 516 } 517 518 Instruction* I = cast<Instruction>(V); 519 Expression exp; 520 switch (I->getOpcode()) { 521 case Instruction::Call: 522 return lookupOrAddCall(cast<CallInst>(I)); 523 case Instruction::FNeg: 524 case Instruction::Add: 525 case Instruction::FAdd: 526 case Instruction::Sub: 527 case Instruction::FSub: 528 case Instruction::Mul: 529 case Instruction::FMul: 530 case Instruction::UDiv: 531 case Instruction::SDiv: 532 case Instruction::FDiv: 533 case Instruction::URem: 534 case Instruction::SRem: 535 case Instruction::FRem: 536 case Instruction::Shl: 537 case Instruction::LShr: 538 case Instruction::AShr: 539 case Instruction::And: 540 case Instruction::Or: 541 case Instruction::Xor: 542 case Instruction::ICmp: 543 case Instruction::FCmp: 544 case Instruction::Trunc: 545 case Instruction::ZExt: 546 case Instruction::SExt: 547 case Instruction::FPToUI: 548 case Instruction::FPToSI: 549 case Instruction::UIToFP: 550 case Instruction::SIToFP: 551 case Instruction::FPTrunc: 552 case Instruction::FPExt: 553 case Instruction::PtrToInt: 554 case Instruction::IntToPtr: 555 case Instruction::AddrSpaceCast: 556 case Instruction::BitCast: 557 case Instruction::Select: 558 case Instruction::Freeze: 559 case Instruction::ExtractElement: 560 case Instruction::InsertElement: 561 case Instruction::ShuffleVector: 562 case Instruction::InsertValue: 563 case Instruction::GetElementPtr: 564 exp = createExpr(I); 565 break; 566 case Instruction::ExtractValue: 567 exp = createExtractvalueExpr(cast<ExtractValueInst>(I)); 568 break; 569 case Instruction::PHI: 570 valueNumbering[V] = nextValueNumber; 571 NumberingPhi[nextValueNumber] = cast<PHINode>(V); 572 return nextValueNumber++; 573 default: 574 valueNumbering[V] = nextValueNumber; 575 return nextValueNumber++; 576 } 577 578 uint32_t e = assignExpNewValueNum(exp).first; 579 valueNumbering[V] = e; 580 return e; 581 } 582 583 /// Returns the value number of the specified value. Fails if 584 /// the value has not yet been numbered. 585 uint32_t GVN::ValueTable::lookup(Value *V, bool Verify) const { 586 DenseMap<Value*, uint32_t>::const_iterator VI = valueNumbering.find(V); 587 if (Verify) { 588 assert(VI != valueNumbering.end() && "Value not numbered?"); 589 return VI->second; 590 } 591 return (VI != valueNumbering.end()) ? VI->second : 0; 592 } 593 594 /// Returns the value number of the given comparison, 595 /// assigning it a new number if it did not have one before. Useful when 596 /// we deduced the result of a comparison, but don't immediately have an 597 /// instruction realizing that comparison to hand. 598 uint32_t GVN::ValueTable::lookupOrAddCmp(unsigned Opcode, 599 CmpInst::Predicate Predicate, 600 Value *LHS, Value *RHS) { 601 Expression exp = createCmpExpr(Opcode, Predicate, LHS, RHS); 602 return assignExpNewValueNum(exp).first; 603 } 604 605 /// Remove all entries from the ValueTable. 606 void GVN::ValueTable::clear() { 607 valueNumbering.clear(); 608 expressionNumbering.clear(); 609 NumberingPhi.clear(); 610 PhiTranslateTable.clear(); 611 nextValueNumber = 1; 612 Expressions.clear(); 613 ExprIdx.clear(); 614 nextExprNumber = 0; 615 } 616 617 /// Remove a value from the value numbering. 618 void GVN::ValueTable::erase(Value *V) { 619 uint32_t Num = valueNumbering.lookup(V); 620 valueNumbering.erase(V); 621 // If V is PHINode, V <--> value number is an one-to-one mapping. 622 if (isa<PHINode>(V)) 623 NumberingPhi.erase(Num); 624 } 625 626 /// verifyRemoved - Verify that the value is removed from all internal data 627 /// structures. 628 void GVN::ValueTable::verifyRemoved(const Value *V) const { 629 for (DenseMap<Value*, uint32_t>::const_iterator 630 I = valueNumbering.begin(), E = valueNumbering.end(); I != E; ++I) { 631 assert(I->first != V && "Inst still occurs in value numbering map!"); 632 } 633 } 634 635 //===----------------------------------------------------------------------===// 636 // GVN Pass 637 //===----------------------------------------------------------------------===// 638 639 bool GVN::isPREEnabled() const { 640 return Options.AllowPRE.getValueOr(GVNEnablePRE); 641 } 642 643 bool GVN::isLoadPREEnabled() const { 644 return Options.AllowLoadPRE.getValueOr(GVNEnableLoadPRE); 645 } 646 647 bool GVN::isLoadInLoopPREEnabled() const { 648 return Options.AllowLoadInLoopPRE.getValueOr(GVNEnableLoadInLoopPRE); 649 } 650 651 bool GVN::isLoadPRESplitBackedgeEnabled() const { 652 return Options.AllowLoadPRESplitBackedge.getValueOr( 653 GVNEnableSplitBackedgeInLoadPRE); 654 } 655 656 bool GVN::isMemDepEnabled() const { 657 return Options.AllowMemDep.getValueOr(GVNEnableMemDep); 658 } 659 660 PreservedAnalyses GVN::run(Function &F, FunctionAnalysisManager &AM) { 661 // FIXME: The order of evaluation of these 'getResult' calls is very 662 // significant! Re-ordering these variables will cause GVN when run alone to 663 // be less effective! We should fix memdep and basic-aa to not exhibit this 664 // behavior, but until then don't change the order here. 665 auto &AC = AM.getResult<AssumptionAnalysis>(F); 666 auto &DT = AM.getResult<DominatorTreeAnalysis>(F); 667 auto &TLI = AM.getResult<TargetLibraryAnalysis>(F); 668 auto &AA = AM.getResult<AAManager>(F); 669 auto *MemDep = 670 isMemDepEnabled() ? &AM.getResult<MemoryDependenceAnalysis>(F) : nullptr; 671 auto *LI = AM.getCachedResult<LoopAnalysis>(F); 672 auto *MSSA = AM.getCachedResult<MemorySSAAnalysis>(F); 673 auto &ORE = AM.getResult<OptimizationRemarkEmitterAnalysis>(F); 674 bool Changed = runImpl(F, AC, DT, TLI, AA, MemDep, LI, &ORE, 675 MSSA ? &MSSA->getMSSA() : nullptr); 676 if (!Changed) 677 return PreservedAnalyses::all(); 678 PreservedAnalyses PA; 679 PA.preserve<DominatorTreeAnalysis>(); 680 PA.preserve<GlobalsAA>(); 681 PA.preserve<TargetLibraryAnalysis>(); 682 if (MSSA) 683 PA.preserve<MemorySSAAnalysis>(); 684 if (LI) 685 PA.preserve<LoopAnalysis>(); 686 return PA; 687 } 688 689 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP) 690 LLVM_DUMP_METHOD void GVN::dump(DenseMap<uint32_t, Value*>& d) const { 691 errs() << "{\n"; 692 for (auto &I : d) { 693 errs() << I.first << "\n"; 694 I.second->dump(); 695 } 696 errs() << "}\n"; 697 } 698 #endif 699 700 enum class AvailabilityState : char { 701 /// We know the block *is not* fully available. This is a fixpoint. 702 Unavailable = 0, 703 /// We know the block *is* fully available. This is a fixpoint. 704 Available = 1, 705 /// We do not know whether the block is fully available or not, 706 /// but we are currently speculating that it will be. 707 /// If it would have turned out that the block was, in fact, not fully 708 /// available, this would have been cleaned up into an Unavailable. 709 SpeculativelyAvailable = 2, 710 }; 711 712 /// Return true if we can prove that the value 713 /// we're analyzing is fully available in the specified block. As we go, keep 714 /// track of which blocks we know are fully alive in FullyAvailableBlocks. This 715 /// map is actually a tri-state map with the following values: 716 /// 0) we know the block *is not* fully available. 717 /// 1) we know the block *is* fully available. 718 /// 2) we do not know whether the block is fully available or not, but we are 719 /// currently speculating that it will be. 720 static bool IsValueFullyAvailableInBlock( 721 BasicBlock *BB, 722 DenseMap<BasicBlock *, AvailabilityState> &FullyAvailableBlocks) { 723 SmallVector<BasicBlock *, 32> Worklist; 724 Optional<BasicBlock *> UnavailableBB; 725 726 // The number of times we didn't find an entry for a block in a map and 727 // optimistically inserted an entry marking block as speculatively available. 728 unsigned NumNewNewSpeculativelyAvailableBBs = 0; 729 730 #ifndef NDEBUG 731 SmallSet<BasicBlock *, 32> NewSpeculativelyAvailableBBs; 732 SmallVector<BasicBlock *, 32> AvailableBBs; 733 #endif 734 735 Worklist.emplace_back(BB); 736 while (!Worklist.empty()) { 737 BasicBlock *CurrBB = Worklist.pop_back_val(); // LoadFO - depth-first! 738 // Optimistically assume that the block is Speculatively Available and check 739 // to see if we already know about this block in one lookup. 740 std::pair<DenseMap<BasicBlock *, AvailabilityState>::iterator, bool> IV = 741 FullyAvailableBlocks.try_emplace( 742 CurrBB, AvailabilityState::SpeculativelyAvailable); 743 AvailabilityState &State = IV.first->second; 744 745 // Did the entry already exist for this block? 746 if (!IV.second) { 747 if (State == AvailabilityState::Unavailable) { 748 UnavailableBB = CurrBB; 749 break; // Backpropagate unavailability info. 750 } 751 752 #ifndef NDEBUG 753 AvailableBBs.emplace_back(CurrBB); 754 #endif 755 continue; // Don't recurse further, but continue processing worklist. 756 } 757 758 // No entry found for block. 759 ++NumNewNewSpeculativelyAvailableBBs; 760 bool OutOfBudget = NumNewNewSpeculativelyAvailableBBs > MaxBBSpeculations; 761 762 // If we have exhausted our budget, mark this block as unavailable. 763 // Also, if this block has no predecessors, the value isn't live-in here. 764 if (OutOfBudget || pred_empty(CurrBB)) { 765 MaxBBSpeculationCutoffReachedTimes += (int)OutOfBudget; 766 State = AvailabilityState::Unavailable; 767 UnavailableBB = CurrBB; 768 break; // Backpropagate unavailability info. 769 } 770 771 // Tentatively consider this block as speculatively available. 772 #ifndef NDEBUG 773 NewSpeculativelyAvailableBBs.insert(CurrBB); 774 #endif 775 // And further recurse into block's predecessors, in depth-first order! 776 Worklist.append(pred_begin(CurrBB), pred_end(CurrBB)); 777 } 778 779 #if LLVM_ENABLE_STATS 780 IsValueFullyAvailableInBlockNumSpeculationsMax.updateMax( 781 NumNewNewSpeculativelyAvailableBBs); 782 #endif 783 784 // If the block isn't marked as fixpoint yet 785 // (the Unavailable and Available states are fixpoints) 786 auto MarkAsFixpointAndEnqueueSuccessors = 787 [&](BasicBlock *BB, AvailabilityState FixpointState) { 788 auto It = FullyAvailableBlocks.find(BB); 789 if (It == FullyAvailableBlocks.end()) 790 return; // Never queried this block, leave as-is. 791 switch (AvailabilityState &State = It->second) { 792 case AvailabilityState::Unavailable: 793 case AvailabilityState::Available: 794 return; // Don't backpropagate further, continue processing worklist. 795 case AvailabilityState::SpeculativelyAvailable: // Fix it! 796 State = FixpointState; 797 #ifndef NDEBUG 798 assert(NewSpeculativelyAvailableBBs.erase(BB) && 799 "Found a speculatively available successor leftover?"); 800 #endif 801 // Queue successors for further processing. 802 Worklist.append(succ_begin(BB), succ_end(BB)); 803 return; 804 } 805 }; 806 807 if (UnavailableBB) { 808 // Okay, we have encountered an unavailable block. 809 // Mark speculatively available blocks reachable from UnavailableBB as 810 // unavailable as well. Paths are terminated when they reach blocks not in 811 // FullyAvailableBlocks or they are not marked as speculatively available. 812 Worklist.clear(); 813 Worklist.append(succ_begin(*UnavailableBB), succ_end(*UnavailableBB)); 814 while (!Worklist.empty()) 815 MarkAsFixpointAndEnqueueSuccessors(Worklist.pop_back_val(), 816 AvailabilityState::Unavailable); 817 } 818 819 #ifndef NDEBUG 820 Worklist.clear(); 821 for (BasicBlock *AvailableBB : AvailableBBs) 822 Worklist.append(succ_begin(AvailableBB), succ_end(AvailableBB)); 823 while (!Worklist.empty()) 824 MarkAsFixpointAndEnqueueSuccessors(Worklist.pop_back_val(), 825 AvailabilityState::Available); 826 827 assert(NewSpeculativelyAvailableBBs.empty() && 828 "Must have fixed all the new speculatively available blocks."); 829 #endif 830 831 return !UnavailableBB; 832 } 833 834 /// Given a set of loads specified by ValuesPerBlock, 835 /// construct SSA form, allowing us to eliminate Load. This returns the value 836 /// that should be used at Load's definition site. 837 static Value * 838 ConstructSSAForLoadSet(LoadInst *Load, 839 SmallVectorImpl<AvailableValueInBlock> &ValuesPerBlock, 840 GVN &gvn) { 841 // Check for the fully redundant, dominating load case. In this case, we can 842 // just use the dominating value directly. 843 if (ValuesPerBlock.size() == 1 && 844 gvn.getDominatorTree().properlyDominates(ValuesPerBlock[0].BB, 845 Load->getParent())) { 846 assert(!ValuesPerBlock[0].AV.isUndefValue() && 847 "Dead BB dominate this block"); 848 return ValuesPerBlock[0].MaterializeAdjustedValue(Load, gvn); 849 } 850 851 // Otherwise, we have to construct SSA form. 852 SmallVector<PHINode*, 8> NewPHIs; 853 SSAUpdater SSAUpdate(&NewPHIs); 854 SSAUpdate.Initialize(Load->getType(), Load->getName()); 855 856 for (const AvailableValueInBlock &AV : ValuesPerBlock) { 857 BasicBlock *BB = AV.BB; 858 859 if (AV.AV.isUndefValue()) 860 continue; 861 862 if (SSAUpdate.HasValueForBlock(BB)) 863 continue; 864 865 // If the value is the load that we will be eliminating, and the block it's 866 // available in is the block that the load is in, then don't add it as 867 // SSAUpdater will resolve the value to the relevant phi which may let it 868 // avoid phi construction entirely if there's actually only one value. 869 if (BB == Load->getParent() && 870 ((AV.AV.isSimpleValue() && AV.AV.getSimpleValue() == Load) || 871 (AV.AV.isCoercedLoadValue() && AV.AV.getCoercedLoadValue() == Load))) 872 continue; 873 874 SSAUpdate.AddAvailableValue(BB, AV.MaterializeAdjustedValue(Load, gvn)); 875 } 876 877 // Perform PHI construction. 878 return SSAUpdate.GetValueInMiddleOfBlock(Load->getParent()); 879 } 880 881 Value *AvailableValue::MaterializeAdjustedValue(LoadInst *Load, 882 Instruction *InsertPt, 883 GVN &gvn) const { 884 Value *Res; 885 Type *LoadTy = Load->getType(); 886 const DataLayout &DL = Load->getModule()->getDataLayout(); 887 if (isSimpleValue()) { 888 Res = getSimpleValue(); 889 if (Res->getType() != LoadTy) { 890 Res = getStoreValueForLoad(Res, Offset, LoadTy, InsertPt, DL); 891 892 LLVM_DEBUG(dbgs() << "GVN COERCED NONLOCAL VAL:\nOffset: " << Offset 893 << " " << *getSimpleValue() << '\n' 894 << *Res << '\n' 895 << "\n\n\n"); 896 } 897 } else if (isCoercedLoadValue()) { 898 LoadInst *Load = getCoercedLoadValue(); 899 if (Load->getType() == LoadTy && Offset == 0) { 900 Res = Load; 901 } else { 902 Res = getLoadValueForLoad(Load, Offset, LoadTy, InsertPt, DL); 903 // We would like to use gvn.markInstructionForDeletion here, but we can't 904 // because the load is already memoized into the leader map table that GVN 905 // tracks. It is potentially possible to remove the load from the table, 906 // but then there all of the operations based on it would need to be 907 // rehashed. Just leave the dead load around. 908 gvn.getMemDep().removeInstruction(Load); 909 LLVM_DEBUG(dbgs() << "GVN COERCED NONLOCAL LOAD:\nOffset: " << Offset 910 << " " << *getCoercedLoadValue() << '\n' 911 << *Res << '\n' 912 << "\n\n\n"); 913 } 914 } else if (isMemIntrinValue()) { 915 Res = getMemInstValueForLoad(getMemIntrinValue(), Offset, LoadTy, 916 InsertPt, DL); 917 LLVM_DEBUG(dbgs() << "GVN COERCED NONLOCAL MEM INTRIN:\nOffset: " << Offset 918 << " " << *getMemIntrinValue() << '\n' 919 << *Res << '\n' 920 << "\n\n\n"); 921 } else { 922 llvm_unreachable("Should not materialize value from dead block"); 923 } 924 assert(Res && "failed to materialize?"); 925 return Res; 926 } 927 928 static bool isLifetimeStart(const Instruction *Inst) { 929 if (const IntrinsicInst* II = dyn_cast<IntrinsicInst>(Inst)) 930 return II->getIntrinsicID() == Intrinsic::lifetime_start; 931 return false; 932 } 933 934 /// Try to locate the three instruction involved in a missed 935 /// load-elimination case that is due to an intervening store. 936 static void reportMayClobberedLoad(LoadInst *Load, MemDepResult DepInfo, 937 DominatorTree *DT, 938 OptimizationRemarkEmitter *ORE) { 939 using namespace ore; 940 941 User *OtherAccess = nullptr; 942 943 OptimizationRemarkMissed R(DEBUG_TYPE, "LoadClobbered", Load); 944 R << "load of type " << NV("Type", Load->getType()) << " not eliminated" 945 << setExtraArgs(); 946 947 for (auto *U : Load->getPointerOperand()->users()) 948 if (U != Load && (isa<LoadInst>(U) || isa<StoreInst>(U)) && 949 DT->dominates(cast<Instruction>(U), Load)) { 950 // FIXME: for now give up if there are multiple memory accesses that 951 // dominate the load. We need further analysis to decide which one is 952 // that we're forwarding from. 953 if (OtherAccess) 954 OtherAccess = nullptr; 955 else 956 OtherAccess = U; 957 } 958 959 if (OtherAccess) 960 R << " in favor of " << NV("OtherAccess", OtherAccess); 961 962 R << " because it is clobbered by " << NV("ClobberedBy", DepInfo.getInst()); 963 964 ORE->emit(R); 965 } 966 967 bool GVN::AnalyzeLoadAvailability(LoadInst *Load, MemDepResult DepInfo, 968 Value *Address, AvailableValue &Res) { 969 assert((DepInfo.isDef() || DepInfo.isClobber()) && 970 "expected a local dependence"); 971 assert(Load->isUnordered() && "rules below are incorrect for ordered access"); 972 973 const DataLayout &DL = Load->getModule()->getDataLayout(); 974 975 Instruction *DepInst = DepInfo.getInst(); 976 if (DepInfo.isClobber()) { 977 // If the dependence is to a store that writes to a superset of the bits 978 // read by the load, we can extract the bits we need for the load from the 979 // stored value. 980 if (StoreInst *DepSI = dyn_cast<StoreInst>(DepInst)) { 981 // Can't forward from non-atomic to atomic without violating memory model. 982 if (Address && Load->isAtomic() <= DepSI->isAtomic()) { 983 int Offset = 984 analyzeLoadFromClobberingStore(Load->getType(), Address, DepSI, DL); 985 if (Offset != -1) { 986 Res = AvailableValue::get(DepSI->getValueOperand(), Offset); 987 return true; 988 } 989 } 990 } 991 992 // Check to see if we have something like this: 993 // load i32* P 994 // load i8* (P+1) 995 // if we have this, replace the later with an extraction from the former. 996 if (LoadInst *DepLoad = dyn_cast<LoadInst>(DepInst)) { 997 // If this is a clobber and L is the first instruction in its block, then 998 // we have the first instruction in the entry block. 999 // Can't forward from non-atomic to atomic without violating memory model. 1000 if (DepLoad != Load && Address && 1001 Load->isAtomic() <= DepLoad->isAtomic()) { 1002 int Offset = analyzeLoadFromClobberingLoad(Load->getType(), Address, 1003 DepLoad, DL); 1004 1005 if (Offset != -1) { 1006 Res = AvailableValue::getLoad(DepLoad, Offset); 1007 return true; 1008 } 1009 } 1010 } 1011 1012 // If the clobbering value is a memset/memcpy/memmove, see if we can 1013 // forward a value on from it. 1014 if (MemIntrinsic *DepMI = dyn_cast<MemIntrinsic>(DepInst)) { 1015 if (Address && !Load->isAtomic()) { 1016 int Offset = analyzeLoadFromClobberingMemInst(Load->getType(), Address, 1017 DepMI, DL); 1018 if (Offset != -1) { 1019 Res = AvailableValue::getMI(DepMI, Offset); 1020 return true; 1021 } 1022 } 1023 } 1024 // Nothing known about this clobber, have to be conservative 1025 LLVM_DEBUG( 1026 // fast print dep, using operator<< on instruction is too slow. 1027 dbgs() << "GVN: load "; Load->printAsOperand(dbgs()); 1028 dbgs() << " is clobbered by " << *DepInst << '\n';); 1029 if (ORE->allowExtraAnalysis(DEBUG_TYPE)) 1030 reportMayClobberedLoad(Load, DepInfo, DT, ORE); 1031 1032 return false; 1033 } 1034 assert(DepInfo.isDef() && "follows from above"); 1035 1036 // Loading the allocation -> undef. 1037 if (isa<AllocaInst>(DepInst) || isMallocLikeFn(DepInst, TLI) || 1038 isAlignedAllocLikeFn(DepInst, TLI) || 1039 // Loading immediately after lifetime begin -> undef. 1040 isLifetimeStart(DepInst)) { 1041 Res = AvailableValue::get(UndefValue::get(Load->getType())); 1042 return true; 1043 } 1044 1045 // Loading from calloc (which zero initializes memory) -> zero 1046 if (isCallocLikeFn(DepInst, TLI)) { 1047 Res = AvailableValue::get(Constant::getNullValue(Load->getType())); 1048 return true; 1049 } 1050 1051 if (StoreInst *S = dyn_cast<StoreInst>(DepInst)) { 1052 // Reject loads and stores that are to the same address but are of 1053 // different types if we have to. If the stored value is convertable to 1054 // the loaded value, we can reuse it. 1055 if (!canCoerceMustAliasedValueToLoad(S->getValueOperand(), Load->getType(), 1056 DL)) 1057 return false; 1058 1059 // Can't forward from non-atomic to atomic without violating memory model. 1060 if (S->isAtomic() < Load->isAtomic()) 1061 return false; 1062 1063 Res = AvailableValue::get(S->getValueOperand()); 1064 return true; 1065 } 1066 1067 if (LoadInst *LD = dyn_cast<LoadInst>(DepInst)) { 1068 // If the types mismatch and we can't handle it, reject reuse of the load. 1069 // If the stored value is larger or equal to the loaded value, we can reuse 1070 // it. 1071 if (!canCoerceMustAliasedValueToLoad(LD, Load->getType(), DL)) 1072 return false; 1073 1074 // Can't forward from non-atomic to atomic without violating memory model. 1075 if (LD->isAtomic() < Load->isAtomic()) 1076 return false; 1077 1078 Res = AvailableValue::getLoad(LD); 1079 return true; 1080 } 1081 1082 // Unknown def - must be conservative 1083 LLVM_DEBUG( 1084 // fast print dep, using operator<< on instruction is too slow. 1085 dbgs() << "GVN: load "; Load->printAsOperand(dbgs()); 1086 dbgs() << " has unknown def " << *DepInst << '\n';); 1087 return false; 1088 } 1089 1090 void GVN::AnalyzeLoadAvailability(LoadInst *Load, LoadDepVect &Deps, 1091 AvailValInBlkVect &ValuesPerBlock, 1092 UnavailBlkVect &UnavailableBlocks) { 1093 // Filter out useless results (non-locals, etc). Keep track of the blocks 1094 // where we have a value available in repl, also keep track of whether we see 1095 // dependencies that produce an unknown value for the load (such as a call 1096 // that could potentially clobber the load). 1097 unsigned NumDeps = Deps.size(); 1098 for (unsigned i = 0, e = NumDeps; i != e; ++i) { 1099 BasicBlock *DepBB = Deps[i].getBB(); 1100 MemDepResult DepInfo = Deps[i].getResult(); 1101 1102 if (DeadBlocks.count(DepBB)) { 1103 // Dead dependent mem-op disguise as a load evaluating the same value 1104 // as the load in question. 1105 ValuesPerBlock.push_back(AvailableValueInBlock::getUndef(DepBB)); 1106 continue; 1107 } 1108 1109 if (!DepInfo.isDef() && !DepInfo.isClobber()) { 1110 UnavailableBlocks.push_back(DepBB); 1111 continue; 1112 } 1113 1114 // The address being loaded in this non-local block may not be the same as 1115 // the pointer operand of the load if PHI translation occurs. Make sure 1116 // to consider the right address. 1117 Value *Address = Deps[i].getAddress(); 1118 1119 AvailableValue AV; 1120 if (AnalyzeLoadAvailability(Load, DepInfo, Address, AV)) { 1121 // subtlety: because we know this was a non-local dependency, we know 1122 // it's safe to materialize anywhere between the instruction within 1123 // DepInfo and the end of it's block. 1124 ValuesPerBlock.push_back(AvailableValueInBlock::get(DepBB, 1125 std::move(AV))); 1126 } else { 1127 UnavailableBlocks.push_back(DepBB); 1128 } 1129 } 1130 1131 assert(NumDeps == ValuesPerBlock.size() + UnavailableBlocks.size() && 1132 "post condition violation"); 1133 } 1134 1135 void GVN::eliminatePartiallyRedundantLoad( 1136 LoadInst *Load, AvailValInBlkVect &ValuesPerBlock, 1137 MapVector<BasicBlock *, Value *> &AvailableLoads) { 1138 for (const auto &AvailableLoad : AvailableLoads) { 1139 BasicBlock *UnavailableBlock = AvailableLoad.first; 1140 Value *LoadPtr = AvailableLoad.second; 1141 1142 auto *NewLoad = 1143 new LoadInst(Load->getType(), LoadPtr, Load->getName() + ".pre", 1144 Load->isVolatile(), Load->getAlign(), Load->getOrdering(), 1145 Load->getSyncScopeID(), UnavailableBlock->getTerminator()); 1146 NewLoad->setDebugLoc(Load->getDebugLoc()); 1147 if (MSSAU) { 1148 auto *MSSA = MSSAU->getMemorySSA(); 1149 // Get the defining access of the original load or use the load if it is a 1150 // MemoryDef (e.g. because it is volatile). The inserted loads are 1151 // guaranteed to load from the same definition. 1152 auto *LoadAcc = MSSA->getMemoryAccess(Load); 1153 auto *DefiningAcc = 1154 isa<MemoryDef>(LoadAcc) ? LoadAcc : LoadAcc->getDefiningAccess(); 1155 auto *NewAccess = MSSAU->createMemoryAccessInBB( 1156 NewLoad, DefiningAcc, NewLoad->getParent(), 1157 MemorySSA::BeforeTerminator); 1158 if (auto *NewDef = dyn_cast<MemoryDef>(NewAccess)) 1159 MSSAU->insertDef(NewDef, /*RenameUses=*/true); 1160 else 1161 MSSAU->insertUse(cast<MemoryUse>(NewAccess), /*RenameUses=*/true); 1162 } 1163 1164 // Transfer the old load's AA tags to the new load. 1165 AAMDNodes Tags; 1166 Load->getAAMetadata(Tags); 1167 if (Tags) 1168 NewLoad->setAAMetadata(Tags); 1169 1170 if (auto *MD = Load->getMetadata(LLVMContext::MD_invariant_load)) 1171 NewLoad->setMetadata(LLVMContext::MD_invariant_load, MD); 1172 if (auto *InvGroupMD = Load->getMetadata(LLVMContext::MD_invariant_group)) 1173 NewLoad->setMetadata(LLVMContext::MD_invariant_group, InvGroupMD); 1174 if (auto *RangeMD = Load->getMetadata(LLVMContext::MD_range)) 1175 NewLoad->setMetadata(LLVMContext::MD_range, RangeMD); 1176 if (auto *AccessMD = Load->getMetadata(LLVMContext::MD_access_group)) 1177 if (LI && 1178 LI->getLoopFor(Load->getParent()) == LI->getLoopFor(UnavailableBlock)) 1179 NewLoad->setMetadata(LLVMContext::MD_access_group, AccessMD); 1180 1181 // We do not propagate the old load's debug location, because the new 1182 // load now lives in a different BB, and we want to avoid a jumpy line 1183 // table. 1184 // FIXME: How do we retain source locations without causing poor debugging 1185 // behavior? 1186 1187 // Add the newly created load. 1188 ValuesPerBlock.push_back( 1189 AvailableValueInBlock::get(UnavailableBlock, NewLoad)); 1190 MD->invalidateCachedPointerInfo(LoadPtr); 1191 LLVM_DEBUG(dbgs() << "GVN INSERTED " << *NewLoad << '\n'); 1192 } 1193 1194 // Perform PHI construction. 1195 Value *V = ConstructSSAForLoadSet(Load, ValuesPerBlock, *this); 1196 Load->replaceAllUsesWith(V); 1197 if (isa<PHINode>(V)) 1198 V->takeName(Load); 1199 if (Instruction *I = dyn_cast<Instruction>(V)) 1200 I->setDebugLoc(Load->getDebugLoc()); 1201 if (V->getType()->isPtrOrPtrVectorTy()) 1202 MD->invalidateCachedPointerInfo(V); 1203 markInstructionForDeletion(Load); 1204 ORE->emit([&]() { 1205 return OptimizationRemark(DEBUG_TYPE, "LoadPRE", Load) 1206 << "load eliminated by PRE"; 1207 }); 1208 } 1209 1210 bool GVN::PerformLoadPRE(LoadInst *Load, AvailValInBlkVect &ValuesPerBlock, 1211 UnavailBlkVect &UnavailableBlocks) { 1212 // Okay, we have *some* definitions of the value. This means that the value 1213 // is available in some of our (transitive) predecessors. Lets think about 1214 // doing PRE of this load. This will involve inserting a new load into the 1215 // predecessor when it's not available. We could do this in general, but 1216 // prefer to not increase code size. As such, we only do this when we know 1217 // that we only have to insert *one* load (which means we're basically moving 1218 // the load, not inserting a new one). 1219 1220 SmallPtrSet<BasicBlock *, 4> Blockers(UnavailableBlocks.begin(), 1221 UnavailableBlocks.end()); 1222 1223 // Let's find the first basic block with more than one predecessor. Walk 1224 // backwards through predecessors if needed. 1225 BasicBlock *LoadBB = Load->getParent(); 1226 BasicBlock *TmpBB = LoadBB; 1227 1228 // Check that there is no implicit control flow instructions above our load in 1229 // its block. If there is an instruction that doesn't always pass the 1230 // execution to the following instruction, then moving through it may become 1231 // invalid. For example: 1232 // 1233 // int arr[LEN]; 1234 // int index = ???; 1235 // ... 1236 // guard(0 <= index && index < LEN); 1237 // use(arr[index]); 1238 // 1239 // It is illegal to move the array access to any point above the guard, 1240 // because if the index is out of bounds we should deoptimize rather than 1241 // access the array. 1242 // Check that there is no guard in this block above our instruction. 1243 bool MustEnsureSafetyOfSpeculativeExecution = 1244 ICF->isDominatedByICFIFromSameBlock(Load); 1245 1246 while (TmpBB->getSinglePredecessor()) { 1247 TmpBB = TmpBB->getSinglePredecessor(); 1248 if (TmpBB == LoadBB) // Infinite (unreachable) loop. 1249 return false; 1250 if (Blockers.count(TmpBB)) 1251 return false; 1252 1253 // If any of these blocks has more than one successor (i.e. if the edge we 1254 // just traversed was critical), then there are other paths through this 1255 // block along which the load may not be anticipated. Hoisting the load 1256 // above this block would be adding the load to execution paths along 1257 // which it was not previously executed. 1258 if (TmpBB->getTerminator()->getNumSuccessors() != 1) 1259 return false; 1260 1261 // Check that there is no implicit control flow in a block above. 1262 MustEnsureSafetyOfSpeculativeExecution = 1263 MustEnsureSafetyOfSpeculativeExecution || ICF->hasICF(TmpBB); 1264 } 1265 1266 assert(TmpBB); 1267 LoadBB = TmpBB; 1268 1269 // Check to see how many predecessors have the loaded value fully 1270 // available. 1271 MapVector<BasicBlock *, Value *> PredLoads; 1272 DenseMap<BasicBlock *, AvailabilityState> FullyAvailableBlocks; 1273 for (const AvailableValueInBlock &AV : ValuesPerBlock) 1274 FullyAvailableBlocks[AV.BB] = AvailabilityState::Available; 1275 for (BasicBlock *UnavailableBB : UnavailableBlocks) 1276 FullyAvailableBlocks[UnavailableBB] = AvailabilityState::Unavailable; 1277 1278 SmallVector<BasicBlock *, 4> CriticalEdgePred; 1279 for (BasicBlock *Pred : predecessors(LoadBB)) { 1280 // If any predecessor block is an EH pad that does not allow non-PHI 1281 // instructions before the terminator, we can't PRE the load. 1282 if (Pred->getTerminator()->isEHPad()) { 1283 LLVM_DEBUG( 1284 dbgs() << "COULD NOT PRE LOAD BECAUSE OF AN EH PAD PREDECESSOR '" 1285 << Pred->getName() << "': " << *Load << '\n'); 1286 return false; 1287 } 1288 1289 if (IsValueFullyAvailableInBlock(Pred, FullyAvailableBlocks)) { 1290 continue; 1291 } 1292 1293 if (Pred->getTerminator()->getNumSuccessors() != 1) { 1294 if (isa<IndirectBrInst>(Pred->getTerminator())) { 1295 LLVM_DEBUG( 1296 dbgs() << "COULD NOT PRE LOAD BECAUSE OF INDBR CRITICAL EDGE '" 1297 << Pred->getName() << "': " << *Load << '\n'); 1298 return false; 1299 } 1300 1301 // FIXME: Can we support the fallthrough edge? 1302 if (isa<CallBrInst>(Pred->getTerminator())) { 1303 LLVM_DEBUG( 1304 dbgs() << "COULD NOT PRE LOAD BECAUSE OF CALLBR CRITICAL EDGE '" 1305 << Pred->getName() << "': " << *Load << '\n'); 1306 return false; 1307 } 1308 1309 if (LoadBB->isEHPad()) { 1310 LLVM_DEBUG( 1311 dbgs() << "COULD NOT PRE LOAD BECAUSE OF AN EH PAD CRITICAL EDGE '" 1312 << Pred->getName() << "': " << *Load << '\n'); 1313 return false; 1314 } 1315 1316 // Do not split backedge as it will break the canonical loop form. 1317 if (!isLoadPRESplitBackedgeEnabled()) 1318 if (DT->dominates(LoadBB, Pred)) { 1319 LLVM_DEBUG( 1320 dbgs() 1321 << "COULD NOT PRE LOAD BECAUSE OF A BACKEDGE CRITICAL EDGE '" 1322 << Pred->getName() << "': " << *Load << '\n'); 1323 return false; 1324 } 1325 1326 CriticalEdgePred.push_back(Pred); 1327 } else { 1328 // Only add the predecessors that will not be split for now. 1329 PredLoads[Pred] = nullptr; 1330 } 1331 } 1332 1333 // Decide whether PRE is profitable for this load. 1334 unsigned NumUnavailablePreds = PredLoads.size() + CriticalEdgePred.size(); 1335 assert(NumUnavailablePreds != 0 && 1336 "Fully available value should already be eliminated!"); 1337 1338 // If this load is unavailable in multiple predecessors, reject it. 1339 // FIXME: If we could restructure the CFG, we could make a common pred with 1340 // all the preds that don't have an available Load and insert a new load into 1341 // that one block. 1342 if (NumUnavailablePreds != 1) 1343 return false; 1344 1345 // Now we know where we will insert load. We must ensure that it is safe 1346 // to speculatively execute the load at that points. 1347 if (MustEnsureSafetyOfSpeculativeExecution) { 1348 if (CriticalEdgePred.size()) 1349 if (!isSafeToSpeculativelyExecute(Load, LoadBB->getFirstNonPHI(), DT)) 1350 return false; 1351 for (auto &PL : PredLoads) 1352 if (!isSafeToSpeculativelyExecute(Load, PL.first->getTerminator(), DT)) 1353 return false; 1354 } 1355 1356 // Split critical edges, and update the unavailable predecessors accordingly. 1357 for (BasicBlock *OrigPred : CriticalEdgePred) { 1358 BasicBlock *NewPred = splitCriticalEdges(OrigPred, LoadBB); 1359 assert(!PredLoads.count(OrigPred) && "Split edges shouldn't be in map!"); 1360 PredLoads[NewPred] = nullptr; 1361 LLVM_DEBUG(dbgs() << "Split critical edge " << OrigPred->getName() << "->" 1362 << LoadBB->getName() << '\n'); 1363 } 1364 1365 // Check if the load can safely be moved to all the unavailable predecessors. 1366 bool CanDoPRE = true; 1367 const DataLayout &DL = Load->getModule()->getDataLayout(); 1368 SmallVector<Instruction*, 8> NewInsts; 1369 for (auto &PredLoad : PredLoads) { 1370 BasicBlock *UnavailablePred = PredLoad.first; 1371 1372 // Do PHI translation to get its value in the predecessor if necessary. The 1373 // returned pointer (if non-null) is guaranteed to dominate UnavailablePred. 1374 // We do the translation for each edge we skipped by going from Load's block 1375 // to LoadBB, otherwise we might miss pieces needing translation. 1376 1377 // If all preds have a single successor, then we know it is safe to insert 1378 // the load on the pred (?!?), so we can insert code to materialize the 1379 // pointer if it is not available. 1380 Value *LoadPtr = Load->getPointerOperand(); 1381 BasicBlock *Cur = Load->getParent(); 1382 while (Cur != LoadBB) { 1383 PHITransAddr Address(LoadPtr, DL, AC); 1384 LoadPtr = Address.PHITranslateWithInsertion( 1385 Cur, Cur->getSinglePredecessor(), *DT, NewInsts); 1386 if (!LoadPtr) { 1387 CanDoPRE = false; 1388 break; 1389 } 1390 Cur = Cur->getSinglePredecessor(); 1391 } 1392 1393 if (LoadPtr) { 1394 PHITransAddr Address(LoadPtr, DL, AC); 1395 LoadPtr = Address.PHITranslateWithInsertion(LoadBB, UnavailablePred, *DT, 1396 NewInsts); 1397 } 1398 // If we couldn't find or insert a computation of this phi translated value, 1399 // we fail PRE. 1400 if (!LoadPtr) { 1401 LLVM_DEBUG(dbgs() << "COULDN'T INSERT PHI TRANSLATED VALUE OF: " 1402 << *Load->getPointerOperand() << "\n"); 1403 CanDoPRE = false; 1404 break; 1405 } 1406 1407 PredLoad.second = LoadPtr; 1408 } 1409 1410 if (!CanDoPRE) { 1411 while (!NewInsts.empty()) { 1412 // Erase instructions generated by the failed PHI translation before 1413 // trying to number them. PHI translation might insert instructions 1414 // in basic blocks other than the current one, and we delete them 1415 // directly, as markInstructionForDeletion only allows removing from the 1416 // current basic block. 1417 NewInsts.pop_back_val()->eraseFromParent(); 1418 } 1419 // HINT: Don't revert the edge-splitting as following transformation may 1420 // also need to split these critical edges. 1421 return !CriticalEdgePred.empty(); 1422 } 1423 1424 // Okay, we can eliminate this load by inserting a reload in the predecessor 1425 // and using PHI construction to get the value in the other predecessors, do 1426 // it. 1427 LLVM_DEBUG(dbgs() << "GVN REMOVING PRE LOAD: " << *Load << '\n'); 1428 LLVM_DEBUG(if (!NewInsts.empty()) dbgs() << "INSERTED " << NewInsts.size() 1429 << " INSTS: " << *NewInsts.back() 1430 << '\n'); 1431 1432 // Assign value numbers to the new instructions. 1433 for (Instruction *I : NewInsts) { 1434 // Instructions that have been inserted in predecessor(s) to materialize 1435 // the load address do not retain their original debug locations. Doing 1436 // so could lead to confusing (but correct) source attributions. 1437 I->updateLocationAfterHoist(); 1438 1439 // FIXME: We really _ought_ to insert these value numbers into their 1440 // parent's availability map. However, in doing so, we risk getting into 1441 // ordering issues. If a block hasn't been processed yet, we would be 1442 // marking a value as AVAIL-IN, which isn't what we intend. 1443 VN.lookupOrAdd(I); 1444 } 1445 1446 eliminatePartiallyRedundantLoad(Load, ValuesPerBlock, PredLoads); 1447 ++NumPRELoad; 1448 return true; 1449 } 1450 1451 bool GVN::performLoopLoadPRE(LoadInst *Load, AvailValInBlkVect &ValuesPerBlock, 1452 UnavailBlkVect &UnavailableBlocks) { 1453 if (!LI) 1454 return false; 1455 1456 const Loop *L = LI->getLoopFor(Load->getParent()); 1457 // TODO: Generalize to other loop blocks that dominate the latch. 1458 if (!L || L->getHeader() != Load->getParent()) 1459 return false; 1460 1461 BasicBlock *Preheader = L->getLoopPreheader(); 1462 BasicBlock *Latch = L->getLoopLatch(); 1463 if (!Preheader || !Latch) 1464 return false; 1465 1466 Value *LoadPtr = Load->getPointerOperand(); 1467 // Must be available in preheader. 1468 if (!L->isLoopInvariant(LoadPtr)) 1469 return false; 1470 1471 // We plan to hoist the load to preheader without introducing a new fault. 1472 // In order to do it, we need to prove that we cannot side-exit the loop 1473 // once loop header is first entered before execution of the load. 1474 if (ICF->isDominatedByICFIFromSameBlock(Load)) 1475 return false; 1476 1477 BasicBlock *LoopBlock = nullptr; 1478 for (auto *Blocker : UnavailableBlocks) { 1479 // Blockers from outside the loop are handled in preheader. 1480 if (!L->contains(Blocker)) 1481 continue; 1482 1483 // Only allow one loop block. Loop header is not less frequently executed 1484 // than each loop block, and likely it is much more frequently executed. But 1485 // in case of multiple loop blocks, we need extra information (such as block 1486 // frequency info) to understand whether it is profitable to PRE into 1487 // multiple loop blocks. 1488 if (LoopBlock) 1489 return false; 1490 1491 // Do not sink into inner loops. This may be non-profitable. 1492 if (L != LI->getLoopFor(Blocker)) 1493 return false; 1494 1495 // Blocks that dominate the latch execute on every single iteration, maybe 1496 // except the last one. So PREing into these blocks doesn't make much sense 1497 // in most cases. But the blocks that do not necessarily execute on each 1498 // iteration are sometimes much colder than the header, and this is when 1499 // PRE is potentially profitable. 1500 if (DT->dominates(Blocker, Latch)) 1501 return false; 1502 1503 // Make sure that the terminator itself doesn't clobber. 1504 if (Blocker->getTerminator()->mayWriteToMemory()) 1505 return false; 1506 1507 LoopBlock = Blocker; 1508 } 1509 1510 if (!LoopBlock) 1511 return false; 1512 1513 // Make sure the memory at this pointer cannot be freed, therefore we can 1514 // safely reload from it after clobber. 1515 if (LoadPtr->canBeFreed()) 1516 return false; 1517 1518 // TODO: Support critical edge splitting if blocker has more than 1 successor. 1519 MapVector<BasicBlock *, Value *> AvailableLoads; 1520 AvailableLoads[LoopBlock] = LoadPtr; 1521 AvailableLoads[Preheader] = LoadPtr; 1522 1523 LLVM_DEBUG(dbgs() << "GVN REMOVING PRE LOOP LOAD: " << *Load << '\n'); 1524 eliminatePartiallyRedundantLoad(Load, ValuesPerBlock, AvailableLoads); 1525 ++NumPRELoopLoad; 1526 return true; 1527 } 1528 1529 static void reportLoadElim(LoadInst *Load, Value *AvailableValue, 1530 OptimizationRemarkEmitter *ORE) { 1531 using namespace ore; 1532 1533 ORE->emit([&]() { 1534 return OptimizationRemark(DEBUG_TYPE, "LoadElim", Load) 1535 << "load of type " << NV("Type", Load->getType()) << " eliminated" 1536 << setExtraArgs() << " in favor of " 1537 << NV("InfavorOfValue", AvailableValue); 1538 }); 1539 } 1540 1541 /// Attempt to eliminate a load whose dependencies are 1542 /// non-local by performing PHI construction. 1543 bool GVN::processNonLocalLoad(LoadInst *Load) { 1544 // non-local speculations are not allowed under asan. 1545 if (Load->getParent()->getParent()->hasFnAttribute( 1546 Attribute::SanitizeAddress) || 1547 Load->getParent()->getParent()->hasFnAttribute( 1548 Attribute::SanitizeHWAddress)) 1549 return false; 1550 1551 // Step 1: Find the non-local dependencies of the load. 1552 LoadDepVect Deps; 1553 MD->getNonLocalPointerDependency(Load, Deps); 1554 1555 // If we had to process more than one hundred blocks to find the 1556 // dependencies, this load isn't worth worrying about. Optimizing 1557 // it will be too expensive. 1558 unsigned NumDeps = Deps.size(); 1559 if (NumDeps > MaxNumDeps) 1560 return false; 1561 1562 // If we had a phi translation failure, we'll have a single entry which is a 1563 // clobber in the current block. Reject this early. 1564 if (NumDeps == 1 && 1565 !Deps[0].getResult().isDef() && !Deps[0].getResult().isClobber()) { 1566 LLVM_DEBUG(dbgs() << "GVN: non-local load "; Load->printAsOperand(dbgs()); 1567 dbgs() << " has unknown dependencies\n";); 1568 return false; 1569 } 1570 1571 bool Changed = false; 1572 // If this load follows a GEP, see if we can PRE the indices before analyzing. 1573 if (GetElementPtrInst *GEP = 1574 dyn_cast<GetElementPtrInst>(Load->getOperand(0))) { 1575 for (GetElementPtrInst::op_iterator OI = GEP->idx_begin(), 1576 OE = GEP->idx_end(); 1577 OI != OE; ++OI) 1578 if (Instruction *I = dyn_cast<Instruction>(OI->get())) 1579 Changed |= performScalarPRE(I); 1580 } 1581 1582 // Step 2: Analyze the availability of the load 1583 AvailValInBlkVect ValuesPerBlock; 1584 UnavailBlkVect UnavailableBlocks; 1585 AnalyzeLoadAvailability(Load, Deps, ValuesPerBlock, UnavailableBlocks); 1586 1587 // If we have no predecessors that produce a known value for this load, exit 1588 // early. 1589 if (ValuesPerBlock.empty()) 1590 return Changed; 1591 1592 // Step 3: Eliminate fully redundancy. 1593 // 1594 // If all of the instructions we depend on produce a known value for this 1595 // load, then it is fully redundant and we can use PHI insertion to compute 1596 // its value. Insert PHIs and remove the fully redundant value now. 1597 if (UnavailableBlocks.empty()) { 1598 LLVM_DEBUG(dbgs() << "GVN REMOVING NONLOCAL LOAD: " << *Load << '\n'); 1599 1600 // Perform PHI construction. 1601 Value *V = ConstructSSAForLoadSet(Load, ValuesPerBlock, *this); 1602 Load->replaceAllUsesWith(V); 1603 1604 if (isa<PHINode>(V)) 1605 V->takeName(Load); 1606 if (Instruction *I = dyn_cast<Instruction>(V)) 1607 // If instruction I has debug info, then we should not update it. 1608 // Also, if I has a null DebugLoc, then it is still potentially incorrect 1609 // to propagate Load's DebugLoc because Load may not post-dominate I. 1610 if (Load->getDebugLoc() && Load->getParent() == I->getParent()) 1611 I->setDebugLoc(Load->getDebugLoc()); 1612 if (V->getType()->isPtrOrPtrVectorTy()) 1613 MD->invalidateCachedPointerInfo(V); 1614 markInstructionForDeletion(Load); 1615 ++NumGVNLoad; 1616 reportLoadElim(Load, V, ORE); 1617 return true; 1618 } 1619 1620 // Step 4: Eliminate partial redundancy. 1621 if (!isPREEnabled() || !isLoadPREEnabled()) 1622 return Changed; 1623 if (!isLoadInLoopPREEnabled() && LI && LI->getLoopFor(Load->getParent())) 1624 return Changed; 1625 1626 return Changed || PerformLoadPRE(Load, ValuesPerBlock, UnavailableBlocks) || 1627 performLoopLoadPRE(Load, ValuesPerBlock, UnavailableBlocks); 1628 } 1629 1630 static bool impliesEquivalanceIfTrue(CmpInst* Cmp) { 1631 if (Cmp->getPredicate() == CmpInst::Predicate::ICMP_EQ) 1632 return true; 1633 1634 // Floating point comparisons can be equal, but not equivalent. Cases: 1635 // NaNs for unordered operators 1636 // +0.0 vs 0.0 for all operators 1637 if (Cmp->getPredicate() == CmpInst::Predicate::FCMP_OEQ || 1638 (Cmp->getPredicate() == CmpInst::Predicate::FCMP_UEQ && 1639 Cmp->getFastMathFlags().noNaNs())) { 1640 Value *LHS = Cmp->getOperand(0); 1641 Value *RHS = Cmp->getOperand(1); 1642 // If we can prove either side non-zero, then equality must imply 1643 // equivalence. 1644 // FIXME: We should do this optimization if 'no signed zeros' is 1645 // applicable via an instruction-level fast-math-flag or some other 1646 // indicator that relaxed FP semantics are being used. 1647 if (isa<ConstantFP>(LHS) && !cast<ConstantFP>(LHS)->isZero()) 1648 return true; 1649 if (isa<ConstantFP>(RHS) && !cast<ConstantFP>(RHS)->isZero()) 1650 return true;; 1651 // TODO: Handle vector floating point constants 1652 } 1653 return false; 1654 } 1655 1656 static bool impliesEquivalanceIfFalse(CmpInst* Cmp) { 1657 if (Cmp->getPredicate() == CmpInst::Predicate::ICMP_NE) 1658 return true; 1659 1660 // Floating point comparisons can be equal, but not equivelent. Cases: 1661 // NaNs for unordered operators 1662 // +0.0 vs 0.0 for all operators 1663 if ((Cmp->getPredicate() == CmpInst::Predicate::FCMP_ONE && 1664 Cmp->getFastMathFlags().noNaNs()) || 1665 Cmp->getPredicate() == CmpInst::Predicate::FCMP_UNE) { 1666 Value *LHS = Cmp->getOperand(0); 1667 Value *RHS = Cmp->getOperand(1); 1668 // If we can prove either side non-zero, then equality must imply 1669 // equivalence. 1670 // FIXME: We should do this optimization if 'no signed zeros' is 1671 // applicable via an instruction-level fast-math-flag or some other 1672 // indicator that relaxed FP semantics are being used. 1673 if (isa<ConstantFP>(LHS) && !cast<ConstantFP>(LHS)->isZero()) 1674 return true; 1675 if (isa<ConstantFP>(RHS) && !cast<ConstantFP>(RHS)->isZero()) 1676 return true;; 1677 // TODO: Handle vector floating point constants 1678 } 1679 return false; 1680 } 1681 1682 1683 static bool hasUsersIn(Value *V, BasicBlock *BB) { 1684 for (User *U : V->users()) 1685 if (isa<Instruction>(U) && 1686 cast<Instruction>(U)->getParent() == BB) 1687 return true; 1688 return false; 1689 } 1690 1691 bool GVN::processAssumeIntrinsic(AssumeInst *IntrinsicI) { 1692 Value *V = IntrinsicI->getArgOperand(0); 1693 1694 if (ConstantInt *Cond = dyn_cast<ConstantInt>(V)) { 1695 if (Cond->isZero()) { 1696 Type *Int8Ty = Type::getInt8Ty(V->getContext()); 1697 // Insert a new store to null instruction before the load to indicate that 1698 // this code is not reachable. FIXME: We could insert unreachable 1699 // instruction directly because we can modify the CFG. 1700 auto *NewS = new StoreInst(UndefValue::get(Int8Ty), 1701 Constant::getNullValue(Int8Ty->getPointerTo()), 1702 IntrinsicI); 1703 if (MSSAU) { 1704 const MemoryUseOrDef *FirstNonDom = nullptr; 1705 const auto *AL = 1706 MSSAU->getMemorySSA()->getBlockAccesses(IntrinsicI->getParent()); 1707 1708 // If there are accesses in the current basic block, find the first one 1709 // that does not come before NewS. The new memory access is inserted 1710 // after the found access or before the terminator if no such access is 1711 // found. 1712 if (AL) { 1713 for (auto &Acc : *AL) { 1714 if (auto *Current = dyn_cast<MemoryUseOrDef>(&Acc)) 1715 if (!Current->getMemoryInst()->comesBefore(NewS)) { 1716 FirstNonDom = Current; 1717 break; 1718 } 1719 } 1720 } 1721 1722 // This added store is to null, so it will never executed and we can 1723 // just use the LiveOnEntry def as defining access. 1724 auto *NewDef = 1725 FirstNonDom ? MSSAU->createMemoryAccessBefore( 1726 NewS, MSSAU->getMemorySSA()->getLiveOnEntryDef(), 1727 const_cast<MemoryUseOrDef *>(FirstNonDom)) 1728 : MSSAU->createMemoryAccessInBB( 1729 NewS, MSSAU->getMemorySSA()->getLiveOnEntryDef(), 1730 NewS->getParent(), MemorySSA::BeforeTerminator); 1731 1732 MSSAU->insertDef(cast<MemoryDef>(NewDef), /*RenameUses=*/false); 1733 } 1734 } 1735 if (isAssumeWithEmptyBundle(*IntrinsicI)) 1736 markInstructionForDeletion(IntrinsicI); 1737 return false; 1738 } else if (isa<Constant>(V)) { 1739 // If it's not false, and constant, it must evaluate to true. This means our 1740 // assume is assume(true), and thus, pointless, and we don't want to do 1741 // anything more here. 1742 return false; 1743 } 1744 1745 Constant *True = ConstantInt::getTrue(V->getContext()); 1746 bool Changed = false; 1747 1748 for (BasicBlock *Successor : successors(IntrinsicI->getParent())) { 1749 BasicBlockEdge Edge(IntrinsicI->getParent(), Successor); 1750 1751 // This property is only true in dominated successors, propagateEquality 1752 // will check dominance for us. 1753 Changed |= propagateEquality(V, True, Edge, false); 1754 } 1755 1756 // We can replace assume value with true, which covers cases like this: 1757 // call void @llvm.assume(i1 %cmp) 1758 // br i1 %cmp, label %bb1, label %bb2 ; will change %cmp to true 1759 ReplaceOperandsWithMap[V] = True; 1760 1761 // Similarly, after assume(!NotV) we know that NotV == false. 1762 Value *NotV; 1763 if (match(V, m_Not(m_Value(NotV)))) 1764 ReplaceOperandsWithMap[NotV] = ConstantInt::getFalse(V->getContext()); 1765 1766 // If we find an equality fact, canonicalize all dominated uses in this block 1767 // to one of the two values. We heuristically choice the "oldest" of the 1768 // two where age is determined by value number. (Note that propagateEquality 1769 // above handles the cross block case.) 1770 // 1771 // Key case to cover are: 1772 // 1) 1773 // %cmp = fcmp oeq float 3.000000e+00, %0 ; const on lhs could happen 1774 // call void @llvm.assume(i1 %cmp) 1775 // ret float %0 ; will change it to ret float 3.000000e+00 1776 // 2) 1777 // %load = load float, float* %addr 1778 // %cmp = fcmp oeq float %load, %0 1779 // call void @llvm.assume(i1 %cmp) 1780 // ret float %load ; will change it to ret float %0 1781 if (auto *CmpI = dyn_cast<CmpInst>(V)) { 1782 if (impliesEquivalanceIfTrue(CmpI)) { 1783 Value *CmpLHS = CmpI->getOperand(0); 1784 Value *CmpRHS = CmpI->getOperand(1); 1785 // Heuristically pick the better replacement -- the choice of heuristic 1786 // isn't terribly important here, but the fact we canonicalize on some 1787 // replacement is for exposing other simplifications. 1788 // TODO: pull this out as a helper function and reuse w/existing 1789 // (slightly different) logic. 1790 if (isa<Constant>(CmpLHS) && !isa<Constant>(CmpRHS)) 1791 std::swap(CmpLHS, CmpRHS); 1792 if (!isa<Instruction>(CmpLHS) && isa<Instruction>(CmpRHS)) 1793 std::swap(CmpLHS, CmpRHS); 1794 if ((isa<Argument>(CmpLHS) && isa<Argument>(CmpRHS)) || 1795 (isa<Instruction>(CmpLHS) && isa<Instruction>(CmpRHS))) { 1796 // Move the 'oldest' value to the right-hand side, using the value 1797 // number as a proxy for age. 1798 uint32_t LVN = VN.lookupOrAdd(CmpLHS); 1799 uint32_t RVN = VN.lookupOrAdd(CmpRHS); 1800 if (LVN < RVN) 1801 std::swap(CmpLHS, CmpRHS); 1802 } 1803 1804 // Handle degenerate case where we either haven't pruned a dead path or a 1805 // removed a trivial assume yet. 1806 if (isa<Constant>(CmpLHS) && isa<Constant>(CmpRHS)) 1807 return Changed; 1808 1809 LLVM_DEBUG(dbgs() << "Replacing dominated uses of " 1810 << *CmpLHS << " with " 1811 << *CmpRHS << " in block " 1812 << IntrinsicI->getParent()->getName() << "\n"); 1813 1814 1815 // Setup the replacement map - this handles uses within the same block 1816 if (hasUsersIn(CmpLHS, IntrinsicI->getParent())) 1817 ReplaceOperandsWithMap[CmpLHS] = CmpRHS; 1818 1819 // NOTE: The non-block local cases are handled by the call to 1820 // propagateEquality above; this block is just about handling the block 1821 // local cases. TODO: There's a bunch of logic in propagateEqualiy which 1822 // isn't duplicated for the block local case, can we share it somehow? 1823 } 1824 } 1825 return Changed; 1826 } 1827 1828 static void patchAndReplaceAllUsesWith(Instruction *I, Value *Repl) { 1829 patchReplacementInstruction(I, Repl); 1830 I->replaceAllUsesWith(Repl); 1831 } 1832 1833 /// Attempt to eliminate a load, first by eliminating it 1834 /// locally, and then attempting non-local elimination if that fails. 1835 bool GVN::processLoad(LoadInst *L) { 1836 if (!MD) 1837 return false; 1838 1839 // This code hasn't been audited for ordered or volatile memory access 1840 if (!L->isUnordered()) 1841 return false; 1842 1843 if (L->use_empty()) { 1844 markInstructionForDeletion(L); 1845 return true; 1846 } 1847 1848 // ... to a pointer that has been loaded from before... 1849 MemDepResult Dep = MD->getDependency(L); 1850 1851 // If it is defined in another block, try harder. 1852 if (Dep.isNonLocal()) 1853 return processNonLocalLoad(L); 1854 1855 // Only handle the local case below 1856 if (!Dep.isDef() && !Dep.isClobber()) { 1857 // This might be a NonFuncLocal or an Unknown 1858 LLVM_DEBUG( 1859 // fast print dep, using operator<< on instruction is too slow. 1860 dbgs() << "GVN: load "; L->printAsOperand(dbgs()); 1861 dbgs() << " has unknown dependence\n";); 1862 return false; 1863 } 1864 1865 AvailableValue AV; 1866 if (AnalyzeLoadAvailability(L, Dep, L->getPointerOperand(), AV)) { 1867 Value *AvailableValue = AV.MaterializeAdjustedValue(L, L, *this); 1868 1869 // Replace the load! 1870 patchAndReplaceAllUsesWith(L, AvailableValue); 1871 markInstructionForDeletion(L); 1872 if (MSSAU) 1873 MSSAU->removeMemoryAccess(L); 1874 ++NumGVNLoad; 1875 reportLoadElim(L, AvailableValue, ORE); 1876 // Tell MDA to reexamine the reused pointer since we might have more 1877 // information after forwarding it. 1878 if (MD && AvailableValue->getType()->isPtrOrPtrVectorTy()) 1879 MD->invalidateCachedPointerInfo(AvailableValue); 1880 return true; 1881 } 1882 1883 return false; 1884 } 1885 1886 /// Return a pair the first field showing the value number of \p Exp and the 1887 /// second field showing whether it is a value number newly created. 1888 std::pair<uint32_t, bool> 1889 GVN::ValueTable::assignExpNewValueNum(Expression &Exp) { 1890 uint32_t &e = expressionNumbering[Exp]; 1891 bool CreateNewValNum = !e; 1892 if (CreateNewValNum) { 1893 Expressions.push_back(Exp); 1894 if (ExprIdx.size() < nextValueNumber + 1) 1895 ExprIdx.resize(nextValueNumber * 2); 1896 e = nextValueNumber; 1897 ExprIdx[nextValueNumber++] = nextExprNumber++; 1898 } 1899 return {e, CreateNewValNum}; 1900 } 1901 1902 /// Return whether all the values related with the same \p num are 1903 /// defined in \p BB. 1904 bool GVN::ValueTable::areAllValsInBB(uint32_t Num, const BasicBlock *BB, 1905 GVN &Gvn) { 1906 LeaderTableEntry *Vals = &Gvn.LeaderTable[Num]; 1907 while (Vals && Vals->BB == BB) 1908 Vals = Vals->Next; 1909 return !Vals; 1910 } 1911 1912 /// Wrap phiTranslateImpl to provide caching functionality. 1913 uint32_t GVN::ValueTable::phiTranslate(const BasicBlock *Pred, 1914 const BasicBlock *PhiBlock, uint32_t Num, 1915 GVN &Gvn) { 1916 auto FindRes = PhiTranslateTable.find({Num, Pred}); 1917 if (FindRes != PhiTranslateTable.end()) 1918 return FindRes->second; 1919 uint32_t NewNum = phiTranslateImpl(Pred, PhiBlock, Num, Gvn); 1920 PhiTranslateTable.insert({{Num, Pred}, NewNum}); 1921 return NewNum; 1922 } 1923 1924 // Return true if the value number \p Num and NewNum have equal value. 1925 // Return false if the result is unknown. 1926 bool GVN::ValueTable::areCallValsEqual(uint32_t Num, uint32_t NewNum, 1927 const BasicBlock *Pred, 1928 const BasicBlock *PhiBlock, GVN &Gvn) { 1929 CallInst *Call = nullptr; 1930 LeaderTableEntry *Vals = &Gvn.LeaderTable[Num]; 1931 while (Vals) { 1932 Call = dyn_cast<CallInst>(Vals->Val); 1933 if (Call && Call->getParent() == PhiBlock) 1934 break; 1935 Vals = Vals->Next; 1936 } 1937 1938 if (AA->doesNotAccessMemory(Call)) 1939 return true; 1940 1941 if (!MD || !AA->onlyReadsMemory(Call)) 1942 return false; 1943 1944 MemDepResult local_dep = MD->getDependency(Call); 1945 if (!local_dep.isNonLocal()) 1946 return false; 1947 1948 const MemoryDependenceResults::NonLocalDepInfo &deps = 1949 MD->getNonLocalCallDependency(Call); 1950 1951 // Check to see if the Call has no function local clobber. 1952 for (const NonLocalDepEntry &D : deps) { 1953 if (D.getResult().isNonFuncLocal()) 1954 return true; 1955 } 1956 return false; 1957 } 1958 1959 /// Translate value number \p Num using phis, so that it has the values of 1960 /// the phis in BB. 1961 uint32_t GVN::ValueTable::phiTranslateImpl(const BasicBlock *Pred, 1962 const BasicBlock *PhiBlock, 1963 uint32_t Num, GVN &Gvn) { 1964 if (PHINode *PN = NumberingPhi[Num]) { 1965 for (unsigned i = 0; i != PN->getNumIncomingValues(); ++i) { 1966 if (PN->getParent() == PhiBlock && PN->getIncomingBlock(i) == Pred) 1967 if (uint32_t TransVal = lookup(PN->getIncomingValue(i), false)) 1968 return TransVal; 1969 } 1970 return Num; 1971 } 1972 1973 // If there is any value related with Num is defined in a BB other than 1974 // PhiBlock, it cannot depend on a phi in PhiBlock without going through 1975 // a backedge. We can do an early exit in that case to save compile time. 1976 if (!areAllValsInBB(Num, PhiBlock, Gvn)) 1977 return Num; 1978 1979 if (Num >= ExprIdx.size() || ExprIdx[Num] == 0) 1980 return Num; 1981 Expression Exp = Expressions[ExprIdx[Num]]; 1982 1983 for (unsigned i = 0; i < Exp.varargs.size(); i++) { 1984 // For InsertValue and ExtractValue, some varargs are index numbers 1985 // instead of value numbers. Those index numbers should not be 1986 // translated. 1987 if ((i > 1 && Exp.opcode == Instruction::InsertValue) || 1988 (i > 0 && Exp.opcode == Instruction::ExtractValue) || 1989 (i > 1 && Exp.opcode == Instruction::ShuffleVector)) 1990 continue; 1991 Exp.varargs[i] = phiTranslate(Pred, PhiBlock, Exp.varargs[i], Gvn); 1992 } 1993 1994 if (Exp.commutative) { 1995 assert(Exp.varargs.size() >= 2 && "Unsupported commutative instruction!"); 1996 if (Exp.varargs[0] > Exp.varargs[1]) { 1997 std::swap(Exp.varargs[0], Exp.varargs[1]); 1998 uint32_t Opcode = Exp.opcode >> 8; 1999 if (Opcode == Instruction::ICmp || Opcode == Instruction::FCmp) 2000 Exp.opcode = (Opcode << 8) | 2001 CmpInst::getSwappedPredicate( 2002 static_cast<CmpInst::Predicate>(Exp.opcode & 255)); 2003 } 2004 } 2005 2006 if (uint32_t NewNum = expressionNumbering[Exp]) { 2007 if (Exp.opcode == Instruction::Call && NewNum != Num) 2008 return areCallValsEqual(Num, NewNum, Pred, PhiBlock, Gvn) ? NewNum : Num; 2009 return NewNum; 2010 } 2011 return Num; 2012 } 2013 2014 /// Erase stale entry from phiTranslate cache so phiTranslate can be computed 2015 /// again. 2016 void GVN::ValueTable::eraseTranslateCacheEntry(uint32_t Num, 2017 const BasicBlock &CurrBlock) { 2018 for (const BasicBlock *Pred : predecessors(&CurrBlock)) 2019 PhiTranslateTable.erase({Num, Pred}); 2020 } 2021 2022 // In order to find a leader for a given value number at a 2023 // specific basic block, we first obtain the list of all Values for that number, 2024 // and then scan the list to find one whose block dominates the block in 2025 // question. This is fast because dominator tree queries consist of only 2026 // a few comparisons of DFS numbers. 2027 Value *GVN::findLeader(const BasicBlock *BB, uint32_t num) { 2028 LeaderTableEntry Vals = LeaderTable[num]; 2029 if (!Vals.Val) return nullptr; 2030 2031 Value *Val = nullptr; 2032 if (DT->dominates(Vals.BB, BB)) { 2033 Val = Vals.Val; 2034 if (isa<Constant>(Val)) return Val; 2035 } 2036 2037 LeaderTableEntry* Next = Vals.Next; 2038 while (Next) { 2039 if (DT->dominates(Next->BB, BB)) { 2040 if (isa<Constant>(Next->Val)) return Next->Val; 2041 if (!Val) Val = Next->Val; 2042 } 2043 2044 Next = Next->Next; 2045 } 2046 2047 return Val; 2048 } 2049 2050 /// There is an edge from 'Src' to 'Dst'. Return 2051 /// true if every path from the entry block to 'Dst' passes via this edge. In 2052 /// particular 'Dst' must not be reachable via another edge from 'Src'. 2053 static bool isOnlyReachableViaThisEdge(const BasicBlockEdge &E, 2054 DominatorTree *DT) { 2055 // While in theory it is interesting to consider the case in which Dst has 2056 // more than one predecessor, because Dst might be part of a loop which is 2057 // only reachable from Src, in practice it is pointless since at the time 2058 // GVN runs all such loops have preheaders, which means that Dst will have 2059 // been changed to have only one predecessor, namely Src. 2060 const BasicBlock *Pred = E.getEnd()->getSinglePredecessor(); 2061 assert((!Pred || Pred == E.getStart()) && 2062 "No edge between these basic blocks!"); 2063 return Pred != nullptr; 2064 } 2065 2066 void GVN::assignBlockRPONumber(Function &F) { 2067 BlockRPONumber.clear(); 2068 uint32_t NextBlockNumber = 1; 2069 ReversePostOrderTraversal<Function *> RPOT(&F); 2070 for (BasicBlock *BB : RPOT) 2071 BlockRPONumber[BB] = NextBlockNumber++; 2072 InvalidBlockRPONumbers = false; 2073 } 2074 2075 bool GVN::replaceOperandsForInBlockEquality(Instruction *Instr) const { 2076 bool Changed = false; 2077 for (unsigned OpNum = 0; OpNum < Instr->getNumOperands(); ++OpNum) { 2078 Value *Operand = Instr->getOperand(OpNum); 2079 auto it = ReplaceOperandsWithMap.find(Operand); 2080 if (it != ReplaceOperandsWithMap.end()) { 2081 LLVM_DEBUG(dbgs() << "GVN replacing: " << *Operand << " with " 2082 << *it->second << " in instruction " << *Instr << '\n'); 2083 Instr->setOperand(OpNum, it->second); 2084 Changed = true; 2085 } 2086 } 2087 return Changed; 2088 } 2089 2090 /// The given values are known to be equal in every block 2091 /// dominated by 'Root'. Exploit this, for example by replacing 'LHS' with 2092 /// 'RHS' everywhere in the scope. Returns whether a change was made. 2093 /// If DominatesByEdge is false, then it means that we will propagate the RHS 2094 /// value starting from the end of Root.Start. 2095 bool GVN::propagateEquality(Value *LHS, Value *RHS, const BasicBlockEdge &Root, 2096 bool DominatesByEdge) { 2097 SmallVector<std::pair<Value*, Value*>, 4> Worklist; 2098 Worklist.push_back(std::make_pair(LHS, RHS)); 2099 bool Changed = false; 2100 // For speed, compute a conservative fast approximation to 2101 // DT->dominates(Root, Root.getEnd()); 2102 const bool RootDominatesEnd = isOnlyReachableViaThisEdge(Root, DT); 2103 2104 while (!Worklist.empty()) { 2105 std::pair<Value*, Value*> Item = Worklist.pop_back_val(); 2106 LHS = Item.first; RHS = Item.second; 2107 2108 if (LHS == RHS) 2109 continue; 2110 assert(LHS->getType() == RHS->getType() && "Equality but unequal types!"); 2111 2112 // Don't try to propagate equalities between constants. 2113 if (isa<Constant>(LHS) && isa<Constant>(RHS)) 2114 continue; 2115 2116 // Prefer a constant on the right-hand side, or an Argument if no constants. 2117 if (isa<Constant>(LHS) || (isa<Argument>(LHS) && !isa<Constant>(RHS))) 2118 std::swap(LHS, RHS); 2119 assert((isa<Argument>(LHS) || isa<Instruction>(LHS)) && "Unexpected value!"); 2120 2121 // If there is no obvious reason to prefer the left-hand side over the 2122 // right-hand side, ensure the longest lived term is on the right-hand side, 2123 // so the shortest lived term will be replaced by the longest lived. 2124 // This tends to expose more simplifications. 2125 uint32_t LVN = VN.lookupOrAdd(LHS); 2126 if ((isa<Argument>(LHS) && isa<Argument>(RHS)) || 2127 (isa<Instruction>(LHS) && isa<Instruction>(RHS))) { 2128 // Move the 'oldest' value to the right-hand side, using the value number 2129 // as a proxy for age. 2130 uint32_t RVN = VN.lookupOrAdd(RHS); 2131 if (LVN < RVN) { 2132 std::swap(LHS, RHS); 2133 LVN = RVN; 2134 } 2135 } 2136 2137 // If value numbering later sees that an instruction in the scope is equal 2138 // to 'LHS' then ensure it will be turned into 'RHS'. In order to preserve 2139 // the invariant that instructions only occur in the leader table for their 2140 // own value number (this is used by removeFromLeaderTable), do not do this 2141 // if RHS is an instruction (if an instruction in the scope is morphed into 2142 // LHS then it will be turned into RHS by the next GVN iteration anyway, so 2143 // using the leader table is about compiling faster, not optimizing better). 2144 // The leader table only tracks basic blocks, not edges. Only add to if we 2145 // have the simple case where the edge dominates the end. 2146 if (RootDominatesEnd && !isa<Instruction>(RHS)) 2147 addToLeaderTable(LVN, RHS, Root.getEnd()); 2148 2149 // Replace all occurrences of 'LHS' with 'RHS' everywhere in the scope. As 2150 // LHS always has at least one use that is not dominated by Root, this will 2151 // never do anything if LHS has only one use. 2152 if (!LHS->hasOneUse()) { 2153 unsigned NumReplacements = 2154 DominatesByEdge 2155 ? replaceDominatedUsesWith(LHS, RHS, *DT, Root) 2156 : replaceDominatedUsesWith(LHS, RHS, *DT, Root.getStart()); 2157 2158 Changed |= NumReplacements > 0; 2159 NumGVNEqProp += NumReplacements; 2160 // Cached information for anything that uses LHS will be invalid. 2161 if (MD) 2162 MD->invalidateCachedPointerInfo(LHS); 2163 } 2164 2165 // Now try to deduce additional equalities from this one. For example, if 2166 // the known equality was "(A != B)" == "false" then it follows that A and B 2167 // are equal in the scope. Only boolean equalities with an explicit true or 2168 // false RHS are currently supported. 2169 if (!RHS->getType()->isIntegerTy(1)) 2170 // Not a boolean equality - bail out. 2171 continue; 2172 ConstantInt *CI = dyn_cast<ConstantInt>(RHS); 2173 if (!CI) 2174 // RHS neither 'true' nor 'false' - bail out. 2175 continue; 2176 // Whether RHS equals 'true'. Otherwise it equals 'false'. 2177 bool isKnownTrue = CI->isMinusOne(); 2178 bool isKnownFalse = !isKnownTrue; 2179 2180 // If "A && B" is known true then both A and B are known true. If "A || B" 2181 // is known false then both A and B are known false. 2182 Value *A, *B; 2183 if ((isKnownTrue && match(LHS, m_LogicalAnd(m_Value(A), m_Value(B)))) || 2184 (isKnownFalse && match(LHS, m_LogicalOr(m_Value(A), m_Value(B))))) { 2185 Worklist.push_back(std::make_pair(A, RHS)); 2186 Worklist.push_back(std::make_pair(B, RHS)); 2187 continue; 2188 } 2189 2190 // If we are propagating an equality like "(A == B)" == "true" then also 2191 // propagate the equality A == B. When propagating a comparison such as 2192 // "(A >= B)" == "true", replace all instances of "A < B" with "false". 2193 if (CmpInst *Cmp = dyn_cast<CmpInst>(LHS)) { 2194 Value *Op0 = Cmp->getOperand(0), *Op1 = Cmp->getOperand(1); 2195 2196 // If "A == B" is known true, or "A != B" is known false, then replace 2197 // A with B everywhere in the scope. For floating point operations, we 2198 // have to be careful since equality does not always imply equivalance. 2199 if ((isKnownTrue && impliesEquivalanceIfTrue(Cmp)) || 2200 (isKnownFalse && impliesEquivalanceIfFalse(Cmp))) 2201 Worklist.push_back(std::make_pair(Op0, Op1)); 2202 2203 // If "A >= B" is known true, replace "A < B" with false everywhere. 2204 CmpInst::Predicate NotPred = Cmp->getInversePredicate(); 2205 Constant *NotVal = ConstantInt::get(Cmp->getType(), isKnownFalse); 2206 // Since we don't have the instruction "A < B" immediately to hand, work 2207 // out the value number that it would have and use that to find an 2208 // appropriate instruction (if any). 2209 uint32_t NextNum = VN.getNextUnusedValueNumber(); 2210 uint32_t Num = VN.lookupOrAddCmp(Cmp->getOpcode(), NotPred, Op0, Op1); 2211 // If the number we were assigned was brand new then there is no point in 2212 // looking for an instruction realizing it: there cannot be one! 2213 if (Num < NextNum) { 2214 Value *NotCmp = findLeader(Root.getEnd(), Num); 2215 if (NotCmp && isa<Instruction>(NotCmp)) { 2216 unsigned NumReplacements = 2217 DominatesByEdge 2218 ? replaceDominatedUsesWith(NotCmp, NotVal, *DT, Root) 2219 : replaceDominatedUsesWith(NotCmp, NotVal, *DT, 2220 Root.getStart()); 2221 Changed |= NumReplacements > 0; 2222 NumGVNEqProp += NumReplacements; 2223 // Cached information for anything that uses NotCmp will be invalid. 2224 if (MD) 2225 MD->invalidateCachedPointerInfo(NotCmp); 2226 } 2227 } 2228 // Ensure that any instruction in scope that gets the "A < B" value number 2229 // is replaced with false. 2230 // The leader table only tracks basic blocks, not edges. Only add to if we 2231 // have the simple case where the edge dominates the end. 2232 if (RootDominatesEnd) 2233 addToLeaderTable(Num, NotVal, Root.getEnd()); 2234 2235 continue; 2236 } 2237 } 2238 2239 return Changed; 2240 } 2241 2242 /// When calculating availability, handle an instruction 2243 /// by inserting it into the appropriate sets 2244 bool GVN::processInstruction(Instruction *I) { 2245 // Ignore dbg info intrinsics. 2246 if (isa<DbgInfoIntrinsic>(I)) 2247 return false; 2248 2249 // If the instruction can be easily simplified then do so now in preference 2250 // to value numbering it. Value numbering often exposes redundancies, for 2251 // example if it determines that %y is equal to %x then the instruction 2252 // "%z = and i32 %x, %y" becomes "%z = and i32 %x, %x" which we now simplify. 2253 const DataLayout &DL = I->getModule()->getDataLayout(); 2254 if (Value *V = SimplifyInstruction(I, {DL, TLI, DT, AC})) { 2255 bool Changed = false; 2256 if (!I->use_empty()) { 2257 // Simplification can cause a special instruction to become not special. 2258 // For example, devirtualization to a willreturn function. 2259 ICF->removeUsersOf(I); 2260 I->replaceAllUsesWith(V); 2261 Changed = true; 2262 } 2263 if (isInstructionTriviallyDead(I, TLI)) { 2264 markInstructionForDeletion(I); 2265 Changed = true; 2266 } 2267 if (Changed) { 2268 if (MD && V->getType()->isPtrOrPtrVectorTy()) 2269 MD->invalidateCachedPointerInfo(V); 2270 ++NumGVNSimpl; 2271 return true; 2272 } 2273 } 2274 2275 if (auto *Assume = dyn_cast<AssumeInst>(I)) 2276 return processAssumeIntrinsic(Assume); 2277 2278 if (LoadInst *Load = dyn_cast<LoadInst>(I)) { 2279 if (processLoad(Load)) 2280 return true; 2281 2282 unsigned Num = VN.lookupOrAdd(Load); 2283 addToLeaderTable(Num, Load, Load->getParent()); 2284 return false; 2285 } 2286 2287 // For conditional branches, we can perform simple conditional propagation on 2288 // the condition value itself. 2289 if (BranchInst *BI = dyn_cast<BranchInst>(I)) { 2290 if (!BI->isConditional()) 2291 return false; 2292 2293 if (isa<Constant>(BI->getCondition())) 2294 return processFoldableCondBr(BI); 2295 2296 Value *BranchCond = BI->getCondition(); 2297 BasicBlock *TrueSucc = BI->getSuccessor(0); 2298 BasicBlock *FalseSucc = BI->getSuccessor(1); 2299 // Avoid multiple edges early. 2300 if (TrueSucc == FalseSucc) 2301 return false; 2302 2303 BasicBlock *Parent = BI->getParent(); 2304 bool Changed = false; 2305 2306 Value *TrueVal = ConstantInt::getTrue(TrueSucc->getContext()); 2307 BasicBlockEdge TrueE(Parent, TrueSucc); 2308 Changed |= propagateEquality(BranchCond, TrueVal, TrueE, true); 2309 2310 Value *FalseVal = ConstantInt::getFalse(FalseSucc->getContext()); 2311 BasicBlockEdge FalseE(Parent, FalseSucc); 2312 Changed |= propagateEquality(BranchCond, FalseVal, FalseE, true); 2313 2314 return Changed; 2315 } 2316 2317 // For switches, propagate the case values into the case destinations. 2318 if (SwitchInst *SI = dyn_cast<SwitchInst>(I)) { 2319 Value *SwitchCond = SI->getCondition(); 2320 BasicBlock *Parent = SI->getParent(); 2321 bool Changed = false; 2322 2323 // Remember how many outgoing edges there are to every successor. 2324 SmallDenseMap<BasicBlock *, unsigned, 16> SwitchEdges; 2325 for (unsigned i = 0, n = SI->getNumSuccessors(); i != n; ++i) 2326 ++SwitchEdges[SI->getSuccessor(i)]; 2327 2328 for (SwitchInst::CaseIt i = SI->case_begin(), e = SI->case_end(); 2329 i != e; ++i) { 2330 BasicBlock *Dst = i->getCaseSuccessor(); 2331 // If there is only a single edge, propagate the case value into it. 2332 if (SwitchEdges.lookup(Dst) == 1) { 2333 BasicBlockEdge E(Parent, Dst); 2334 Changed |= propagateEquality(SwitchCond, i->getCaseValue(), E, true); 2335 } 2336 } 2337 return Changed; 2338 } 2339 2340 // Instructions with void type don't return a value, so there's 2341 // no point in trying to find redundancies in them. 2342 if (I->getType()->isVoidTy()) 2343 return false; 2344 2345 uint32_t NextNum = VN.getNextUnusedValueNumber(); 2346 unsigned Num = VN.lookupOrAdd(I); 2347 2348 // Allocations are always uniquely numbered, so we can save time and memory 2349 // by fast failing them. 2350 if (isa<AllocaInst>(I) || I->isTerminator() || isa<PHINode>(I)) { 2351 addToLeaderTable(Num, I, I->getParent()); 2352 return false; 2353 } 2354 2355 // If the number we were assigned was a brand new VN, then we don't 2356 // need to do a lookup to see if the number already exists 2357 // somewhere in the domtree: it can't! 2358 if (Num >= NextNum) { 2359 addToLeaderTable(Num, I, I->getParent()); 2360 return false; 2361 } 2362 2363 // Perform fast-path value-number based elimination of values inherited from 2364 // dominators. 2365 Value *Repl = findLeader(I->getParent(), Num); 2366 if (!Repl) { 2367 // Failure, just remember this instance for future use. 2368 addToLeaderTable(Num, I, I->getParent()); 2369 return false; 2370 } else if (Repl == I) { 2371 // If I was the result of a shortcut PRE, it might already be in the table 2372 // and the best replacement for itself. Nothing to do. 2373 return false; 2374 } 2375 2376 // Remove it! 2377 patchAndReplaceAllUsesWith(I, Repl); 2378 if (MD && Repl->getType()->isPtrOrPtrVectorTy()) 2379 MD->invalidateCachedPointerInfo(Repl); 2380 markInstructionForDeletion(I); 2381 return true; 2382 } 2383 2384 /// runOnFunction - This is the main transformation entry point for a function. 2385 bool GVN::runImpl(Function &F, AssumptionCache &RunAC, DominatorTree &RunDT, 2386 const TargetLibraryInfo &RunTLI, AAResults &RunAA, 2387 MemoryDependenceResults *RunMD, LoopInfo *LI, 2388 OptimizationRemarkEmitter *RunORE, MemorySSA *MSSA) { 2389 AC = &RunAC; 2390 DT = &RunDT; 2391 VN.setDomTree(DT); 2392 TLI = &RunTLI; 2393 VN.setAliasAnalysis(&RunAA); 2394 MD = RunMD; 2395 ImplicitControlFlowTracking ImplicitCFT; 2396 ICF = &ImplicitCFT; 2397 this->LI = LI; 2398 VN.setMemDep(MD); 2399 ORE = RunORE; 2400 InvalidBlockRPONumbers = true; 2401 MemorySSAUpdater Updater(MSSA); 2402 MSSAU = MSSA ? &Updater : nullptr; 2403 2404 bool Changed = false; 2405 bool ShouldContinue = true; 2406 2407 DomTreeUpdater DTU(DT, DomTreeUpdater::UpdateStrategy::Eager); 2408 // Merge unconditional branches, allowing PRE to catch more 2409 // optimization opportunities. 2410 for (Function::iterator FI = F.begin(), FE = F.end(); FI != FE; ) { 2411 BasicBlock *BB = &*FI++; 2412 2413 bool removedBlock = MergeBlockIntoPredecessor(BB, &DTU, LI, MSSAU, MD); 2414 if (removedBlock) 2415 ++NumGVNBlocks; 2416 2417 Changed |= removedBlock; 2418 } 2419 2420 unsigned Iteration = 0; 2421 while (ShouldContinue) { 2422 LLVM_DEBUG(dbgs() << "GVN iteration: " << Iteration << "\n"); 2423 ShouldContinue = iterateOnFunction(F); 2424 Changed |= ShouldContinue; 2425 ++Iteration; 2426 } 2427 2428 if (isPREEnabled()) { 2429 // Fabricate val-num for dead-code in order to suppress assertion in 2430 // performPRE(). 2431 assignValNumForDeadCode(); 2432 bool PREChanged = true; 2433 while (PREChanged) { 2434 PREChanged = performPRE(F); 2435 Changed |= PREChanged; 2436 } 2437 } 2438 2439 // FIXME: Should perform GVN again after PRE does something. PRE can move 2440 // computations into blocks where they become fully redundant. Note that 2441 // we can't do this until PRE's critical edge splitting updates memdep. 2442 // Actually, when this happens, we should just fully integrate PRE into GVN. 2443 2444 cleanupGlobalSets(); 2445 // Do not cleanup DeadBlocks in cleanupGlobalSets() as it's called for each 2446 // iteration. 2447 DeadBlocks.clear(); 2448 2449 if (MSSA && VerifyMemorySSA) 2450 MSSA->verifyMemorySSA(); 2451 2452 return Changed; 2453 } 2454 2455 bool GVN::processBlock(BasicBlock *BB) { 2456 // FIXME: Kill off InstrsToErase by doing erasing eagerly in a helper function 2457 // (and incrementing BI before processing an instruction). 2458 assert(InstrsToErase.empty() && 2459 "We expect InstrsToErase to be empty across iterations"); 2460 if (DeadBlocks.count(BB)) 2461 return false; 2462 2463 // Clearing map before every BB because it can be used only for single BB. 2464 ReplaceOperandsWithMap.clear(); 2465 bool ChangedFunction = false; 2466 2467 // Since we may not have visited the input blocks of the phis, we can't 2468 // use our normal hash approach for phis. Instead, simply look for 2469 // obvious duplicates. The first pass of GVN will tend to create 2470 // identical phis, and the second or later passes can eliminate them. 2471 ChangedFunction |= EliminateDuplicatePHINodes(BB); 2472 2473 for (BasicBlock::iterator BI = BB->begin(), BE = BB->end(); 2474 BI != BE;) { 2475 if (!ReplaceOperandsWithMap.empty()) 2476 ChangedFunction |= replaceOperandsForInBlockEquality(&*BI); 2477 ChangedFunction |= processInstruction(&*BI); 2478 2479 if (InstrsToErase.empty()) { 2480 ++BI; 2481 continue; 2482 } 2483 2484 // If we need some instructions deleted, do it now. 2485 NumGVNInstr += InstrsToErase.size(); 2486 2487 // Avoid iterator invalidation. 2488 bool AtStart = BI == BB->begin(); 2489 if (!AtStart) 2490 --BI; 2491 2492 for (auto *I : InstrsToErase) { 2493 assert(I->getParent() == BB && "Removing instruction from wrong block?"); 2494 LLVM_DEBUG(dbgs() << "GVN removed: " << *I << '\n'); 2495 salvageKnowledge(I, AC); 2496 salvageDebugInfo(*I); 2497 if (MD) MD->removeInstruction(I); 2498 if (MSSAU) 2499 MSSAU->removeMemoryAccess(I); 2500 LLVM_DEBUG(verifyRemoved(I)); 2501 ICF->removeInstruction(I); 2502 I->eraseFromParent(); 2503 } 2504 InstrsToErase.clear(); 2505 2506 if (AtStart) 2507 BI = BB->begin(); 2508 else 2509 ++BI; 2510 } 2511 2512 return ChangedFunction; 2513 } 2514 2515 // Instantiate an expression in a predecessor that lacked it. 2516 bool GVN::performScalarPREInsertion(Instruction *Instr, BasicBlock *Pred, 2517 BasicBlock *Curr, unsigned int ValNo) { 2518 // Because we are going top-down through the block, all value numbers 2519 // will be available in the predecessor by the time we need them. Any 2520 // that weren't originally present will have been instantiated earlier 2521 // in this loop. 2522 bool success = true; 2523 for (unsigned i = 0, e = Instr->getNumOperands(); i != e; ++i) { 2524 Value *Op = Instr->getOperand(i); 2525 if (isa<Argument>(Op) || isa<Constant>(Op) || isa<GlobalValue>(Op)) 2526 continue; 2527 // This could be a newly inserted instruction, in which case, we won't 2528 // find a value number, and should give up before we hurt ourselves. 2529 // FIXME: Rewrite the infrastructure to let it easier to value number 2530 // and process newly inserted instructions. 2531 if (!VN.exists(Op)) { 2532 success = false; 2533 break; 2534 } 2535 uint32_t TValNo = 2536 VN.phiTranslate(Pred, Curr, VN.lookup(Op), *this); 2537 if (Value *V = findLeader(Pred, TValNo)) { 2538 Instr->setOperand(i, V); 2539 } else { 2540 success = false; 2541 break; 2542 } 2543 } 2544 2545 // Fail out if we encounter an operand that is not available in 2546 // the PRE predecessor. This is typically because of loads which 2547 // are not value numbered precisely. 2548 if (!success) 2549 return false; 2550 2551 Instr->insertBefore(Pred->getTerminator()); 2552 Instr->setName(Instr->getName() + ".pre"); 2553 Instr->setDebugLoc(Instr->getDebugLoc()); 2554 2555 ICF->insertInstructionTo(Instr, Pred); 2556 2557 unsigned Num = VN.lookupOrAdd(Instr); 2558 VN.add(Instr, Num); 2559 2560 // Update the availability map to include the new instruction. 2561 addToLeaderTable(Num, Instr, Pred); 2562 return true; 2563 } 2564 2565 bool GVN::performScalarPRE(Instruction *CurInst) { 2566 if (isa<AllocaInst>(CurInst) || CurInst->isTerminator() || 2567 isa<PHINode>(CurInst) || CurInst->getType()->isVoidTy() || 2568 CurInst->mayReadFromMemory() || CurInst->mayHaveSideEffects() || 2569 isa<DbgInfoIntrinsic>(CurInst)) 2570 return false; 2571 2572 // Don't do PRE on compares. The PHI would prevent CodeGenPrepare from 2573 // sinking the compare again, and it would force the code generator to 2574 // move the i1 from processor flags or predicate registers into a general 2575 // purpose register. 2576 if (isa<CmpInst>(CurInst)) 2577 return false; 2578 2579 // Don't do PRE on GEPs. The inserted PHI would prevent CodeGenPrepare from 2580 // sinking the addressing mode computation back to its uses. Extending the 2581 // GEP's live range increases the register pressure, and therefore it can 2582 // introduce unnecessary spills. 2583 // 2584 // This doesn't prevent Load PRE. PHI translation will make the GEP available 2585 // to the load by moving it to the predecessor block if necessary. 2586 if (isa<GetElementPtrInst>(CurInst)) 2587 return false; 2588 2589 if (auto *CallB = dyn_cast<CallBase>(CurInst)) { 2590 // We don't currently value number ANY inline asm calls. 2591 if (CallB->isInlineAsm()) 2592 return false; 2593 // Don't do PRE on convergent calls. 2594 if (CallB->isConvergent()) 2595 return false; 2596 } 2597 2598 uint32_t ValNo = VN.lookup(CurInst); 2599 2600 // Look for the predecessors for PRE opportunities. We're 2601 // only trying to solve the basic diamond case, where 2602 // a value is computed in the successor and one predecessor, 2603 // but not the other. We also explicitly disallow cases 2604 // where the successor is its own predecessor, because they're 2605 // more complicated to get right. 2606 unsigned NumWith = 0; 2607 unsigned NumWithout = 0; 2608 BasicBlock *PREPred = nullptr; 2609 BasicBlock *CurrentBlock = CurInst->getParent(); 2610 2611 // Update the RPO numbers for this function. 2612 if (InvalidBlockRPONumbers) 2613 assignBlockRPONumber(*CurrentBlock->getParent()); 2614 2615 SmallVector<std::pair<Value *, BasicBlock *>, 8> predMap; 2616 for (BasicBlock *P : predecessors(CurrentBlock)) { 2617 // We're not interested in PRE where blocks with predecessors that are 2618 // not reachable. 2619 if (!DT->isReachableFromEntry(P)) { 2620 NumWithout = 2; 2621 break; 2622 } 2623 // It is not safe to do PRE when P->CurrentBlock is a loop backedge, and 2624 // when CurInst has operand defined in CurrentBlock (so it may be defined 2625 // by phi in the loop header). 2626 assert(BlockRPONumber.count(P) && BlockRPONumber.count(CurrentBlock) && 2627 "Invalid BlockRPONumber map."); 2628 if (BlockRPONumber[P] >= BlockRPONumber[CurrentBlock] && 2629 llvm::any_of(CurInst->operands(), [&](const Use &U) { 2630 if (auto *Inst = dyn_cast<Instruction>(U.get())) 2631 return Inst->getParent() == CurrentBlock; 2632 return false; 2633 })) { 2634 NumWithout = 2; 2635 break; 2636 } 2637 2638 uint32_t TValNo = VN.phiTranslate(P, CurrentBlock, ValNo, *this); 2639 Value *predV = findLeader(P, TValNo); 2640 if (!predV) { 2641 predMap.push_back(std::make_pair(static_cast<Value *>(nullptr), P)); 2642 PREPred = P; 2643 ++NumWithout; 2644 } else if (predV == CurInst) { 2645 /* CurInst dominates this predecessor. */ 2646 NumWithout = 2; 2647 break; 2648 } else { 2649 predMap.push_back(std::make_pair(predV, P)); 2650 ++NumWith; 2651 } 2652 } 2653 2654 // Don't do PRE when it might increase code size, i.e. when 2655 // we would need to insert instructions in more than one pred. 2656 if (NumWithout > 1 || NumWith == 0) 2657 return false; 2658 2659 // We may have a case where all predecessors have the instruction, 2660 // and we just need to insert a phi node. Otherwise, perform 2661 // insertion. 2662 Instruction *PREInstr = nullptr; 2663 2664 if (NumWithout != 0) { 2665 if (!isSafeToSpeculativelyExecute(CurInst)) { 2666 // It is only valid to insert a new instruction if the current instruction 2667 // is always executed. An instruction with implicit control flow could 2668 // prevent us from doing it. If we cannot speculate the execution, then 2669 // PRE should be prohibited. 2670 if (ICF->isDominatedByICFIFromSameBlock(CurInst)) 2671 return false; 2672 } 2673 2674 // Don't do PRE across indirect branch. 2675 if (isa<IndirectBrInst>(PREPred->getTerminator())) 2676 return false; 2677 2678 // Don't do PRE across callbr. 2679 // FIXME: Can we do this across the fallthrough edge? 2680 if (isa<CallBrInst>(PREPred->getTerminator())) 2681 return false; 2682 2683 // We can't do PRE safely on a critical edge, so instead we schedule 2684 // the edge to be split and perform the PRE the next time we iterate 2685 // on the function. 2686 unsigned SuccNum = GetSuccessorNumber(PREPred, CurrentBlock); 2687 if (isCriticalEdge(PREPred->getTerminator(), SuccNum)) { 2688 toSplit.push_back(std::make_pair(PREPred->getTerminator(), SuccNum)); 2689 return false; 2690 } 2691 // We need to insert somewhere, so let's give it a shot 2692 PREInstr = CurInst->clone(); 2693 if (!performScalarPREInsertion(PREInstr, PREPred, CurrentBlock, ValNo)) { 2694 // If we failed insertion, make sure we remove the instruction. 2695 LLVM_DEBUG(verifyRemoved(PREInstr)); 2696 PREInstr->deleteValue(); 2697 return false; 2698 } 2699 } 2700 2701 // Either we should have filled in the PRE instruction, or we should 2702 // not have needed insertions. 2703 assert(PREInstr != nullptr || NumWithout == 0); 2704 2705 ++NumGVNPRE; 2706 2707 // Create a PHI to make the value available in this block. 2708 PHINode *Phi = 2709 PHINode::Create(CurInst->getType(), predMap.size(), 2710 CurInst->getName() + ".pre-phi", &CurrentBlock->front()); 2711 for (unsigned i = 0, e = predMap.size(); i != e; ++i) { 2712 if (Value *V = predMap[i].first) { 2713 // If we use an existing value in this phi, we have to patch the original 2714 // value because the phi will be used to replace a later value. 2715 patchReplacementInstruction(CurInst, V); 2716 Phi->addIncoming(V, predMap[i].second); 2717 } else 2718 Phi->addIncoming(PREInstr, PREPred); 2719 } 2720 2721 VN.add(Phi, ValNo); 2722 // After creating a new PHI for ValNo, the phi translate result for ValNo will 2723 // be changed, so erase the related stale entries in phi translate cache. 2724 VN.eraseTranslateCacheEntry(ValNo, *CurrentBlock); 2725 addToLeaderTable(ValNo, Phi, CurrentBlock); 2726 Phi->setDebugLoc(CurInst->getDebugLoc()); 2727 CurInst->replaceAllUsesWith(Phi); 2728 if (MD && Phi->getType()->isPtrOrPtrVectorTy()) 2729 MD->invalidateCachedPointerInfo(Phi); 2730 VN.erase(CurInst); 2731 removeFromLeaderTable(ValNo, CurInst, CurrentBlock); 2732 2733 LLVM_DEBUG(dbgs() << "GVN PRE removed: " << *CurInst << '\n'); 2734 if (MD) 2735 MD->removeInstruction(CurInst); 2736 if (MSSAU) 2737 MSSAU->removeMemoryAccess(CurInst); 2738 LLVM_DEBUG(verifyRemoved(CurInst)); 2739 // FIXME: Intended to be markInstructionForDeletion(CurInst), but it causes 2740 // some assertion failures. 2741 ICF->removeInstruction(CurInst); 2742 CurInst->eraseFromParent(); 2743 ++NumGVNInstr; 2744 2745 return true; 2746 } 2747 2748 /// Perform a purely local form of PRE that looks for diamond 2749 /// control flow patterns and attempts to perform simple PRE at the join point. 2750 bool GVN::performPRE(Function &F) { 2751 bool Changed = false; 2752 for (BasicBlock *CurrentBlock : depth_first(&F.getEntryBlock())) { 2753 // Nothing to PRE in the entry block. 2754 if (CurrentBlock == &F.getEntryBlock()) 2755 continue; 2756 2757 // Don't perform PRE on an EH pad. 2758 if (CurrentBlock->isEHPad()) 2759 continue; 2760 2761 for (BasicBlock::iterator BI = CurrentBlock->begin(), 2762 BE = CurrentBlock->end(); 2763 BI != BE;) { 2764 Instruction *CurInst = &*BI++; 2765 Changed |= performScalarPRE(CurInst); 2766 } 2767 } 2768 2769 if (splitCriticalEdges()) 2770 Changed = true; 2771 2772 return Changed; 2773 } 2774 2775 /// Split the critical edge connecting the given two blocks, and return 2776 /// the block inserted to the critical edge. 2777 BasicBlock *GVN::splitCriticalEdges(BasicBlock *Pred, BasicBlock *Succ) { 2778 // GVN does not require loop-simplify, do not try to preserve it if it is not 2779 // possible. 2780 BasicBlock *BB = SplitCriticalEdge( 2781 Pred, Succ, 2782 CriticalEdgeSplittingOptions(DT, LI, MSSAU).unsetPreserveLoopSimplify()); 2783 if (BB) { 2784 if (MD) 2785 MD->invalidateCachedPredecessors(); 2786 InvalidBlockRPONumbers = true; 2787 } 2788 return BB; 2789 } 2790 2791 /// Split critical edges found during the previous 2792 /// iteration that may enable further optimization. 2793 bool GVN::splitCriticalEdges() { 2794 if (toSplit.empty()) 2795 return false; 2796 2797 bool Changed = false; 2798 do { 2799 std::pair<Instruction *, unsigned> Edge = toSplit.pop_back_val(); 2800 Changed |= SplitCriticalEdge(Edge.first, Edge.second, 2801 CriticalEdgeSplittingOptions(DT, LI, MSSAU)) != 2802 nullptr; 2803 } while (!toSplit.empty()); 2804 if (Changed) { 2805 if (MD) 2806 MD->invalidateCachedPredecessors(); 2807 InvalidBlockRPONumbers = true; 2808 } 2809 return Changed; 2810 } 2811 2812 /// Executes one iteration of GVN 2813 bool GVN::iterateOnFunction(Function &F) { 2814 cleanupGlobalSets(); 2815 2816 // Top-down walk of the dominator tree 2817 bool Changed = false; 2818 // Needed for value numbering with phi construction to work. 2819 // RPOT walks the graph in its constructor and will not be invalidated during 2820 // processBlock. 2821 ReversePostOrderTraversal<Function *> RPOT(&F); 2822 2823 for (BasicBlock *BB : RPOT) 2824 Changed |= processBlock(BB); 2825 2826 return Changed; 2827 } 2828 2829 void GVN::cleanupGlobalSets() { 2830 VN.clear(); 2831 LeaderTable.clear(); 2832 BlockRPONumber.clear(); 2833 TableAllocator.Reset(); 2834 ICF->clear(); 2835 InvalidBlockRPONumbers = true; 2836 } 2837 2838 /// Verify that the specified instruction does not occur in our 2839 /// internal data structures. 2840 void GVN::verifyRemoved(const Instruction *Inst) const { 2841 VN.verifyRemoved(Inst); 2842 2843 // Walk through the value number scope to make sure the instruction isn't 2844 // ferreted away in it. 2845 for (const auto &I : LeaderTable) { 2846 const LeaderTableEntry *Node = &I.second; 2847 assert(Node->Val != Inst && "Inst still in value numbering scope!"); 2848 2849 while (Node->Next) { 2850 Node = Node->Next; 2851 assert(Node->Val != Inst && "Inst still in value numbering scope!"); 2852 } 2853 } 2854 } 2855 2856 /// BB is declared dead, which implied other blocks become dead as well. This 2857 /// function is to add all these blocks to "DeadBlocks". For the dead blocks' 2858 /// live successors, update their phi nodes by replacing the operands 2859 /// corresponding to dead blocks with UndefVal. 2860 void GVN::addDeadBlock(BasicBlock *BB) { 2861 SmallVector<BasicBlock *, 4> NewDead; 2862 SmallSetVector<BasicBlock *, 4> DF; 2863 2864 NewDead.push_back(BB); 2865 while (!NewDead.empty()) { 2866 BasicBlock *D = NewDead.pop_back_val(); 2867 if (DeadBlocks.count(D)) 2868 continue; 2869 2870 // All blocks dominated by D are dead. 2871 SmallVector<BasicBlock *, 8> Dom; 2872 DT->getDescendants(D, Dom); 2873 DeadBlocks.insert(Dom.begin(), Dom.end()); 2874 2875 // Figure out the dominance-frontier(D). 2876 for (BasicBlock *B : Dom) { 2877 for (BasicBlock *S : successors(B)) { 2878 if (DeadBlocks.count(S)) 2879 continue; 2880 2881 bool AllPredDead = true; 2882 for (BasicBlock *P : predecessors(S)) 2883 if (!DeadBlocks.count(P)) { 2884 AllPredDead = false; 2885 break; 2886 } 2887 2888 if (!AllPredDead) { 2889 // S could be proved dead later on. That is why we don't update phi 2890 // operands at this moment. 2891 DF.insert(S); 2892 } else { 2893 // While S is not dominated by D, it is dead by now. This could take 2894 // place if S already have a dead predecessor before D is declared 2895 // dead. 2896 NewDead.push_back(S); 2897 } 2898 } 2899 } 2900 } 2901 2902 // For the dead blocks' live successors, update their phi nodes by replacing 2903 // the operands corresponding to dead blocks with UndefVal. 2904 for (BasicBlock *B : DF) { 2905 if (DeadBlocks.count(B)) 2906 continue; 2907 2908 // First, split the critical edges. This might also create additional blocks 2909 // to preserve LoopSimplify form and adjust edges accordingly. 2910 SmallVector<BasicBlock *, 4> Preds(predecessors(B)); 2911 for (BasicBlock *P : Preds) { 2912 if (!DeadBlocks.count(P)) 2913 continue; 2914 2915 if (llvm::is_contained(successors(P), B) && 2916 isCriticalEdge(P->getTerminator(), B)) { 2917 if (BasicBlock *S = splitCriticalEdges(P, B)) 2918 DeadBlocks.insert(P = S); 2919 } 2920 } 2921 2922 // Now undef the incoming values from the dead predecessors. 2923 for (BasicBlock *P : predecessors(B)) { 2924 if (!DeadBlocks.count(P)) 2925 continue; 2926 for (PHINode &Phi : B->phis()) { 2927 Phi.setIncomingValueForBlock(P, UndefValue::get(Phi.getType())); 2928 if (MD) 2929 MD->invalidateCachedPointerInfo(&Phi); 2930 } 2931 } 2932 } 2933 } 2934 2935 // If the given branch is recognized as a foldable branch (i.e. conditional 2936 // branch with constant condition), it will perform following analyses and 2937 // transformation. 2938 // 1) If the dead out-coming edge is a critical-edge, split it. Let 2939 // R be the target of the dead out-coming edge. 2940 // 1) Identify the set of dead blocks implied by the branch's dead outcoming 2941 // edge. The result of this step will be {X| X is dominated by R} 2942 // 2) Identify those blocks which haves at least one dead predecessor. The 2943 // result of this step will be dominance-frontier(R). 2944 // 3) Update the PHIs in DF(R) by replacing the operands corresponding to 2945 // dead blocks with "UndefVal" in an hope these PHIs will optimized away. 2946 // 2947 // Return true iff *NEW* dead code are found. 2948 bool GVN::processFoldableCondBr(BranchInst *BI) { 2949 if (!BI || BI->isUnconditional()) 2950 return false; 2951 2952 // If a branch has two identical successors, we cannot declare either dead. 2953 if (BI->getSuccessor(0) == BI->getSuccessor(1)) 2954 return false; 2955 2956 ConstantInt *Cond = dyn_cast<ConstantInt>(BI->getCondition()); 2957 if (!Cond) 2958 return false; 2959 2960 BasicBlock *DeadRoot = 2961 Cond->getZExtValue() ? BI->getSuccessor(1) : BI->getSuccessor(0); 2962 if (DeadBlocks.count(DeadRoot)) 2963 return false; 2964 2965 if (!DeadRoot->getSinglePredecessor()) 2966 DeadRoot = splitCriticalEdges(BI->getParent(), DeadRoot); 2967 2968 addDeadBlock(DeadRoot); 2969 return true; 2970 } 2971 2972 // performPRE() will trigger assert if it comes across an instruction without 2973 // associated val-num. As it normally has far more live instructions than dead 2974 // instructions, it makes more sense just to "fabricate" a val-number for the 2975 // dead code than checking if instruction involved is dead or not. 2976 void GVN::assignValNumForDeadCode() { 2977 for (BasicBlock *BB : DeadBlocks) { 2978 for (Instruction &Inst : *BB) { 2979 unsigned ValNum = VN.lookupOrAdd(&Inst); 2980 addToLeaderTable(ValNum, &Inst, BB); 2981 } 2982 } 2983 } 2984 2985 class llvm::gvn::GVNLegacyPass : public FunctionPass { 2986 public: 2987 static char ID; // Pass identification, replacement for typeid 2988 2989 explicit GVNLegacyPass(bool NoMemDepAnalysis = !GVNEnableMemDep) 2990 : FunctionPass(ID), Impl(GVNOptions().setMemDep(!NoMemDepAnalysis)) { 2991 initializeGVNLegacyPassPass(*PassRegistry::getPassRegistry()); 2992 } 2993 2994 bool runOnFunction(Function &F) override { 2995 if (skipFunction(F)) 2996 return false; 2997 2998 auto *LIWP = getAnalysisIfAvailable<LoopInfoWrapperPass>(); 2999 3000 auto *MSSAWP = getAnalysisIfAvailable<MemorySSAWrapperPass>(); 3001 return Impl.runImpl( 3002 F, getAnalysis<AssumptionCacheTracker>().getAssumptionCache(F), 3003 getAnalysis<DominatorTreeWrapperPass>().getDomTree(), 3004 getAnalysis<TargetLibraryInfoWrapperPass>().getTLI(F), 3005 getAnalysis<AAResultsWrapperPass>().getAAResults(), 3006 Impl.isMemDepEnabled() 3007 ? &getAnalysis<MemoryDependenceWrapperPass>().getMemDep() 3008 : nullptr, 3009 LIWP ? &LIWP->getLoopInfo() : nullptr, 3010 &getAnalysis<OptimizationRemarkEmitterWrapperPass>().getORE(), 3011 MSSAWP ? &MSSAWP->getMSSA() : nullptr); 3012 } 3013 3014 void getAnalysisUsage(AnalysisUsage &AU) const override { 3015 AU.addRequired<AssumptionCacheTracker>(); 3016 AU.addRequired<DominatorTreeWrapperPass>(); 3017 AU.addRequired<TargetLibraryInfoWrapperPass>(); 3018 AU.addRequired<LoopInfoWrapperPass>(); 3019 if (Impl.isMemDepEnabled()) 3020 AU.addRequired<MemoryDependenceWrapperPass>(); 3021 AU.addRequired<AAResultsWrapperPass>(); 3022 AU.addPreserved<DominatorTreeWrapperPass>(); 3023 AU.addPreserved<GlobalsAAWrapperPass>(); 3024 AU.addPreserved<TargetLibraryInfoWrapperPass>(); 3025 AU.addPreserved<LoopInfoWrapperPass>(); 3026 AU.addRequired<OptimizationRemarkEmitterWrapperPass>(); 3027 AU.addPreserved<MemorySSAWrapperPass>(); 3028 } 3029 3030 private: 3031 GVN Impl; 3032 }; 3033 3034 char GVNLegacyPass::ID = 0; 3035 3036 INITIALIZE_PASS_BEGIN(GVNLegacyPass, "gvn", "Global Value Numbering", false, false) 3037 INITIALIZE_PASS_DEPENDENCY(AssumptionCacheTracker) 3038 INITIALIZE_PASS_DEPENDENCY(MemoryDependenceWrapperPass) 3039 INITIALIZE_PASS_DEPENDENCY(DominatorTreeWrapperPass) 3040 INITIALIZE_PASS_DEPENDENCY(TargetLibraryInfoWrapperPass) 3041 INITIALIZE_PASS_DEPENDENCY(AAResultsWrapperPass) 3042 INITIALIZE_PASS_DEPENDENCY(GlobalsAAWrapperPass) 3043 INITIALIZE_PASS_DEPENDENCY(OptimizationRemarkEmitterWrapperPass) 3044 INITIALIZE_PASS_END(GVNLegacyPass, "gvn", "Global Value Numbering", false, false) 3045 3046 // The public interface to this file... 3047 FunctionPass *llvm::createGVNPass(bool NoMemDepAnalysis) { 3048 return new GVNLegacyPass(NoMemDepAnalysis); 3049 } 3050