[SCEV] Compute exit counts for unsigned IVs using mustprogress semantics
authorPhilip Reames <listmail@philipreames.com>
Mon, 7 Jun 2021 18:16:23 +0000 (11:16 -0700)
committerPhilip Reames <listmail@philipreames.com>
Mon, 7 Jun 2021 18:24:00 +0000 (11:24 -0700)
commit38540d71c74c2f63a77993e7bfb6e52e4b0da0fc
tree6e5ed181a0778e8fa8bb6ec988677438a2d034c1
parent00b6463b269f6815a8a110b9208f483e239ce1e2
[SCEV] Compute exit counts for unsigned IVs using mustprogress semantics

The motivation here is simple loops with unsigned induction variables w/non-one steps. A toy example would be:
for (unsigned i = 0; i < N; i += 2) { body; }

Given C/C++ semantics, we do not get the nuw flag on the induction variable. Given that lack, we currently can't compute a bound for this loop. We can do better for many cases, depending on the contents of "body".

The basic intuition behind this patch is as follows:
* A step which evenly divides the iteration space must wrap through the same numbers repeatedly. And thus, we can ignore potential cornercases where we exit after the n-th wrap through uint32_max.
* Per C++ rules, infinite loops without side effects are UB. We already have code in SCEV which relies on this.  In LLVM, this is tied to the mustprogress attribute.

Together, these let us conclude that the trip count of this loop must come before unsigned overflow unless the body would form a well defined infinite loop.

A couple notes for those reading along:
* I reused the loop properties code which is overly conservative for this case. I may follow up in another patch to generalize it for the actual UB rules.
* We could cache the n(s/u)w facts. I left that out because doing a pre-patch which cached existing inference showed a lot of diffs I had trouble fully explaining. I plan to get back to this, but I don't want it on the critical path.

Differential Revision: https://reviews.llvm.org/D103118
llvm/include/llvm/Analysis/ScalarEvolution.h
llvm/lib/Analysis/ScalarEvolution.cpp
llvm/test/Analysis/ScalarEvolution/lt-overflow.ll [new file with mode: 0644]