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