LLVM Loop Terminology (and Canonical Forms)

Loop Definition

Loops are an important concept for a code optimizer. In LLVM, detectionof loops in a control-flow graph is done byLoopInfo. It is basedon the following definition.

A loop is a subset of nodes from the control-flow graph (CFG; wherenodes represent basic blocks) with the following properties:

  1. The induced subgraph (which is the subgraph that contains all theedges from the CFG within the loop) is strongly connected(every node is reachable from all others).

  2. All edges from outside the subset into the subset point to the samenode, called theheader. As a consequence, the header dominatesall nodes in the loop (i.e. every execution path to any of the loop’snode will have to pass through the header).

  3. The loop is the maximum subset with these properties. That is, noadditional nodes from the CFG can be added such that the inducedsubgraph would still be strongly connected and the header wouldremain the same.

In computer science literature, this is often called anatural loop.In LLVM, a more generalized definition is called acycle.

Terminology

The definition of a loop comes with some additional terminology:

  • Anentering block (orloop predecessor) is a non-loop nodethat has an edge into the loop (necessarily the header). If there isonly one entering block, and its only edge is to theheader, it is also called the loop’spreheader. The preheaderdominates the loop without itself being part of the loop.

  • Alatch is a loop node that has an edge to the header.

  • Abackedge is an edge from a latch to the header.

  • Anexiting edge is an edge from inside the loop to a node outsideof the loop. The source of such an edge is called anexiting block, itstarget is anexit block.

_images/loop-terminology.svg

Important Notes

This loop definition has some noteworthy consequences:

  • A node can be the header of at most one loop. As such, a loop can beidentified by its header. Due to the header being the only entry intoa loop, it can be called a Single-Entry-Multiple-Exits (SEME) region.

  • For basic blocks that are not reachable from the function’s entry, theconcept of loops is undefined. This follows from the concept ofdominance being undefined as well.

  • The smallest loop consists of a single basic block that branches toitself. In this case that block is the header, latch (and exitingblock if it has another edge to a different block) at the same time.A single block that has no branch to itself is not considered a loop,even though it is trivially strongly connected.

_images/loop-single.svg

In this case, the role of header, exiting block and latch fall to thesame node.LoopInfo reports this as:

$optinput.ll-passes='print<loops>'Loop at depth 1 containing: %for.body<header><latch><exiting>
  • Loops can be nested inside each other. That is, a loop’s node set canbe a subset of another loop with a different loop header. The loophierarchy in a function forms a forest: Each top-level loop is theroot of the tree of the loops nested inside it.

_images/loop-nested.svg
  • It is not possible that two loops share only a few of their nodes.Two loops are either disjoint or one is nested inside the other. Inthe example below the left and right subsets both violate themaximality condition. Only the merge of both sets is considered a loop.

_images/loop-nonmaximal.svg
  • It is also possible that two logical loops share a header, but areconsidered a single loop by LLVM:

for(inti=0;i<128;++i)for(intj=0;j<128;++j)body(i,j);

which might be represented in LLVM-IR as follows. Note that there isonly a single header and hence just a single loop.

_images/loop-merge.svg

TheLoopSimplify pass willdetect the loop and ensure separate headers for the outer and inner loop.

_images/loop-separate.svg
  • A cycle in the CFG does not imply there is a loop. The example belowshows such a CFG, where there is no header node that dominates allother nodes in the cycle. This is calledirreducible control-flow.

_images/loop-irreducible.svg

The term reducible results from the ability to collapse the CFG into asingle node by successively replacing one of three base structures witha single node: A sequential execution of basic blocks, acyclic conditionalbranches (or switches), and a basic block looping on itself.Wikipediahas a more formal definition, which basically says that every cycle hasa dominating header.

  • Irreducible control-flow can occur at any level of the loop nesting.That is, a loop that itself does not contain any loops can still havecyclic control flow in its body; a loop that is not nested insideanother loop can still be part of an outer cycle; and there can beadditional cycles between any two loops where one is contained in the other.However, an LLVMcycle covers both, loops andirreducible control flow.

  • TheFixIrreduciblepass can transform irreducible control flow into loops by insertingnew loop headers. It is not included in any default optimization passpipeline, but is required for some back-end targets.

  • Exiting edges are not the only way to break out of a loop. Otherpossibilities are unreachable terminators, [[noreturn]] functions,exceptions, signals, and your computer’s power button.

  • A basic block “inside” the loop that does not have a path back to theloop (i.e. to a latch or header) is not considered part of the loop.This is illustrated by the following code.

