[OpenMPIRBuilder] Implement CreateCanonicalLoop.

Authored by Meinersbur on Nov 9 2020, 12:13 PM.


[OpenMPIRBuilder] Implement CreateCanonicalLoop.

CreateCanonicalLoop generates a standardized control flow structure for OpenMP canonical for loops. The structure can be consumed by loop-associated directives such as worksharing-loop, distribute, simd etc. as well as loop transformations such as tile and unroll.

This is a first design without considering all complexities yet. The control-flow emits more basic block than strictly necessary, but these will be optimized by CFGSimplify anyway, provide a nice separation of concerns and might later be useful with more complex scenarios. I successfully implemented a basic tile construct using this API, which is not part of this patch.

The fundamental building block is the CreateCanonicalLoop that only takes the loop trip count and operates on the logical iteration spaces only. An overloaded CreateCanonicalLoop for using LB, UB, Increment is provided as well, but at least for C++, Clang will need to implement a loop counter to logical induction variable mapping anyway, since iterator overload resolution cannot be done in LLVMFrontend.

As there currently is no user for CreateCanonicalLoop, it is only called from unittests. Similarly, CanonicalLoopInfo::eraseFromParent() is used in my file implementation and might be generally useful for implementing loop-associated constructs, but is not used in this patch itself.

The following non-exhaustive list describes not yet covered items:

  • collapse clause (including non-rectangular and non-perfectly nested); idea is to provide a OpenMPIRBuilder::collapseLoopNest method consuming multiple nested loops and returning a new CanonicalLoopInfo that can be used for loop-associated directives.
  • simarly: ordered clause for DOACROSS loops
  • branch weights
  • Cancellation point (?)
  • AllocaIP
  • break statement (if needed at all)
  • Exceptions (if not completely handled in the front-end)
    • Using it in Clang; this requires implementing at least one loop-associated construct.
  • ...

Reviewed By: jdoerfert

Differential Revision: https://reviews.llvm.org/D90830


MeinersburNov 9 2020, 1:03 PM
Differential Revision
D90830: [OpenMPIRBuilder] Implement CreateCanonicalLoop.
rG9d02e0e38df5: [mlir][std] Add ExpandOps pass.