e460634820
gcc/ 2015-11-14 Jakub Jelinek <jakub@redhat.com> * omp-low.c (lower_omp_ordered): Add argument to GOMP_SMD_ORDERED_* internal calls - 0 if ordered simd and 1 for ordered threads simd. * tree-vectorizer.c (adjust_simduid_builtins): If GOMP_SIMD_ORDERED_* argument is 1, replace it with GOMP_ordered_* call instead of removing it. gcc/c/ 2015-11-14 Jakub Jelinek <jakub@redhat.com> * c-typeck.c (c_finish_omp_clauses): Don't mark GOMP_MAP_FIRSTPRIVATE_POINTER decls addressable. gcc/cp/ 2015-11-14 Jakub Jelinek <jakub@redhat.com> * semantics.c (finish_omp_clauses): Don't mark GOMP_MAP_FIRSTPRIVATE_POINTER decls addressable. libgomp/ 2015-11-14 Jakub Jelinek <jakub@redhat.com> Aldy Hernandez <aldyh@redhat.com> Ilya Verbin <ilya.verbin@intel.com> * ordered.c (gomp_doacross_init, GOMP_doacross_post, GOMP_doacross_wait, gomp_doacross_ull_init, GOMP_doacross_ull_post, GOMP_doacross_ull_wait): For GFS_GUIDED don't divide number of iterators or IV by chunk size. * parallel.c (gomp_resolve_num_threads): Don't assume that if thr->ts.team is non-NULL, then pool must be non-NULL. * libgomp-plugin.h (GOMP_PLUGIN_target_task_completion): Declare. * libgomp.map (GOMP_PLUGIN_1.1): New symbol version, export GOMP_PLUGIN_target_task_completion. * Makefile.am (libgomp_la_SOURCES): Add priority_queue.c. * Makefile.in: Regenerate. * libgomp.h: Shuffle prototypes and forward definitions around so priority queues can be defined. (enum gomp_task_kind): Add GOMP_TASK_ASYNC_RUNNING. (enum gomp_target_task_state): New enum. (struct gomp_target_task): Add state, tgt, task and team fields. (gomp_create_target_task): Change return type to bool, add state argument. (gomp_target_task_fn): Change return type to bool. (struct gomp_device_descr): Add async_run_func. (struct gomp_task): Remove children, next_child, prev_child, next_queue, prev_queue, next_taskgroup, prev_taskgroup. Add pnode field. (struct gomp_taskgroup): Remove children. Add taskgroup_queue. (struct gomp_team): Change task_queue type to a priority queue. (splay_compare): Define inline. (priority_queue_offset): New. (priority_node_to_task): New. (task_to_priority_node): New. * oacc-mem.c: Do not include splay-tree.h. * priority_queue.c: New file. * priority_queue.h: New file. * splay-tree.c: Do not include splay-tree.h. (splay_tree_foreach_internal): New. (splay_tree_foreach): New. * splay-tree.h: Become re-entrant if splay_tree_prefix is defined. (splay_tree_callback): Define typedef. * target.c (splay_compare): Move to libgomp.h. (GOMP_target): Don't adjust *thr in any way around running offloaded task. (GOMP_target_ext): Likewise. Handle target nowait. (GOMP_target_update_ext, GOMP_target_enter_exit_data): Check return value from gomp_create_target_task, if false, fallthrough as if no dependencies exist. (gomp_target_task_fn): Change return type to bool, return true if the task should have another part scheduled later. Handle target nowait. (gomp_load_plugin_for_device): Initialize async_run. * task.c (gomp_init_task): Initialize children_queue. (gomp_clear_parent_in_list): New. (gomp_clear_parent_in_tree): New. (gomp_clear_parent): Handle priorities. (GOMP_task): Likewise. (priority_queue_move_task_first, gomp_target_task_completion, GOMP_PLUGIN_target_task_completion): New functions. (gomp_create_target_task): Use priority queues. Change return type to bool, add state argument, return false if for async {{enter,exit} data,update} constructs no dependencies need to be waited for, handle target nowait. Set task->fn to NULL instead of gomp_target_task_fn. (verify_children_queue): Remove. (priority_list_upgrade_task): New. (priority_queue_upgrade_task): New. (verify_task_queue): Remove. (priority_list_downgrade_task): New. (priority_queue_downgrade_task): New. (gomp_task_run_pre): Use priority queues. Abstract code out to priority_queue_downgrade_task. (gomp_task_run_post_handle_dependers): Use priority queues. (gomp_task_run_post_remove_parent): Likewise. (gomp_task_run_post_remove_taskgroup): Likewise. (gomp_barrier_handle_tasks): Likewise. Handle target nowait target tasks specially. (GOMP_taskwait): Likewise. (gomp_task_maybe_wait_for_dependencies): Likewise. Abstract code to priority-queue_upgrade_task. (GOMP_taskgroup_start): Use priority queues. (GOMP_taskgroup_end): Likewise. Handle target nowait target tasks specially. If taskgroup is NULL, and thr->ts.level is 0, act as a barrier. * taskloop.c (GOMP_taskloop): Handle priorities. * team.c (gomp_new_team): Call priority_queue_init. (free_team): Call priority_queue_free. (gomp_free_thread): Call gomp_team_end if thr->ts.team is artificial team created for target nowait in implicit parallel region. (gomp_team_start): For nested check, test thr->ts.level instead of thr->ts.team != NULL. * testsuite/libgomp.c/doacross-3.c: New test. * testsuite/libgomp.c/ordered-5.c: New test. * testsuite/libgomp.c/priority.c: New test. * testsuite/libgomp.c/target-31.c: New test. * testsuite/libgomp.c/target-32.c: New test. * testsuite/libgomp.c/target-33.c: New test. * testsuite/libgomp.c/target-34.c: New test. liboffloadmic/ 2015-11-14 Ilya Verbin <ilya.verbin@intel.com> * runtime/offload_host.cpp (task_completion_callback): New variable. (offload_proxy_task_completed_ooo): Call task_completion_callback. (__offload_register_task_callback): New function. * runtime/offload_host.h (__offload_register_task_callback): New declaration. * plugin/libgomp-plugin-intelmic.cpp (offload): Add async_data argument, handle async offloading. (register_main_image): Call register_main_image. (GOMP_OFFLOAD_init_device, get_target_table, GOMP_OFFLOAD_alloc, GOMP_OFFLOAD_free, GOMP_OFFLOAD_host2dev, GOMP_OFFLOAD_dev2host, GOMP_OFFLOAD_dev2dev) Adjust offload callers. (GOMP_OFFLOAD_async_run): New function. (GOMP_OFFLOAD_run): Implement using GOMP_OFFLOAD_async_run. From-SVN: r230381
786 lines
22 KiB
C
786 lines
22 KiB
C
/* Copyright (C) 2005-2015 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/>. */
|
|
|
|
/* This file handles the ORDERED construct. */
|
|
|
|
#include "libgomp.h"
|
|
#include <stdarg.h>
|
|
#include <string.h>
|
|
#include "doacross.h"
|
|
|
|
|
|
/* This function is called when first allocating an iteration block. That
|
|
is, the thread is not currently on the queue. The work-share lock must
|
|
be held on entry. */
|
|
|
|
void
|
|
gomp_ordered_first (void)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_team *team = thr->ts.team;
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
unsigned index;
|
|
|
|
/* Work share constructs can be orphaned. */
|
|
if (team == NULL || team->nthreads == 1)
|
|
return;
|
|
|
|
index = ws->ordered_cur + ws->ordered_num_used;
|
|
if (index >= team->nthreads)
|
|
index -= team->nthreads;
|
|
ws->ordered_team_ids[index] = thr->ts.team_id;
|
|
|
|
/* If this is the first and only thread in the queue, then there is
|
|
no one to release us when we get to our ordered section. Post to
|
|
our own release queue now so that we won't block later. */
|
|
if (ws->ordered_num_used++ == 0)
|
|
gomp_sem_post (team->ordered_release[thr->ts.team_id]);
|
|
}
|
|
|
|
/* This function is called when completing the last iteration block. That
|
|
is, there are no more iterations to perform and so the thread should be
|
|
removed from the queue entirely. Because of the way ORDERED blocks are
|
|
managed, it follows that we currently own access to the ORDERED block,
|
|
and should now pass it on to the next thread. The work-share lock must
|
|
be held on entry. */
|
|
|
|
void
|
|
gomp_ordered_last (void)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_team *team = thr->ts.team;
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
unsigned next_id;
|
|
|
|
/* Work share constructs can be orphaned. */
|
|
if (team == NULL || team->nthreads == 1)
|
|
return;
|
|
|
|
/* We're no longer the owner. */
|
|
ws->ordered_owner = -1;
|
|
|
|
/* If we're not the last thread in the queue, then wake the next. */
|
|
if (--ws->ordered_num_used > 0)
|
|
{
|
|
unsigned next = ws->ordered_cur + 1;
|
|
if (next == team->nthreads)
|
|
next = 0;
|
|
ws->ordered_cur = next;
|
|
|
|
next_id = ws->ordered_team_ids[next];
|
|
gomp_sem_post (team->ordered_release[next_id]);
|
|
}
|
|
}
|
|
|
|
|
|
/* This function is called when allocating a subsequent allocation block.
|
|
That is, we're done with the current iteration block and we're allocating
|
|
another. This is the logical combination of a call to gomp_ordered_last
|
|
followed by a call to gomp_ordered_first. The work-share lock must be
|
|
held on entry. */
|
|
|
|
void
|
|
gomp_ordered_next (void)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_team *team = thr->ts.team;
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
unsigned index, next_id;
|
|
|
|
/* Work share constructs can be orphaned. */
|
|
if (team == NULL || team->nthreads == 1)
|
|
return;
|
|
|
|
/* We're no longer the owner. */
|
|
ws->ordered_owner = -1;
|
|
|
|
/* If there's only one thread in the queue, that must be us. */
|
|
if (ws->ordered_num_used == 1)
|
|
{
|
|
/* We have a similar situation as in gomp_ordered_first
|
|
where we need to post to our own release semaphore. */
|
|
gomp_sem_post (team->ordered_release[thr->ts.team_id]);
|
|
return;
|
|
}
|
|
|
|
/* If the queue is entirely full, then we move ourself to the end of
|
|
the queue merely by incrementing ordered_cur. Only if it's not
|
|
full do we have to write our id. */
|
|
if (ws->ordered_num_used < team->nthreads)
|
|
{
|
|
index = ws->ordered_cur + ws->ordered_num_used;
|
|
if (index >= team->nthreads)
|
|
index -= team->nthreads;
|
|
ws->ordered_team_ids[index] = thr->ts.team_id;
|
|
}
|
|
|
|
index = ws->ordered_cur + 1;
|
|
if (index == team->nthreads)
|
|
index = 0;
|
|
ws->ordered_cur = index;
|
|
|
|
next_id = ws->ordered_team_ids[index];
|
|
gomp_sem_post (team->ordered_release[next_id]);
|
|
}
|
|
|
|
|
|
/* This function is called when a statically scheduled loop is first
|
|
being created. */
|
|
|
|
void
|
|
gomp_ordered_static_init (void)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_team *team = thr->ts.team;
|
|
|
|
if (team == NULL || team->nthreads == 1)
|
|
return;
|
|
|
|
gomp_sem_post (team->ordered_release[0]);
|
|
}
|
|
|
|
/* This function is called when a statically scheduled loop is moving to
|
|
the next allocation block. Static schedules are not first come first
|
|
served like the others, so we're to move to the numerically next thread,
|
|
not the next thread on a list. The work-share lock should *not* be held
|
|
on entry. */
|
|
|
|
void
|
|
gomp_ordered_static_next (void)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_team *team = thr->ts.team;
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
unsigned id = thr->ts.team_id;
|
|
|
|
if (team == NULL || team->nthreads == 1)
|
|
return;
|
|
|
|
ws->ordered_owner = -1;
|
|
|
|
/* This thread currently owns the lock. Increment the owner. */
|
|
if (++id == team->nthreads)
|
|
id = 0;
|
|
ws->ordered_team_ids[0] = id;
|
|
gomp_sem_post (team->ordered_release[id]);
|
|
}
|
|
|
|
/* This function is called when we need to assert that the thread owns the
|
|
ordered section. Due to the problem of posted-but-not-waited semaphores,
|
|
this needs to happen before completing a loop iteration. */
|
|
|
|
void
|
|
gomp_ordered_sync (void)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_team *team = thr->ts.team;
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
|
|
/* Work share constructs can be orphaned. But this clearly means that
|
|
we are the only thread, and so we automatically own the section. */
|
|
if (team == NULL || team->nthreads == 1)
|
|
return;
|
|
|
|
/* ??? I believe it to be safe to access this data without taking the
|
|
ws->lock. The only presumed race condition is with the previous
|
|
thread on the queue incrementing ordered_cur such that it points
|
|
to us, concurrently with our check below. But our team_id is
|
|
already present in the queue, and the other thread will always
|
|
post to our release semaphore. So the two cases are that we will
|
|
either win the race an momentarily block on the semaphore, or lose
|
|
the race and find the semaphore already unlocked and so not block.
|
|
Either way we get correct results.
|
|
However, there is an implicit flush on entry to an ordered region,
|
|
so we do need to have a barrier here. If we were taking a lock
|
|
this could be MEMMODEL_RELEASE since the acquire would be coverd
|
|
by the lock. */
|
|
|
|
__atomic_thread_fence (MEMMODEL_ACQ_REL);
|
|
if (ws->ordered_owner != thr->ts.team_id)
|
|
{
|
|
gomp_sem_wait (team->ordered_release[thr->ts.team_id]);
|
|
ws->ordered_owner = thr->ts.team_id;
|
|
}
|
|
}
|
|
|
|
/* This function is called by user code when encountering the start of an
|
|
ORDERED block. We must check to see if the current thread is at the
|
|
head of the queue, and if not, block. */
|
|
|
|
#ifdef HAVE_ATTRIBUTE_ALIAS
|
|
extern void GOMP_ordered_start (void)
|
|
__attribute__((alias ("gomp_ordered_sync")));
|
|
#else
|
|
void
|
|
GOMP_ordered_start (void)
|
|
{
|
|
gomp_ordered_sync ();
|
|
}
|
|
#endif
|
|
|
|
/* This function is called by user code when encountering the end of an
|
|
ORDERED block. With the current ORDERED implementation there's nothing
|
|
for us to do.
|
|
|
|
However, the current implementation has a flaw in that it does not allow
|
|
the next thread into the ORDERED section immediately after the current
|
|
thread exits the ORDERED section in its last iteration. The existance
|
|
of this function allows the implementation to change. */
|
|
|
|
void
|
|
GOMP_ordered_end (void)
|
|
{
|
|
}
|
|
|
|
/* DOACROSS initialization. */
|
|
|
|
#define MAX_COLLAPSED_BITS (__SIZEOF_LONG__ * __CHAR_BIT__)
|
|
|
|
void
|
|
gomp_doacross_init (unsigned ncounts, long *counts, long chunk_size)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_team *team = thr->ts.team;
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
unsigned int i, bits[MAX_COLLAPSED_BITS], num_bits = 0;
|
|
unsigned long ent, num_ents, elt_sz, shift_sz;
|
|
struct gomp_doacross_work_share *doacross;
|
|
|
|
if (team == NULL || team->nthreads == 1)
|
|
return;
|
|
|
|
for (i = 0; i < ncounts; i++)
|
|
{
|
|
/* If any count is 0, GOMP_doacross_{post,wait} can't be called. */
|
|
if (counts[i] == 0)
|
|
return;
|
|
|
|
if (num_bits <= MAX_COLLAPSED_BITS)
|
|
{
|
|
unsigned int this_bits;
|
|
if (counts[i] == 1)
|
|
this_bits = 1;
|
|
else
|
|
this_bits = __SIZEOF_LONG__ * __CHAR_BIT__
|
|
- __builtin_clzl (counts[i] - 1);
|
|
if (num_bits + this_bits <= MAX_COLLAPSED_BITS)
|
|
{
|
|
bits[i] = this_bits;
|
|
num_bits += this_bits;
|
|
}
|
|
else
|
|
num_bits = MAX_COLLAPSED_BITS + 1;
|
|
}
|
|
}
|
|
|
|
if (ws->sched == GFS_STATIC)
|
|
num_ents = team->nthreads;
|
|
else if (ws->sched == GFS_GUIDED)
|
|
num_ents = counts[0];
|
|
else
|
|
num_ents = (counts[0] - 1) / chunk_size + 1;
|
|
if (num_bits <= MAX_COLLAPSED_BITS)
|
|
{
|
|
elt_sz = sizeof (unsigned long);
|
|
shift_sz = ncounts * sizeof (unsigned int);
|
|
}
|
|
else
|
|
{
|
|
elt_sz = sizeof (unsigned long) * ncounts;
|
|
shift_sz = 0;
|
|
}
|
|
elt_sz = (elt_sz + 63) & ~63UL;
|
|
|
|
doacross = gomp_malloc (sizeof (*doacross) + 63 + num_ents * elt_sz
|
|
+ shift_sz);
|
|
doacross->chunk_size = chunk_size;
|
|
doacross->elt_sz = elt_sz;
|
|
doacross->ncounts = ncounts;
|
|
doacross->flattened = false;
|
|
doacross->array = (unsigned char *)
|
|
((((uintptr_t) (doacross + 1)) + 63 + shift_sz)
|
|
& ~(uintptr_t) 63);
|
|
if (num_bits <= MAX_COLLAPSED_BITS)
|
|
{
|
|
unsigned int shift_count = 0;
|
|
doacross->flattened = true;
|
|
for (i = ncounts; i > 0; i--)
|
|
{
|
|
doacross->shift_counts[i - 1] = shift_count;
|
|
shift_count += bits[i - 1];
|
|
}
|
|
for (ent = 0; ent < num_ents; ent++)
|
|
*(unsigned long *) (doacross->array + ent * elt_sz) = 0;
|
|
}
|
|
else
|
|
for (ent = 0; ent < num_ents; ent++)
|
|
memset (doacross->array + ent * elt_sz, '\0',
|
|
sizeof (unsigned long) * ncounts);
|
|
if (ws->sched == GFS_STATIC && chunk_size == 0)
|
|
{
|
|
unsigned long q = counts[0] / num_ents;
|
|
unsigned long t = counts[0] % num_ents;
|
|
doacross->boundary = t * (q + 1);
|
|
doacross->q = q;
|
|
doacross->t = t;
|
|
}
|
|
ws->doacross = doacross;
|
|
}
|
|
|
|
/* DOACROSS POST operation. */
|
|
|
|
void
|
|
GOMP_doacross_post (long *counts)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
struct gomp_doacross_work_share *doacross = ws->doacross;
|
|
unsigned long ent;
|
|
unsigned int i;
|
|
|
|
if (__builtin_expect (doacross == NULL, 0))
|
|
{
|
|
__sync_synchronize ();
|
|
return;
|
|
}
|
|
|
|
if (__builtin_expect (ws->sched == GFS_STATIC, 1))
|
|
ent = thr->ts.team_id;
|
|
else if (ws->sched == GFS_GUIDED)
|
|
ent = counts[0];
|
|
else
|
|
ent = counts[0] / doacross->chunk_size;
|
|
unsigned long *array = (unsigned long *) (doacross->array
|
|
+ ent * doacross->elt_sz);
|
|
|
|
if (__builtin_expect (doacross->flattened, 1))
|
|
{
|
|
unsigned long flattened
|
|
= (unsigned long) counts[0] << doacross->shift_counts[0];
|
|
|
|
for (i = 1; i < doacross->ncounts; i++)
|
|
flattened |= (unsigned long) counts[i]
|
|
<< doacross->shift_counts[i];
|
|
flattened++;
|
|
if (flattened == __atomic_load_n (array, MEMMODEL_ACQUIRE))
|
|
__atomic_thread_fence (MEMMODEL_RELEASE);
|
|
else
|
|
__atomic_store_n (array, flattened, MEMMODEL_RELEASE);
|
|
return;
|
|
}
|
|
|
|
__atomic_thread_fence (MEMMODEL_ACQUIRE);
|
|
for (i = doacross->ncounts; i-- > 0; )
|
|
{
|
|
if (counts[i] + 1UL != __atomic_load_n (&array[i], MEMMODEL_RELAXED))
|
|
__atomic_store_n (&array[i], counts[i] + 1UL, MEMMODEL_RELEASE);
|
|
}
|
|
}
|
|
|
|
/* DOACROSS WAIT operation. */
|
|
|
|
void
|
|
GOMP_doacross_wait (long first, ...)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
struct gomp_doacross_work_share *doacross = ws->doacross;
|
|
va_list ap;
|
|
unsigned long ent;
|
|
unsigned int i;
|
|
|
|
if (__builtin_expect (doacross == NULL, 0))
|
|
{
|
|
__sync_synchronize ();
|
|
return;
|
|
}
|
|
|
|
if (__builtin_expect (ws->sched == GFS_STATIC, 1))
|
|
{
|
|
if (ws->chunk_size == 0)
|
|
{
|
|
if (first < doacross->boundary)
|
|
ent = first / (doacross->q + 1);
|
|
else
|
|
ent = (first - doacross->boundary) / doacross->q
|
|
+ doacross->t;
|
|
}
|
|
else
|
|
ent = first / ws->chunk_size % thr->ts.team->nthreads;
|
|
}
|
|
else if (ws->sched == GFS_GUIDED)
|
|
ent = first;
|
|
else
|
|
ent = first / doacross->chunk_size;
|
|
unsigned long *array = (unsigned long *) (doacross->array
|
|
+ ent * doacross->elt_sz);
|
|
|
|
if (__builtin_expect (doacross->flattened, 1))
|
|
{
|
|
unsigned long flattened
|
|
= (unsigned long) first << doacross->shift_counts[0];
|
|
unsigned long cur;
|
|
|
|
va_start (ap, first);
|
|
for (i = 1; i < doacross->ncounts; i++)
|
|
flattened |= (unsigned long) va_arg (ap, long)
|
|
<< doacross->shift_counts[i];
|
|
cur = __atomic_load_n (array, MEMMODEL_ACQUIRE);
|
|
if (flattened < cur)
|
|
{
|
|
__atomic_thread_fence (MEMMODEL_RELEASE);
|
|
va_end (ap);
|
|
return;
|
|
}
|
|
doacross_spin (array, flattened, cur);
|
|
__atomic_thread_fence (MEMMODEL_RELEASE);
|
|
va_end (ap);
|
|
return;
|
|
}
|
|
|
|
do
|
|
{
|
|
va_start (ap, first);
|
|
for (i = 0; i < doacross->ncounts; i++)
|
|
{
|
|
unsigned long thisv
|
|
= (unsigned long) (i ? va_arg (ap, long) : first) + 1;
|
|
unsigned long cur = __atomic_load_n (&array[i], MEMMODEL_RELAXED);
|
|
if (thisv < cur)
|
|
{
|
|
i = doacross->ncounts;
|
|
break;
|
|
}
|
|
if (thisv > cur)
|
|
break;
|
|
}
|
|
va_end (ap);
|
|
if (i == doacross->ncounts)
|
|
break;
|
|
cpu_relax ();
|
|
}
|
|
while (1);
|
|
__sync_synchronize ();
|
|
}
|
|
|
|
typedef unsigned long long gomp_ull;
|
|
|
|
void
|
|
gomp_doacross_ull_init (unsigned ncounts, gomp_ull *counts, gomp_ull chunk_size)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_team *team = thr->ts.team;
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
unsigned int i, bits[MAX_COLLAPSED_BITS], num_bits = 0;
|
|
unsigned long ent, num_ents, elt_sz, shift_sz;
|
|
struct gomp_doacross_work_share *doacross;
|
|
|
|
if (team == NULL || team->nthreads == 1)
|
|
return;
|
|
|
|
for (i = 0; i < ncounts; i++)
|
|
{
|
|
/* If any count is 0, GOMP_doacross_{post,wait} can't be called. */
|
|
if (counts[i] == 0)
|
|
return;
|
|
|
|
if (num_bits <= MAX_COLLAPSED_BITS)
|
|
{
|
|
unsigned int this_bits;
|
|
if (counts[i] == 1)
|
|
this_bits = 1;
|
|
else
|
|
this_bits = __SIZEOF_LONG_LONG__ * __CHAR_BIT__
|
|
- __builtin_clzll (counts[i] - 1);
|
|
if (num_bits + this_bits <= MAX_COLLAPSED_BITS)
|
|
{
|
|
bits[i] = this_bits;
|
|
num_bits += this_bits;
|
|
}
|
|
else
|
|
num_bits = MAX_COLLAPSED_BITS + 1;
|
|
}
|
|
}
|
|
|
|
if (ws->sched == GFS_STATIC)
|
|
num_ents = team->nthreads;
|
|
else if (ws->sched == GFS_GUIDED)
|
|
num_ents = counts[0];
|
|
else
|
|
num_ents = (counts[0] - 1) / chunk_size + 1;
|
|
if (num_bits <= MAX_COLLAPSED_BITS)
|
|
{
|
|
elt_sz = sizeof (unsigned long);
|
|
shift_sz = ncounts * sizeof (unsigned int);
|
|
}
|
|
else
|
|
{
|
|
if (sizeof (gomp_ull) == sizeof (unsigned long))
|
|
elt_sz = sizeof (gomp_ull) * ncounts;
|
|
else if (sizeof (gomp_ull) == 2 * sizeof (unsigned long))
|
|
elt_sz = sizeof (unsigned long) * 2 * ncounts;
|
|
else
|
|
abort ();
|
|
shift_sz = 0;
|
|
}
|
|
elt_sz = (elt_sz + 63) & ~63UL;
|
|
|
|
doacross = gomp_malloc (sizeof (*doacross) + 63 + num_ents * elt_sz
|
|
+ shift_sz);
|
|
doacross->chunk_size_ull = chunk_size;
|
|
doacross->elt_sz = elt_sz;
|
|
doacross->ncounts = ncounts;
|
|
doacross->flattened = false;
|
|
doacross->boundary = 0;
|
|
doacross->array = (unsigned char *)
|
|
((((uintptr_t) (doacross + 1)) + 63 + shift_sz)
|
|
& ~(uintptr_t) 63);
|
|
if (num_bits <= MAX_COLLAPSED_BITS)
|
|
{
|
|
unsigned int shift_count = 0;
|
|
doacross->flattened = true;
|
|
for (i = ncounts; i > 0; i--)
|
|
{
|
|
doacross->shift_counts[i - 1] = shift_count;
|
|
shift_count += bits[i - 1];
|
|
}
|
|
for (ent = 0; ent < num_ents; ent++)
|
|
*(unsigned long *) (doacross->array + ent * elt_sz) = 0;
|
|
}
|
|
else
|
|
for (ent = 0; ent < num_ents; ent++)
|
|
memset (doacross->array + ent * elt_sz, '\0',
|
|
sizeof (unsigned long) * ncounts);
|
|
if (ws->sched == GFS_STATIC && chunk_size == 0)
|
|
{
|
|
gomp_ull q = counts[0] / num_ents;
|
|
gomp_ull t = counts[0] % num_ents;
|
|
doacross->boundary_ull = t * (q + 1);
|
|
doacross->q_ull = q;
|
|
doacross->t = t;
|
|
}
|
|
ws->doacross = doacross;
|
|
}
|
|
|
|
/* DOACROSS POST operation. */
|
|
|
|
void
|
|
GOMP_doacross_ull_post (gomp_ull *counts)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
struct gomp_doacross_work_share *doacross = ws->doacross;
|
|
unsigned long ent;
|
|
unsigned int i;
|
|
|
|
if (__builtin_expect (doacross == NULL, 0))
|
|
{
|
|
__sync_synchronize ();
|
|
return;
|
|
}
|
|
|
|
if (__builtin_expect (ws->sched == GFS_STATIC, 1))
|
|
ent = thr->ts.team_id;
|
|
else if (ws->sched == GFS_GUIDED)
|
|
ent = counts[0];
|
|
else
|
|
ent = counts[0] / doacross->chunk_size_ull;
|
|
|
|
if (__builtin_expect (doacross->flattened, 1))
|
|
{
|
|
unsigned long *array = (unsigned long *) (doacross->array
|
|
+ ent * doacross->elt_sz);
|
|
gomp_ull flattened
|
|
= counts[0] << doacross->shift_counts[0];
|
|
|
|
for (i = 1; i < doacross->ncounts; i++)
|
|
flattened |= counts[i] << doacross->shift_counts[i];
|
|
flattened++;
|
|
if (flattened == __atomic_load_n (array, MEMMODEL_ACQUIRE))
|
|
__atomic_thread_fence (MEMMODEL_RELEASE);
|
|
else
|
|
__atomic_store_n (array, flattened, MEMMODEL_RELEASE);
|
|
return;
|
|
}
|
|
|
|
__atomic_thread_fence (MEMMODEL_ACQUIRE);
|
|
if (sizeof (gomp_ull) == sizeof (unsigned long))
|
|
{
|
|
gomp_ull *array = (gomp_ull *) (doacross->array
|
|
+ ent * doacross->elt_sz);
|
|
|
|
for (i = doacross->ncounts; i-- > 0; )
|
|
{
|
|
if (counts[i] + 1UL != __atomic_load_n (&array[i], MEMMODEL_RELAXED))
|
|
__atomic_store_n (&array[i], counts[i] + 1UL, MEMMODEL_RELEASE);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
unsigned long *array = (unsigned long *) (doacross->array
|
|
+ ent * doacross->elt_sz);
|
|
|
|
for (i = doacross->ncounts; i-- > 0; )
|
|
{
|
|
gomp_ull cull = counts[i] + 1UL;
|
|
unsigned long c = (unsigned long) cull;
|
|
if (c != __atomic_load_n (&array[2 * i + 1], MEMMODEL_RELAXED))
|
|
__atomic_store_n (&array[2 * i + 1], c, MEMMODEL_RELEASE);
|
|
c = cull >> (__SIZEOF_LONG_LONG__ * __CHAR_BIT__ / 2);
|
|
if (c != __atomic_load_n (&array[2 * i], MEMMODEL_RELAXED))
|
|
__atomic_store_n (&array[2 * i], c, MEMMODEL_RELEASE);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* DOACROSS WAIT operation. */
|
|
|
|
void
|
|
GOMP_doacross_ull_wait (gomp_ull first, ...)
|
|
{
|
|
struct gomp_thread *thr = gomp_thread ();
|
|
struct gomp_work_share *ws = thr->ts.work_share;
|
|
struct gomp_doacross_work_share *doacross = ws->doacross;
|
|
va_list ap;
|
|
unsigned long ent;
|
|
unsigned int i;
|
|
|
|
if (__builtin_expect (doacross == NULL, 0))
|
|
{
|
|
__sync_synchronize ();
|
|
return;
|
|
}
|
|
|
|
if (__builtin_expect (ws->sched == GFS_STATIC, 1))
|
|
{
|
|
if (ws->chunk_size_ull == 0)
|
|
{
|
|
if (first < doacross->boundary_ull)
|
|
ent = first / (doacross->q_ull + 1);
|
|
else
|
|
ent = (first - doacross->boundary_ull) / doacross->q_ull
|
|
+ doacross->t;
|
|
}
|
|
else
|
|
ent = first / ws->chunk_size_ull % thr->ts.team->nthreads;
|
|
}
|
|
else if (ws->sched == GFS_GUIDED)
|
|
ent = first;
|
|
else
|
|
ent = first / doacross->chunk_size_ull;
|
|
|
|
if (__builtin_expect (doacross->flattened, 1))
|
|
{
|
|
unsigned long *array = (unsigned long *) (doacross->array
|
|
+ ent * doacross->elt_sz);
|
|
gomp_ull flattened = first << doacross->shift_counts[0];
|
|
unsigned long cur;
|
|
|
|
va_start (ap, first);
|
|
for (i = 1; i < doacross->ncounts; i++)
|
|
flattened |= va_arg (ap, gomp_ull)
|
|
<< doacross->shift_counts[i];
|
|
cur = __atomic_load_n (array, MEMMODEL_ACQUIRE);
|
|
if (flattened < cur)
|
|
{
|
|
__atomic_thread_fence (MEMMODEL_RELEASE);
|
|
va_end (ap);
|
|
return;
|
|
}
|
|
doacross_spin (array, flattened, cur);
|
|
__atomic_thread_fence (MEMMODEL_RELEASE);
|
|
va_end (ap);
|
|
return;
|
|
}
|
|
|
|
if (sizeof (gomp_ull) == sizeof (unsigned long))
|
|
{
|
|
gomp_ull *array = (gomp_ull *) (doacross->array
|
|
+ ent * doacross->elt_sz);
|
|
do
|
|
{
|
|
va_start (ap, first);
|
|
for (i = 0; i < doacross->ncounts; i++)
|
|
{
|
|
gomp_ull thisv
|
|
= (i ? va_arg (ap, gomp_ull) : first) + 1;
|
|
gomp_ull cur = __atomic_load_n (&array[i], MEMMODEL_RELAXED);
|
|
if (thisv < cur)
|
|
{
|
|
i = doacross->ncounts;
|
|
break;
|
|
}
|
|
if (thisv > cur)
|
|
break;
|
|
}
|
|
va_end (ap);
|
|
if (i == doacross->ncounts)
|
|
break;
|
|
cpu_relax ();
|
|
}
|
|
while (1);
|
|
}
|
|
else
|
|
{
|
|
unsigned long *array = (unsigned long *) (doacross->array
|
|
+ ent * doacross->elt_sz);
|
|
do
|
|
{
|
|
va_start (ap, first);
|
|
for (i = 0; i < doacross->ncounts; i++)
|
|
{
|
|
gomp_ull thisv
|
|
= (i ? va_arg (ap, gomp_ull) : first) + 1;
|
|
unsigned long t
|
|
= thisv >> (__SIZEOF_LONG_LONG__ * __CHAR_BIT__ / 2);
|
|
unsigned long cur
|
|
= __atomic_load_n (&array[2 * i], MEMMODEL_RELAXED);
|
|
if (t < cur)
|
|
{
|
|
i = doacross->ncounts;
|
|
break;
|
|
}
|
|
if (t > cur)
|
|
break;
|
|
t = thisv;
|
|
cur = __atomic_load_n (&array[2 * i + 1], MEMMODEL_RELAXED);
|
|
if (t < cur)
|
|
{
|
|
i = doacross->ncounts;
|
|
break;
|
|
}
|
|
if (t > cur)
|
|
break;
|
|
}
|
|
va_end (ap);
|
|
if (i == doacross->ncounts)
|
|
break;
|
|
cpu_relax ();
|
|
}
|
|
while (1);
|
|
}
|
|
__sync_synchronize ();
|
|
}
|