diff options
| author | Pranav Bhandarkar <pranav.bhandarkar@amd.com> | 2025-10-22 12:18:56 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-10-22 12:18:56 -0500 |
| commit | e2ad55499197db540d09e7201b9b80366a0908c3 (patch) | |
| tree | 6468aa546e52a7b428dabbbdb1ed36d01426c67b /llvm/lib/Bitcode | |
| parent | 866879f80342b857a8b911c804189c43ac4fc334 (diff) | |
| download | llvm-e2ad55499197db540d09e7201b9b80366a0908c3.zip llvm-e2ad55499197db540d09e7201b9b80366a0908c3.tar.gz llvm-e2ad55499197db540d09e7201b9b80366a0908c3.tar.bz2 | |
[Flang][mlir] - Translation of delayed privatization for deferred target-tasks (#155348)
This PR adds support for translation of the private clause on deferred
target tasks - that is `omp.target` operations with the `nowait` clause.
An offloading call for a deferred target-task is not blocking - the
offloading (target-generating) host task continues its execution after issuing the offloading
call. Therefore, the key problem we need to solve is to ensure that the
data needed for private variables to be initialized in the target task
persists even after the host task has completed.
We do this in a new pass called `PrepareForOMPOffloadPrivatizationPass`.
For a privatized variable that needs its host counterpart for
initialization (such as the shape of the data from the descriptor when
an allocatable is privatized or the value of the data when an
allocatable is firstprivatized),
- the pass allocates memory on the heap.
- it then initializes this memory by using the `init` and `copy` (for
firstprivate) regions of the corresponding `omp::PrivateClauseOp`.
- Finally the memory allocated on the heap is freed using the `dealloc`
region of the same `omp::PrivateClauseOp` instance. This step is not
straightforward though, because we cannot simply free the memory that's
going to be used by another thread without any synchronization. So, for
deallocation, we create a `omp.task` after the `omp.target` and
synchronize the two with a dummy dependency (using the `depend` clause).
In this newly created `omp.task` we do the deallocation.
Diffstat (limited to 'llvm/lib/Bitcode')
0 files changed, 0 insertions, 0 deletions