for(unsignedi=0;i<=n;++i){if(c1){// When reaching this block, we will have exited the loop.do_something();break;}if(c2){// abort(), never returns, so we have exited the loop.abort();}if(c3){// The unreachable allows the compiler to assume that this will not rejoin the loop.do_something();__builtin_unreachable();}if(c4){// This statically infinite loop is not nested because control-flow will not continue with the for-loop.while(true){do_something();}}}
  • There is no requirement for the control flow to eventually leave theloop, i.e. a loop can be infinite. Astatically infinite loop is aloop that has no exiting edges. Adynamically infinite loop hasexiting edges, but it is possible to be never taken. This may happenonly under some circumstances, such as when n == UINT_MAX in the codebelow.

for(unsignedi=0;i<=n;++i)body(i);

It is possible for the optimizer to turn a dynamically infinite loopinto a statically infinite loop, for instance when it can prove that theexiting condition is always false. Because the exiting edge is nevertaken, the optimizer can change the conditional branch into anunconditional one.

If a is loop is annotated withllvm.loop.mustprogress metadata,the compiler is allowed to assume that it will eventually terminate, evenif it cannot prove it. For instance, it may remove a mustprogress-loopthat does not have any side-effect in its body even though the programcould be stuck in that loop forever. Languages such as C andC++ have suchforward-progress guarantees for some loops. Also see themustprogress andwillreturn function attributes, as well asthe olderllvm.sideeffect intrinsic.

  • The number of executions of the loop header before leaving the loop istheloop trip count (oriteration count). If the loop shouldnot be executed at all, aloop guard must skip the entire loop:

_images/loop-guard.svg

Since the first thing a loop header might do is to check whether thereis another execution and if not, immediately exit without doing any work(also seeRotated Loops), loop trip count is notthe best measure of a loop’s number of iterations. For instance, thenumber of header executions of the code below for a non-positive n(before loop rotation) is 1, even though the loop body is not executedat all.

for(inti=0;i<n;++i)body(i);

A better measure is thebackedge-taken count, which is the number oftimes any of the backedges is taken before the loop. It is one less thanthe trip count for executions that enter the header.

LoopInfo

LoopInfo is the core analysis for obtaining information about loops.There are few key implications of the definitions given above whichare important for working successfully with this interface.

  • LoopInfo does not contain information about non-loop cycles. As aresult, it is not suitable for any algorithm which requires completecycle detection for correctness.

  • LoopInfo provides an interface for enumerating all top level loops(e.g. those not contained in any other loop). From there, you maywalk the tree of sub-loops rooted in that top level loop.

  • Loops which become statically unreachable during optimizationmustbe removed from LoopInfo. If this can not be done for some reason,then the optimization isrequired to preserve the staticreachability of the loop.

Loop Simplify Form

The Loop Simplify Form is a canonical form that makesseveral analyses and transformations simpler and more effective.It is ensured by the LoopSimplify(-loop-simplify) pass and is automaticallyadded by the pass managers when scheduling a LoopPass.This pass is implemented inLoopSimplify.h.When it is successful, the loop has:

  • A preheader.

  • A single backedge (which implies that there is a single latch).

  • Dedicated exits. That is, no exit block for the loophas a predecessor that is outside the loop. This impliesthat all exit blocks are dominated by the loop header.

Loop Closed SSA (LCSSA)

A program is in Loop Closed SSA Form if it is in SSA formand all values that are defined in a loop are used only insidethis loop.

Programs written in LLVM IR are always in SSA form but not necessarilyin LCSSA. To achieve the latter, for each value that is live across theloop boundary, single entry PHI nodes are inserted to each of the exit blocks[1] in order to “close” these values inside the loop.In particular, consider the following loop:

c=...;for(...){if(c)X1=...elseX2=...X3=phi(X1,X2);// X3 defined}...=X3+4;// X3 used, i.e. live// outside the loop

In the inner loop, the X3 is defined inside the loop, but usedoutside of it. In Loop Closed SSA form, this would be represented as follows:

c=...;for(...){if(c)X1=...elseX2=...X3=phi(X1,X2);}X4=phi(X3);...=X4+4;

This is still valid LLVM; the extra phi nodes are purely redundant,but all LoopPass’es are required to preserve them.This form is ensured by the LCSSA (-lcssa)pass and is added automatically by the LoopPassManager whenscheduling a LoopPass.After the loop optimizations are done, these extra phi nodeswill be deleted by-instcombine.

Note that an exit block is outside of a loop, so how can such a phi “close”the value inside the loop since it uses it outside of it ? First of all,for phi nodes, asmentioned in the LangRef:“the use of each incoming value is deemed to occur on the edge from thecorresponding predecessor block to the current block”. Now, anedge to an exit block is considered outside of the loop becauseif we take that edge, it leads us clearly out of the loop.

However, an edge doesn’t actually contain any IR, so in source code,we have to choose a convention of whether the use happens inthe current block or in the respective predecessor. For LCSSA’s purpose,we consider the use happens in the latter (so as to consider theuse inside)[2].

The major benefit of LCSSA is that it makes many other loop optimizationssimpler.

First of all, a simple observation is that if one needs to see allthe outside users, they can just iterate over all the (loop closing)PHI nodes in the exit blocks (the alternative would be toscan the def-use chain[3] of all instructions in the loop).

Then, consider for examplesimple-loop-unswitch ing the loop above.Because it is in LCSSA form, we know that any value defined inside ofthe loop will be used either only inside the loop or in a loop closingPHI node. In this case, the only loop closing PHI node is X4.This means that we can just copy the loop and change the X4accordingly, like so:

c=...;if(c){for(...){if(true)X1=...elseX2=...X3=phi(X1,X2);}}else{for(...){if(false)X1'=...elseX2'=...X3'=phi(X1',X2');}}X4=phi(X3,X3')

Now, all uses of X4 will get the updated value (in general,if a loop is in LCSSA form, in any loop transformation,we only need to update the loop closing PHI nodes for the changesto take effect). If we did not have Loop Closed SSA form, it means that X3 couldpossibly be used outside the loop. So, we would have to introduce theX4 (which is the new X3) and replace all uses of X3 with that.However, we should note that because LLVM keeps a def-use chain[3] for each Value, we wouldn’t needto perform data-flow analysis to find and replace all the uses(there is even a utility function, replaceAllUsesWith(),that performs this transformation by iterating the def-use chain).

Another important advantage is that the behavior of all usesof an induction variable is the same. Without this, you need todistinguish the case when the variable is used outside ofthe loop it is defined in, for example:

for(i=0;i<100;i++){for(j=0;j<100;j++){k=i+j;use(k);// use 1}use(k);// use 2}

Looking from the outer loop with the normal SSA form, the first use of kis not well-behaved, while the second one is an induction variable withbase 100 and step 1. Although, in practice, and in the LLVM context,such cases can be handled effectively by SCEV. Scalar Evolution(scalar-evolution) or SCEV, is a(analysis) pass that analyzes and categorizes the evolution of scalarexpressions in loops.

In general, it’s easier to use SCEV in loops that are in LCSSA form.The evolution of a scalar (loop-variant) expression thatSCEV can analyze is, by definition, relative to a loop.An expression is represented in LLVM by anllvm::Instruction.If the expression is inside two (or more) loops (which can onlyhappen if the loops are nested, like in the example above) and you wantto get an analysis of its evolution (from SCEV),you have to also specify relative to what Loop you want it.Specifically, you have to usegetSCEVAtScope().

However, if all loops are in LCSSA form, each expression is actuallyrepresented by two different llvm::Instructions. One inside the loopand one outside, which is the loop-closing PHI node and representsthe value of the expression after the last iteration (effectively,we break each loop-variant expression into two expressions and so, everyexpression is at most in one loop). You can now just usegetSCEV().and which of these two llvm::Instructions you pass to it disambiguatesthe context / scope / relative loop.

Footnotes

[1]

To insert these loop-closing PHI nodes, one has to(re-)compute dominance frontiers (if the loop has multiple exits).

[2]

Considering the point of use of a PHI entry valueto be in the respective predecessor is a convention across the whole LLVM.The reason is mostly practical; for example it preserves the dominanceproperty of SSA. It is also just an overapproximation of the actualnumber of uses; the incoming block could branch to another block in whichcase the value is not actually used but there are no side-effects (it mightincrease its live range which is not relevant in LCSSA though).Furthermore, we can gain some intuition if we consider liveness:A PHI isusually inserted in the current block because the value can’tbe used from this point and onwards (i.e. the current block is a dominancefrontier). It doesn’t make sense to consider that the value is used inthe current block (because of the PHI) since the value stops being livebefore the PHI. In some sense the PHI definition just “replaces” the originalvalue definition and doesn’t actually use it. It should be stressed thatthis analogy is only used as an example and does not pose any strictrequirements. For example, the value might dominate the current blockbut we can still insert a PHI (as we do with LCSSA PHI nodes)anduse the original value afterwards (in which case the two live ranges overlap,although in LCSSA (the whole point is that) we never do that).

[3](1,2)

A property of SSA is that there exists a def-use chainfor each definition, which is a list of all the uses of this definition.LLVM implements this property by keeping a list of all the uses of a Valuein an internal data structure.

“More Canonical” Loops

Rotated Loops

Loops are rotated by the LoopRotate (loop-rotate)pass, which converts loops into do/while style loops and isimplemented inLoopRotation.h. Example:

voidtest(intn){for(inti=0;i<n;i+=1)// Loop body}

is transformed to:

voidtest(intn){inti=0;do{// Loop bodyi+=1;}while(i<n);}

Warning: This transformation is valid only if the compilercan prove that the loop body will be executed at least once. Otherwise,it has to insert a guard which will test it at runtime. In the exampleabove, that would be:

voidtest(intn){inti=0;if(n>0){do{// Loop bodyi+=1;}while(i<n);}}

It’s important to understand the effect of loop rotationat the LLVM IR level. We follow with the previous examplesin LLVM IR while also providing a graphical representationof the control-flow graphs (CFG). You can get the same graphicalresults by utilizing theview-cfg pass.

The initialfor loop could be translated to:

define void @test(i32 %n) {entry:  br label %for.headerfor.header:  %i = phi i32 [ 0, %entry ], [ %i.next, %latch ]  %cond = icmp slt i32 %i, %n  br i1 %cond, label %body, label %exitbody:  ; Loop body  br label %latchlatch:  %i.next = add nsw i32 %i, 1  br label %for.headerexit:  ret void}
_images/loop-terminology-initial-loop.png

Before we explain how LoopRotate will actuallytransform this loop, here’s how we could convertit (by hand) to a do-while style loop.

define void @test(i32 %n) {entry:  br label %bodybody:  %i = phi i32 [ 0, %entry ], [ %i.next, %latch ]  ; Loop body  br label %latchlatch:  %i.next = add nsw i32 %i, 1  %cond = icmp slt i32 %i.next, %n  br i1 %cond, label %body, label %exitexit:  ret void}
_images/loop-terminology-rotated-loop.png

Note two things:

  • The condition check was moved to the “bottom” of the loop, i.e.the latch. This is something that LoopRotate does by copying the headerof the loop to the latch.

  • The compiler in this case can’t deduce that the loop willdefinitely execute at least once so the above transformationis not valid. As mentioned above, a guard has to be inserted,which is something that LoopRotate will do.

This is how LoopRotate transforms this loop:

define void @test(i32 %n) {entry:  %guard_cond = icmp slt i32 0, %n  br i1 %guard_cond, label %loop.preheader, label %exitloop.preheader:  br label %bodybody:  %i2 = phi i32 [ 0, %loop.preheader ], [ %i.next, %latch ]  br label %latchlatch:  %i.next = add nsw i32 %i2, 1  %cond = icmp slt i32 %i.next, %n  br i1 %cond, label %body, label %loop.exitloop.exit:  br label %exitexit:  ret void}
_images/loop-terminology-guarded-loop.png

The result is a little bit more complicated than we may expectbecause LoopRotate ensures that the loop is inLoop Simplify Formafter rotation.In this case, it inserted the %loop.preheader basic block sothat the loop has a preheader and it introduced the %loop.exitbasic block so that the loop has dedicated exits(otherwise, %exit would be jumped from both %latch and %entry,but %entry is not contained in the loop).Note that a loop has to be in Loop Simplify Form beforehandtoo for LoopRotate to be applied successfully.

The main advantage of this form is that it allows hoistinginvariant instructions, especially loads, into the preheader.That could be done in non-rotated loops as well but withsome disadvantages. Let’s illustrate them with an example:

for(inti=0;i<n;++i){autov=*p;use(v);}

We assume that loading from p is invariant and use(v) is somestatement that uses v.If we wanted to execute the load only once we could move it“out” of the loop body, resulting in this:

autov=*p;for(inti=0;i<n;++i){use(v);}

However, now, in the case that n <= 0, in the initial form,the loop body would never execute, and so, the load wouldnever execute. This is a problem mainly for semantic reasons.Consider the case in which n <= 0 and loading from p is invalid.In the initial program there would be no error. However, with thistransformation we would introduce one, effectively breakingthe initial semantics.

To avoid both of these problems, we can insert a guard:

if(n>0){// loop guardautov=*p;for(inti=0;i<n;++i){use(v);}}

This is certainly better but it could be improved slightly. Noticethat the check for whether n is bigger than 0 is executed twice (andn does not change in between). Once when we check the guard conditionand once in the first execution of the loop. To avoid that, we coulddo an unconditional first execution and insert the loop conditionin the end. This effectively means transforming the loop into a do-while loop:

if(0<n){autov=*p;do{use(v);++i;}while(i<n);}

Note that LoopRotate does not generally do suchhoisting. Rather, it is an enabling transformation for otherpasses like Loop-Invariant Code Motion (-licm).