]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - contrib/llvm/include/llvm/Analysis/ScalarEvolution.h
Merge llvm, clang, lld, lldb, compiler-rt and libc++ r304149, and update
[FreeBSD/FreeBSD.git] / contrib / llvm / include / llvm / Analysis / ScalarEvolution.h
1 //===- llvm/Analysis/ScalarEvolution.h - Scalar Evolution -------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // The ScalarEvolution class is an LLVM pass which can be used to analyze and
11 // categorize scalar expressions in loops.  It specializes in recognizing
12 // general induction variables, representing them with the abstract and opaque
13 // SCEV class.  Given this analysis, trip counts of loops and other important
14 // properties can be obtained.
15 //
16 // This analysis is primarily useful for induction variable substitution and
17 // strength reduction.
18 //
19 //===----------------------------------------------------------------------===//
20
21 #ifndef LLVM_ANALYSIS_SCALAREVOLUTION_H
22 #define LLVM_ANALYSIS_SCALAREVOLUTION_H
23
24 #include "llvm/ADT/DenseSet.h"
25 #include "llvm/ADT/FoldingSet.h"
26 #include "llvm/ADT/SetVector.h"
27 #include "llvm/Analysis/LoopInfo.h"
28 #include "llvm/IR/ConstantRange.h"
29 #include "llvm/IR/Instructions.h"
30 #include "llvm/IR/Operator.h"
31 #include "llvm/IR/PassManager.h"
32 #include "llvm/IR/ValueHandle.h"
33 #include "llvm/IR/ValueMap.h"
34 #include "llvm/Pass.h"
35 #include "llvm/Support/Allocator.h"
36 #include "llvm/Support/DataTypes.h"
37
38 namespace llvm {
39 class APInt;
40 class AssumptionCache;
41 class Constant;
42 class ConstantInt;
43 class DominatorTree;
44 class Type;
45 class ScalarEvolution;
46 class DataLayout;
47 class TargetLibraryInfo;
48 class LLVMContext;
49 class Operator;
50 class SCEV;
51 class SCEVAddRecExpr;
52 class SCEVConstant;
53 class SCEVExpander;
54 class SCEVPredicate;
55 class SCEVUnknown;
56 class Function;
57
58 template <> struct FoldingSetTrait<SCEV>;
59 template <> struct FoldingSetTrait<SCEVPredicate>;
60
61 /// This class represents an analyzed expression in the program.  These are
62 /// opaque objects that the client is not allowed to do much with directly.
63 ///
64 class SCEV : public FoldingSetNode {
65   friend struct FoldingSetTrait<SCEV>;
66
67   /// A reference to an Interned FoldingSetNodeID for this node.  The
68   /// ScalarEvolution's BumpPtrAllocator holds the data.
69   FoldingSetNodeIDRef FastID;
70
71   // The SCEV baseclass this node corresponds to
72   const unsigned short SCEVType;
73
74 protected:
75   /// This field is initialized to zero and may be used in subclasses to store
76   /// miscellaneous information.
77   unsigned short SubclassData;
78
79 private:
80   SCEV(const SCEV &) = delete;
81   void operator=(const SCEV &) = delete;
82
83 public:
84   /// NoWrapFlags are bitfield indices into SubclassData.
85   ///
86   /// Add and Mul expressions may have no-unsigned-wrap <NUW> or
87   /// no-signed-wrap <NSW> properties, which are derived from the IR
88   /// operator. NSW is a misnomer that we use to mean no signed overflow or
89   /// underflow.
90   ///
91   /// AddRec expressions may have a no-self-wraparound <NW> property if, in
92   /// the integer domain, abs(step) * max-iteration(loop) <=
93   /// unsigned-max(bitwidth).  This means that the recurrence will never reach
94   /// its start value if the step is non-zero.  Computing the same value on
95   /// each iteration is not considered wrapping, and recurrences with step = 0
96   /// are trivially <NW>.  <NW> is independent of the sign of step and the
97   /// value the add recurrence starts with.
98   ///
99   /// Note that NUW and NSW are also valid properties of a recurrence, and
100   /// either implies NW. For convenience, NW will be set for a recurrence
101   /// whenever either NUW or NSW are set.
102   enum NoWrapFlags {
103     FlagAnyWrap = 0,    // No guarantee.
104     FlagNW = (1 << 0),  // No self-wrap.
105     FlagNUW = (1 << 1), // No unsigned wrap.
106     FlagNSW = (1 << 2), // No signed wrap.
107     NoWrapMask = (1 << 3) - 1
108   };
109
110   explicit SCEV(const FoldingSetNodeIDRef ID, unsigned SCEVTy)
111       : FastID(ID), SCEVType(SCEVTy), SubclassData(0) {}
112
113   unsigned getSCEVType() const { return SCEVType; }
114
115   /// Return the LLVM type of this SCEV expression.
116   ///
117   Type *getType() const;
118
119   /// Return true if the expression is a constant zero.
120   ///
121   bool isZero() const;
122
123   /// Return true if the expression is a constant one.
124   ///
125   bool isOne() const;
126
127   /// Return true if the expression is a constant all-ones value.
128   ///
129   bool isAllOnesValue() const;
130
131   /// Return true if the specified scev is negated, but not a constant.
132   bool isNonConstantNegative() const;
133
134   /// Print out the internal representation of this scalar to the specified
135   /// stream.  This should really only be used for debugging purposes.
136   void print(raw_ostream &OS) const;
137
138   /// This method is used for debugging.
139   ///
140   void dump() const;
141 };
142
143 // Specialize FoldingSetTrait for SCEV to avoid needing to compute
144 // temporary FoldingSetNodeID values.
145 template <> struct FoldingSetTrait<SCEV> : DefaultFoldingSetTrait<SCEV> {
146   static void Profile(const SCEV &X, FoldingSetNodeID &ID) { ID = X.FastID; }
147   static bool Equals(const SCEV &X, const FoldingSetNodeID &ID, unsigned IDHash,
148                      FoldingSetNodeID &TempID) {
149     return ID == X.FastID;
150   }
151   static unsigned ComputeHash(const SCEV &X, FoldingSetNodeID &TempID) {
152     return X.FastID.ComputeHash();
153   }
154 };
155
156 inline raw_ostream &operator<<(raw_ostream &OS, const SCEV &S) {
157   S.print(OS);
158   return OS;
159 }
160
161 /// An object of this class is returned by queries that could not be answered.
162 /// For example, if you ask for the number of iterations of a linked-list
163 /// traversal loop, you will get one of these.  None of the standard SCEV
164 /// operations are valid on this class, it is just a marker.
165 struct SCEVCouldNotCompute : public SCEV {
166   SCEVCouldNotCompute();
167
168   /// Methods for support type inquiry through isa, cast, and dyn_cast:
169   static bool classof(const SCEV *S);
170 };
171
172 /// This class represents an assumption made using SCEV expressions which can
173 /// be checked at run-time.
174 class SCEVPredicate : public FoldingSetNode {
175   friend struct FoldingSetTrait<SCEVPredicate>;
176
177   /// A reference to an Interned FoldingSetNodeID for this node.  The
178   /// ScalarEvolution's BumpPtrAllocator holds the data.
179   FoldingSetNodeIDRef FastID;
180
181 public:
182   enum SCEVPredicateKind { P_Union, P_Equal, P_Wrap };
183
184 protected:
185   SCEVPredicateKind Kind;
186   ~SCEVPredicate() = default;
187   SCEVPredicate(const SCEVPredicate &) = default;
188   SCEVPredicate &operator=(const SCEVPredicate &) = default;
189
190 public:
191   SCEVPredicate(const FoldingSetNodeIDRef ID, SCEVPredicateKind Kind);
192
193   SCEVPredicateKind getKind() const { return Kind; }
194
195   /// Returns the estimated complexity of this predicate.  This is roughly
196   /// measured in the number of run-time checks required.
197   virtual unsigned getComplexity() const { return 1; }
198
199   /// Returns true if the predicate is always true. This means that no
200   /// assumptions were made and nothing needs to be checked at run-time.
201   virtual bool isAlwaysTrue() const = 0;
202
203   /// Returns true if this predicate implies \p N.
204   virtual bool implies(const SCEVPredicate *N) const = 0;
205
206   /// Prints a textual representation of this predicate with an indentation of
207   /// \p Depth.
208   virtual void print(raw_ostream &OS, unsigned Depth = 0) const = 0;
209
210   /// Returns the SCEV to which this predicate applies, or nullptr if this is
211   /// a SCEVUnionPredicate.
212   virtual const SCEV *getExpr() const = 0;
213 };
214
215 inline raw_ostream &operator<<(raw_ostream &OS, const SCEVPredicate &P) {
216   P.print(OS);
217   return OS;
218 }
219
220 // Specialize FoldingSetTrait for SCEVPredicate to avoid needing to compute
221 // temporary FoldingSetNodeID values.
222 template <>
223 struct FoldingSetTrait<SCEVPredicate> : DefaultFoldingSetTrait<SCEVPredicate> {
224
225   static void Profile(const SCEVPredicate &X, FoldingSetNodeID &ID) {
226     ID = X.FastID;
227   }
228
229   static bool Equals(const SCEVPredicate &X, const FoldingSetNodeID &ID,
230                      unsigned IDHash, FoldingSetNodeID &TempID) {
231     return ID == X.FastID;
232   }
233   static unsigned ComputeHash(const SCEVPredicate &X,
234                               FoldingSetNodeID &TempID) {
235     return X.FastID.ComputeHash();
236   }
237 };
238
239 /// This class represents an assumption that two SCEV expressions are equal,
240 /// and this can be checked at run-time. We assume that the left hand side is
241 /// a SCEVUnknown and the right hand side a constant.
242 class SCEVEqualPredicate final : public SCEVPredicate {
243   /// We assume that LHS == RHS, where LHS is a SCEVUnknown and RHS a
244   /// constant.
245   const SCEVUnknown *LHS;
246   const SCEVConstant *RHS;
247
248 public:
249   SCEVEqualPredicate(const FoldingSetNodeIDRef ID, const SCEVUnknown *LHS,
250                      const SCEVConstant *RHS);
251
252   /// Implementation of the SCEVPredicate interface
253   bool implies(const SCEVPredicate *N) const override;
254   void print(raw_ostream &OS, unsigned Depth = 0) const override;
255   bool isAlwaysTrue() const override;
256   const SCEV *getExpr() const override;
257
258   /// Returns the left hand side of the equality.
259   const SCEVUnknown *getLHS() const { return LHS; }
260
261   /// Returns the right hand side of the equality.
262   const SCEVConstant *getRHS() const { return RHS; }
263
264   /// Methods for support type inquiry through isa, cast, and dyn_cast:
265   static inline bool classof(const SCEVPredicate *P) {
266     return P->getKind() == P_Equal;
267   }
268 };
269
270 /// This class represents an assumption made on an AddRec expression. Given an
271 /// affine AddRec expression {a,+,b}, we assume that it has the nssw or nusw
272 /// flags (defined below) in the first X iterations of the loop, where X is a
273 /// SCEV expression returned by getPredicatedBackedgeTakenCount).
274 ///
275 /// Note that this does not imply that X is equal to the backedge taken
276 /// count. This means that if we have a nusw predicate for i32 {0,+,1} with a
277 /// predicated backedge taken count of X, we only guarantee that {0,+,1} has
278 /// nusw in the first X iterations. {0,+,1} may still wrap in the loop if we
279 /// have more than X iterations.
280 class SCEVWrapPredicate final : public SCEVPredicate {
281 public:
282   /// Similar to SCEV::NoWrapFlags, but with slightly different semantics
283   /// for FlagNUSW. The increment is considered to be signed, and a + b
284   /// (where b is the increment) is considered to wrap if:
285   ///    zext(a + b) != zext(a) + sext(b)
286   ///
287   /// If Signed is a function that takes an n-bit tuple and maps to the
288   /// integer domain as the tuples value interpreted as twos complement,
289   /// and Unsigned a function that takes an n-bit tuple and maps to the
290   /// integer domain as as the base two value of input tuple, then a + b
291   /// has IncrementNUSW iff:
292   ///
293   /// 0 <= Unsigned(a) + Signed(b) < 2^n
294   ///
295   /// The IncrementNSSW flag has identical semantics with SCEV::FlagNSW.
296   ///
297   /// Note that the IncrementNUSW flag is not commutative: if base + inc
298   /// has IncrementNUSW, then inc + base doesn't neccessarily have this
299   /// property. The reason for this is that this is used for sign/zero
300   /// extending affine AddRec SCEV expressions when a SCEVWrapPredicate is
301   /// assumed. A {base,+,inc} expression is already non-commutative with
302   /// regards to base and inc, since it is interpreted as:
303   ///     (((base + inc) + inc) + inc) ...
304   enum IncrementWrapFlags {
305     IncrementAnyWrap = 0,     // No guarantee.
306     IncrementNUSW = (1 << 0), // No unsigned with signed increment wrap.
307     IncrementNSSW = (1 << 1), // No signed with signed increment wrap
308                               // (equivalent with SCEV::NSW)
309     IncrementNoWrapMask = (1 << 2) - 1
310   };
311
312   /// Convenient IncrementWrapFlags manipulation methods.
313   LLVM_NODISCARD static SCEVWrapPredicate::IncrementWrapFlags
314   clearFlags(SCEVWrapPredicate::IncrementWrapFlags Flags,
315              SCEVWrapPredicate::IncrementWrapFlags OffFlags) {
316     assert((Flags & IncrementNoWrapMask) == Flags && "Invalid flags value!");
317     assert((OffFlags & IncrementNoWrapMask) == OffFlags &&
318            "Invalid flags value!");
319     return (SCEVWrapPredicate::IncrementWrapFlags)(Flags & ~OffFlags);
320   }
321
322   LLVM_NODISCARD static SCEVWrapPredicate::IncrementWrapFlags
323   maskFlags(SCEVWrapPredicate::IncrementWrapFlags Flags, int Mask) {
324     assert((Flags & IncrementNoWrapMask) == Flags && "Invalid flags value!");
325     assert((Mask & IncrementNoWrapMask) == Mask && "Invalid mask value!");
326
327     return (SCEVWrapPredicate::IncrementWrapFlags)(Flags & Mask);
328   }
329
330   LLVM_NODISCARD static SCEVWrapPredicate::IncrementWrapFlags
331   setFlags(SCEVWrapPredicate::IncrementWrapFlags Flags,
332            SCEVWrapPredicate::IncrementWrapFlags OnFlags) {
333     assert((Flags & IncrementNoWrapMask) == Flags && "Invalid flags value!");
334     assert((OnFlags & IncrementNoWrapMask) == OnFlags &&
335            "Invalid flags value!");
336
337     return (SCEVWrapPredicate::IncrementWrapFlags)(Flags | OnFlags);
338   }
339
340   /// Returns the set of SCEVWrapPredicate no wrap flags implied by a
341   /// SCEVAddRecExpr.
342   LLVM_NODISCARD static SCEVWrapPredicate::IncrementWrapFlags
343   getImpliedFlags(const SCEVAddRecExpr *AR, ScalarEvolution &SE);
344
345 private:
346   const SCEVAddRecExpr *AR;
347   IncrementWrapFlags Flags;
348
349 public:
350   explicit SCEVWrapPredicate(const FoldingSetNodeIDRef ID,
351                              const SCEVAddRecExpr *AR,
352                              IncrementWrapFlags Flags);
353
354   /// Returns the set assumed no overflow flags.
355   IncrementWrapFlags getFlags() const { return Flags; }
356   /// Implementation of the SCEVPredicate interface
357   const SCEV *getExpr() const override;
358   bool implies(const SCEVPredicate *N) const override;
359   void print(raw_ostream &OS, unsigned Depth = 0) const override;
360   bool isAlwaysTrue() const override;
361
362   /// Methods for support type inquiry through isa, cast, and dyn_cast:
363   static inline bool classof(const SCEVPredicate *P) {
364     return P->getKind() == P_Wrap;
365   }
366 };
367
368 /// This class represents a composition of other SCEV predicates, and is the
369 /// class that most clients will interact with.  This is equivalent to a
370 /// logical "AND" of all the predicates in the union.
371 ///
372 /// NB! Unlike other SCEVPredicate sub-classes this class does not live in the
373 /// ScalarEvolution::Preds folding set.  This is why the \c add function is sound.
374 class SCEVUnionPredicate final : public SCEVPredicate {
375 private:
376   typedef DenseMap<const SCEV *, SmallVector<const SCEVPredicate *, 4>>
377       PredicateMap;
378
379   /// Vector with references to all predicates in this union.
380   SmallVector<const SCEVPredicate *, 16> Preds;
381   /// Maps SCEVs to predicates for quick look-ups.
382   PredicateMap SCEVToPreds;
383
384 public:
385   SCEVUnionPredicate();
386
387   const SmallVectorImpl<const SCEVPredicate *> &getPredicates() const {
388     return Preds;
389   }
390
391   /// Adds a predicate to this union.
392   void add(const SCEVPredicate *N);
393
394   /// Returns a reference to a vector containing all predicates which apply to
395   /// \p Expr.
396   ArrayRef<const SCEVPredicate *> getPredicatesForExpr(const SCEV *Expr);
397
398   /// Implementation of the SCEVPredicate interface
399   bool isAlwaysTrue() const override;
400   bool implies(const SCEVPredicate *N) const override;
401   void print(raw_ostream &OS, unsigned Depth) const override;
402   const SCEV *getExpr() const override;
403
404   /// We estimate the complexity of a union predicate as the size number of
405   /// predicates in the union.
406   unsigned getComplexity() const override { return Preds.size(); }
407
408   /// Methods for support type inquiry through isa, cast, and dyn_cast:
409   static inline bool classof(const SCEVPredicate *P) {
410     return P->getKind() == P_Union;
411   }
412 };
413
414 /// The main scalar evolution driver. Because client code (intentionally)
415 /// can't do much with the SCEV objects directly, they must ask this class
416 /// for services.
417 class ScalarEvolution {
418 public:
419   /// An enum describing the relationship between a SCEV and a loop.
420   enum LoopDisposition {
421     LoopVariant,   ///< The SCEV is loop-variant (unknown).
422     LoopInvariant, ///< The SCEV is loop-invariant.
423     LoopComputable ///< The SCEV varies predictably with the loop.
424   };
425
426   /// An enum describing the relationship between a SCEV and a basic block.
427   enum BlockDisposition {
428     DoesNotDominateBlock,  ///< The SCEV does not dominate the block.
429     DominatesBlock,        ///< The SCEV dominates the block.
430     ProperlyDominatesBlock ///< The SCEV properly dominates the block.
431   };
432
433   /// Convenient NoWrapFlags manipulation that hides enum casts and is
434   /// visible in the ScalarEvolution name space.
435   LLVM_NODISCARD static SCEV::NoWrapFlags maskFlags(SCEV::NoWrapFlags Flags,
436                                                     int Mask) {
437     return (SCEV::NoWrapFlags)(Flags & Mask);
438   }
439   LLVM_NODISCARD static SCEV::NoWrapFlags setFlags(SCEV::NoWrapFlags Flags,
440                                                    SCEV::NoWrapFlags OnFlags) {
441     return (SCEV::NoWrapFlags)(Flags | OnFlags);
442   }
443   LLVM_NODISCARD static SCEV::NoWrapFlags
444   clearFlags(SCEV::NoWrapFlags Flags, SCEV::NoWrapFlags OffFlags) {
445     return (SCEV::NoWrapFlags)(Flags & ~OffFlags);
446   }
447
448 private:
449   /// A CallbackVH to arrange for ScalarEvolution to be notified whenever a
450   /// Value is deleted.
451   class SCEVCallbackVH final : public CallbackVH {
452     ScalarEvolution *SE;
453     void deleted() override;
454     void allUsesReplacedWith(Value *New) override;
455
456   public:
457     SCEVCallbackVH(Value *V, ScalarEvolution *SE = nullptr);
458   };
459
460   friend class SCEVCallbackVH;
461   friend class SCEVExpander;
462   friend class SCEVUnknown;
463
464   /// The function we are analyzing.
465   ///
466   Function &F;
467
468   /// Does the module have any calls to the llvm.experimental.guard intrinsic
469   /// at all?  If this is false, we avoid doing work that will only help if
470   /// thare are guards present in the IR.
471   ///
472   bool HasGuards;
473
474   /// The target library information for the target we are targeting.
475   ///
476   TargetLibraryInfo &TLI;
477
478   /// The tracker for @llvm.assume intrinsics in this function.
479   AssumptionCache &AC;
480
481   /// The dominator tree.
482   ///
483   DominatorTree &DT;
484
485   /// The loop information for the function we are currently analyzing.
486   ///
487   LoopInfo &LI;
488
489   /// This SCEV is used to represent unknown trip counts and things.
490   std::unique_ptr<SCEVCouldNotCompute> CouldNotCompute;
491
492   /// The typedef for HasRecMap.
493   ///
494   typedef DenseMap<const SCEV *, bool> HasRecMapType;
495
496   /// This is a cache to record whether a SCEV contains any scAddRecExpr.
497   HasRecMapType HasRecMap;
498
499   /// The typedef for ExprValueMap.
500   ///
501   typedef std::pair<Value *, ConstantInt *> ValueOffsetPair;
502   typedef DenseMap<const SCEV *, SetVector<ValueOffsetPair>> ExprValueMapType;
503
504   /// ExprValueMap -- This map records the original values from which
505   /// the SCEV expr is generated from.
506   ///
507   /// We want to represent the mapping as SCEV -> ValueOffsetPair instead
508   /// of SCEV -> Value:
509   /// Suppose we know S1 expands to V1, and
510   ///  S1 = S2 + C_a
511   ///  S3 = S2 + C_b
512   /// where C_a and C_b are different SCEVConstants. Then we'd like to
513   /// expand S3 as V1 - C_a + C_b instead of expanding S2 literally.
514   /// It is helpful when S2 is a complex SCEV expr.
515   ///
516   /// In order to do that, we represent ExprValueMap as a mapping from
517   /// SCEV to ValueOffsetPair. We will save both S1->{V1, 0} and
518   /// S2->{V1, C_a} into the map when we create SCEV for V1. When S3
519   /// is expanded, it will first expand S2 to V1 - C_a because of
520   /// S2->{V1, C_a} in the map, then expand S3 to V1 - C_a + C_b.
521   ///
522   /// Note: S->{V, Offset} in the ExprValueMap means S can be expanded
523   /// to V - Offset.
524   ExprValueMapType ExprValueMap;
525
526   /// The typedef for ValueExprMap.
527   ///
528   typedef DenseMap<SCEVCallbackVH, const SCEV *, DenseMapInfo<Value *>>
529       ValueExprMapType;
530
531   /// This is a cache of the values we have analyzed so far.
532   ///
533   ValueExprMapType ValueExprMap;
534
535   /// Mark predicate values currently being processed by isImpliedCond.
536   SmallPtrSet<Value *, 6> PendingLoopPredicates;
537
538   /// Set to true by isLoopBackedgeGuardedByCond when we're walking the set of
539   /// conditions dominating the backedge of a loop.
540   bool WalkingBEDominatingConds;
541
542   /// Set to true by isKnownPredicateViaSplitting when we're trying to prove a
543   /// predicate by splitting it into a set of independent predicates.
544   bool ProvingSplitPredicate;
545
546   /// Memoized values for the GetMinTrailingZeros
547   DenseMap<const SCEV *, uint32_t> MinTrailingZerosCache;
548
549   /// Private helper method for the GetMinTrailingZeros method
550   uint32_t GetMinTrailingZerosImpl(const SCEV *S);
551
552   /// Information about the number of loop iterations for which a loop exit's
553   /// branch condition evaluates to the not-taken path.  This is a temporary
554   /// pair of exact and max expressions that are eventually summarized in
555   /// ExitNotTakenInfo and BackedgeTakenInfo.
556   struct ExitLimit {
557     const SCEV *ExactNotTaken; // The exit is not taken exactly this many times
558     const SCEV *MaxNotTaken; // The exit is not taken at most this many times
559     bool MaxOrZero; // Not taken either exactly MaxNotTaken or zero times
560
561     /// A set of predicate guards for this ExitLimit. The result is only valid
562     /// if all of the predicates in \c Predicates evaluate to 'true' at
563     /// run-time.
564     SmallPtrSet<const SCEVPredicate *, 4> Predicates;
565
566     void addPredicate(const SCEVPredicate *P) {
567       assert(!isa<SCEVUnionPredicate>(P) && "Only add leaf predicates here!");
568       Predicates.insert(P);
569     }
570
571     /*implicit*/ ExitLimit(const SCEV *E);
572
573     ExitLimit(
574         const SCEV *E, const SCEV *M, bool MaxOrZero,
575         ArrayRef<const SmallPtrSetImpl<const SCEVPredicate *> *> PredSetList);
576
577     ExitLimit(const SCEV *E, const SCEV *M, bool MaxOrZero,
578               const SmallPtrSetImpl<const SCEVPredicate *> &PredSet);
579
580     ExitLimit(const SCEV *E, const SCEV *M, bool MaxOrZero);
581
582     /// Test whether this ExitLimit contains any computed information, or
583     /// whether it's all SCEVCouldNotCompute values.
584     bool hasAnyInfo() const {
585       return !isa<SCEVCouldNotCompute>(ExactNotTaken) ||
586              !isa<SCEVCouldNotCompute>(MaxNotTaken);
587     }
588
589     /// Test whether this ExitLimit contains all information.
590     bool hasFullInfo() const {
591       return !isa<SCEVCouldNotCompute>(ExactNotTaken);
592     }
593   };
594
595   /// Information about the number of times a particular loop exit may be
596   /// reached before exiting the loop.
597   struct ExitNotTakenInfo {
598     PoisoningVH<BasicBlock> ExitingBlock;
599     const SCEV *ExactNotTaken;
600     std::unique_ptr<SCEVUnionPredicate> Predicate;
601     bool hasAlwaysTruePredicate() const {
602       return !Predicate || Predicate->isAlwaysTrue();
603     }
604
605     explicit ExitNotTakenInfo(PoisoningVH<BasicBlock> ExitingBlock,
606                               const SCEV *ExactNotTaken,
607                               std::unique_ptr<SCEVUnionPredicate> Predicate)
608         : ExitingBlock(ExitingBlock), ExactNotTaken(ExactNotTaken),
609           Predicate(std::move(Predicate)) {}
610   };
611
612   /// Information about the backedge-taken count of a loop. This currently
613   /// includes an exact count and a maximum count.
614   ///
615   class BackedgeTakenInfo {
616     /// A list of computable exits and their not-taken counts.  Loops almost
617     /// never have more than one computable exit.
618     SmallVector<ExitNotTakenInfo, 1> ExitNotTaken;
619
620     /// The pointer part of \c MaxAndComplete is an expression indicating the
621     /// least maximum backedge-taken count of the loop that is known, or a
622     /// SCEVCouldNotCompute. This expression is only valid if the predicates
623     /// associated with all loop exits are true.
624     ///
625     /// The integer part of \c MaxAndComplete is a boolean indicating if \c
626     /// ExitNotTaken has an element for every exiting block in the loop.
627     PointerIntPair<const SCEV *, 1> MaxAndComplete;
628
629     /// True iff the backedge is taken either exactly Max or zero times.
630     bool MaxOrZero;
631
632     /// \name Helper projection functions on \c MaxAndComplete.
633     /// @{
634     bool isComplete() const { return MaxAndComplete.getInt(); }
635     const SCEV *getMax() const { return MaxAndComplete.getPointer(); }
636     /// @}
637
638   public:
639     BackedgeTakenInfo() : MaxAndComplete(nullptr, 0) {}
640
641     BackedgeTakenInfo(BackedgeTakenInfo &&) = default;
642     BackedgeTakenInfo &operator=(BackedgeTakenInfo &&) = default;
643
644     typedef std::pair<BasicBlock *, ExitLimit> EdgeExitInfo;
645
646     /// Initialize BackedgeTakenInfo from a list of exact exit counts.
647     BackedgeTakenInfo(SmallVectorImpl<EdgeExitInfo> &&ExitCounts, bool Complete,
648                       const SCEV *MaxCount, bool MaxOrZero);
649
650     /// Test whether this BackedgeTakenInfo contains any computed information,
651     /// or whether it's all SCEVCouldNotCompute values.
652     bool hasAnyInfo() const {
653       return !ExitNotTaken.empty() || !isa<SCEVCouldNotCompute>(getMax());
654     }
655
656     /// Test whether this BackedgeTakenInfo contains complete information.
657     bool hasFullInfo() const { return isComplete(); }
658
659     /// Return an expression indicating the exact *backedge-taken*
660     /// count of the loop if it is known or SCEVCouldNotCompute
661     /// otherwise.  If execution makes it to the backedge on every
662     /// iteration (i.e. there are no abnormal exists like exception
663     /// throws and thread exits) then this is the number of times the
664     /// loop header will execute minus one.
665     ///
666     /// If the SCEV predicate associated with the answer can be different
667     /// from AlwaysTrue, we must add a (non null) Predicates argument.
668     /// The SCEV predicate associated with the answer will be added to
669     /// Predicates. A run-time check needs to be emitted for the SCEV
670     /// predicate in order for the answer to be valid.
671     ///
672     /// Note that we should always know if we need to pass a predicate
673     /// argument or not from the way the ExitCounts vector was computed.
674     /// If we allowed SCEV predicates to be generated when populating this
675     /// vector, this information can contain them and therefore a
676     /// SCEVPredicate argument should be added to getExact.
677     const SCEV *getExact(ScalarEvolution *SE,
678                          SCEVUnionPredicate *Predicates = nullptr) const;
679
680     /// Return the number of times this loop exit may fall through to the back
681     /// edge, or SCEVCouldNotCompute. The loop is guaranteed not to exit via
682     /// this block before this number of iterations, but may exit via another
683     /// block.
684     const SCEV *getExact(BasicBlock *ExitingBlock, ScalarEvolution *SE) const;
685
686     /// Get the max backedge taken count for the loop.
687     const SCEV *getMax(ScalarEvolution *SE) const;
688
689     /// Return true if the number of times this backedge is taken is either the
690     /// value returned by getMax or zero.
691     bool isMaxOrZero(ScalarEvolution *SE) const;
692
693     /// Return true if any backedge taken count expressions refer to the given
694     /// subexpression.
695     bool hasOperand(const SCEV *S, ScalarEvolution *SE) const;
696
697     /// Invalidate this result and free associated memory.
698     void clear();
699   };
700
701   /// Cache the backedge-taken count of the loops for this function as they
702   /// are computed.
703   DenseMap<const Loop *, BackedgeTakenInfo> BackedgeTakenCounts;
704
705   /// Cache the predicated backedge-taken count of the loops for this
706   /// function as they are computed.
707   DenseMap<const Loop *, BackedgeTakenInfo> PredicatedBackedgeTakenCounts;
708
709   /// This map contains entries for all of the PHI instructions that we
710   /// attempt to compute constant evolutions for.  This allows us to avoid
711   /// potentially expensive recomputation of these properties.  An instruction
712   /// maps to null if we are unable to compute its exit value.
713   DenseMap<PHINode *, Constant *> ConstantEvolutionLoopExitValue;
714
715   /// This map contains entries for all the expressions that we attempt to
716   /// compute getSCEVAtScope information for, which can be expensive in
717   /// extreme cases.
718   DenseMap<const SCEV *, SmallVector<std::pair<const Loop *, const SCEV *>, 2>>
719       ValuesAtScopes;
720
721   /// Memoized computeLoopDisposition results.
722   DenseMap<const SCEV *,
723            SmallVector<PointerIntPair<const Loop *, 2, LoopDisposition>, 2>>
724       LoopDispositions;
725
726   struct LoopProperties {
727     /// Set to true if the loop contains no instruction that can have side
728     /// effects (i.e. via throwing an exception, volatile or atomic access).
729     bool HasNoAbnormalExits;
730
731     /// Set to true if the loop contains no instruction that can abnormally exit
732     /// the loop (i.e. via throwing an exception, by terminating the thread
733     /// cleanly or by infinite looping in a called function).  Strictly
734     /// speaking, the last one is not leaving the loop, but is identical to
735     /// leaving the loop for reasoning about undefined behavior.
736     bool HasNoSideEffects;
737   };
738
739   /// Cache for \c getLoopProperties.
740   DenseMap<const Loop *, LoopProperties> LoopPropertiesCache;
741
742   /// Return a \c LoopProperties instance for \p L, creating one if necessary.
743   LoopProperties getLoopProperties(const Loop *L);
744
745   bool loopHasNoSideEffects(const Loop *L) {
746     return getLoopProperties(L).HasNoSideEffects;
747   }
748
749   bool loopHasNoAbnormalExits(const Loop *L) {
750     return getLoopProperties(L).HasNoAbnormalExits;
751   }
752
753   /// Compute a LoopDisposition value.
754   LoopDisposition computeLoopDisposition(const SCEV *S, const Loop *L);
755
756   /// Memoized computeBlockDisposition results.
757   DenseMap<
758       const SCEV *,
759       SmallVector<PointerIntPair<const BasicBlock *, 2, BlockDisposition>, 2>>
760       BlockDispositions;
761
762   /// Compute a BlockDisposition value.
763   BlockDisposition computeBlockDisposition(const SCEV *S, const BasicBlock *BB);
764
765   /// Memoized results from getRange
766   DenseMap<const SCEV *, ConstantRange> UnsignedRanges;
767
768   /// Memoized results from getRange
769   DenseMap<const SCEV *, ConstantRange> SignedRanges;
770
771   /// Used to parameterize getRange
772   enum RangeSignHint { HINT_RANGE_UNSIGNED, HINT_RANGE_SIGNED };
773
774   /// Set the memoized range for the given SCEV.
775   const ConstantRange &setRange(const SCEV *S, RangeSignHint Hint,
776                                 ConstantRange CR) {
777     DenseMap<const SCEV *, ConstantRange> &Cache =
778         Hint == HINT_RANGE_UNSIGNED ? UnsignedRanges : SignedRanges;
779
780     auto Pair = Cache.try_emplace(S, std::move(CR));
781     if (!Pair.second)
782       Pair.first->second = std::move(CR);
783     return Pair.first->second;
784   }
785
786   /// Determine the range for a particular SCEV.
787   ConstantRange getRange(const SCEV *S, RangeSignHint Hint);
788
789   /// Determines the range for the affine SCEVAddRecExpr {\p Start,+,\p Stop}.
790   /// Helper for \c getRange.
791   ConstantRange getRangeForAffineAR(const SCEV *Start, const SCEV *Stop,
792                                     const SCEV *MaxBECount, unsigned BitWidth);
793
794   /// Try to compute a range for the affine SCEVAddRecExpr {\p Start,+,\p
795   /// Stop} by "factoring out" a ternary expression from the add recurrence.
796   /// Helper called by \c getRange.
797   ConstantRange getRangeViaFactoring(const SCEV *Start, const SCEV *Stop,
798                                      const SCEV *MaxBECount, unsigned BitWidth);
799
800   /// We know that there is no SCEV for the specified value.  Analyze the
801   /// expression.
802   const SCEV *createSCEV(Value *V);
803
804   /// Provide the special handling we need to analyze PHI SCEVs.
805   const SCEV *createNodeForPHI(PHINode *PN);
806
807   /// Helper function called from createNodeForPHI.
808   const SCEV *createAddRecFromPHI(PHINode *PN);
809
810   /// A helper function for createAddRecFromPHI to handle simple cases.
811   const SCEV *createSimpleAffineAddRec(PHINode *PN, Value *BEValueV,
812                                             Value *StartValueV);
813
814   /// Helper function called from createNodeForPHI.
815   const SCEV *createNodeFromSelectLikePHI(PHINode *PN);
816
817   /// Provide special handling for a select-like instruction (currently this
818   /// is either a select instruction or a phi node).  \p I is the instruction
819   /// being processed, and it is assumed equivalent to "Cond ? TrueVal :
820   /// FalseVal".
821   const SCEV *createNodeForSelectOrPHI(Instruction *I, Value *Cond,
822                                        Value *TrueVal, Value *FalseVal);
823
824   /// Provide the special handling we need to analyze GEP SCEVs.
825   const SCEV *createNodeForGEP(GEPOperator *GEP);
826
827   /// Implementation code for getSCEVAtScope; called at most once for each
828   /// SCEV+Loop pair.
829   ///
830   const SCEV *computeSCEVAtScope(const SCEV *S, const Loop *L);
831
832   /// This looks up computed SCEV values for all instructions that depend on
833   /// the given instruction and removes them from the ValueExprMap map if they
834   /// reference SymName. This is used during PHI resolution.
835   void forgetSymbolicName(Instruction *I, const SCEV *SymName);
836
837   /// Return the BackedgeTakenInfo for the given loop, lazily computing new
838   /// values if the loop hasn't been analyzed yet. The returned result is
839   /// guaranteed not to be predicated.
840   const BackedgeTakenInfo &getBackedgeTakenInfo(const Loop *L);
841
842   /// Similar to getBackedgeTakenInfo, but will add predicates as required
843   /// with the purpose of returning complete information.
844   const BackedgeTakenInfo &getPredicatedBackedgeTakenInfo(const Loop *L);
845
846   /// Compute the number of times the specified loop will iterate.
847   /// If AllowPredicates is set, we will create new SCEV predicates as
848   /// necessary in order to return an exact answer.
849   BackedgeTakenInfo computeBackedgeTakenCount(const Loop *L,
850                                               bool AllowPredicates = false);
851
852   /// Compute the number of times the backedge of the specified loop will
853   /// execute if it exits via the specified block. If AllowPredicates is set,
854   /// this call will try to use a minimal set of SCEV predicates in order to
855   /// return an exact answer.
856   ExitLimit computeExitLimit(const Loop *L, BasicBlock *ExitingBlock,
857                              bool AllowPredicates = false);
858
859   /// Compute the number of times the backedge of the specified loop will
860   /// execute if its exit condition were a conditional branch of ExitCond,
861   /// TBB, and FBB.
862   ///
863   /// \p ControlsExit is true if ExitCond directly controls the exit
864   /// branch. In this case, we can assume that the loop exits only if the
865   /// condition is true and can infer that failing to meet the condition prior
866   /// to integer wraparound results in undefined behavior.
867   ///
868   /// If \p AllowPredicates is set, this call will try to use a minimal set of
869   /// SCEV predicates in order to return an exact answer.
870   ExitLimit computeExitLimitFromCond(const Loop *L, Value *ExitCond,
871                                      BasicBlock *TBB, BasicBlock *FBB,
872                                      bool ControlsExit,
873                                      bool AllowPredicates = false);
874
875   // Helper functions for computeExitLimitFromCond to avoid exponential time
876   // complexity.
877
878   class ExitLimitCache {
879     // It may look like we need key on the whole (L, TBB, FBB, ControlsExit,
880     // AllowPredicates) tuple, but recursive calls to
881     // computeExitLimitFromCondCached from computeExitLimitFromCondImpl only
882     // vary the in \c ExitCond and \c ControlsExit parameters.  We remember the
883     // initial values of the other values to assert our assumption.
884     SmallDenseMap<PointerIntPair<Value *, 1>, ExitLimit> TripCountMap;
885
886     const Loop *L;
887     BasicBlock *TBB;
888     BasicBlock *FBB;
889     bool AllowPredicates;
890
891   public:
892     ExitLimitCache(const Loop *L, BasicBlock *TBB, BasicBlock *FBB,
893                    bool AllowPredicates)
894         : L(L), TBB(TBB), FBB(FBB), AllowPredicates(AllowPredicates) {}
895
896     Optional<ExitLimit> find(const Loop *L, Value *ExitCond, BasicBlock *TBB,
897                              BasicBlock *FBB, bool ControlsExit,
898                              bool AllowPredicates);
899
900     void insert(const Loop *L, Value *ExitCond, BasicBlock *TBB,
901                 BasicBlock *FBB, bool ControlsExit, bool AllowPredicates,
902                 const ExitLimit &EL);
903   };
904
905   typedef ExitLimitCache ExitLimitCacheTy;
906   ExitLimit computeExitLimitFromCondCached(ExitLimitCacheTy &Cache,
907                                            const Loop *L, Value *ExitCond,
908                                            BasicBlock *TBB, BasicBlock *FBB,
909                                            bool ControlsExit,
910                                            bool AllowPredicates);
911   ExitLimit computeExitLimitFromCondImpl(ExitLimitCacheTy &Cache, const Loop *L,
912                                          Value *ExitCond, BasicBlock *TBB,
913                                          BasicBlock *FBB, bool ControlsExit,
914                                          bool AllowPredicates);
915
916   /// Compute the number of times the backedge of the specified loop will
917   /// execute if its exit condition were a conditional branch of the ICmpInst
918   /// ExitCond, TBB, and FBB. If AllowPredicates is set, this call will try
919   /// to use a minimal set of SCEV predicates in order to return an exact
920   /// answer.
921   ExitLimit computeExitLimitFromICmp(const Loop *L, ICmpInst *ExitCond,
922                                      BasicBlock *TBB, BasicBlock *FBB,
923                                      bool IsSubExpr,
924                                      bool AllowPredicates = false);
925
926   /// Compute the number of times the backedge of the specified loop will
927   /// execute if its exit condition were a switch with a single exiting case
928   /// to ExitingBB.
929   ExitLimit computeExitLimitFromSingleExitSwitch(const Loop *L,
930                                                  SwitchInst *Switch,
931                                                  BasicBlock *ExitingBB,
932                                                  bool IsSubExpr);
933
934   /// Given an exit condition of 'icmp op load X, cst', try to see if we can
935   /// compute the backedge-taken count.
936   ExitLimit computeLoadConstantCompareExitLimit(LoadInst *LI, Constant *RHS,
937                                                 const Loop *L,
938                                                 ICmpInst::Predicate p);
939
940   /// Compute the exit limit of a loop that is controlled by a
941   /// "(IV >> 1) != 0" type comparison.  We cannot compute the exact trip
942   /// count in these cases (since SCEV has no way of expressing them), but we
943   /// can still sometimes compute an upper bound.
944   ///
945   /// Return an ExitLimit for a loop whose backedge is guarded by `LHS Pred
946   /// RHS`.
947   ExitLimit computeShiftCompareExitLimit(Value *LHS, Value *RHS, const Loop *L,
948                                          ICmpInst::Predicate Pred);
949
950   /// If the loop is known to execute a constant number of times (the
951   /// condition evolves only from constants), try to evaluate a few iterations
952   /// of the loop until we get the exit condition gets a value of ExitWhen
953   /// (true or false).  If we cannot evaluate the exit count of the loop,
954   /// return CouldNotCompute.
955   const SCEV *computeExitCountExhaustively(const Loop *L, Value *Cond,
956                                            bool ExitWhen);
957
958   /// Return the number of times an exit condition comparing the specified
959   /// value to zero will execute.  If not computable, return CouldNotCompute.
960   /// If AllowPredicates is set, this call will try to use a minimal set of
961   /// SCEV predicates in order to return an exact answer.
962   ExitLimit howFarToZero(const SCEV *V, const Loop *L, bool IsSubExpr,
963                          bool AllowPredicates = false);
964
965   /// Return the number of times an exit condition checking the specified
966   /// value for nonzero will execute.  If not computable, return
967   /// CouldNotCompute.
968   ExitLimit howFarToNonZero(const SCEV *V, const Loop *L);
969
970   /// Return the number of times an exit condition containing the specified
971   /// less-than comparison will execute.  If not computable, return
972   /// CouldNotCompute.
973   ///
974   /// \p isSigned specifies whether the less-than is signed.
975   ///
976   /// \p ControlsExit is true when the LHS < RHS condition directly controls
977   /// the branch (loops exits only if condition is true). In this case, we can
978   /// use NoWrapFlags to skip overflow checks.
979   ///
980   /// If \p AllowPredicates is set, this call will try to use a minimal set of
981   /// SCEV predicates in order to return an exact answer.
982   ExitLimit howManyLessThans(const SCEV *LHS, const SCEV *RHS, const Loop *L,
983                              bool isSigned, bool ControlsExit,
984                              bool AllowPredicates = false);
985
986   ExitLimit howManyGreaterThans(const SCEV *LHS, const SCEV *RHS, const Loop *L,
987                                 bool isSigned, bool IsSubExpr,
988                                 bool AllowPredicates = false);
989
990   /// Return a predecessor of BB (which may not be an immediate predecessor)
991   /// which has exactly one successor from which BB is reachable, or null if
992   /// no such block is found.
993   std::pair<BasicBlock *, BasicBlock *>
994   getPredecessorWithUniqueSuccessorForBB(BasicBlock *BB);
995
996   /// Test whether the condition described by Pred, LHS, and RHS is true
997   /// whenever the given FoundCondValue value evaluates to true.
998   bool isImpliedCond(ICmpInst::Predicate Pred, const SCEV *LHS, const SCEV *RHS,
999                      Value *FoundCondValue, bool Inverse);
1000
1001   /// Test whether the condition described by Pred, LHS, and RHS is true
1002   /// whenever the condition described by FoundPred, FoundLHS, FoundRHS is
1003   /// true.
1004   bool isImpliedCond(ICmpInst::Predicate Pred, const SCEV *LHS, const SCEV *RHS,
1005                      ICmpInst::Predicate FoundPred, const SCEV *FoundLHS,
1006                      const SCEV *FoundRHS);
1007
1008   /// Test whether the condition described by Pred, LHS, and RHS is true
1009   /// whenever the condition described by Pred, FoundLHS, and FoundRHS is
1010   /// true.
1011   bool isImpliedCondOperands(ICmpInst::Predicate Pred, const SCEV *LHS,
1012                              const SCEV *RHS, const SCEV *FoundLHS,
1013                              const SCEV *FoundRHS);
1014
1015   /// Test whether the condition described by Pred, LHS, and RHS is true
1016   /// whenever the condition described by Pred, FoundLHS, and FoundRHS is
1017   /// true. Here LHS is an operation that includes FoundLHS as one of its
1018   /// arguments.
1019   bool isImpliedViaOperations(ICmpInst::Predicate Pred,
1020                               const SCEV *LHS, const SCEV *RHS,
1021                               const SCEV *FoundLHS, const SCEV *FoundRHS,
1022                               unsigned Depth = 0);
1023
1024   /// Test whether the condition described by Pred, LHS, and RHS is true.
1025   /// Use only simple non-recursive types of checks, such as range analysis etc.
1026   bool isKnownViaSimpleReasoning(ICmpInst::Predicate Pred,
1027                                  const SCEV *LHS, const SCEV *RHS);
1028
1029   /// Test whether the condition described by Pred, LHS, and RHS is true
1030   /// whenever the condition described by Pred, FoundLHS, and FoundRHS is
1031   /// true.
1032   bool isImpliedCondOperandsHelper(ICmpInst::Predicate Pred, const SCEV *LHS,
1033                                    const SCEV *RHS, const SCEV *FoundLHS,
1034                                    const SCEV *FoundRHS);
1035
1036   /// Test whether the condition described by Pred, LHS, and RHS is true
1037   /// whenever the condition described by Pred, FoundLHS, and FoundRHS is
1038   /// true.  Utility function used by isImpliedCondOperands.  Tries to get
1039   /// cases like "X `sgt` 0 => X - 1 `sgt` -1".
1040   bool isImpliedCondOperandsViaRanges(ICmpInst::Predicate Pred, const SCEV *LHS,
1041                                       const SCEV *RHS, const SCEV *FoundLHS,
1042                                       const SCEV *FoundRHS);
1043
1044   /// Return true if the condition denoted by \p LHS \p Pred \p RHS is implied
1045   /// by a call to \c @llvm.experimental.guard in \p BB.
1046   bool isImpliedViaGuard(BasicBlock *BB, ICmpInst::Predicate Pred,
1047                          const SCEV *LHS, const SCEV *RHS);
1048
1049   /// Test whether the condition described by Pred, LHS, and RHS is true
1050   /// whenever the condition described by Pred, FoundLHS, and FoundRHS is
1051   /// true.
1052   ///
1053   /// This routine tries to rule out certain kinds of integer overflow, and
1054   /// then tries to reason about arithmetic properties of the predicates.
1055   bool isImpliedCondOperandsViaNoOverflow(ICmpInst::Predicate Pred,
1056                                           const SCEV *LHS, const SCEV *RHS,
1057                                           const SCEV *FoundLHS,
1058                                           const SCEV *FoundRHS);
1059
1060   /// If we know that the specified Phi is in the header of its containing
1061   /// loop, we know the loop executes a constant number of times, and the PHI
1062   /// node is just a recurrence involving constants, fold it.
1063   Constant *getConstantEvolutionLoopExitValue(PHINode *PN, const APInt &BEs,
1064                                               const Loop *L);
1065
1066   /// Test if the given expression is known to satisfy the condition described
1067   /// by Pred and the known constant ranges of LHS and RHS.
1068   ///
1069   bool isKnownPredicateViaConstantRanges(ICmpInst::Predicate Pred,
1070                                          const SCEV *LHS, const SCEV *RHS);
1071
1072   /// Try to prove the condition described by "LHS Pred RHS" by ruling out
1073   /// integer overflow.
1074   ///
1075   /// For instance, this will return true for "A s< (A + C)<nsw>" if C is
1076   /// positive.
1077   bool isKnownPredicateViaNoOverflow(ICmpInst::Predicate Pred, const SCEV *LHS,
1078                                      const SCEV *RHS);
1079
1080   /// Try to split Pred LHS RHS into logical conjunctions (and's) and try to
1081   /// prove them individually.
1082   bool isKnownPredicateViaSplitting(ICmpInst::Predicate Pred, const SCEV *LHS,
1083                                     const SCEV *RHS);
1084
1085   /// Try to match the Expr as "(L + R)<Flags>".
1086   bool splitBinaryAdd(const SCEV *Expr, const SCEV *&L, const SCEV *&R,
1087                       SCEV::NoWrapFlags &Flags);
1088
1089   /// Compute \p LHS - \p RHS and returns the result as an APInt if it is a
1090   /// constant, and None if it isn't.
1091   ///
1092   /// This is intended to be a cheaper version of getMinusSCEV.  We can be
1093   /// frugal here since we just bail out of actually constructing and
1094   /// canonicalizing an expression in the cases where the result isn't going
1095   /// to be a constant.
1096   Optional<APInt> computeConstantDifference(const SCEV *LHS, const SCEV *RHS);
1097
1098   /// Drop memoized information computed for S.
1099   void forgetMemoizedResults(const SCEV *S);
1100
1101   /// Return an existing SCEV for V if there is one, otherwise return nullptr.
1102   const SCEV *getExistingSCEV(Value *V);
1103
1104   /// Return false iff given SCEV contains a SCEVUnknown with NULL value-
1105   /// pointer.
1106   bool checkValidity(const SCEV *S) const;
1107
1108   /// Return true if `ExtendOpTy`({`Start`,+,`Step`}) can be proved to be
1109   /// equal to {`ExtendOpTy`(`Start`),+,`ExtendOpTy`(`Step`)}.  This is
1110   /// equivalent to proving no signed (resp. unsigned) wrap in
1111   /// {`Start`,+,`Step`} if `ExtendOpTy` is `SCEVSignExtendExpr`
1112   /// (resp. `SCEVZeroExtendExpr`).
1113   ///
1114   template <typename ExtendOpTy>
1115   bool proveNoWrapByVaryingStart(const SCEV *Start, const SCEV *Step,
1116                                  const Loop *L);
1117
1118   /// Try to prove NSW or NUW on \p AR relying on ConstantRange manipulation.
1119   SCEV::NoWrapFlags proveNoWrapViaConstantRanges(const SCEVAddRecExpr *AR);
1120
1121   bool isMonotonicPredicateImpl(const SCEVAddRecExpr *LHS,
1122                                 ICmpInst::Predicate Pred, bool &Increasing);
1123
1124   /// Return SCEV no-wrap flags that can be proven based on reasoning about
1125   /// how poison produced from no-wrap flags on this value (e.g. a nuw add)
1126   /// would trigger undefined behavior on overflow.
1127   SCEV::NoWrapFlags getNoWrapFlagsFromUB(const Value *V);
1128
1129   /// Return true if the SCEV corresponding to \p I is never poison.  Proving
1130   /// this is more complex than proving that just \p I is never poison, since
1131   /// SCEV commons expressions across control flow, and you can have cases
1132   /// like:
1133   ///
1134   ///   idx0 = a + b;
1135   ///   ptr[idx0] = 100;
1136   ///   if (<condition>) {
1137   ///     idx1 = a +nsw b;
1138   ///     ptr[idx1] = 200;
1139   ///   }
1140   ///
1141   /// where the SCEV expression (+ a b) is guaranteed to not be poison (and
1142   /// hence not sign-overflow) only if "<condition>" is true.  Since both
1143   /// `idx0` and `idx1` will be mapped to the same SCEV expression, (+ a b),
1144   /// it is not okay to annotate (+ a b) with <nsw> in the above example.
1145   bool isSCEVExprNeverPoison(const Instruction *I);
1146
1147   /// This is like \c isSCEVExprNeverPoison but it specifically works for
1148   /// instructions that will get mapped to SCEV add recurrences.  Return true
1149   /// if \p I will never generate poison under the assumption that \p I is an
1150   /// add recurrence on the loop \p L.
1151   bool isAddRecNeverPoison(const Instruction *I, const Loop *L);
1152
1153 public:
1154   ScalarEvolution(Function &F, TargetLibraryInfo &TLI, AssumptionCache &AC,
1155                   DominatorTree &DT, LoopInfo &LI);
1156   ~ScalarEvolution();
1157   ScalarEvolution(ScalarEvolution &&Arg);
1158
1159   LLVMContext &getContext() const { return F.getContext(); }
1160
1161   /// Test if values of the given type are analyzable within the SCEV
1162   /// framework. This primarily includes integer types, and it can optionally
1163   /// include pointer types if the ScalarEvolution class has access to
1164   /// target-specific information.
1165   bool isSCEVable(Type *Ty) const;
1166
1167   /// Return the size in bits of the specified type, for which isSCEVable must
1168   /// return true.
1169   uint64_t getTypeSizeInBits(Type *Ty) const;
1170
1171   /// Return a type with the same bitwidth as the given type and which
1172   /// represents how SCEV will treat the given type, for which isSCEVable must
1173   /// return true. For pointer types, this is the pointer-sized integer type.
1174   Type *getEffectiveSCEVType(Type *Ty) const;
1175
1176   // Returns a wider type among {Ty1, Ty2}.
1177   Type *getWiderType(Type *Ty1, Type *Ty2) const;
1178
1179   /// Return true if the SCEV is a scAddRecExpr or it contains
1180   /// scAddRecExpr. The result will be cached in HasRecMap.
1181   ///
1182   bool containsAddRecurrence(const SCEV *S);
1183
1184   /// Return the Value set from which the SCEV expr is generated.
1185   SetVector<ValueOffsetPair> *getSCEVValues(const SCEV *S);
1186
1187   /// Erase Value from ValueExprMap and ExprValueMap.
1188   void eraseValueFromMap(Value *V);
1189
1190   /// Return a SCEV expression for the full generality of the specified
1191   /// expression.
1192   const SCEV *getSCEV(Value *V);
1193
1194   const SCEV *getConstant(ConstantInt *V);
1195   const SCEV *getConstant(const APInt &Val);
1196   const SCEV *getConstant(Type *Ty, uint64_t V, bool isSigned = false);
1197   const SCEV *getTruncateExpr(const SCEV *Op, Type *Ty);
1198
1199   typedef SmallDenseMap<std::pair<const SCEV *, Type *>, const SCEV *, 8>
1200       ExtendCacheTy;
1201   const SCEV *getZeroExtendExpr(const SCEV *Op, Type *Ty);
1202   const SCEV *getZeroExtendExprCached(const SCEV *Op, Type *Ty,
1203                                       ExtendCacheTy &Cache);
1204   const SCEV *getZeroExtendExprImpl(const SCEV *Op, Type *Ty,
1205                                     ExtendCacheTy &Cache);
1206
1207   const SCEV *getSignExtendExpr(const SCEV *Op, Type *Ty);
1208   const SCEV *getSignExtendExprCached(const SCEV *Op, Type *Ty,
1209                                       ExtendCacheTy &Cache);
1210   const SCEV *getSignExtendExprImpl(const SCEV *Op, Type *Ty,
1211                                     ExtendCacheTy &Cache);
1212   const SCEV *getAnyExtendExpr(const SCEV *Op, Type *Ty);
1213   const SCEV *getAddExpr(SmallVectorImpl<const SCEV *> &Ops,
1214                          SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap,
1215                          unsigned Depth = 0);
1216   const SCEV *getAddExpr(const SCEV *LHS, const SCEV *RHS,
1217                          SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap) {
1218     SmallVector<const SCEV *, 2> Ops = {LHS, RHS};
1219     return getAddExpr(Ops, Flags);
1220   }
1221   const SCEV *getAddExpr(const SCEV *Op0, const SCEV *Op1, const SCEV *Op2,
1222                          SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap) {
1223     SmallVector<const SCEV *, 3> Ops = {Op0, Op1, Op2};
1224     return getAddExpr(Ops, Flags);
1225   }
1226   const SCEV *getMulExpr(SmallVectorImpl<const SCEV *> &Ops,
1227                          SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap);
1228   const SCEV *getMulExpr(const SCEV *LHS, const SCEV *RHS,
1229                          SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap) {
1230     SmallVector<const SCEV *, 2> Ops = {LHS, RHS};
1231     return getMulExpr(Ops, Flags);
1232   }
1233   const SCEV *getMulExpr(const SCEV *Op0, const SCEV *Op1, const SCEV *Op2,
1234                          SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap) {
1235     SmallVector<const SCEV *, 3> Ops = {Op0, Op1, Op2};
1236     return getMulExpr(Ops, Flags);
1237   }
1238   const SCEV *getUDivExpr(const SCEV *LHS, const SCEV *RHS);
1239   const SCEV *getUDivExactExpr(const SCEV *LHS, const SCEV *RHS);
1240   const SCEV *getAddRecExpr(const SCEV *Start, const SCEV *Step, const Loop *L,
1241                             SCEV::NoWrapFlags Flags);
1242   const SCEV *getAddRecExpr(SmallVectorImpl<const SCEV *> &Operands,
1243                             const Loop *L, SCEV::NoWrapFlags Flags);
1244   const SCEV *getAddRecExpr(const SmallVectorImpl<const SCEV *> &Operands,
1245                             const Loop *L, SCEV::NoWrapFlags Flags) {
1246     SmallVector<const SCEV *, 4> NewOp(Operands.begin(), Operands.end());
1247     return getAddRecExpr(NewOp, L, Flags);
1248   }
1249   /// Returns an expression for a GEP
1250   ///
1251   /// \p GEP The GEP. The indices contained in the GEP itself are ignored,
1252   /// instead we use IndexExprs.
1253   /// \p IndexExprs The expressions for the indices.
1254   const SCEV *getGEPExpr(GEPOperator *GEP,
1255                          const SmallVectorImpl<const SCEV *> &IndexExprs);
1256   const SCEV *getSMaxExpr(const SCEV *LHS, const SCEV *RHS);
1257   const SCEV *getSMaxExpr(SmallVectorImpl<const SCEV *> &Operands);
1258   const SCEV *getUMaxExpr(const SCEV *LHS, const SCEV *RHS);
1259   const SCEV *getUMaxExpr(SmallVectorImpl<const SCEV *> &Operands);
1260   const SCEV *getSMinExpr(const SCEV *LHS, const SCEV *RHS);
1261   const SCEV *getUMinExpr(const SCEV *LHS, const SCEV *RHS);
1262   const SCEV *getUnknown(Value *V);
1263   const SCEV *getCouldNotCompute();
1264
1265   /// Return a SCEV for the constant 0 of a specific type.
1266   const SCEV *getZero(Type *Ty) { return getConstant(Ty, 0); }
1267
1268   /// Return a SCEV for the constant 1 of a specific type.
1269   const SCEV *getOne(Type *Ty) { return getConstant(Ty, 1); }
1270
1271   /// Return an expression for sizeof AllocTy that is type IntTy
1272   ///
1273   const SCEV *getSizeOfExpr(Type *IntTy, Type *AllocTy);
1274
1275   /// Return an expression for offsetof on the given field with type IntTy
1276   ///
1277   const SCEV *getOffsetOfExpr(Type *IntTy, StructType *STy, unsigned FieldNo);
1278
1279   /// Return the SCEV object corresponding to -V.
1280   ///
1281   const SCEV *getNegativeSCEV(const SCEV *V,
1282                               SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap);
1283
1284   /// Return the SCEV object corresponding to ~V.
1285   ///
1286   const SCEV *getNotSCEV(const SCEV *V);
1287
1288   /// Return LHS-RHS.  Minus is represented in SCEV as A+B*-1.
1289   const SCEV *getMinusSCEV(const SCEV *LHS, const SCEV *RHS,
1290                            SCEV::NoWrapFlags Flags = SCEV::FlagAnyWrap);
1291
1292   /// Return a SCEV corresponding to a conversion of the input value to the
1293   /// specified type.  If the type must be extended, it is zero extended.
1294   const SCEV *getTruncateOrZeroExtend(const SCEV *V, Type *Ty);
1295
1296   /// Return a SCEV corresponding to a conversion of the input value to the
1297   /// specified type.  If the type must be extended, it is sign extended.
1298   const SCEV *getTruncateOrSignExtend(const SCEV *V, Type *Ty);
1299
1300   /// Return a SCEV corresponding to a conversion of the input value to the
1301   /// specified type.  If the type must be extended, it is zero extended.  The
1302   /// conversion must not be narrowing.
1303   const SCEV *getNoopOrZeroExtend(const SCEV *V, Type *Ty);
1304
1305   /// Return a SCEV corresponding to a conversion of the input value to the
1306   /// specified type.  If the type must be extended, it is sign extended.  The
1307   /// conversion must not be narrowing.
1308   const SCEV *getNoopOrSignExtend(const SCEV *V, Type *Ty);
1309
1310   /// Return a SCEV corresponding to a conversion of the input value to the
1311   /// specified type. If the type must be extended, it is extended with
1312   /// unspecified bits. The conversion must not be narrowing.
1313   const SCEV *getNoopOrAnyExtend(const SCEV *V, Type *Ty);
1314
1315   /// Return a SCEV corresponding to a conversion of the input value to the
1316   /// specified type.  The conversion must not be widening.
1317   const SCEV *getTruncateOrNoop(const SCEV *V, Type *Ty);
1318
1319   /// Promote the operands to the wider of the types using zero-extension, and
1320   /// then perform a umax operation with them.
1321   const SCEV *getUMaxFromMismatchedTypes(const SCEV *LHS, const SCEV *RHS);
1322
1323   /// Promote the operands to the wider of the types using zero-extension, and
1324   /// then perform a umin operation with them.
1325   const SCEV *getUMinFromMismatchedTypes(const SCEV *LHS, const SCEV *RHS);
1326
1327   /// Transitively follow the chain of pointer-type operands until reaching a
1328   /// SCEV that does not have a single pointer operand. This returns a
1329   /// SCEVUnknown pointer for well-formed pointer-type expressions, but corner
1330   /// cases do exist.
1331   const SCEV *getPointerBase(const SCEV *V);
1332
1333   /// Return a SCEV expression for the specified value at the specified scope
1334   /// in the program.  The L value specifies a loop nest to evaluate the
1335   /// expression at, where null is the top-level or a specified loop is
1336   /// immediately inside of the loop.
1337   ///
1338   /// This method can be used to compute the exit value for a variable defined
1339   /// in a loop by querying what the value will hold in the parent loop.
1340   ///
1341   /// In the case that a relevant loop exit value cannot be computed, the
1342   /// original value V is returned.
1343   const SCEV *getSCEVAtScope(const SCEV *S, const Loop *L);
1344
1345   /// This is a convenience function which does getSCEVAtScope(getSCEV(V), L).
1346   const SCEV *getSCEVAtScope(Value *V, const Loop *L);
1347
1348   /// Test whether entry to the loop is protected by a conditional between LHS
1349   /// and RHS.  This is used to help avoid max expressions in loop trip
1350   /// counts, and to eliminate casts.
1351   bool isLoopEntryGuardedByCond(const Loop *L, ICmpInst::Predicate Pred,
1352                                 const SCEV *LHS, const SCEV *RHS);
1353
1354   /// Test whether the backedge of the loop is protected by a conditional
1355   /// between LHS and RHS.  This is used to to eliminate casts.
1356   bool isLoopBackedgeGuardedByCond(const Loop *L, ICmpInst::Predicate Pred,
1357                                    const SCEV *LHS, const SCEV *RHS);
1358
1359   /// Returns the maximum trip count of the loop if it is a single-exit
1360   /// loop and we can compute a small maximum for that loop.
1361   ///
1362   /// Implemented in terms of the \c getSmallConstantTripCount overload with
1363   /// the single exiting block passed to it. See that routine for details.
1364   unsigned getSmallConstantTripCount(const Loop *L);
1365
1366   /// Returns the maximum trip count of this loop as a normal unsigned
1367   /// value. Returns 0 if the trip count is unknown or not constant. This
1368   /// "trip count" assumes that control exits via ExitingBlock. More
1369   /// precisely, it is the number of times that control may reach ExitingBlock
1370   /// before taking the branch. For loops with multiple exits, it may not be
1371   /// the number times that the loop header executes if the loop exits
1372   /// prematurely via another branch.
1373   unsigned getSmallConstantTripCount(const Loop *L, BasicBlock *ExitingBlock);
1374
1375   /// Returns the upper bound of the loop trip count as a normal unsigned
1376   /// value.
1377   /// Returns 0 if the trip count is unknown or not constant.
1378   unsigned getSmallConstantMaxTripCount(const Loop *L);
1379
1380   /// Returns the largest constant divisor of the trip count of the
1381   /// loop if it is a single-exit loop and we can compute a small maximum for
1382   /// that loop.
1383   ///
1384   /// Implemented in terms of the \c getSmallConstantTripMultiple overload with
1385   /// the single exiting block passed to it. See that routine for details.
1386   unsigned getSmallConstantTripMultiple(const Loop *L);
1387
1388   /// Returns the largest constant divisor of the trip count of this loop as a
1389   /// normal unsigned value, if possible. This means that the actual trip
1390   /// count is always a multiple of the returned value (don't forget the trip
1391   /// count could very well be zero as well!). As explained in the comments
1392   /// for getSmallConstantTripCount, this assumes that control exits the loop
1393   /// via ExitingBlock.
1394   unsigned getSmallConstantTripMultiple(const Loop *L,
1395                                         BasicBlock *ExitingBlock);
1396
1397   /// Get the expression for the number of loop iterations for which this loop
1398   /// is guaranteed not to exit via ExitingBlock. Otherwise return
1399   /// SCEVCouldNotCompute.
1400   const SCEV *getExitCount(const Loop *L, BasicBlock *ExitingBlock);
1401
1402   /// If the specified loop has a predictable backedge-taken count, return it,
1403   /// otherwise return a SCEVCouldNotCompute object. The backedge-taken count is
1404   /// the number of times the loop header will be branched to from within the
1405   /// loop, assuming there are no abnormal exists like exception throws. This is
1406   /// one less than the trip count of the loop, since it doesn't count the first
1407   /// iteration, when the header is branched to from outside the loop.
1408   ///
1409   /// Note that it is not valid to call this method on a loop without a
1410   /// loop-invariant backedge-taken count (see
1411   /// hasLoopInvariantBackedgeTakenCount).
1412   ///
1413   const SCEV *getBackedgeTakenCount(const Loop *L);
1414
1415   /// Similar to getBackedgeTakenCount, except it will add a set of
1416   /// SCEV predicates to Predicates that are required to be true in order for
1417   /// the answer to be correct. Predicates can be checked with run-time
1418   /// checks and can be used to perform loop versioning.
1419   const SCEV *getPredicatedBackedgeTakenCount(const Loop *L,
1420                                               SCEVUnionPredicate &Predicates);
1421
1422   /// When successful, this returns a SCEVConstant that is greater than or equal
1423   /// to (i.e. a "conservative over-approximation") of the value returend by
1424   /// getBackedgeTakenCount.  If such a value cannot be computed, it returns the
1425   /// SCEVCouldNotCompute object.
1426   const SCEV *getMaxBackedgeTakenCount(const Loop *L);
1427
1428   /// Return true if the backedge taken count is either the value returned by
1429   /// getMaxBackedgeTakenCount or zero.
1430   bool isBackedgeTakenCountMaxOrZero(const Loop *L);
1431
1432   /// Return true if the specified loop has an analyzable loop-invariant
1433   /// backedge-taken count.
1434   bool hasLoopInvariantBackedgeTakenCount(const Loop *L);
1435
1436   /// This method should be called by the client when it has changed a loop in
1437   /// a way that may effect ScalarEvolution's ability to compute a trip count,
1438   /// or if the loop is deleted.  This call is potentially expensive for large
1439   /// loop bodies.
1440   void forgetLoop(const Loop *L);
1441
1442   /// This method should be called by the client when it has changed a value
1443   /// in a way that may effect its value, or which may disconnect it from a
1444   /// def-use chain linking it to a loop.
1445   void forgetValue(Value *V);
1446
1447   /// Called when the client has changed the disposition of values in
1448   /// this loop.
1449   ///
1450   /// We don't have a way to invalidate per-loop dispositions. Clear and
1451   /// recompute is simpler.
1452   void forgetLoopDispositions(const Loop *L) { LoopDispositions.clear(); }
1453
1454   /// Determine the minimum number of zero bits that S is guaranteed to end in
1455   /// (at every loop iteration).  It is, at the same time, the minimum number
1456   /// of times S is divisible by 2.  For example, given {4,+,8} it returns 2.
1457   /// If S is guaranteed to be 0, it returns the bitwidth of S.
1458   uint32_t GetMinTrailingZeros(const SCEV *S);
1459
1460   /// Determine the unsigned range for a particular SCEV.
1461   ///
1462   ConstantRange getUnsignedRange(const SCEV *S) {
1463     return getRange(S, HINT_RANGE_UNSIGNED);
1464   }
1465
1466   /// Determine the signed range for a particular SCEV.
1467   ///
1468   ConstantRange getSignedRange(const SCEV *S) {
1469     return getRange(S, HINT_RANGE_SIGNED);
1470   }
1471
1472   /// Test if the given expression is known to be negative.
1473   ///
1474   bool isKnownNegative(const SCEV *S);
1475
1476   /// Test if the given expression is known to be positive.
1477   ///
1478   bool isKnownPositive(const SCEV *S);
1479
1480   /// Test if the given expression is known to be non-negative.
1481   ///
1482   bool isKnownNonNegative(const SCEV *S);
1483
1484   /// Test if the given expression is known to be non-positive.
1485   ///
1486   bool isKnownNonPositive(const SCEV *S);
1487
1488   /// Test if the given expression is known to be non-zero.
1489   ///
1490   bool isKnownNonZero(const SCEV *S);
1491
1492   /// Test if the given expression is known to satisfy the condition described
1493   /// by Pred, LHS, and RHS.
1494   ///
1495   bool isKnownPredicate(ICmpInst::Predicate Pred, const SCEV *LHS,
1496                         const SCEV *RHS);
1497
1498   /// Return true if, for all loop invariant X, the predicate "LHS `Pred` X"
1499   /// is monotonically increasing or decreasing.  In the former case set
1500   /// `Increasing` to true and in the latter case set `Increasing` to false.
1501   ///
1502   /// A predicate is said to be monotonically increasing if may go from being
1503   /// false to being true as the loop iterates, but never the other way
1504   /// around.  A predicate is said to be monotonically decreasing if may go
1505   /// from being true to being false as the loop iterates, but never the other
1506   /// way around.
1507   bool isMonotonicPredicate(const SCEVAddRecExpr *LHS, ICmpInst::Predicate Pred,
1508                             bool &Increasing);
1509
1510   /// Return true if the result of the predicate LHS `Pred` RHS is loop
1511   /// invariant with respect to L.  Set InvariantPred, InvariantLHS and
1512   /// InvariantLHS so that InvariantLHS `InvariantPred` InvariantRHS is the
1513   /// loop invariant form of LHS `Pred` RHS.
1514   bool isLoopInvariantPredicate(ICmpInst::Predicate Pred, const SCEV *LHS,
1515                                 const SCEV *RHS, const Loop *L,
1516                                 ICmpInst::Predicate &InvariantPred,
1517                                 const SCEV *&InvariantLHS,
1518                                 const SCEV *&InvariantRHS);
1519
1520   /// Simplify LHS and RHS in a comparison with predicate Pred. Return true
1521   /// iff any changes were made. If the operands are provably equal or
1522   /// unequal, LHS and RHS are set to the same value and Pred is set to either
1523   /// ICMP_EQ or ICMP_NE.
1524   ///
1525   bool SimplifyICmpOperands(ICmpInst::Predicate &Pred, const SCEV *&LHS,
1526                             const SCEV *&RHS, unsigned Depth = 0);
1527
1528   /// Return the "disposition" of the given SCEV with respect to the given
1529   /// loop.
1530   LoopDisposition getLoopDisposition(const SCEV *S, const Loop *L);
1531
1532   /// Return true if the value of the given SCEV is unchanging in the
1533   /// specified loop.
1534   bool isLoopInvariant(const SCEV *S, const Loop *L);
1535
1536   /// Determine if the SCEV can be evaluated at loop's entry. It is true if it
1537   /// doesn't depend on a SCEVUnknown of an instruction which is dominated by
1538   /// the header of loop L.
1539   bool isAvailableAtLoopEntry(const SCEV *S, const Loop *L, DominatorTree &DT,
1540                               LoopInfo &LI);
1541
1542   /// Return true if the given SCEV changes value in a known way in the
1543   /// specified loop.  This property being true implies that the value is
1544   /// variant in the loop AND that we can emit an expression to compute the
1545   /// value of the expression at any particular loop iteration.
1546   bool hasComputableLoopEvolution(const SCEV *S, const Loop *L);
1547
1548   /// Return the "disposition" of the given SCEV with respect to the given
1549   /// block.
1550   BlockDisposition getBlockDisposition(const SCEV *S, const BasicBlock *BB);
1551
1552   /// Return true if elements that makes up the given SCEV dominate the
1553   /// specified basic block.
1554   bool dominates(const SCEV *S, const BasicBlock *BB);
1555
1556   /// Return true if elements that makes up the given SCEV properly dominate
1557   /// the specified basic block.
1558   bool properlyDominates(const SCEV *S, const BasicBlock *BB);
1559
1560   /// Test whether the given SCEV has Op as a direct or indirect operand.
1561   bool hasOperand(const SCEV *S, const SCEV *Op) const;
1562
1563   /// Return the size of an element read or written by Inst.
1564   const SCEV *getElementSize(Instruction *Inst);
1565
1566   /// Compute the array dimensions Sizes from the set of Terms extracted from
1567   /// the memory access function of this SCEVAddRecExpr (second step of
1568   /// delinearization).
1569   void findArrayDimensions(SmallVectorImpl<const SCEV *> &Terms,
1570                            SmallVectorImpl<const SCEV *> &Sizes,
1571                            const SCEV *ElementSize);
1572
1573   void print(raw_ostream &OS) const;
1574   void verify() const;
1575   bool invalidate(Function &F, const PreservedAnalyses &PA,
1576                   FunctionAnalysisManager::Invalidator &Inv);
1577
1578   /// Collect parametric terms occurring in step expressions (first step of
1579   /// delinearization).
1580   void collectParametricTerms(const SCEV *Expr,
1581                               SmallVectorImpl<const SCEV *> &Terms);
1582
1583   /// Return in Subscripts the access functions for each dimension in Sizes
1584   /// (third step of delinearization).
1585   void computeAccessFunctions(const SCEV *Expr,
1586                               SmallVectorImpl<const SCEV *> &Subscripts,
1587                               SmallVectorImpl<const SCEV *> &Sizes);
1588
1589   /// Split this SCEVAddRecExpr into two vectors of SCEVs representing the
1590   /// subscripts and sizes of an array access.
1591   ///
1592   /// The delinearization is a 3 step process: the first two steps compute the
1593   /// sizes of each subscript and the third step computes the access functions
1594   /// for the delinearized array:
1595   ///
1596   /// 1. Find the terms in the step functions
1597   /// 2. Compute the array size
1598   /// 3. Compute the access function: divide the SCEV by the array size
1599   ///    starting with the innermost dimensions found in step 2. The Quotient
1600   ///    is the SCEV to be divided in the next step of the recursion. The
1601   ///    Remainder is the subscript of the innermost dimension. Loop over all
1602   ///    array dimensions computed in step 2.
1603   ///
1604   /// To compute a uniform array size for several memory accesses to the same
1605   /// object, one can collect in step 1 all the step terms for all the memory
1606   /// accesses, and compute in step 2 a unique array shape. This guarantees
1607   /// that the array shape will be the same across all memory accesses.
1608   ///
1609   /// FIXME: We could derive the result of steps 1 and 2 from a description of
1610   /// the array shape given in metadata.
1611   ///
1612   /// Example:
1613   ///
1614   /// A[][n][m]
1615   ///
1616   /// for i
1617   ///   for j
1618   ///     for k
1619   ///       A[j+k][2i][5i] =
1620   ///
1621   /// The initial SCEV:
1622   ///
1623   /// A[{{{0,+,2*m+5}_i, +, n*m}_j, +, n*m}_k]
1624   ///
1625   /// 1. Find the different terms in the step functions:
1626   /// -> [2*m, 5, n*m, n*m]
1627   ///
1628   /// 2. Compute the array size: sort and unique them
1629   /// -> [n*m, 2*m, 5]
1630   /// find the GCD of all the terms = 1
1631   /// divide by the GCD and erase constant terms
1632   /// -> [n*m, 2*m]
1633   /// GCD = m
1634   /// divide by GCD -> [n, 2]
1635   /// remove constant terms
1636   /// -> [n]
1637   /// size of the array is A[unknown][n][m]
1638   ///
1639   /// 3. Compute the access function
1640   /// a. Divide {{{0,+,2*m+5}_i, +, n*m}_j, +, n*m}_k by the innermost size m
1641   /// Quotient: {{{0,+,2}_i, +, n}_j, +, n}_k
1642   /// Remainder: {{{0,+,5}_i, +, 0}_j, +, 0}_k
1643   /// The remainder is the subscript of the innermost array dimension: [5i].
1644   ///
1645   /// b. Divide Quotient: {{{0,+,2}_i, +, n}_j, +, n}_k by next outer size n
1646   /// Quotient: {{{0,+,0}_i, +, 1}_j, +, 1}_k
1647   /// Remainder: {{{0,+,2}_i, +, 0}_j, +, 0}_k
1648   /// The Remainder is the subscript of the next array dimension: [2i].
1649   ///
1650   /// The subscript of the outermost dimension is the Quotient: [j+k].
1651   ///
1652   /// Overall, we have: A[][n][m], and the access function: A[j+k][2i][5i].
1653   void delinearize(const SCEV *Expr, SmallVectorImpl<const SCEV *> &Subscripts,
1654                    SmallVectorImpl<const SCEV *> &Sizes,
1655                    const SCEV *ElementSize);
1656
1657   /// Return the DataLayout associated with the module this SCEV instance is
1658   /// operating on.
1659   const DataLayout &getDataLayout() const {
1660     return F.getParent()->getDataLayout();
1661   }
1662
1663   const SCEVPredicate *getEqualPredicate(const SCEVUnknown *LHS,
1664                                          const SCEVConstant *RHS);
1665
1666   const SCEVPredicate *
1667   getWrapPredicate(const SCEVAddRecExpr *AR,
1668                    SCEVWrapPredicate::IncrementWrapFlags AddedFlags);
1669
1670   /// Re-writes the SCEV according to the Predicates in \p A.
1671   const SCEV *rewriteUsingPredicate(const SCEV *S, const Loop *L,
1672                                     SCEVUnionPredicate &A);
1673   /// Tries to convert the \p S expression to an AddRec expression,
1674   /// adding additional predicates to \p Preds as required.
1675   const SCEVAddRecExpr *convertSCEVToAddRecWithPredicates(
1676       const SCEV *S, const Loop *L,
1677       SmallPtrSetImpl<const SCEVPredicate *> &Preds);
1678
1679 private:
1680   /// Compute the backedge taken count knowing the interval difference, the
1681   /// stride and presence of the equality in the comparison.
1682   const SCEV *computeBECount(const SCEV *Delta, const SCEV *Stride,
1683                              bool Equality);
1684
1685   /// Verify if an linear IV with positive stride can overflow when in a
1686   /// less-than comparison, knowing the invariant term of the comparison,
1687   /// the stride and the knowledge of NSW/NUW flags on the recurrence.
1688   bool doesIVOverflowOnLT(const SCEV *RHS, const SCEV *Stride, bool IsSigned,
1689                           bool NoWrap);
1690
1691   /// Verify if an linear IV with negative stride can overflow when in a
1692   /// greater-than comparison, knowing the invariant term of the comparison,
1693   /// the stride and the knowledge of NSW/NUW flags on the recurrence.
1694   bool doesIVOverflowOnGT(const SCEV *RHS, const SCEV *Stride, bool IsSigned,
1695                           bool NoWrap);
1696
1697   /// Get add expr already created or create a new one
1698   const SCEV *getOrCreateAddExpr(SmallVectorImpl<const SCEV *> &Ops,
1699                                  SCEV::NoWrapFlags Flags);
1700
1701 private:
1702   FoldingSet<SCEV> UniqueSCEVs;
1703   FoldingSet<SCEVPredicate> UniquePreds;
1704   BumpPtrAllocator SCEVAllocator;
1705
1706   /// The head of a linked list of all SCEVUnknown values that have been
1707   /// allocated. This is used by releaseMemory to locate them all and call
1708   /// their destructors.
1709   SCEVUnknown *FirstUnknown;
1710 };
1711
1712 /// Analysis pass that exposes the \c ScalarEvolution for a function.
1713 class ScalarEvolutionAnalysis
1714     : public AnalysisInfoMixin<ScalarEvolutionAnalysis> {
1715   friend AnalysisInfoMixin<ScalarEvolutionAnalysis>;
1716   static AnalysisKey Key;
1717
1718 public:
1719   typedef ScalarEvolution Result;
1720
1721   ScalarEvolution run(Function &F, FunctionAnalysisManager &AM);
1722 };
1723
1724 /// Printer pass for the \c ScalarEvolutionAnalysis results.
1725 class ScalarEvolutionPrinterPass
1726     : public PassInfoMixin<ScalarEvolutionPrinterPass> {
1727   raw_ostream &OS;
1728
1729 public:
1730   explicit ScalarEvolutionPrinterPass(raw_ostream &OS) : OS(OS) {}
1731   PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM);
1732 };
1733
1734 class ScalarEvolutionWrapperPass : public FunctionPass {
1735   std::unique_ptr<ScalarEvolution> SE;
1736
1737 public:
1738   static char ID;
1739
1740   ScalarEvolutionWrapperPass();
1741
1742   ScalarEvolution &getSE() { return *SE; }
1743   const ScalarEvolution &getSE() const { return *SE; }
1744
1745   bool runOnFunction(Function &F) override;
1746   void releaseMemory() override;
1747   void getAnalysisUsage(AnalysisUsage &AU) const override;
1748   void print(raw_ostream &OS, const Module * = nullptr) const override;
1749   void verifyAnalysis() const override;
1750 };
1751
1752 /// An interface layer with SCEV used to manage how we see SCEV expressions
1753 /// for values in the context of existing predicates. We can add new
1754 /// predicates, but we cannot remove them.
1755 ///
1756 /// This layer has multiple purposes:
1757 ///   - provides a simple interface for SCEV versioning.
1758 ///   - guarantees that the order of transformations applied on a SCEV
1759 ///     expression for a single Value is consistent across two different
1760 ///     getSCEV calls. This means that, for example, once we've obtained
1761 ///     an AddRec expression for a certain value through expression
1762 ///     rewriting, we will continue to get an AddRec expression for that
1763 ///     Value.
1764 ///   - lowers the number of expression rewrites.
1765 class PredicatedScalarEvolution {
1766 public:
1767   PredicatedScalarEvolution(ScalarEvolution &SE, Loop &L);
1768   const SCEVUnionPredicate &getUnionPredicate() const;
1769
1770   /// Returns the SCEV expression of V, in the context of the current SCEV
1771   /// predicate.  The order of transformations applied on the expression of V
1772   /// returned by ScalarEvolution is guaranteed to be preserved, even when
1773   /// adding new predicates.
1774   const SCEV *getSCEV(Value *V);
1775
1776   /// Get the (predicated) backedge count for the analyzed loop.
1777   const SCEV *getBackedgeTakenCount();
1778
1779   /// Adds a new predicate.
1780   void addPredicate(const SCEVPredicate &Pred);
1781
1782   /// Attempts to produce an AddRecExpr for V by adding additional SCEV
1783   /// predicates. If we can't transform the expression into an AddRecExpr we
1784   /// return nullptr and not add additional SCEV predicates to the current
1785   /// context.
1786   const SCEVAddRecExpr *getAsAddRec(Value *V);
1787
1788   /// Proves that V doesn't overflow by adding SCEV predicate.
1789   void setNoOverflow(Value *V, SCEVWrapPredicate::IncrementWrapFlags Flags);
1790
1791   /// Returns true if we've proved that V doesn't wrap by means of a SCEV
1792   /// predicate.
1793   bool hasNoOverflow(Value *V, SCEVWrapPredicate::IncrementWrapFlags Flags);
1794
1795   /// Returns the ScalarEvolution analysis used.
1796   ScalarEvolution *getSE() const { return &SE; }
1797
1798   /// We need to explicitly define the copy constructor because of FlagsMap.
1799   PredicatedScalarEvolution(const PredicatedScalarEvolution &);
1800
1801   /// Print the SCEV mappings done by the Predicated Scalar Evolution.
1802   /// The printed text is indented by \p Depth.
1803   void print(raw_ostream &OS, unsigned Depth) const;
1804
1805 private:
1806   /// Increments the version number of the predicate.  This needs to be called
1807   /// every time the SCEV predicate changes.
1808   void updateGeneration();
1809
1810   /// Holds a SCEV and the version number of the SCEV predicate used to
1811   /// perform the rewrite of the expression.
1812   typedef std::pair<unsigned, const SCEV *> RewriteEntry;
1813
1814   /// Maps a SCEV to the rewrite result of that SCEV at a certain version
1815   /// number. If this number doesn't match the current Generation, we will
1816   /// need to do a rewrite. To preserve the transformation order of previous
1817   /// rewrites, we will rewrite the previous result instead of the original
1818   /// SCEV.
1819   DenseMap<const SCEV *, RewriteEntry> RewriteMap;
1820
1821   /// Records what NoWrap flags we've added to a Value *.
1822   ValueMap<Value *, SCEVWrapPredicate::IncrementWrapFlags> FlagsMap;
1823
1824   /// The ScalarEvolution analysis.
1825   ScalarEvolution &SE;
1826
1827   /// The analyzed Loop.
1828   const Loop &L;
1829
1830   /// The SCEVPredicate that forms our context. We will rewrite all
1831   /// expressions assuming that this predicate true.
1832   SCEVUnionPredicate Preds;
1833
1834   /// Marks the version of the SCEV predicate used. When rewriting a SCEV
1835   /// expression we mark it with the version of the predicate. We use this to
1836   /// figure out if the predicate has changed from the last rewrite of the
1837   /// SCEV. If so, we need to perform a new rewrite.
1838   unsigned Generation;
1839
1840   /// The backedge taken count.
1841   const SCEV *BackedgeCount;
1842 };
1843 }
1844
1845 #endif