Commit 343f573
committed
[MLIR][OpenMP] Normalize lowering of omp.loop_nest
This patch refactors the translation of `omp.loop_nest` operations into LLVM IR
so that it is handled similarly to other operations. Before this change, the
responsibility of translating the loop nest fell into each loop wrapper,
causing code duplication. This patch centralizes that handling of the loop.
One consequence of this was fixing an issue lowering non-inclusive `omp.simd`
loops.
As a result, it is now expected that the handling of composite constructs is
performed collaboratively among translating functions for each operation
involved. At the moment, only `do/for simd` is supported by ignoring SIMD
information, and this behavior is preserved.
The translation of loop wrapper operations needs access to the
`llvm::CanonicalLoopInfo` loop information structure in order to apply
transformations to it. This is now created in the nested call to
`convertOmpLoopNest`, so it needs to be passed up to all associated loop
wrapper translation functions. This is done via the creation of an
`OpenMPLoopInfoStackFrame` within `convertHostOrTargetOperation`, associated to
the outermost loop wrapper. This structure is updated by `convertOmpLoopNest`,
making the result available to all loop wrappers after their body has been
translated.1 parent 9b52d9e commit 343f573
File tree
4 files changed
+285
-285
lines changed- mlir
- include/mlir/Target/LLVMIR
- lib/Target/LLVMIR/Dialect/OpenMP
- test/Target/LLVMIR
4 files changed
+285
-285
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
290 | 290 | | |
291 | 291 | | |
292 | 292 | | |
293 | | - | |
294 | | - | |
| 293 | + | |
295 | 294 | | |
296 | 295 | | |
297 | 296 | | |
298 | 297 | | |
299 | | - | |
| 298 | + | |
300 | 299 | | |
301 | 300 | | |
302 | 301 | | |
| |||
0 commit comments