From 998f1156a51010490c2e918defb4517706916c92 Mon Sep 17 00:00:00 2001 From: Thomas Schwinge Date: Wed, 15 Mar 2023 13:34:02 +0100 Subject: [PATCH] Given OpenACC 'async', defer 'free' of non-contiguous array support data structures Fix-up for og12 commit 15d0f61a7fecdc8fd12857c40879ea3730f6d99f "Merge non-contiguous array support patches". libgomp/ * oacc-parallel.c (GOACC_parallel_keyed): Given OpenACC 'async', defer 'free' of non-contiguous array support data structures. * target.c (gomp_map_vars_internal): Likewise. --- libgomp/oacc-parallel.c | 5 ++++- libgomp/target.c | 6 +++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/libgomp/oacc-parallel.c b/libgomp/oacc-parallel.c index 9cd99b4a0b4..136702d6e61 100644 --- a/libgomp/oacc-parallel.c +++ b/libgomp/oacc-parallel.c @@ -470,7 +470,10 @@ GOACC_parallel_keyed (int flags_m, void (*fn) (void *), = goacc_map_vars (acc_dev, aq, mapnum, hostaddrs, NULL, sizes, kinds, nca_info, true, GOMP_MAP_VARS_TARGET); - free (nca_info); + if (aq == NULL) + free (nca_info); + else + acc_dev->openacc.async.queue_callback_func (aq, free, nca_info); if (profiling_p) { diff --git a/libgomp/target.c b/libgomp/target.c index 96ece0b31fd..aaa597f6610 100644 --- a/libgomp/target.c +++ b/libgomp/target.c @@ -1938,7 +1938,11 @@ gomp_map_vars_internal (struct gomp_device_descr *devicep, (nca, target_ptrblock); gomp_copy_host2dev (devicep, aq, target_ptrblock, ptrblock, nca->ptrblock_size, false, cbufp); - free (ptrblock); + if (aq) + /* Free once the transfer has completed. */ + devicep->openacc.async.queue_callback_func (aq, free, ptrblock); + else + free (ptrblock); } } } -- 2.25.1