diff options
author | Tobias Burnus <tburnus@baylibre.com> | 2024-05-29 15:14:38 +0200 |
---|---|---|
committer | Tobias Burnus <tburnus@baylibre.com> | 2024-05-29 15:14:38 +0200 |
commit | 4ccb3366ade6ec9493f8ca20ab73b0da4b9816db (patch) | |
tree | a4f4a29116032e83903cb2352021873e7f5bfb2a /include | |
parent | 19c491d1848a8410559247183597096778967edf (diff) | |
download | gcc-4ccb3366ade6ec9493f8ca20ab73b0da4b9816db.zip gcc-4ccb3366ade6ec9493f8ca20ab73b0da4b9816db.tar.gz gcc-4ccb3366ade6ec9493f8ca20ab73b0da4b9816db.tar.bz2 |
libgomp: Enable USM for some nvptx devices
A few high-end nvptx devices support the attribute
CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS; for those, unified shared
memory is supported in hardware. This patch enables support for those -
if all installed nvptx devices have this feature (as the capabilities
are per device type).
This exposes a bug in gomp_copy_back_icvs as it did before use
omp_get_mapped_ptr to find mapped variables, but that returns
the unchanged pointer in cased of shared memory. But in this case,
we have a few actually mapped pointers - like the ICV variables.
Additionally, there was a mismatch with regards to '-1' for the
device number as gomp_copy_back_icvs and omp_get_mapped_ptr count
differently. Hence, do the lookup manually.
include/ChangeLog:
* cuda/cuda.h (CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS): Add.
libgomp/ChangeLog:
* libgomp.texi (nvptx): Update USM description.
* plugin/plugin-nvptx.c (GOMP_OFFLOAD_get_num_devices):
Claim support when requesting USM and all devices support
CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS.
* target.c (gomp_copy_back_icvs): Fix device ptr lookup.
(gomp_target_init): Set GOMP_OFFLOAD_CAP_SHARED_MEM is the
devices supports USM.
Diffstat (limited to 'include')
-rw-r--r-- | include/cuda/cuda.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/include/cuda/cuda.h b/include/cuda/cuda.h index 0dca4b3..804d08c 100644 --- a/include/cuda/cuda.h +++ b/include/cuda/cuda.h @@ -83,7 +83,8 @@ typedef enum { CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_MULTIPROCESSOR = 39, CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT = 40, CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING = 41, - CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_MULTIPROCESSOR = 82 + CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_MULTIPROCESSOR = 82, + CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS = 88 } CUdevice_attribute; enum { |