[Mlir-commits] [mlir] [OpenMP][MLIR] Add omp.distribute op to the OMP dialect (PR #67720)

Sergio Afonso llvmlistbot at llvm.org
Fri Sep 29 06:01:15 PDT 2023


skatrak wrote:

I just created a dependent PR with lowering support from PFT to create instances of this operation, in case that helps understanding the representation proposed here.

> Could you share your plan before we proceed? The proposal of this patch of a distribute operation that (possibly) nests a loop seems to be different from both the existing worksharing-loop design (that includes the loop) and the canonical-loop proposal under design where the distribute will accept a CLI value that represents a loop.

You make a good point about this representation being slightly different. Since the canonical loop proposal might take some time before it's finished, in my opinion a good approach could be to mimic `omp.wsloop` and produce new loop index variables as entry block arguments to this op's region. I guess in that case we'd have to think about what the `omp.distribute` loop bounds and steps would be for a given nested Do loop, and how they might interoperate with the corresponding `omp.wsloop` operation.

https://github.com/llvm/llvm-project/pull/67720


More information about the Mlir-commits mailing list