]> git.ipfire.org Git - thirdparty/gcc.git/blame - libgomp/libgomp.h
Daily bump.
[thirdparty/gcc.git] / libgomp / libgomp.h
CommitLineData
a945c346 1/* Copyright (C) 2005-2024 Free Software Foundation, Inc.
953ff289
DN
2 Contributed by Richard Henderson <rth@redhat.com>.
3
f1f3453e
TS
4 This file is part of the GNU Offloading and Multi Processing Library
5 (libgomp).
953ff289
DN
6
7 Libgomp is free software; you can redistribute it and/or modify it
748086b7
JJ
8 under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3, or (at your option)
10 any later version.
953ff289
DN
11
12 Libgomp is distributed in the hope that it will be useful, but WITHOUT ANY
13 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
748086b7 14 FOR A PARTICULAR PURPOSE. See the GNU General Public License for
953ff289
DN
15 more details.
16
748086b7
JJ
17 Under Section 7 of GPL version 3, you are granted additional
18 permissions described in the GCC Runtime Library Exception, version
19 3.1, as published by the Free Software Foundation.
953ff289 20
748086b7
JJ
21 You should have received a copy of the GNU General Public License and
22 a copy of the GCC Runtime Library Exception along with this program;
23 see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
24 <http://www.gnu.org/licenses/>. */
953ff289
DN
25
26/* This file contains data types and function declarations that are not
41dbbb37
TS
27 part of the official OpenACC or OpenMP user interfaces. There are
28 declarations in here that are part of the GNU Offloading and Multi
29 Processing ABI, in that the compiler is required to know about them
30 and use them.
953ff289
DN
31
32 The convention is that the all caps prefix "GOMP" is used group items
33 that are part of the external ABI, and the lower case prefix "gomp"
34 is used group items that are completely private to the library. */
35
36#ifndef LIBGOMP_H
37#define LIBGOMP_H 1
38
d9a6bd32
JJ
39#ifndef _LIBGOMP_CHECKING_
40/* Define to 1 to perform internal sanity checks. */
41#define _LIBGOMP_CHECKING_ 0
42#endif
43
953ff289 44#include "config.h"
810f316d 45#include <stdint.h>
41dbbb37 46#include "libgomp-plugin.h"
ec00d3fa 47#include "gomp-constants.h"
953ff289 48
6103184e 49#ifdef HAVE_PTHREAD_H
953ff289 50#include <pthread.h>
6103184e 51#endif
953ff289 52#include <stdbool.h>
acf0174b 53#include <stdlib.h>
41dbbb37 54#include <stdarg.h>
953ff289 55
e4606348
JJ
56/* Needed for memset in priority_queue.c. */
57#if _LIBGOMP_CHECKING_
58# ifdef STRING_WITH_STRINGS
59# include <string.h>
60# include <strings.h>
61# else
62# ifdef HAVE_STRING_H
63# include <string.h>
64# else
65# ifdef HAVE_STRINGS_H
66# include <strings.h>
67# endif
68# endif
69# endif
70#endif
71
953ff289
DN
72#ifdef HAVE_ATTRIBUTE_VISIBILITY
73# pragma GCC visibility push(hidden)
74#endif
75
cef86eb2
RH
76/* If we were a C++ library, we'd get this from <std/atomic>. */
77enum memmodel
78{
79 MEMMODEL_RELAXED = 0,
80 MEMMODEL_CONSUME = 1,
81 MEMMODEL_ACQUIRE = 2,
82 MEMMODEL_RELEASE = 3,
83 MEMMODEL_ACQ_REL = 4,
84 MEMMODEL_SEQ_CST = 5
85};
86
e4606348
JJ
87/* alloc.c */
88
28567c40 89#if defined(HAVE_ALIGNED_ALLOC) \
28567c40
JJ
90 || defined(HAVE_POSIX_MEMALIGN) \
91 || defined(HAVE_MEMALIGN)
92/* Defined if gomp_aligned_alloc doesn't use fallback version
93 and free can be used instead of gomp_aligned_free. */
94#define GOMP_HAVE_EFFICIENT_ALIGNED_ALLOC 1
95#endif
96
17da2c74
JJ
97#if defined(GOMP_HAVE_EFFICIENT_ALIGNED_ALLOC) && !defined(__AMDGCN__)
98#define GOMP_USE_ALIGNED_WORK_SHARES 1
99#endif
100
e4606348
JJ
101extern void *gomp_malloc (size_t) __attribute__((malloc));
102extern void *gomp_malloc_cleared (size_t) __attribute__((malloc));
103extern void *gomp_realloc (void *, size_t);
28567c40
JJ
104extern void *gomp_aligned_alloc (size_t, size_t)
105 __attribute__((malloc, alloc_size (2)));
106extern void gomp_aligned_free (void *);
e4606348
JJ
107
108/* Avoid conflicting prototypes of alloca() in system headers by using
109 GCC's builtin alloca(). */
110#define gomp_alloca(x) __builtin_alloca(x)
111
cee16451
AS
112/* Optimized allocators for team-specific data that will die with the team. */
113
114#ifdef __AMDGCN__
f6fff8a6 115#include "libgomp-gcn.h"
cee16451 116/* The arena is initialized in config/gcn/team.c. */
cee16451
AS
117
118static inline void * __attribute__((malloc))
119team_malloc (size_t size)
120{
121 /* 4-byte align the size. */
122 size = (size + 3) & ~3;
123
124 /* Allocate directly from the arena.
125 The compiler does not support DS atomics, yet. */
126 void *result;
127 asm ("ds_add_rtn_u64 %0, %1, %2\n\ts_waitcnt 0"
128 : "=v"(result) : "v"(TEAM_ARENA_FREE), "v"(size), "e"(1L) : "memory");
129
130 /* Handle OOM. */
131 if (result + size > *(void * __lds *)TEAM_ARENA_END)
132 {
133 /* While this is experimental, let's make sure we know when OOM
134 happens. */
f6fff8a6
AS
135 const char msg[] = "GCN team arena exhausted;"
136 " configure with GCN_TEAM_ARENA_SIZE=bytes\n";
cee16451
AS
137 write (2, msg, sizeof(msg)-1);
138
139 /* Fall back to using the heap (slowly). */
140 result = gomp_malloc (size);
141 }
142 return result;
143}
144
145static inline void * __attribute__((malloc))
146team_malloc_cleared (size_t size)
147{
148 char *result = team_malloc (size);
149
150 /* Clear the allocated memory. */
151 __builtin_memset (result, 0, size);
152
153 return result;
154}
155
156static inline void
157team_free (void *ptr)
158{
159 /* The whole arena is freed when the kernel exits.
160 However, if we fell back to using heap then we should free it.
161 It would be better if this function could be a no-op, but at least
162 LDS loads are cheap. */
163 if (ptr < *(void * __lds *)TEAM_ARENA_START
164 || ptr >= *(void * __lds *)TEAM_ARENA_END)
165 free (ptr);
166}
167#else
168#define team_malloc(...) gomp_malloc (__VA_ARGS__)
169#define team_malloc_cleared(...) gomp_malloc_cleared (__VA_ARGS__)
170#define team_free(...) free (__VA_ARGS__)
171#endif
172
e4606348
JJ
173/* error.c */
174
175extern void gomp_vdebug (int, const char *, va_list);
176extern void gomp_debug (int, const char *, ...)
177 __attribute__ ((format (printf, 2, 3)));
178#define gomp_vdebug(KIND, FMT, VALIST) \
179 do { \
180 if (__builtin_expect (gomp_debug_var, 0)) \
181 (gomp_vdebug) ((KIND), (FMT), (VALIST)); \
182 } while (0)
183#define gomp_debug(KIND, ...) \
184 do { \
185 if (__builtin_expect (gomp_debug_var, 0)) \
186 (gomp_debug) ((KIND), __VA_ARGS__); \
187 } while (0)
188extern void gomp_verror (const char *, va_list);
189extern void gomp_error (const char *, ...)
190 __attribute__ ((format (printf, 1, 2)));
191extern void gomp_vfatal (const char *, va_list)
192 __attribute__ ((noreturn));
193extern void gomp_fatal (const char *, ...)
194 __attribute__ ((noreturn, format (printf, 1, 2)));
195
196struct gomp_task;
197struct gomp_taskgroup;
198struct htab;
199
200#include "priority_queue.h"
953ff289
DN
201#include "sem.h"
202#include "mutex.h"
203#include "bar.h"
6103184e 204#include "simple-bar.h"
a68ab351 205#include "ptrlock.h"
953ff289
DN
206
207
208/* This structure contains the data to control one work-sharing construct,
209 either a LOOP (FOR/DO) or a SECTIONS. */
210
211enum gomp_schedule_type
212{
a68ab351 213 GFS_RUNTIME,
953ff289
DN
214 GFS_STATIC,
215 GFS_DYNAMIC,
216 GFS_GUIDED,
28567c40
JJ
217 GFS_AUTO,
218 GFS_MONOTONIC = 0x80000000U
953ff289
DN
219};
220
d9a6bd32
JJ
221struct gomp_doacross_work_share
222{
223 union {
224 /* chunk_size copy, as ws->chunk_size is multiplied by incr for
225 GFS_DYNAMIC. */
226 long chunk_size;
227 /* Likewise, but for ull implementation. */
228 unsigned long long chunk_size_ull;
229 /* For schedule(static,0) this is the number
230 of iterations assigned to the last thread, i.e. number of
231 iterations / number of threads. */
232 long q;
233 /* Likewise, but for ull implementation. */
234 unsigned long long q_ull;
235 };
236 /* Size of each array entry (padded to cache line size). */
237 unsigned long elt_sz;
238 /* Number of dimensions in sink vectors. */
239 unsigned int ncounts;
240 /* True if the iterations can be flattened. */
241 bool flattened;
242 /* Actual array (of elt_sz sized units), aligned to cache line size.
243 This is indexed by team_id for GFS_STATIC and outermost iteration
244 / chunk_size for other schedules. */
245 unsigned char *array;
246 /* These two are only used for schedule(static,0). */
247 /* This one is number of iterations % number of threads. */
248 long t;
249 union {
250 /* And this one is cached t * (q + 1). */
251 long boundary;
252 /* Likewise, but for the ull implementation. */
253 unsigned long long boundary_ull;
254 };
28567c40
JJ
255 /* Pointer to extra memory if needed for lastprivate(conditional). */
256 void *extra;
d9a6bd32
JJ
257 /* Array of shift counts for each dimension if they can be flattened. */
258 unsigned int shift_counts[];
259};
260
c7abdf46
JJ
261/* Like struct gomp_work_share, but only the 1st cacheline of it plus
262 flexible array at the end.
263 Keep in sync with struct gomp_work_share. */
264struct gomp_work_share_1st_cacheline
265{
266 enum gomp_schedule_type sched;
267 int mode;
268 union {
269 struct {
270 long chunk_size, end, incr;
271 };
272 struct {
273 unsigned long long chunk_size_ull, end_ull, incr_ull;
274 };
275 };
276 union {
277 unsigned *ordered_team_ids;
278 struct gomp_doacross_work_share *doacross;
279 };
280 unsigned ordered_num_used, ordered_owner, ordered_cur;
281 struct gomp_work_share *next_alloc;
282 char pad[];
283};
284
953ff289
DN
285struct gomp_work_share
286{
287 /* This member records the SCHEDULE clause to be used for this construct.
288 The user specification of "runtime" will already have been resolved.
289 If this is a SECTIONS construct, this value will always be DYNAMIC. */
290 enum gomp_schedule_type sched;
291
a68ab351 292 int mode;
953ff289 293
a68ab351
JJ
294 union {
295 struct {
296 /* This is the chunk_size argument to the SCHEDULE clause. */
297 long chunk_size;
298
299 /* This is the iteration end point. If this is a SECTIONS construct,
300 this is the number of contained sections. */
301 long end;
302
303 /* This is the iteration step. If this is a SECTIONS construct, this
304 is always 1. */
305 long incr;
306 };
307
308 struct {
309 /* The same as above, but for the unsigned long long loop variants. */
310 unsigned long long chunk_size_ull;
311 unsigned long long end_ull;
312 unsigned long long incr_ull;
313 };
314 };
315
d9a6bd32
JJ
316 union {
317 /* This is a circular queue that details which threads will be allowed
318 into the ordered region and in which order. When a thread allocates
319 iterations on which it is going to work, it also registers itself at
320 the end of the array. When a thread reaches the ordered region, it
321 checks to see if it is the one at the head of the queue. If not, it
322 blocks on its RELEASE semaphore. */
323 unsigned *ordered_team_ids;
324
325 /* This is a pointer to DOACROSS work share data. */
326 struct gomp_doacross_work_share *doacross;
327 };
a68ab351
JJ
328
329 /* This is the number of threads that have registered themselves in
330 the circular queue ordered_team_ids. */
331 unsigned ordered_num_used;
332
333 /* This is the team_id of the currently acknowledged owner of the ordered
334 section, or -1u if the ordered section has not been acknowledged by
335 any thread. This is distinguished from the thread that is *allowed*
336 to take the section next. */
337 unsigned ordered_owner;
338
339 /* This is the index into the circular queue ordered_team_ids of the
340 current thread that's allowed into the ordered reason. */
341 unsigned ordered_cur;
953ff289 342
a68ab351
JJ
343 /* This is a chain of allocated gomp_work_share blocks, valid only
344 in the first gomp_work_share struct in the block. */
345 struct gomp_work_share *next_alloc;
346
347 /* The above fields are written once during workshare initialization,
348 or related to ordered worksharing. Make sure the following fields
349 are in a different cache line. */
953ff289
DN
350
351 /* This lock protects the update of the following members. */
17da2c74 352#ifdef GOMP_USE_ALIGNED_WORK_SHARES
a68ab351 353 gomp_mutex_t lock __attribute__((aligned (64)));
c7abdf46
JJ
354#else
355 char pad[64 - offsetof (struct gomp_work_share_1st_cacheline, pad)];
356 gomp_mutex_t lock;
357#endif
a68ab351
JJ
358
359 /* This is the count of the number of threads that have exited the work
360 share construct. If the construct was marked nowait, they have moved on
361 to other work; otherwise they're blocked on a barrier. The last member
362 of the team to exit the work share construct must deallocate it. */
363 unsigned threads_completed;
953ff289
DN
364
365 union {
366 /* This is the next iteration value to be allocated. In the case of
367 GFS_STATIC loops, this the iteration start point and never changes. */
368 long next;
369
a68ab351
JJ
370 /* The same, but with unsigned long long type. */
371 unsigned long long next_ull;
372
953ff289
DN
373 /* This is the returned data structure for SINGLE COPYPRIVATE. */
374 void *copyprivate;
375 };
376
a68ab351
JJ
377 union {
378 /* Link to gomp_work_share struct for next work sharing construct
379 encountered after this one. */
380 gomp_ptrlock_t next_ws;
953ff289 381
a68ab351
JJ
382 /* gomp_work_share structs are chained in the free work share cache
383 through this. */
384 struct gomp_work_share *next_free;
385 };
953ff289 386
28567c40
JJ
387 /* Task reductions for this work-sharing construct. */
388 uintptr_t *task_reductions;
389
a68ab351
JJ
390 /* If only few threads are in the team, ordered_team_ids can point
391 to this array which fills the padding at the end of this struct. */
392 unsigned inline_ordered_team_ids[0];
953ff289
DN
393};
394
c7abdf46
JJ
395extern char gomp_workshare_struct_check1
396 [offsetof (struct gomp_work_share_1st_cacheline, next_alloc)
397 == offsetof (struct gomp_work_share, next_alloc) ? 1 : -1];
398extern char gomp_workshare_struct_check2
399 [offsetof (struct gomp_work_share, lock) == 64 ? 1 : -1];
400
953ff289
DN
401/* This structure contains all of the thread-local data associated with
402 a thread team. This is the data that must be saved when a thread
403 encounters a nested PARALLEL construct. */
404
405struct gomp_team_state
406{
407 /* This is the team of which the thread is currently a member. */
408 struct gomp_team *team;
409
410 /* This is the work share construct which this thread is currently
411 processing. Recall that with NOWAIT, not all threads may be
a68ab351 412 processing the same construct. */
953ff289
DN
413 struct gomp_work_share *work_share;
414
a68ab351
JJ
415 /* This is the previous work share construct or NULL if there wasn't any.
416 When all threads are done with the current work sharing construct,
417 the previous one can be freed. The current one can't, as its
418 next_ws field is used. */
419 struct gomp_work_share *last_work_share;
420
953ff289
DN
421 /* This is the ID of this thread within the team. This value is
422 guaranteed to be between 0 and N-1, where N is the number of
423 threads in the team. */
424 unsigned team_id;
425
a68ab351
JJ
426 /* Nesting level. */
427 unsigned level;
428
429 /* Active nesting level. Only active parallel regions are counted. */
430 unsigned active_level;
431
acf0174b
JJ
432 /* Place-partition-var, offset and length into gomp_places_list array. */
433 unsigned place_partition_off;
434 unsigned place_partition_len;
435
800bcc8c
JJ
436 /* Def-allocator-var ICV. */
437 uintptr_t def_allocator;
438
a68ab351
JJ
439#ifdef HAVE_SYNC_BUILTINS
440 /* Number of single stmts encountered. */
441 unsigned long single_count;
442#endif
953ff289
DN
443
444 /* For GFS_RUNTIME loops that resolved to GFS_STATIC, this is the
445 trip number through the loop. So first time a particular loop
446 is encountered this number is 0, the second time through the loop
447 is 1, etc. This is unused when the compiler knows in advance that
448 the loop is statically scheduled. */
449 unsigned long static_trip;
450};
451
acf0174b
JJ
452struct target_mem_desc;
453
9f2fca56
MV
454enum gomp_icvs
455{
456 GOMP_ICV_NTEAMS = 1,
457 GOMP_ICV_SCHEDULE = 2,
458 GOMP_ICV_SCHEDULE_CHUNK_SIZE = 3,
459 GOMP_ICV_DYNAMIC = 4,
460 GOMP_ICV_TEAMS_THREAD_LIMIT = 5,
461 GOMP_ICV_THREAD_LIMIT = 6,
462 GOMP_ICV_NTHREADS = 7,
463 GOMP_ICV_NTHREADS_LIST = 8,
464 GOMP_ICV_NTHREADS_LIST_LEN = 9,
465 GOMP_ICV_BIND = 10,
466 GOMP_ICV_BIND_LIST = 11,
467 GOMP_ICV_BIND_LIST_LEN = 12,
468 GOMP_ICV_MAX_ACTIVE_LEVELS = 13,
469 GOMP_ICV_WAIT_POLICY = 14,
470 GOMP_ICV_STACKSIZE = 15,
471 GOMP_ICV_DEFAULT_DEVICE = 16,
472 GOMP_ICV_CANCELLATION = 17,
473 GOMP_ICV_DISPLAY_AFFINITY = 18,
474 GOMP_ICV_TARGET_OFFLOAD = 19,
475 GOMP_ICV_MAX_TASK_PRIORITY = 20,
476 GOMP_ICV_ALLOCATOR = 21
477};
478
479enum gomp_device_num
480{
481 GOMP_DEVICE_NUM_FOR_DEV = -1,
482 GOMP_DEVICE_NUM_FOR_ALL = -2,
483 GOMP_DEVICE_NUM_FOR_NO_SUFFIX = -3
484};
485
acf0174b 486/* These are the OpenMP 4.0 Internal Control Variables described in
a68ab351
JJ
487 section 2.3.1. Those described as having one copy per task are
488 stored within the structure; those described as having one copy
489 for the whole program are (naturally) global variables. */
acf0174b 490
a68ab351 491struct gomp_task_icv
953ff289 492{
a68ab351
JJ
493 unsigned long nthreads_var;
494 enum gomp_schedule_type run_sched_var;
d9a6bd32 495 int run_sched_chunk_size;
acf0174b
JJ
496 int default_device_var;
497 unsigned int thread_limit_var;
a68ab351 498 bool dyn_var;
6fae7eda 499 unsigned char max_active_levels_var;
acf0174b
JJ
500 char bind_var;
501 /* Internal ICV. */
502 struct target_mem_desc *target_data;
a68ab351 503};
953ff289 504
9f2fca56
MV
505enum gomp_env_suffix
506{
507 GOMP_ENV_SUFFIX_UNKNOWN = 0,
508 GOMP_ENV_SUFFIX_NONE = 1,
509 GOMP_ENV_SUFFIX_DEV = 2,
510 GOMP_ENV_SUFFIX_ALL = 4,
511 GOMP_ENV_SUFFIX_DEV_X = 8
512};
513
514/* Struct that contains all ICVs for which we need to store initial values.
515 Keeping the initial values is needed for omp_display_env. Moreover initial
516 _DEV and _ALL variants of environment variables are also used to determine
517 actually used values for devices and for the host. */
518struct gomp_initial_icvs
519{
520 unsigned long *nthreads_var_list;
521 char *bind_var_list;
522 unsigned long nthreads_var;
523 unsigned long nthreads_var_list_len;
524 unsigned long bind_var_list_len;
525 unsigned long stacksize;
526 int run_sched_chunk_size;
527 int default_device_var;
528 int nteams_var;
529 int teams_thread_limit_var;
530 int wait_policy;
531 unsigned int thread_limit_var;
532 enum gomp_schedule_type run_sched_var;
533 bool dyn_var;
534 unsigned char max_active_levels_var;
535 char bind_var;
536};
537
538struct gomp_default_icv
539{
540 unsigned long nthreads_var;
541 enum gomp_schedule_type run_sched_var;
542 int run_sched_chunk_size;
543 int default_device_var;
544 unsigned int thread_limit_var;
545 int nteams_var;
546 int teams_thread_limit_var;
547 bool dyn_var;
548 unsigned char max_active_levels_var;
549 char bind_var;
550};
551
552/* DEVICE_NUM "-1" is reserved for "_DEV" icvs.
553 DEVICE_NUM "-2" is reserved for "_ALL" icvs.
554 DEVICE_NUM "-3" is reserved for ICVs without suffix.
555 Non-negative DEVICE_NUM is for "_DEV_X" icvs. */
556struct gomp_icv_list
557{
558 int device_num;
559 uint32_t flags;
560 struct gomp_initial_icvs icvs;
561 struct gomp_icv_list *next;
562};
563
564struct gomp_offload_icvs
565{
566 int device_num;
567 int default_device;
568 int nteams;
569 int teams_thread_limit;
570};
571
572struct gomp_offload_icv_list
573{
574 int device_num;
575 struct gomp_offload_icvs icvs;
576 struct gomp_offload_icv_list *next;
577};
578
1bfc07d1
KCY
579enum gomp_target_offload_t
580{
581 GOMP_TARGET_OFFLOAD_DEFAULT,
582 GOMP_TARGET_OFFLOAD_MANDATORY,
583 GOMP_TARGET_OFFLOAD_DISABLED
584};
585
6fae7eda 586#define gomp_supported_active_levels UCHAR_MAX
8949b985 587
a68ab351 588extern struct gomp_task_icv gomp_global_icv;
a68ab351 589#ifndef HAVE_SYNC_BUILTINS
acf0174b 590extern gomp_mutex_t gomp_managed_threads_lock;
a68ab351 591#endif
acf0174b 592extern bool gomp_cancel_var;
1bfc07d1 593extern enum gomp_target_offload_t gomp_target_offload_var;
e4606348 594extern int gomp_max_task_priority_var;
a68ab351
JJ
595extern unsigned long long gomp_spin_count_var, gomp_throttled_spin_count_var;
596extern unsigned long gomp_available_cpus, gomp_managed_threads;
20906c66 597extern unsigned long *gomp_nthreads_var_list, gomp_nthreads_var_list_len;
acf0174b
JJ
598extern char *gomp_bind_var_list;
599extern unsigned long gomp_bind_var_list_len;
600extern void **gomp_places_list;
601extern unsigned long gomp_places_list_len;
6103184e 602extern unsigned int gomp_num_teams_var;
07dd3bcd
JJ
603extern int gomp_nteams_var;
604extern int gomp_teams_thread_limit_var;
41dbbb37 605extern int gomp_debug_var;
28567c40
JJ
606extern bool gomp_display_affinity_var;
607extern char *gomp_affinity_format_var;
608extern size_t gomp_affinity_format_len;
800bcc8c 609extern uintptr_t gomp_def_allocator;
9f2fca56
MV
610extern const struct gomp_default_icv gomp_default_icv_values;
611extern struct gomp_icv_list *gomp_initial_icv_list;
612extern struct gomp_offload_icv_list *gomp_offload_icv_list;
41dbbb37
TS
613extern int goacc_device_num;
614extern char *goacc_device_type;
ec00d3fa 615extern int goacc_default_dims[GOMP_DIM_MAX];
953ff289 616
a68ab351
JJ
617enum gomp_task_kind
618{
d9a6bd32 619 /* Implicit task. */
a68ab351 620 GOMP_TASK_IMPLICIT,
d9a6bd32
JJ
621 /* Undeferred task. */
622 GOMP_TASK_UNDEFERRED,
623 /* Task created by GOMP_task and waiting to be run. */
a68ab351 624 GOMP_TASK_WAITING,
d9a6bd32 625 /* Task currently executing or scheduled and about to execute. */
e4606348
JJ
626 GOMP_TASK_TIED,
627 /* Used for target tasks that have vars mapped and async run started,
628 but not yet completed. Once that completes, they will be readded
629 into the queues as GOMP_TASK_WAITING in order to perform the var
630 unmapping. */
d656bfda
KCY
631 GOMP_TASK_ASYNC_RUNNING,
632 /* Task that has finished executing but is waiting for its
633 completion event to be fulfilled. */
634 GOMP_TASK_DETACHED
a68ab351 635};
953ff289 636
acf0174b
JJ
637struct gomp_task_depend_entry
638{
d9a6bd32 639 /* Address of dependency. */
acf0174b
JJ
640 void *addr;
641 struct gomp_task_depend_entry *next;
642 struct gomp_task_depend_entry *prev;
d9a6bd32 643 /* Task that provides the dependency in ADDR. */
acf0174b 644 struct gomp_task *task;
2c16eb31
JJ
645 /* Depend entry is of type "IN" (1) or "INOUTSET" (2). */
646 unsigned char is_in;
acf0174b 647 bool redundant;
0494285a 648 bool redundant_out;
acf0174b
JJ
649};
650
651struct gomp_dependers_vec
652{
653 size_t n_elem;
654 size_t allocated;
655 struct gomp_task *elem[];
656};
657
0494285a
JJ
658/* Used when in GOMP_taskwait or in gomp_task_maybe_wait_for_dependencies. */
659
660struct gomp_taskwait
661{
662 bool in_taskwait;
663 bool in_depend_wait;
e4606348 664 /* Number of tasks we are waiting for. */
0494285a 665 size_t n_depend;
0494285a
JJ
666 gomp_sem_t taskwait_sem;
667};
668
a68ab351 669/* This structure describes a "task" to be run by a thread. */
953ff289 670
a68ab351
JJ
671struct gomp_task
672{
e4606348 673 /* Parent of this task. */
a68ab351 674 struct gomp_task *parent;
e4606348
JJ
675 /* Children of this task. */
676 struct priority_queue children_queue;
d9a6bd32 677 /* Taskgroup this task belongs in. */
acf0174b 678 struct gomp_taskgroup *taskgroup;
d9a6bd32 679 /* Tasks that depend on this task. */
acf0174b
JJ
680 struct gomp_dependers_vec *dependers;
681 struct htab *depend_hash;
0494285a 682 struct gomp_taskwait *taskwait;
7f78783d
JJ
683 /* Last depend({,in}out:omp_all_memory) child if any. */
684 struct gomp_task *depend_all_memory;
d9a6bd32 685 /* Number of items in DEPEND. */
acf0174b 686 size_t depend_count;
e4606348
JJ
687 /* Number of tasks this task depends on. Once this counter reaches
688 0, we have no unsatisfied dependencies, and this task can be put
689 into the various queues to be scheduled. */
acf0174b 690 size_t num_dependees;
e4606348 691
d656bfda
KCY
692 union {
693 /* Valid only if deferred_p is false. */
694 gomp_sem_t *completion_sem;
695 /* Valid only if deferred_p is true. Set to the team that executes the
696 task if the task is detached and the completion event has yet to be
697 fulfilled. */
698 struct gomp_team *detach_team;
699 };
700 bool deferred_p;
701
e4606348
JJ
702 /* Priority of this task. */
703 int priority;
704 /* The priority node for this task in each of the different queues.
705 We put this here to avoid allocating space for each priority
706 node. Then we play offsetof() games to convert between pnode[]
707 entries and the gomp_task in which they reside. */
708 struct priority_node pnode[3];
709
a68ab351
JJ
710 struct gomp_task_icv icv;
711 void (*fn) (void *);
712 void *fn_data;
713 enum gomp_task_kind kind;
5f836cbb 714 bool in_tied_task;
20906c66 715 bool final_task;
acf0174b 716 bool copy_ctors_done;
d9a6bd32
JJ
717 /* Set for undeferred tasks with unsatisfied dependencies which
718 block further execution of their parent until the dependencies
719 are satisfied. */
0494285a 720 bool parent_depends_on;
d9a6bd32
JJ
721 /* Dependencies provided and/or needed for this task. DEPEND_COUNT
722 is the number of items available. */
acf0174b
JJ
723 struct gomp_task_depend_entry depend[];
724};
725
e4606348
JJ
726/* This structure describes a single #pragma omp taskgroup. */
727
acf0174b
JJ
728struct gomp_taskgroup
729{
730 struct gomp_taskgroup *prev;
e4606348
JJ
731 /* Queue of tasks that belong in this taskgroup. */
732 struct priority_queue taskgroup_queue;
28567c40 733 uintptr_t *reductions;
acf0174b
JJ
734 bool in_taskgroup_wait;
735 bool cancelled;
28567c40 736 bool workshare;
acf0174b
JJ
737 gomp_sem_t taskgroup_sem;
738 size_t num_children;
a68ab351
JJ
739};
740
e4606348
JJ
741/* Various state of OpenMP async offloading tasks. */
742enum gomp_target_task_state
743{
744 GOMP_TARGET_TASK_DATA,
745 GOMP_TARGET_TASK_BEFORE_MAP,
746 GOMP_TARGET_TASK_FALLBACK,
747 GOMP_TARGET_TASK_READY_TO_RUN,
748 GOMP_TARGET_TASK_RUNNING,
749 GOMP_TARGET_TASK_FINISHED
750};
751
752/* This structure describes a target task. */
753
d9a6bd32
JJ
754struct gomp_target_task
755{
756 struct gomp_device_descr *devicep;
757 void (*fn) (void *);
758 size_t mapnum;
759 size_t *sizes;
760 unsigned short *kinds;
761 unsigned int flags;
e4606348
JJ
762 enum gomp_target_task_state state;
763 struct target_mem_desc *tgt;
764 struct gomp_task *task;
765 struct gomp_team *team;
b2b40051
MJ
766 /* Device-specific target arguments. */
767 void **args;
d9a6bd32
JJ
768 void *hostaddrs[];
769};
770
a68ab351
JJ
771/* This structure describes a "team" of threads. These are the threads
772 that are spawned by a PARALLEL constructs, as well as the work sharing
773 constructs that the team encounters. */
774
775struct gomp_team
776{
953ff289
DN
777 /* This is the number of threads in the current team. */
778 unsigned nthreads;
779
a68ab351
JJ
780 /* This is number of gomp_work_share structs that have been allocated
781 as a block last time. */
782 unsigned work_share_chunk;
783
953ff289
DN
784 /* This is the saved team state that applied to a master thread before
785 the current thread was created. */
786 struct gomp_team_state prev_ts;
787
953ff289
DN
788 /* This semaphore should be used by the master thread instead of its
789 "native" semaphore in the thread structure. Required for nested
790 parallels, as the master is a member of two teams. */
791 gomp_sem_t master_release;
792
a68ab351
JJ
793 /* This points to an array with pointers to the release semaphore
794 of the threads in the team. */
795 gomp_sem_t **ordered_release;
796
acf0174b
JJ
797 /* List of work shares on which gomp_fini_work_share hasn't been
798 called yet. If the team hasn't been cancelled, this should be
799 equal to each thr->ts.work_share, but otherwise it can be a possibly
800 long list of workshares. */
801 struct gomp_work_share *work_shares_to_free;
802
a68ab351
JJ
803 /* List of gomp_work_share structs chained through next_free fields.
804 This is populated and taken off only by the first thread in the
805 team encountering a new work sharing construct, in a critical
806 section. */
807 struct gomp_work_share *work_share_list_alloc;
808
809 /* List of gomp_work_share structs freed by free_work_share. New
810 entries are atomically added to the start of the list, and
811 alloc_work_share can safely only move all but the first entry
812 to work_share_list alloc, as free_work_share can happen concurrently
813 with alloc_work_share. */
814 struct gomp_work_share *work_share_list_free;
815
816#ifdef HAVE_SYNC_BUILTINS
817 /* Number of simple single regions encountered by threads in this
818 team. */
819 unsigned long single_count;
820#else
821 /* Mutex protecting addition of workshares to work_share_list_free. */
822 gomp_mutex_t work_share_list_free_lock;
823#endif
824
825 /* This barrier is used for most synchronization of the team. */
826 gomp_barrier_t barrier;
827
828 /* Initial work shares, to avoid allocating any gomp_work_share
829 structs in the common case. */
830 struct gomp_work_share work_shares[8];
831
832 gomp_mutex_t task_lock;
e4606348
JJ
833 /* Scheduled tasks. */
834 struct priority_queue task_queue;
acf0174b
JJ
835 /* Number of all GOMP_TASK_{WAITING,TIED} tasks in the team. */
836 unsigned int task_count;
837 /* Number of GOMP_TASK_WAITING tasks currently waiting to be scheduled. */
838 unsigned int task_queued_count;
839 /* Number of GOMP_TASK_{WAITING,TIED} tasks currently running
840 directly in gomp_barrier_handle_tasks; tasks spawned
841 from e.g. GOMP_taskwait or GOMP_taskgroup_end don't count, even when
842 that is called from a task run from gomp_barrier_handle_tasks.
843 task_running_count should be always <= team->nthreads,
844 and if current task isn't in_tied_task, then it will be
845 even < team->nthreads. */
846 unsigned int task_running_count;
847 int work_share_cancelled;
848 int team_cancelled;
a68ab351 849
d656bfda 850 /* Number of tasks waiting for their completion event to be fulfilled. */
a6d22fb2
KCY
851 unsigned int task_detach_count;
852
a68ab351
JJ
853 /* This array contains structures for implicit tasks. */
854 struct gomp_task implicit_task[];
953ff289
DN
855};
856
857/* This structure contains all data that is private to libgomp and is
858 allocated per thread. */
859
860struct gomp_thread
861{
862 /* This is the function that the thread should run upon launch. */
863 void (*fn) (void *data);
864 void *data;
865
866 /* This is the current team state for this thread. The ts.team member
867 is NULL only if the thread is idle. */
868 struct gomp_team_state ts;
869
a68ab351
JJ
870 /* This is the task that the thread is currently executing. */
871 struct gomp_task *task;
872
953ff289
DN
873 /* This semaphore is used for ordered loops. */
874 gomp_sem_t release;
a68ab351 875
acf0174b
JJ
876 /* Place this thread is bound to plus one, or zero if not bound
877 to any place. */
878 unsigned int place;
879
880 /* User pthread thread pool */
a68ab351 881 struct gomp_thread_pool *thread_pool;
28567c40 882
fa4fcb11
JJ
883#ifdef LIBGOMP_USE_PTHREADS
884 /* omp_get_num_teams () - 1. */
885 unsigned int num_teams;
886
887 /* omp_get_team_num (). */
888 unsigned int team_num;
889#endif
890
28567c40
JJ
891#if defined(LIBGOMP_USE_PTHREADS) \
892 && (!defined(HAVE_TLS) \
893 || !defined(__GLIBC__) \
894 || !defined(USING_INITIAL_EXEC_TLS))
895 /* pthread_t of the thread containing this gomp_thread.
896 On Linux when using initial-exec TLS,
897 (typeof (pthread_t)) gomp_thread () - pthread_self ()
898 is constant in all threads, so we can optimize and not
899 store it. */
900#define GOMP_NEEDS_THREAD_HANDLE 1
901 pthread_t handle;
902#endif
a68ab351
JJ
903};
904
905
906struct gomp_thread_pool
907{
908 /* This array manages threads spawned from the top level, which will
909 return to the idle loop once the current PARALLEL construct ends. */
910 struct gomp_thread **threads;
911 unsigned threads_size;
912 unsigned threads_used;
e5210c77
SH
913 /* The last team is used for non-nested teams to delay their destruction to
914 make sure all the threads in the team move on to the pool's barrier before
915 the team's barrier is destroyed. */
a68ab351 916 struct gomp_team *last_team;
acf0174b
JJ
917 /* Number of threads running in this contention group. */
918 unsigned long threads_busy;
a68ab351 919
6103184e
AM
920 /* This barrier holds and releases threads waiting in thread pools. */
921 gomp_simple_barrier_t threads_dock;
953ff289
DN
922};
923
acf0174b
JJ
924enum gomp_cancel_kind
925{
926 GOMP_CANCEL_PARALLEL = 1,
927 GOMP_CANCEL_LOOP = 2,
928 GOMP_CANCEL_FOR = GOMP_CANCEL_LOOP,
929 GOMP_CANCEL_DO = GOMP_CANCEL_LOOP,
930 GOMP_CANCEL_SECTIONS = 4,
931 GOMP_CANCEL_TASKGROUP = 8
932};
933
953ff289
DN
934/* ... and here is that TLS data. */
935
6103184e
AM
936#if defined __nvptx__
937extern struct gomp_thread *nvptx_thrs __attribute__((shared));
938static inline struct gomp_thread *gomp_thread (void)
939{
940 int tid;
941 asm ("mov.u32 %0, %%tid.y;" : "=r" (tid));
942 return nvptx_thrs + tid;
943}
fa499995
AS
944#elif defined __AMDGCN__
945static inline struct gomp_thread *gcn_thrs (void)
946{
947 /* The value is at the bottom of LDS. */
948 struct gomp_thread * __lds *thrs = (struct gomp_thread * __lds *)4;
949 return *thrs;
950}
951static inline void set_gcn_thrs (struct gomp_thread *val)
952{
953 /* The value is at the bottom of LDS. */
954 struct gomp_thread * __lds *thrs = (struct gomp_thread * __lds *)4;
955 *thrs = val;
956}
957static inline struct gomp_thread *gomp_thread (void)
958{
959 int tid = __builtin_gcn_dim_pos(1);
960 return gcn_thrs () + tid;
961}
6103184e 962#elif defined HAVE_TLS || defined USE_EMUTLS
953ff289
DN
963extern __thread struct gomp_thread gomp_tls_data;
964static inline struct gomp_thread *gomp_thread (void)
965{
966 return &gomp_tls_data;
967}
968#else
969extern pthread_key_t gomp_tls_key;
970static inline struct gomp_thread *gomp_thread (void)
971{
972 return pthread_getspecific (gomp_tls_key);
973}
974#endif
975
a68ab351
JJ
976extern struct gomp_task_icv *gomp_new_icv (void);
977
978/* Here's how to access the current copy of the ICVs. */
953ff289 979
a68ab351
JJ
980static inline struct gomp_task_icv *gomp_icv (bool write)
981{
982 struct gomp_task *task = gomp_thread ()->task;
983 if (task)
984 return &task->icv;
985 else if (write)
986 return gomp_new_icv ();
987 else
988 return &gomp_global_icv;
989}
d0d1b24d 990
6103184e 991#ifdef LIBGOMP_USE_PTHREADS
d0d1b24d
RH
992/* The attributes to be used during thread creation. */
993extern pthread_attr_t gomp_thread_attr;
953ff289 994
66c59f92 995extern pthread_key_t gomp_thread_destructor;
6103184e 996#endif
66c59f92 997
953ff289
DN
998/* Function prototypes. */
999
a0884cf0
JJ
1000/* affinity.c */
1001
1002extern void gomp_init_affinity (void);
6103184e 1003#ifdef LIBGOMP_USE_PTHREADS
acf0174b 1004extern void gomp_init_thread_affinity (pthread_attr_t *, unsigned int);
6103184e 1005#endif
acf0174b
JJ
1006extern void **gomp_affinity_alloc (unsigned long, bool);
1007extern void gomp_affinity_init_place (void *);
1008extern bool gomp_affinity_add_cpus (void *, unsigned long, unsigned long,
1009 long, bool);
1010extern bool gomp_affinity_remove_cpu (void *, unsigned long);
1011extern bool gomp_affinity_copy_place (void *, void *, long);
1012extern bool gomp_affinity_same_place (void *, void *);
1013extern bool gomp_affinity_finalize_place_list (bool);
1014extern bool gomp_affinity_init_level (int, unsigned long, bool);
1015extern void gomp_affinity_print_place (void *);
d9a6bd32 1016extern void gomp_get_place_proc_ids_8 (int, int64_t *);
28567c40
JJ
1017extern void gomp_display_affinity_place (char *, size_t, size_t *, int);
1018
1019/* affinity-fmt.c */
1020
91df4397 1021extern bool gomp_print_string (const char *str, size_t len);
28567c40
JJ
1022extern void gomp_set_affinity_format (const char *, size_t);
1023extern void gomp_display_string (char *, size_t, size_t *, const char *,
1024 size_t);
1025#ifdef LIBGOMP_USE_PTHREADS
1026typedef pthread_t gomp_thread_handle;
1027#else
1028typedef struct {} gomp_thread_handle;
1029#endif
1030extern size_t gomp_display_affinity (char *, size_t, const char *,
1031 gomp_thread_handle,
1032 struct gomp_team_state *, unsigned int);
1033extern void gomp_display_affinity_thread (gomp_thread_handle,
1034 struct gomp_team_state *,
1035 unsigned int) __attribute__((cold));
a0884cf0 1036
9f2fca56
MV
1037/* env.c */
1038
1039extern struct gomp_icv_list *gomp_get_initial_icv_item (int dev_num);
1040extern bool gomp_get_icv_flag (uint32_t value, enum gomp_icvs icv);
1041
953ff289
DN
1042/* iter.c */
1043
1044extern int gomp_iter_static_next (long *, long *);
1045extern bool gomp_iter_dynamic_next_locked (long *, long *);
1046extern bool gomp_iter_guided_next_locked (long *, long *);
1047
1048#ifdef HAVE_SYNC_BUILTINS
1049extern bool gomp_iter_dynamic_next (long *, long *);
1050extern bool gomp_iter_guided_next (long *, long *);
1051#endif
1052
a68ab351
JJ
1053/* iter_ull.c */
1054
1055extern int gomp_iter_ull_static_next (unsigned long long *,
1056 unsigned long long *);
1057extern bool gomp_iter_ull_dynamic_next_locked (unsigned long long *,
1058 unsigned long long *);
1059extern bool gomp_iter_ull_guided_next_locked (unsigned long long *,
1060 unsigned long long *);
1061
1062#if defined HAVE_SYNC_BUILTINS && defined __LP64__
1063extern bool gomp_iter_ull_dynamic_next (unsigned long long *,
1064 unsigned long long *);
1065extern bool gomp_iter_ull_guided_next (unsigned long long *,
1066 unsigned long long *);
1067#endif
1068
953ff289
DN
1069/* ordered.c */
1070
1071extern void gomp_ordered_first (void);
1072extern void gomp_ordered_last (void);
1073extern void gomp_ordered_next (void);
1074extern void gomp_ordered_static_init (void);
1075extern void gomp_ordered_static_next (void);
1076extern void gomp_ordered_sync (void);
28567c40 1077extern void gomp_doacross_init (unsigned, long *, long, size_t);
d9a6bd32 1078extern void gomp_doacross_ull_init (unsigned, unsigned long long *,
28567c40 1079 unsigned long long, size_t);
953ff289
DN
1080
1081/* parallel.c */
1082
a68ab351 1083extern unsigned gomp_resolve_num_threads (unsigned, unsigned);
953ff289
DN
1084
1085/* proc.c (in config/) */
1086
1087extern void gomp_init_num_threads (void);
1088extern unsigned gomp_dynamic_max_threads (void);
1089
a68ab351
JJ
1090/* task.c */
1091
1092extern void gomp_init_task (struct gomp_task *, struct gomp_task *,
1093 struct gomp_task_icv *);
1094extern void gomp_end_task (void);
1095extern void gomp_barrier_handle_tasks (gomp_barrier_state_t);
d9a6bd32 1096extern void gomp_task_maybe_wait_for_dependencies (void **);
e4606348 1097extern bool gomp_create_target_task (struct gomp_device_descr *,
d9a6bd32
JJ
1098 void (*) (void *), size_t, void **,
1099 size_t *, unsigned short *, unsigned int,
b2b40051
MJ
1100 void **, void **,
1101 enum gomp_target_task_state);
28567c40
JJ
1102extern struct gomp_taskgroup *gomp_parallel_reduction_register (uintptr_t *,
1103 unsigned);
1104extern void gomp_workshare_taskgroup_start (void);
1105extern void gomp_workshare_task_reduction_register (uintptr_t *, uintptr_t *);
a68ab351
JJ
1106
1107static void inline
1108gomp_finish_task (struct gomp_task *task)
1109{
acf0174b
JJ
1110 if (__builtin_expect (task->depend_hash != NULL, 0))
1111 free (task->depend_hash);
a68ab351
JJ
1112}
1113
953ff289
DN
1114/* team.c */
1115
a68ab351 1116extern struct gomp_team *gomp_new_team (unsigned);
953ff289 1117extern void gomp_team_start (void (*) (void *), void *, unsigned,
28567c40
JJ
1118 unsigned, struct gomp_team *,
1119 struct gomp_taskgroup *);
953ff289 1120extern void gomp_team_end (void);
acf0174b 1121extern void gomp_free_thread (void *);
28567c40 1122extern int gomp_pause_host (void);
acf0174b
JJ
1123
1124/* target.c */
1125
41dbbb37 1126extern void gomp_init_targets_once (void);
acf0174b 1127extern int gomp_get_num_devices (void);
e4606348 1128extern bool gomp_target_task_fn (void *);
131d18e9 1129extern void gomp_target_rev (uint64_t, uint64_t, uint64_t, uint64_t, uint64_t,
130c2f3c 1130 int, struct goacc_asyncqueue *);
953ff289 1131
e4606348 1132/* Splay tree definitions. */
41dbbb37
TS
1133typedef struct splay_tree_node_s *splay_tree_node;
1134typedef struct splay_tree_s *splay_tree;
1135typedef struct splay_tree_key_s *splay_tree_key;
1136
d9a6bd32
JJ
1137struct target_var_desc {
1138 /* Splay key. */
1139 splay_tree_key key;
1140 /* True if data should be copied from device to host at the end. */
1141 bool copy_from;
1142 /* True if data always should be copied from device to host at the end. */
1143 bool always_copy_from;
bc4ed079
JB
1144 /* True if this is for OpenACC 'attach'. */
1145 bool is_attach;
972da557
TB
1146 /* If GOMP_MAP_TO_PSET had a NULL pointer; used for Fortran descriptors,
1147 which were initially unallocated. */
1148 bool has_null_ptr_assoc;
d9a6bd32
JJ
1149 /* Relative offset against key host_start. */
1150 uintptr_t offset;
1151 /* Actual length. */
1152 uintptr_t length;
1153};
1154
ea4b23d9 1155struct target_mem_desc;
41dbbb37 1156
275c736e
CLT
1157/* Special value for refcount - mask to indicate existence of special
1158 values. Right now we allocate 3 bits. */
1159#define REFCOUNT_SPECIAL (~(uintptr_t) 0x7)
1160
d9a6bd32 1161/* Special value for refcount - infinity. */
275c736e 1162#define REFCOUNT_INFINITY (REFCOUNT_SPECIAL | 0)
4a38b02b
IV
1163/* Special value for refcount - tgt_offset contains target address of the
1164 artificial pointer to "omp declare target link" object. */
275c736e
CLT
1165#define REFCOUNT_LINK (REFCOUNT_SPECIAL | 1)
1166
1167/* Special value for refcount - structure element sibling list items.
1168 All such key refounts have REFCOUNT_STRUCTELEM bits set, with _FLAG_FIRST
1169 and _FLAG_LAST indicating first and last in the created sibling sequence. */
1170#define REFCOUNT_STRUCTELEM (REFCOUNT_SPECIAL | 4)
1171#define REFCOUNT_STRUCTELEM_P(V) \
1172 (((V) & REFCOUNT_STRUCTELEM) == REFCOUNT_STRUCTELEM)
1173/* The first leading key with _FLAG_FIRST set houses the actual reference count
1174 in the structelem_refcount field. Other siblings point to this counter value
1175 through its structelem_refcount_ptr field. */
1176#define REFCOUNT_STRUCTELEM_FLAG_FIRST (1)
1177/* The last key in the sibling sequence has this set. This is required to
1178 indicate the sequence boundary, when we remove the structure sibling list
1179 from the map. */
1180#define REFCOUNT_STRUCTELEM_FLAG_LAST (2)
1181
1182#define REFCOUNT_STRUCTELEM_FIRST_P(V) \
1183 (REFCOUNT_STRUCTELEM_P (V) && ((V) & REFCOUNT_STRUCTELEM_FLAG_FIRST))
1184#define REFCOUNT_STRUCTELEM_LAST_P(V) \
1185 (REFCOUNT_STRUCTELEM_P (V) && ((V) & REFCOUNT_STRUCTELEM_FLAG_LAST))
d9a6bd32 1186
6c7e076b
JB
1187/* Special offset values. */
1188#define OFFSET_INLINED (~(uintptr_t) 0)
1189#define OFFSET_POINTER (~(uintptr_t) 1)
1190#define OFFSET_STRUCT (~(uintptr_t) 2)
1191
2a656a93
JB
1192/* Auxiliary structure for infrequently-used or API-specific data. */
1193
1194struct splay_tree_aux {
1195 /* Pointer to the original mapping of "omp declare target link" object. */
1196 splay_tree_key link_key;
5d5be7bf
JB
1197 /* For a block with attached pointers, the attachment counters for each.
1198 Only used for OpenACC. */
1199 uintptr_t *attach_count;
2a656a93
JB
1200};
1201
41dbbb37
TS
1202struct splay_tree_key_s {
1203 /* Address of the host object. */
1204 uintptr_t host_start;
1205 /* Address immediately after the host object. */
1206 uintptr_t host_end;
1207 /* Descriptor of the target memory. */
1208 struct target_mem_desc *tgt;
1209 /* Offset from tgt->tgt_start to the start of the target object. */
1210 uintptr_t tgt_offset;
1211 /* Reference count. */
1212 uintptr_t refcount;
275c736e
CLT
1213 union {
1214 /* Dynamic reference count. */
1215 uintptr_t dynamic_refcount;
1216
1217 /* Unified reference count for structure element siblings, this is used
1218 when REFCOUNT_STRUCTELEM_FIRST_P(k->refcount) == true, the first sibling
1219 in a structure element sibling list item sequence. */
1220 uintptr_t structelem_refcount;
1221
1222 /* When REFCOUNT_STRUCTELEM_P (k->refcount) == true, this field points
1223 into the (above) structelem_refcount field of the _FIRST splay_tree_key,
1224 the first key in the created sequence. All structure element siblings
1225 share a single refcount in this manner. Since these two fields won't be
1226 used at the same time, they are stashed in a union. */
1227 uintptr_t *structelem_refcount_ptr;
1228 };
2a656a93 1229 struct splay_tree_aux *aux;
41dbbb37
TS
1230};
1231
e4606348
JJ
1232/* The comparison function. */
1233
1234static inline int
1235splay_compare (splay_tree_key x, splay_tree_key y)
1236{
1237 if (x->host_start == x->host_end
1238 && y->host_start == y->host_end)
1239 return 0;
1240 if (x->host_end <= y->host_start)
1241 return -1;
1242 if (x->host_start >= y->host_end)
1243 return 1;
1244 return 0;
1245}
1246
41dbbb37
TS
1247#include "splay-tree.h"
1248
ea4b23d9
TB
1249/* Reverse offload splay-tree handling (functions only). */
1250
1251struct reverse_splay_tree_key_s {
1252 /* Address of the device object. */
1253 uint64_t dev;
1254 splay_tree_key k;
1255};
1256
1257typedef struct reverse_splay_tree_node_s *reverse_splay_tree_node;
1258typedef struct reverse_splay_tree_s *reverse_splay_tree;
1259typedef struct reverse_splay_tree_key_s *reverse_splay_tree_key;
1260
1261static inline int
1262reverse_splay_compare (reverse_splay_tree_key x, reverse_splay_tree_key y)
1263{
1264 if (x->dev < y->dev)
1265 return -1;
1266 if (x->dev > y->dev)
1267 return 1;
1268 return 0;
1269}
1270
1271#define splay_tree_prefix reverse
d4b6d147 1272#define splay_tree_static
ea4b23d9
TB
1273#include "splay-tree.h"
1274
a49c7d31
KCY
1275/* Indirect target function splay-tree handling. */
1276
1277struct indirect_splay_tree_key_s {
1278 uint64_t host_addr, target_addr;
1279};
1280
1281typedef struct indirect_splay_tree_node_s *indirect_splay_tree_node;
1282typedef struct indirect_splay_tree_s *indirect_splay_tree;
1283typedef struct indirect_splay_tree_key_s *indirect_splay_tree_key;
1284
1285static inline int
1286indirect_splay_compare (indirect_splay_tree_key x, indirect_splay_tree_key y)
1287{
1288 if (x->host_addr < y->host_addr)
1289 return -1;
1290 if (x->host_addr > y->host_addr)
1291 return 1;
1292 return 0;
1293}
1294
1295#define splay_tree_prefix indirect
1296#include "splay-tree.h"
1297
ea4b23d9
TB
1298struct target_mem_desc {
1299 /* Reference count. */
1300 uintptr_t refcount;
1301 /* All the splay nodes allocated together. */
1302 splay_tree_node array;
1303 /* Likewise for the reverse lookup device->host for reverse offload. */
1304 reverse_splay_tree_node rev_array;
1305 /* Start of the target region. */
1306 uintptr_t tgt_start;
1307 /* End of the targer region. */
1308 uintptr_t tgt_end;
1309 /* Handle to free. */
1310 void *to_free;
1311 /* Previous target_mem_desc. */
1312 struct target_mem_desc *prev;
1313 /* Number of items in following list. */
1314 size_t list_count;
1315
1316 /* Corresponding target device descriptor. */
1317 struct gomp_device_descr *device_descr;
1318
1319 /* List of target items to remove (or decrease refcount)
1320 at the end of region. */
1321 struct target_var_desc list[];
1322};
1323
1324
41dbbb37
TS
1325typedef struct acc_dispatch_t
1326{
41dbbb37 1327 /* Execute. */
345a8c17 1328 __typeof (GOMP_OFFLOAD_openacc_exec) *exec_func;
41dbbb37 1329
41dbbb37 1330 /* Create/destroy TLS data. */
dced339c
TS
1331 __typeof (GOMP_OFFLOAD_openacc_create_thread_data) *create_thread_data_func;
1332 __typeof (GOMP_OFFLOAD_openacc_destroy_thread_data)
1333 *destroy_thread_data_func;
1f4c5b9b
CLT
1334
1335 struct {
1336 /* Once created and put into the "active" list, asyncqueues are then never
1337 destructed and removed from the "active" list, other than if the TODO
1338 device is shut down. */
1339 gomp_mutex_t lock;
1340 int nasyncqueue;
1341 struct goacc_asyncqueue **asyncqueue;
1342 struct goacc_asyncqueue_list *active;
1343
1344 __typeof (GOMP_OFFLOAD_openacc_async_construct) *construct_func;
1345 __typeof (GOMP_OFFLOAD_openacc_async_destruct) *destruct_func;
1346 __typeof (GOMP_OFFLOAD_openacc_async_test) *test_func;
1347 __typeof (GOMP_OFFLOAD_openacc_async_synchronize) *synchronize_func;
1348 __typeof (GOMP_OFFLOAD_openacc_async_serialize) *serialize_func;
1349 __typeof (GOMP_OFFLOAD_openacc_async_queue_callback) *queue_callback_func;
1350
1351 __typeof (GOMP_OFFLOAD_openacc_async_exec) *exec_func;
1352 __typeof (GOMP_OFFLOAD_openacc_async_dev2host) *dev2host_func;
1353 __typeof (GOMP_OFFLOAD_openacc_async_host2dev) *host2dev_func;
1354 } async;
41dbbb37 1355
6fc0385c
TS
1356 __typeof (GOMP_OFFLOAD_openacc_get_property) *get_property_func;
1357
41dbbb37
TS
1358 /* NVIDIA target specific routines. */
1359 struct {
345a8c17 1360 __typeof (GOMP_OFFLOAD_openacc_cuda_get_current_device)
dced339c 1361 *get_current_device_func;
345a8c17 1362 __typeof (GOMP_OFFLOAD_openacc_cuda_get_current_context)
dced339c 1363 *get_current_context_func;
345a8c17
TS
1364 __typeof (GOMP_OFFLOAD_openacc_cuda_get_stream) *get_stream_func;
1365 __typeof (GOMP_OFFLOAD_openacc_cuda_set_stream) *set_stream_func;
41dbbb37
TS
1366 } cuda;
1367} acc_dispatch_t;
1368
d84ffc0a
IV
1369/* Various state of the accelerator device. */
1370enum gomp_device_state
1371{
1372 GOMP_DEVICE_UNINITIALIZED,
1373 GOMP_DEVICE_INITIALIZED,
1374 GOMP_DEVICE_FINALIZED
1375};
1376
41dbbb37
TS
1377/* This structure describes accelerator device.
1378 It contains name of the corresponding libgomp plugin, function handlers for
1379 interaction with the device, ID-number of the device, and information about
1380 mapped memory. */
1381struct gomp_device_descr
1382{
1383 /* Immutable data, which is only set during initialization, and which is not
1384 guarded by the lock. */
1385
1386 /* The name of the device. */
1387 const char *name;
1388
1389 /* Capabilities of device (supports OpenACC, OpenMP). */
1390 unsigned int capabilities;
1391
1392 /* This is the ID number of device among devices of the same type. */
1393 int target_id;
1394
1395 /* This is the TYPE of device. */
1396 enum offload_target_type type;
1397
1398 /* Function handlers. */
dced339c
TS
1399 __typeof (GOMP_OFFLOAD_get_name) *get_name_func;
1400 __typeof (GOMP_OFFLOAD_get_caps) *get_caps_func;
1401 __typeof (GOMP_OFFLOAD_get_type) *get_type_func;
1402 __typeof (GOMP_OFFLOAD_get_num_devices) *get_num_devices_func;
1403 __typeof (GOMP_OFFLOAD_init_device) *init_device_func;
1404 __typeof (GOMP_OFFLOAD_fini_device) *fini_device_func;
1405 __typeof (GOMP_OFFLOAD_version) *version_func;
1406 __typeof (GOMP_OFFLOAD_load_image) *load_image_func;
1407 __typeof (GOMP_OFFLOAD_unload_image) *unload_image_func;
1408 __typeof (GOMP_OFFLOAD_alloc) *alloc_func;
1409 __typeof (GOMP_OFFLOAD_free) *free_func;
1410 __typeof (GOMP_OFFLOAD_dev2host) *dev2host_func;
1411 __typeof (GOMP_OFFLOAD_host2dev) *host2dev_func;
25072a47
TB
1412 __typeof (GOMP_OFFLOAD_memcpy2d) *memcpy2d_func;
1413 __typeof (GOMP_OFFLOAD_memcpy3d) *memcpy3d_func;
dced339c
TS
1414 __typeof (GOMP_OFFLOAD_dev2dev) *dev2dev_func;
1415 __typeof (GOMP_OFFLOAD_can_run) *can_run_func;
1416 __typeof (GOMP_OFFLOAD_run) *run_func;
1417 __typeof (GOMP_OFFLOAD_async_run) *async_run_func;
41dbbb37 1418
a51df54e
IV
1419 /* Splay tree containing information about mapped memory regions. */
1420 struct splay_tree_s mem_map;
ea4b23d9 1421 struct reverse_splay_tree_s mem_map_rev;
41dbbb37
TS
1422
1423 /* Mutex for the mutable data. */
1424 gomp_mutex_t lock;
1425
d84ffc0a
IV
1426 /* Current state of the device. OpenACC allows to move from INITIALIZED state
1427 back to UNINITIALIZED state. OpenMP allows only to move from INITIALIZED
1428 to FINALIZED state (at program shutdown). */
1429 enum gomp_device_state state;
41dbbb37 1430
41dbbb37 1431 /* OpenACC-specific data and functions. */
47afc7b4 1432 /* This is mutable because of its mutable target_data member. */
41dbbb37
TS
1433 acc_dispatch_t openacc;
1434};
1435
d9a6bd32
JJ
1436/* Kind of the pragma, for which gomp_map_vars () is called. */
1437enum gomp_map_vars_kind
1438{
9e628024
CLT
1439 GOMP_MAP_VARS_OPENACC = 1,
1440 GOMP_MAP_VARS_TARGET = 2,
1441 GOMP_MAP_VARS_DATA = 4,
1442 GOMP_MAP_VARS_ENTER_DATA = 8
d9a6bd32
JJ
1443};
1444
829c6349
CLT
1445extern void gomp_acc_declare_allocate (bool, size_t, void **, size_t *,
1446 unsigned short *);
1f4c5b9b
CLT
1447struct gomp_coalesce_buf;
1448extern void gomp_copy_host2dev (struct gomp_device_descr *,
1449 struct goacc_asyncqueue *, void *, const void *,
9c41f5b9 1450 size_t, bool, struct gomp_coalesce_buf *);
1f4c5b9b
CLT
1451extern void gomp_copy_dev2host (struct gomp_device_descr *,
1452 struct goacc_asyncqueue *, void *, const void *,
1453 size_t);
5bcd470b 1454extern uintptr_t gomp_map_val (struct target_mem_desc *, void **, size_t);
5d5be7bf
JB
1455extern void gomp_attach_pointer (struct gomp_device_descr *,
1456 struct goacc_asyncqueue *, splay_tree,
1457 splay_tree_key, uintptr_t, size_t,
0ab29cf0 1458 struct gomp_coalesce_buf *, bool);
5d5be7bf
JB
1459extern void gomp_detach_pointer (struct gomp_device_descr *,
1460 struct goacc_asyncqueue *, splay_tree_key,
1461 uintptr_t, bool, struct gomp_coalesce_buf *);
275c736e
CLT
1462extern struct target_mem_desc *goacc_map_vars (struct gomp_device_descr *,
1463 struct goacc_asyncqueue *,
1464 size_t, void **, void **,
1465 size_t *, void *, bool,
1466 enum gomp_map_vars_kind);
1467extern void goacc_unmap_vars (struct target_mem_desc *, bool,
1468 struct goacc_asyncqueue *);
41dbbb37 1469extern void gomp_init_device (struct gomp_device_descr *);
1f4c5b9b 1470extern bool gomp_fini_device (struct gomp_device_descr *);
22be2349 1471extern void gomp_unload_device (struct gomp_device_descr *);
829c6349 1472extern bool gomp_remove_var (struct gomp_device_descr *, splay_tree_key);
1cbd94e8
JB
1473extern void gomp_remove_var_async (struct gomp_device_descr *, splay_tree_key,
1474 struct goacc_asyncqueue *);
41dbbb37 1475
953ff289
DN
1476/* work.c */
1477
28567c40 1478extern void gomp_init_work_share (struct gomp_work_share *, size_t, unsigned);
a68ab351 1479extern void gomp_fini_work_share (struct gomp_work_share *);
28567c40 1480extern bool gomp_work_share_start (size_t);
953ff289 1481extern void gomp_work_share_end (void);
acf0174b 1482extern bool gomp_work_share_end_cancel (void);
953ff289
DN
1483extern void gomp_work_share_end_nowait (void);
1484
a68ab351
JJ
1485static inline void
1486gomp_work_share_init_done (void)
1487{
1488 struct gomp_thread *thr = gomp_thread ();
1489 if (__builtin_expect (thr->ts.last_work_share != NULL, 1))
1490 gomp_ptrlock_set (&thr->ts.last_work_share->next_ws, thr->ts.work_share);
1491}
1492
953ff289
DN
1493#ifdef HAVE_ATTRIBUTE_VISIBILITY
1494# pragma GCC visibility pop
1495#endif
1496
1497/* Now that we're back to default visibility, include the globals. */
1498#include "libgomp_g.h"
1499
1500/* Include omp.h by parts. */
1501#include "omp-lock.h"
1502#define _LIBGOMP_OMP_LOCK_DEFINED 1
1503#include "omp.h.in"
1504
a68ab351
JJ
1505#if !defined (HAVE_ATTRIBUTE_VISIBILITY) \
1506 || !defined (HAVE_ATTRIBUTE_ALIAS) \
876080ff 1507 || !defined (HAVE_AS_SYMVER_DIRECTIVE) \
6d28b933
RO
1508 || !defined (PIC) \
1509 || !defined (HAVE_SYMVER_SYMBOL_RENAMING_RUNTIME_SUPPORT)
a68ab351
JJ
1510# undef LIBGOMP_GNU_SYMBOL_VERSIONING
1511#endif
1512
1513#ifdef LIBGOMP_GNU_SYMBOL_VERSIONING
1514extern void gomp_init_lock_30 (omp_lock_t *) __GOMP_NOTHROW;
1515extern void gomp_destroy_lock_30 (omp_lock_t *) __GOMP_NOTHROW;
1516extern void gomp_set_lock_30 (omp_lock_t *) __GOMP_NOTHROW;
1517extern void gomp_unset_lock_30 (omp_lock_t *) __GOMP_NOTHROW;
1518extern int gomp_test_lock_30 (omp_lock_t *) __GOMP_NOTHROW;
1519extern void gomp_init_nest_lock_30 (omp_nest_lock_t *) __GOMP_NOTHROW;
1520extern void gomp_destroy_nest_lock_30 (omp_nest_lock_t *) __GOMP_NOTHROW;
1521extern void gomp_set_nest_lock_30 (omp_nest_lock_t *) __GOMP_NOTHROW;
1522extern void gomp_unset_nest_lock_30 (omp_nest_lock_t *) __GOMP_NOTHROW;
1523extern int gomp_test_nest_lock_30 (omp_nest_lock_t *) __GOMP_NOTHROW;
1524
1525extern void gomp_init_lock_25 (omp_lock_25_t *) __GOMP_NOTHROW;
1526extern void gomp_destroy_lock_25 (omp_lock_25_t *) __GOMP_NOTHROW;
1527extern void gomp_set_lock_25 (omp_lock_25_t *) __GOMP_NOTHROW;
1528extern void gomp_unset_lock_25 (omp_lock_25_t *) __GOMP_NOTHROW;
1529extern int gomp_test_lock_25 (omp_lock_25_t *) __GOMP_NOTHROW;
1530extern void gomp_init_nest_lock_25 (omp_nest_lock_25_t *) __GOMP_NOTHROW;
1531extern void gomp_destroy_nest_lock_25 (omp_nest_lock_25_t *) __GOMP_NOTHROW;
1532extern void gomp_set_nest_lock_25 (omp_nest_lock_25_t *) __GOMP_NOTHROW;
1533extern void gomp_unset_nest_lock_25 (omp_nest_lock_25_t *) __GOMP_NOTHROW;
1534extern int gomp_test_nest_lock_25 (omp_nest_lock_25_t *) __GOMP_NOTHROW;
1535
a68ab351
JJ
1536# define omp_lock_symver(fn) \
1537 __asm (".symver g" #fn "_30, " #fn "@@OMP_3.0"); \
1538 __asm (".symver g" #fn "_25, " #fn "@OMP_1.0");
1539#else
1540# define gomp_init_lock_30 omp_init_lock
1541# define gomp_destroy_lock_30 omp_destroy_lock
1542# define gomp_set_lock_30 omp_set_lock
1543# define gomp_unset_lock_30 omp_unset_lock
1544# define gomp_test_lock_30 omp_test_lock
1545# define gomp_init_nest_lock_30 omp_init_nest_lock
1546# define gomp_destroy_nest_lock_30 omp_destroy_nest_lock
1547# define gomp_set_nest_lock_30 omp_set_nest_lock
1548# define gomp_unset_nest_lock_30 omp_unset_nest_lock
1549# define gomp_test_nest_lock_30 omp_test_nest_lock
1550#endif
1551
953ff289
DN
1552#ifdef HAVE_ATTRIBUTE_VISIBILITY
1553# define attribute_hidden __attribute__ ((visibility ("hidden")))
1554#else
1555# define attribute_hidden
1556#endif
1557
79a2c428
MS
1558#if __GNUC__ >= 9
1559# define HAVE_ATTRIBUTE_COPY
1560#endif
1561
1562#ifdef HAVE_ATTRIBUTE_COPY
1563# define attribute_copy(arg) __attribute__ ((copy (arg)))
1564#else
1565# define attribute_copy(arg)
1566#endif
1567
953ff289 1568#ifdef HAVE_ATTRIBUTE_ALIAS
9b94fbc7 1569# define strong_alias(fn, al) \
79a2c428 1570 extern __typeof (fn) al __attribute__ ((alias (#fn))) attribute_copy (fn);
9b94fbc7 1571
acf0174b
JJ
1572# define ialias_ulp ialias_str1(__USER_LABEL_PREFIX__)
1573# define ialias_str1(x) ialias_str2(x)
1574# define ialias_str2(x) #x
953ff289
DN
1575# define ialias(fn) \
1576 extern __typeof (fn) gomp_ialias_##fn \
79a2c428 1577 __attribute__ ((alias (#fn))) attribute_hidden attribute_copy (fn);
acf0174b
JJ
1578# define ialias_redirect(fn) \
1579 extern __typeof (fn) fn __asm__ (ialias_ulp "gomp_ialias_" #fn) attribute_hidden;
1580# define ialias_call(fn) gomp_ialias_ ## fn
953ff289
DN
1581#else
1582# define ialias(fn)
acf0174b
JJ
1583# define ialias_redirect(fn)
1584# define ialias_call(fn) fn
953ff289
DN
1585#endif
1586
e4606348
JJ
1587/* Helper function for priority_node_to_task() and
1588 task_to_priority_node().
1589
1590 Return the offset from a task to its priority_node entry. The
1591 priority_node entry is has a type of TYPE. */
1592
1593static inline size_t
1594priority_queue_offset (enum priority_queue_type type)
1595{
1596 return offsetof (struct gomp_task, pnode[(int) type]);
1597}
1598
1599/* Return the task associated with a priority NODE of type TYPE. */
1600
1601static inline struct gomp_task *
1602priority_node_to_task (enum priority_queue_type type,
1603 struct priority_node *node)
1604{
1605 return (struct gomp_task *) ((char *) node - priority_queue_offset (type));
1606}
1607
1608/* Return the priority node of type TYPE for a given TASK. */
1609
1610static inline struct priority_node *
1611task_to_priority_node (enum priority_queue_type type,
1612 struct gomp_task *task)
1613{
1614 return (struct priority_node *) ((char *) task
1615 + priority_queue_offset (type));
1616}
28567c40
JJ
1617
1618#ifdef LIBGOMP_USE_PTHREADS
1619static inline gomp_thread_handle
1620gomp_thread_self (void)
1621{
1622 return pthread_self ();
1623}
1624
1625static inline gomp_thread_handle
1626gomp_thread_to_pthread_t (struct gomp_thread *thr)
1627{
1628 struct gomp_thread *this_thr = gomp_thread ();
1629 if (thr == this_thr)
1630 return pthread_self ();
1631#ifdef GOMP_NEEDS_THREAD_HANDLE
1632 return thr->handle;
1633#else
1634 /* On Linux with initial-exec TLS, the pthread_t of the thread containing
1635 thr can be computed from thr, this_thr and pthread_self (),
1636 as the distance between this_thr and pthread_self () is constant. */
1637 return pthread_self () + ((uintptr_t) thr - (uintptr_t) this_thr);
1638#endif
1639}
1640#else
1641static inline gomp_thread_handle
1642gomp_thread_self (void)
1643{
1644 return (gomp_thread_handle) {};
1645}
1646
1647static inline gomp_thread_handle
1648gomp_thread_to_pthread_t (struct gomp_thread *thr)
1649{
1650 (void) thr;
1651 return gomp_thread_self ();
1652}
1653#endif
1654
953ff289 1655#endif /* LIBGOMP_H */