99835bd68e
The implementation of each allocator is simplified by inheriting from __detail::__allocator_templ. At the moment, none of the implementations diverge in any way, simply passing in the allocator handle to be used when an allocation is made. In the future, const_mem will need special handling added to it to support constant memory space. libgomp/ChangeLog: * omp.h.in: Add omp::allocator::* and ompx::allocator::* allocators. (__detail::__allocator_templ<T, omp_allocator_handle_t>): New struct template. (null_allocator<T>): New struct template. (default_mem<T>): Likewise. (large_cap_mem<T>): Likewise. (const_mem<T>): Likewise. (high_bw_mem<T>): Likewise. (low_lat_mem<T>): Likewise. (cgroup_mem<T>): Likewise. (pteam_mem<T>): Likewise. (thread_mem<T>): Likewise. (ompx::allocator::gnu_pinned_mem<T>): Likewise. * testsuite/libgomp.c++/allocator-1.C: New test. * testsuite/libgomp.c++/allocator-2.C: New test. Signed-off-by: waffl3x <waffl3x@baylibre.com>
567 lines
17 KiB
C
567 lines
17 KiB
C
/* Copyright (C) 2005-2025 Free Software Foundation, Inc.
|
|
Contributed by Richard Henderson <rth@redhat.com>.
|
|
|
|
This file is part of the GNU Offloading and Multi Processing Library
|
|
(libgomp).
|
|
|
|
Libgomp is free software; you can redistribute it and/or modify it
|
|
under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 3, or (at your option)
|
|
any later version.
|
|
|
|
Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
more details.
|
|
|
|
Under Section 7 of GPL version 3, you are granted additional
|
|
permissions described in the GCC Runtime Library Exception, version
|
|
3.1, as published by the Free Software Foundation.
|
|
|
|
You should have received a copy of the GNU General Public License and
|
|
a copy of the GCC Runtime Library Exception along with this program;
|
|
see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
|
|
<http://www.gnu.org/licenses/>. */
|
|
|
|
#ifndef _OMP_H
|
|
#define _OMP_H 1
|
|
|
|
#if defined(__GNUC__) && _OPENMP >= 201811
|
|
# define __GOMP_DEPRECATED_5_0 __attribute__((__deprecated__))
|
|
#else
|
|
# define __GOMP_DEPRECATED_5_0
|
|
#endif
|
|
|
|
#if defined(__GNUC__) && _OPENMP >= 202011
|
|
# define __GOMP_DEPRECATED_5_1 __attribute__((__deprecated__))
|
|
#else
|
|
# define __GOMP_DEPRECATED_5_1
|
|
#endif
|
|
|
|
#ifndef _LIBGOMP_OMP_LOCK_DEFINED
|
|
#define _LIBGOMP_OMP_LOCK_DEFINED 1
|
|
/* These two structures get edited by the libgomp build process to
|
|
reflect the shape of the two types. Their internals are private
|
|
to the library. */
|
|
|
|
typedef struct
|
|
{
|
|
unsigned char _x[@OMP_LOCK_SIZE@]
|
|
__attribute__((__aligned__(@OMP_LOCK_ALIGN@)));
|
|
} omp_lock_t;
|
|
|
|
typedef struct
|
|
{
|
|
unsigned char _x[@OMP_NEST_LOCK_SIZE@]
|
|
__attribute__((__aligned__(@OMP_NEST_LOCK_ALIGN@)));
|
|
} omp_nest_lock_t;
|
|
#endif
|
|
|
|
typedef enum omp_sched_t
|
|
{
|
|
omp_sched_static = 1,
|
|
omp_sched_dynamic = 2,
|
|
omp_sched_guided = 3,
|
|
omp_sched_auto = 4,
|
|
omp_sched_monotonic = 0x80000000U
|
|
} omp_sched_t;
|
|
|
|
typedef enum omp_proc_bind_t
|
|
{
|
|
omp_proc_bind_false = 0,
|
|
omp_proc_bind_true = 1,
|
|
omp_proc_bind_primary = 2,
|
|
omp_proc_bind_master __GOMP_DEPRECATED_5_1
|
|
= omp_proc_bind_primary,
|
|
omp_proc_bind_close = 3,
|
|
omp_proc_bind_spread = 4
|
|
} omp_proc_bind_t;
|
|
|
|
typedef enum omp_sync_hint_t
|
|
{
|
|
omp_sync_hint_none = 0,
|
|
omp_lock_hint_none __GOMP_DEPRECATED_5_0 = omp_sync_hint_none,
|
|
omp_sync_hint_uncontended = 1,
|
|
omp_lock_hint_uncontended __GOMP_DEPRECATED_5_0 = omp_sync_hint_uncontended,
|
|
omp_sync_hint_contended = 2,
|
|
omp_lock_hint_contended __GOMP_DEPRECATED_5_0 = omp_sync_hint_contended,
|
|
omp_sync_hint_nonspeculative = 4,
|
|
omp_lock_hint_nonspeculative __GOMP_DEPRECATED_5_0
|
|
= omp_sync_hint_nonspeculative,
|
|
omp_sync_hint_speculative = 8,
|
|
omp_lock_hint_speculative __GOMP_DEPRECATED_5_0 = omp_sync_hint_speculative
|
|
} omp_sync_hint_t;
|
|
|
|
typedef __GOMP_DEPRECATED_5_0 omp_sync_hint_t omp_lock_hint_t;
|
|
|
|
typedef struct __attribute__((__aligned__ (sizeof (void *)))) omp_depend_t
|
|
{
|
|
char __omp_depend_t__[2 * sizeof (void *)];
|
|
} omp_depend_t;
|
|
|
|
typedef enum omp_pause_resource_t
|
|
{
|
|
omp_pause_soft = 1,
|
|
omp_pause_hard = 2
|
|
} omp_pause_resource_t;
|
|
|
|
typedef __INTPTR_TYPE__ omp_intptr_t;
|
|
typedef __UINTPTR_TYPE__ omp_uintptr_t;
|
|
|
|
#if __cplusplus >= 201103L
|
|
# define __GOMP_UINTPTR_T_ENUM : omp_uintptr_t
|
|
#else
|
|
# define __GOMP_UINTPTR_T_ENUM
|
|
#endif
|
|
|
|
typedef enum omp_memspace_handle_t __GOMP_UINTPTR_T_ENUM
|
|
{
|
|
omp_default_mem_space = 0,
|
|
omp_large_cap_mem_space = 1,
|
|
omp_const_mem_space = 2,
|
|
omp_high_bw_mem_space = 3,
|
|
omp_low_lat_mem_space = 4,
|
|
__omp_memspace_handle_t_max__ = __UINTPTR_MAX__
|
|
} omp_memspace_handle_t;
|
|
|
|
typedef enum omp_allocator_handle_t __GOMP_UINTPTR_T_ENUM
|
|
{
|
|
omp_null_allocator = 0,
|
|
omp_default_mem_alloc = 1,
|
|
omp_large_cap_mem_alloc = 2,
|
|
omp_const_mem_alloc = 3,
|
|
omp_high_bw_mem_alloc = 4,
|
|
omp_low_lat_mem_alloc = 5,
|
|
omp_cgroup_mem_alloc = 6,
|
|
omp_pteam_mem_alloc = 7,
|
|
omp_thread_mem_alloc = 8,
|
|
ompx_gnu_pinned_mem_alloc = 200,
|
|
__omp_allocator_handle_t_max__ = __UINTPTR_MAX__
|
|
} omp_allocator_handle_t;
|
|
|
|
typedef enum omp_alloctrait_key_t
|
|
{
|
|
omp_atk_sync_hint = 1,
|
|
omp_atk_alignment = 2,
|
|
omp_atk_access = 3,
|
|
omp_atk_pool_size = 4,
|
|
omp_atk_fallback = 5,
|
|
omp_atk_fb_data = 6,
|
|
omp_atk_pinned = 7,
|
|
omp_atk_partition = 8
|
|
} omp_alloctrait_key_t;
|
|
|
|
typedef enum omp_alloctrait_value_t
|
|
{
|
|
omp_atv_default = (__UINTPTR_TYPE__) -1,
|
|
omp_atv_false = 0,
|
|
omp_atv_true = 1,
|
|
omp_atv_contended = 3,
|
|
omp_atv_uncontended = 4,
|
|
omp_atv_serialized = 5,
|
|
omp_atv_sequential __GOMP_DEPRECATED_5_1 = omp_atv_serialized,
|
|
omp_atv_private = 6,
|
|
omp_atv_all = 7,
|
|
omp_atv_thread = 8,
|
|
omp_atv_pteam = 9,
|
|
omp_atv_cgroup = 10,
|
|
omp_atv_default_mem_fb = 11,
|
|
omp_atv_null_fb = 12,
|
|
omp_atv_abort_fb = 13,
|
|
omp_atv_allocator_fb = 14,
|
|
omp_atv_environment = 15,
|
|
omp_atv_nearest = 16,
|
|
omp_atv_blocked = 17,
|
|
omp_atv_interleaved = 18
|
|
} omp_alloctrait_value_t;
|
|
|
|
typedef struct omp_alloctrait_t
|
|
{
|
|
omp_alloctrait_key_t key;
|
|
omp_uintptr_t value;
|
|
} omp_alloctrait_t;
|
|
|
|
typedef enum omp_event_handle_t __GOMP_UINTPTR_T_ENUM
|
|
{
|
|
__omp_event_handle_t_max__ = __UINTPTR_MAX__
|
|
} omp_event_handle_t;
|
|
|
|
enum
|
|
{
|
|
omp_initial_device = -1,
|
|
omp_invalid_device = -4
|
|
};
|
|
|
|
typedef enum omp_interop_t __GOMP_UINTPTR_T_ENUM
|
|
{
|
|
omp_interop_none = 0,
|
|
__omp_interop_t_max__ = __UINTPTR_MAX__
|
|
} omp_interop_t;
|
|
|
|
typedef enum omp_interop_fr_t
|
|
{
|
|
omp_ifr_cuda = 1,
|
|
omp_ifr_cuda_driver = 2,
|
|
omp_ifr_opencl = 3,
|
|
omp_ifr_sycl = 4,
|
|
omp_ifr_hip = 5,
|
|
omp_ifr_level_zero = 6,
|
|
omp_ifr_hsa = 7,
|
|
omp_ifr_last = omp_ifr_hsa
|
|
} omp_interop_fr_t;
|
|
|
|
typedef enum omp_interop_property_t
|
|
{
|
|
omp_ipr_fr_id = -1,
|
|
omp_ipr_fr_name = -2,
|
|
omp_ipr_vendor = -3,
|
|
omp_ipr_vendor_name = -4,
|
|
omp_ipr_device_num = -5,
|
|
omp_ipr_platform = -6,
|
|
omp_ipr_device = -7,
|
|
omp_ipr_device_context = -8,
|
|
omp_ipr_targetsync = -9,
|
|
omp_ipr_first = -9
|
|
} omp_interop_property_t;
|
|
|
|
|
|
typedef enum omp_interop_rc_t
|
|
{
|
|
omp_irc_no_value = 1,
|
|
omp_irc_success = 0,
|
|
omp_irc_empty = -1,
|
|
omp_irc_out_of_range = -2,
|
|
omp_irc_type_int = -3,
|
|
omp_irc_type_ptr = -4,
|
|
omp_irc_type_str = -5,
|
|
omp_irc_other = -6
|
|
} omp_interop_rc_t;
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
# define __GOMP_NOTHROW throw ()
|
|
# define __GOMP_DEFAULT_NULL_ALLOCATOR = omp_null_allocator
|
|
# define __GOMP_DEFAULT_NULL = __null
|
|
#else
|
|
# define __GOMP_NOTHROW __attribute__((__nothrow__))
|
|
# define __GOMP_DEFAULT_NULL_ALLOCATOR
|
|
# define __GOMP_DEFAULT_NULL
|
|
#endif
|
|
|
|
extern void omp_set_num_threads (int) __GOMP_NOTHROW;
|
|
extern int omp_get_num_threads (void) __GOMP_NOTHROW;
|
|
extern int omp_get_max_threads (void) __GOMP_NOTHROW;
|
|
extern int omp_get_thread_num (void) __GOMP_NOTHROW;
|
|
extern int omp_get_num_procs (void) __GOMP_NOTHROW;
|
|
|
|
extern int omp_in_parallel (void) __GOMP_NOTHROW;
|
|
|
|
extern void omp_set_dynamic (int) __GOMP_NOTHROW;
|
|
extern int omp_get_dynamic (void) __GOMP_NOTHROW;
|
|
|
|
extern void omp_set_nested (int) __GOMP_NOTHROW __GOMP_DEPRECATED_5_0;
|
|
extern int omp_get_nested (void) __GOMP_NOTHROW __GOMP_DEPRECATED_5_0;
|
|
|
|
extern void omp_init_lock (omp_lock_t *) __GOMP_NOTHROW;
|
|
extern void omp_init_lock_with_hint (omp_lock_t *, omp_sync_hint_t)
|
|
__GOMP_NOTHROW;
|
|
extern void omp_destroy_lock (omp_lock_t *) __GOMP_NOTHROW;
|
|
extern void omp_set_lock (omp_lock_t *) __GOMP_NOTHROW;
|
|
extern void omp_unset_lock (omp_lock_t *) __GOMP_NOTHROW;
|
|
extern int omp_test_lock (omp_lock_t *) __GOMP_NOTHROW;
|
|
|
|
extern void omp_init_nest_lock (omp_nest_lock_t *) __GOMP_NOTHROW;
|
|
extern void omp_init_nest_lock_with_hint (omp_nest_lock_t *, omp_sync_hint_t)
|
|
__GOMP_NOTHROW;
|
|
extern void omp_destroy_nest_lock (omp_nest_lock_t *) __GOMP_NOTHROW;
|
|
extern void omp_set_nest_lock (omp_nest_lock_t *) __GOMP_NOTHROW;
|
|
extern void omp_unset_nest_lock (omp_nest_lock_t *) __GOMP_NOTHROW;
|
|
extern int omp_test_nest_lock (omp_nest_lock_t *) __GOMP_NOTHROW;
|
|
|
|
extern double omp_get_wtime (void) __GOMP_NOTHROW;
|
|
extern double omp_get_wtick (void) __GOMP_NOTHROW;
|
|
|
|
extern void omp_set_schedule (omp_sched_t, int) __GOMP_NOTHROW;
|
|
extern void omp_get_schedule (omp_sched_t *, int *) __GOMP_NOTHROW;
|
|
extern int omp_get_thread_limit (void) __GOMP_NOTHROW;
|
|
extern void omp_set_max_active_levels (int) __GOMP_NOTHROW;
|
|
extern int omp_get_max_active_levels (void) __GOMP_NOTHROW;
|
|
extern int omp_get_supported_active_levels (void) __GOMP_NOTHROW;
|
|
extern int omp_get_level (void) __GOMP_NOTHROW;
|
|
extern int omp_get_ancestor_thread_num (int) __GOMP_NOTHROW;
|
|
extern int omp_get_team_size (int) __GOMP_NOTHROW;
|
|
extern int omp_get_active_level (void) __GOMP_NOTHROW;
|
|
|
|
extern int omp_in_final (void) __GOMP_NOTHROW;
|
|
extern int omp_in_explicit_task (void) __GOMP_NOTHROW;
|
|
|
|
extern int omp_get_cancellation (void) __GOMP_NOTHROW;
|
|
extern omp_proc_bind_t omp_get_proc_bind (void) __GOMP_NOTHROW;
|
|
extern int omp_get_num_places (void) __GOMP_NOTHROW;
|
|
extern int omp_get_place_num_procs (int) __GOMP_NOTHROW;
|
|
extern void omp_get_place_proc_ids (int, int *) __GOMP_NOTHROW;
|
|
extern int omp_get_place_num (void) __GOMP_NOTHROW;
|
|
extern int omp_get_partition_num_places (void) __GOMP_NOTHROW;
|
|
extern void omp_get_partition_place_nums (int *) __GOMP_NOTHROW;
|
|
|
|
extern void omp_set_default_device (int) __GOMP_NOTHROW;
|
|
extern int omp_get_default_device (void) __GOMP_NOTHROW;
|
|
extern int omp_get_num_devices (void) __GOMP_NOTHROW;
|
|
extern int omp_get_device_num (void) __GOMP_NOTHROW;
|
|
extern int omp_get_num_teams (void) __GOMP_NOTHROW;
|
|
extern int omp_get_team_num (void) __GOMP_NOTHROW;
|
|
|
|
extern int omp_is_initial_device (void) __GOMP_NOTHROW;
|
|
extern int omp_get_initial_device (void) __GOMP_NOTHROW;
|
|
extern int omp_get_max_task_priority (void) __GOMP_NOTHROW;
|
|
|
|
extern void omp_fulfill_event (omp_event_handle_t) __GOMP_NOTHROW;
|
|
|
|
extern void omp_set_num_teams (int) __GOMP_NOTHROW;
|
|
extern int omp_get_max_teams (void) __GOMP_NOTHROW;
|
|
extern void omp_set_teams_thread_limit (int) __GOMP_NOTHROW;
|
|
extern int omp_get_teams_thread_limit (void) __GOMP_NOTHROW;
|
|
|
|
extern void *omp_target_alloc (__SIZE_TYPE__, int) __GOMP_NOTHROW;
|
|
extern void omp_target_free (void *, int) __GOMP_NOTHROW;
|
|
extern int omp_target_is_present (const void *, int) __GOMP_NOTHROW;
|
|
extern int omp_target_memcpy (void *, const void *, __SIZE_TYPE__,
|
|
__SIZE_TYPE__, __SIZE_TYPE__, int, int)
|
|
__GOMP_NOTHROW;
|
|
extern int omp_target_memcpy_async (void *, const void *, __SIZE_TYPE__,
|
|
__SIZE_TYPE__, __SIZE_TYPE__, int, int,
|
|
int, omp_depend_t * __GOMP_DEFAULT_NULL)
|
|
__GOMP_NOTHROW;
|
|
extern int omp_target_memcpy_rect (void *, const void *, __SIZE_TYPE__, int,
|
|
const __SIZE_TYPE__ *,
|
|
const __SIZE_TYPE__ *,
|
|
const __SIZE_TYPE__ *,
|
|
const __SIZE_TYPE__ *,
|
|
const __SIZE_TYPE__ *, int, int)
|
|
__GOMP_NOTHROW;
|
|
extern int omp_target_memcpy_rect_async (void *, const void *, __SIZE_TYPE__,
|
|
int, const __SIZE_TYPE__ *,
|
|
const __SIZE_TYPE__ *,
|
|
const __SIZE_TYPE__ *,
|
|
const __SIZE_TYPE__ *,
|
|
const __SIZE_TYPE__ *, int, int, int,
|
|
omp_depend_t * __GOMP_DEFAULT_NULL)
|
|
__GOMP_NOTHROW;
|
|
extern int omp_target_associate_ptr (const void *, const void *, __SIZE_TYPE__,
|
|
__SIZE_TYPE__, int) __GOMP_NOTHROW;
|
|
extern int omp_target_disassociate_ptr (const void *, int) __GOMP_NOTHROW;
|
|
extern void *omp_get_mapped_ptr (const void *, int) __GOMP_NOTHROW;
|
|
extern int omp_target_is_accessible (const void *, __SIZE_TYPE__, int)
|
|
__GOMP_NOTHROW;
|
|
|
|
extern void omp_set_affinity_format (const char *) __GOMP_NOTHROW;
|
|
extern __SIZE_TYPE__ omp_get_affinity_format (char *, __SIZE_TYPE__)
|
|
__GOMP_NOTHROW;
|
|
extern void omp_display_affinity (const char *) __GOMP_NOTHROW;
|
|
extern __SIZE_TYPE__ omp_capture_affinity (char *, __SIZE_TYPE__, const char *)
|
|
__GOMP_NOTHROW;
|
|
|
|
extern int omp_pause_resource (omp_pause_resource_t, int) __GOMP_NOTHROW;
|
|
extern int omp_pause_resource_all (omp_pause_resource_t) __GOMP_NOTHROW;
|
|
|
|
extern omp_allocator_handle_t omp_init_allocator (omp_memspace_handle_t,
|
|
int,
|
|
const omp_alloctrait_t [])
|
|
__GOMP_NOTHROW;
|
|
extern void omp_destroy_allocator (omp_allocator_handle_t) __GOMP_NOTHROW;
|
|
extern void omp_set_default_allocator (omp_allocator_handle_t) __GOMP_NOTHROW;
|
|
extern omp_allocator_handle_t omp_get_default_allocator (void) __GOMP_NOTHROW;
|
|
extern void omp_free (void *,
|
|
omp_allocator_handle_t __GOMP_DEFAULT_NULL_ALLOCATOR)
|
|
__GOMP_NOTHROW;
|
|
extern void *omp_alloc (__SIZE_TYPE__,
|
|
omp_allocator_handle_t __GOMP_DEFAULT_NULL_ALLOCATOR)
|
|
__GOMP_NOTHROW __attribute__((__malloc__, __malloc__ (omp_free),
|
|
__alloc_size__ (1)));
|
|
extern void *omp_aligned_alloc (__SIZE_TYPE__, __SIZE_TYPE__,
|
|
omp_allocator_handle_t
|
|
__GOMP_DEFAULT_NULL_ALLOCATOR)
|
|
__GOMP_NOTHROW __attribute__((__malloc__, __malloc__ (omp_free),
|
|
__alloc_size__ (2), __alloc_align__ (1)));
|
|
extern void *omp_calloc (__SIZE_TYPE__, __SIZE_TYPE__,
|
|
omp_allocator_handle_t __GOMP_DEFAULT_NULL_ALLOCATOR)
|
|
__GOMP_NOTHROW __attribute__((__malloc__, __malloc__ (omp_free),
|
|
__alloc_size__ (1, 2)));
|
|
extern void *omp_aligned_calloc (__SIZE_TYPE__, __SIZE_TYPE__, __SIZE_TYPE__,
|
|
omp_allocator_handle_t
|
|
__GOMP_DEFAULT_NULL_ALLOCATOR)
|
|
__GOMP_NOTHROW __attribute__((__malloc__, __malloc__ (omp_free),
|
|
__alloc_size__ (2, 3), __alloc_align__ (1)));
|
|
extern void *omp_realloc (void *, __SIZE_TYPE__,
|
|
omp_allocator_handle_t __GOMP_DEFAULT_NULL_ALLOCATOR,
|
|
omp_allocator_handle_t __GOMP_DEFAULT_NULL_ALLOCATOR)
|
|
__GOMP_NOTHROW __attribute__((__malloc__ (omp_free), __alloc_size__ (2)));
|
|
|
|
extern void omp_display_env (int) __GOMP_NOTHROW;
|
|
|
|
extern int omp_get_num_interop_properties (const omp_interop_t) __GOMP_NOTHROW;
|
|
|
|
extern omp_intptr_t omp_get_interop_int (const omp_interop_t,
|
|
omp_interop_property_t,
|
|
omp_interop_rc_t * __GOMP_DEFAULT_NULL)
|
|
__GOMP_NOTHROW;
|
|
|
|
extern void *omp_get_interop_ptr (const omp_interop_t, omp_interop_property_t,
|
|
omp_interop_rc_t * __GOMP_DEFAULT_NULL)
|
|
__GOMP_NOTHROW;
|
|
|
|
extern const char *omp_get_interop_str (const omp_interop_t,
|
|
omp_interop_property_t,
|
|
omp_interop_rc_t * __GOMP_DEFAULT_NULL)
|
|
__GOMP_NOTHROW;
|
|
|
|
extern const char *omp_get_interop_name (const omp_interop_t,
|
|
omp_interop_property_t) __GOMP_NOTHROW;
|
|
|
|
extern const char *omp_get_interop_type_desc (const omp_interop_t,
|
|
omp_interop_property_t)
|
|
__GOMP_NOTHROW;
|
|
|
|
extern const char *omp_get_interop_rc_desc (const omp_interop_t,
|
|
omp_interop_rc_t) __GOMP_NOTHROW;
|
|
|
|
extern int omp_get_device_from_uid (const char *) __GOMP_NOTHROW;
|
|
extern const char *omp_get_uid_from_device (int) __GOMP_NOTHROW;
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#if __cplusplus >= 201103L
|
|
|
|
/* std::__throw_bad_alloc and std::__throw_bad_array_new_length. */
|
|
#include <bits/functexcept.h>
|
|
|
|
namespace omp
|
|
{
|
|
namespace allocator
|
|
{
|
|
|
|
namespace __detail
|
|
{
|
|
|
|
template<typename __T, omp_allocator_handle_t __Handle>
|
|
struct __allocator_templ
|
|
{
|
|
using value_type = __T;
|
|
using pointer = __T*;
|
|
using const_pointer = const __T*;
|
|
using size_type = __SIZE_TYPE__;
|
|
using difference_type = __PTRDIFF_TYPE__;
|
|
|
|
__T*
|
|
allocate (size_type __n)
|
|
{
|
|
if (__SIZE_MAX__ / sizeof(__T) < __n)
|
|
std::__throw_bad_array_new_length ();
|
|
void *__p = omp_aligned_alloc (alignof(__T), __n * sizeof(__T), __Handle);
|
|
if (!__p)
|
|
std::__throw_bad_alloc ();
|
|
return static_cast<__T*>(__p);
|
|
}
|
|
|
|
void
|
|
deallocate (__T *__p, size_type) __GOMP_NOTHROW
|
|
{
|
|
omp_free (static_cast<void*>(__p), __Handle);
|
|
}
|
|
};
|
|
|
|
template<typename __T, typename __U, omp_allocator_handle_t __Handle>
|
|
constexpr bool
|
|
operator== (const __allocator_templ<__T, __Handle>&,
|
|
const __allocator_templ<__U, __Handle>&) __GOMP_NOTHROW
|
|
{
|
|
return true;
|
|
}
|
|
|
|
template<typename __T, omp_allocator_handle_t __Handle,
|
|
typename __U, omp_allocator_handle_t __UHandle>
|
|
constexpr bool
|
|
operator== (const __allocator_templ<__T, __Handle>&,
|
|
const __allocator_templ<__U, __UHandle>&) __GOMP_NOTHROW
|
|
{
|
|
return false;
|
|
}
|
|
|
|
template<typename __T, typename __U, omp_allocator_handle_t __Handle>
|
|
constexpr bool
|
|
operator!= (const __allocator_templ<__T, __Handle>&,
|
|
const __allocator_templ<__U, __Handle>&) __GOMP_NOTHROW
|
|
{
|
|
return false;
|
|
}
|
|
|
|
template<typename __T, omp_allocator_handle_t __Handle,
|
|
typename __U, omp_allocator_handle_t __UHandle>
|
|
constexpr bool
|
|
operator!= (const __allocator_templ<__T, __Handle>&,
|
|
const __allocator_templ<__U, __UHandle>&) __GOMP_NOTHROW
|
|
{
|
|
return true;
|
|
}
|
|
|
|
} /* namespace __detail */
|
|
|
|
template<typename __T>
|
|
struct null_allocator
|
|
: __detail::__allocator_templ<__T, omp_null_allocator> {};
|
|
|
|
template<typename __T>
|
|
struct default_mem
|
|
: __detail::__allocator_templ<__T, omp_default_mem_alloc> {};
|
|
|
|
template<typename __T>
|
|
struct large_cap_mem
|
|
: __detail::__allocator_templ<__T, omp_large_cap_mem_alloc> {};
|
|
|
|
template<typename __T>
|
|
struct const_mem
|
|
: __detail::__allocator_templ<__T, omp_const_mem_alloc> {};
|
|
|
|
template<typename __T>
|
|
struct high_bw_mem
|
|
: __detail::__allocator_templ<__T, omp_high_bw_mem_alloc> {};
|
|
|
|
template<typename __T>
|
|
struct low_lat_mem
|
|
: __detail::__allocator_templ<__T, omp_low_lat_mem_alloc> {};
|
|
|
|
template<typename __T>
|
|
struct cgroup_mem
|
|
: __detail::__allocator_templ<__T, omp_cgroup_mem_alloc> {};
|
|
|
|
template<typename __T>
|
|
struct pteam_mem
|
|
: __detail::__allocator_templ<__T, omp_pteam_mem_alloc> {};
|
|
|
|
template<typename __T>
|
|
struct thread_mem
|
|
: __detail::__allocator_templ<__T, omp_thread_mem_alloc> {};
|
|
|
|
} /* namespace allocator */
|
|
|
|
} /* namespace omp */
|
|
|
|
namespace ompx
|
|
{
|
|
|
|
namespace allocator
|
|
{
|
|
|
|
template<typename __T>
|
|
struct gnu_pinned_mem
|
|
: omp::allocator::__detail::__allocator_templ<__T, ompx_gnu_pinned_mem_alloc> {};
|
|
|
|
} /* namespace allocator */
|
|
|
|
} /* namespace ompx */
|
|
|
|
#endif /* __cplusplus */
|
|
|
|
#endif /* _OMP_H */
|