17#include "kmp_wait_release.h"
18#include "kmp_taskdeps.h"
20#include "ompt-specific.h"
32#ifdef KMP_SUPPORT_GRAPH_OUTPUT
33static std::atomic<kmp_int32> kmp_node_id_seed = 0;
36static void __kmp_init_node(kmp_depnode_t *node,
bool on_stack) {
37 node->dn.successors = NULL;
40 for (
int i = 0; i < MAX_MTX_DEPS; ++i)
41 node->dn.mtx_locks[i] = NULL;
42 node->dn.mtx_num_locks = 0;
43 __kmp_init_lock(&node->dn.lock);
48 KMP_ATOMIC_ST_RLX(&node->dn.nrefs, on_stack ? 3 : 2);
49#ifdef KMP_SUPPORT_GRAPH_OUTPUT
50 node->dn.id = KMP_ATOMIC_INC(&kmp_node_id_seed);
52#if USE_ITT_BUILD && USE_ITT_NOTIFY
53 __itt_sync_create(node,
"OMP task dep node", NULL, 0);
57static inline kmp_depnode_t *__kmp_node_ref(kmp_depnode_t *node) {
58 KMP_ATOMIC_ADD(&node->dn.nrefs, 2);
62enum { KMP_DEPHASH_OTHER_SIZE = 97, KMP_DEPHASH_MASTER_SIZE = 997 };
64size_t sizes[] = {997, 2003, 4001, 8191, 16001, 32003, 64007, 131071, 270029};
65const size_t MAX_GEN = 8;
67static inline size_t __kmp_dephash_hash(kmp_intptr_t addr,
size_t hsize) {
70 return ((addr >> 6) ^ (addr >> 2)) % hsize;
73static kmp_dephash_t *__kmp_dephash_extend(kmp_info_t *thread,
74 kmp_dephash_t *current_dephash) {
77 size_t gen = current_dephash->generation + 1;
79 return current_dephash;
80 size_t new_size = sizes[gen];
82 size_t size_to_allocate =
83 new_size *
sizeof(kmp_dephash_entry_t *) +
sizeof(kmp_dephash_t);
86 h = (kmp_dephash_t *)__kmp_fast_allocate(thread, size_to_allocate);
88 h = (kmp_dephash_t *)__kmp_thread_malloc(thread, size_to_allocate);
92 h->nelements = current_dephash->nelements;
93 h->buckets = (kmp_dephash_entry **)(h + 1);
96 h->last_all = current_dephash->last_all;
99 for (
size_t i = 0; i < new_size; i++) {
100 h->buckets[i] = NULL;
104 for (
size_t i = 0; i < current_dephash->size; i++) {
105 kmp_dephash_entry_t *next, *entry;
106 for (entry = current_dephash->buckets[i]; entry; entry = next) {
107 next = entry->next_in_bucket;
110 size_t new_bucket = __kmp_dephash_hash(entry->addr, h->size);
111 entry->next_in_bucket = h->buckets[new_bucket];
112 if (entry->next_in_bucket) {
115 h->buckets[new_bucket] = entry;
121 __kmp_fast_free(thread, current_dephash);
123 __kmp_thread_free(thread, current_dephash);
129static kmp_dephash_t *__kmp_dephash_create(kmp_info_t *thread,
130 kmp_taskdata_t *current_task) {
135 if (current_task->td_flags.tasktype == TASK_IMPLICIT)
136 h_size = KMP_DEPHASH_MASTER_SIZE;
138 h_size = KMP_DEPHASH_OTHER_SIZE;
140 size_t size = h_size *
sizeof(kmp_dephash_entry_t *) +
sizeof(kmp_dephash_t);
143 h = (kmp_dephash_t *)__kmp_fast_allocate(thread, size);
145 h = (kmp_dephash_t *)__kmp_thread_malloc(thread, size);
152 h->buckets = (kmp_dephash_entry **)(h + 1);
155 for (
size_t i = 0; i < h_size; i++)
161static kmp_dephash_entry *__kmp_dephash_find(kmp_info_t *thread,
162 kmp_dephash_t **hash,
164 kmp_dephash_t *h = *hash;
165 if (h->nelements != 0 && h->nconflicts / h->size >= 1) {
166 *hash = __kmp_dephash_extend(thread, h);
169 size_t bucket = __kmp_dephash_hash(addr, h->size);
171 kmp_dephash_entry_t *entry;
172 for (entry = h->buckets[bucket]; entry; entry = entry->next_in_bucket)
173 if (entry->addr == addr)
179 entry = (kmp_dephash_entry_t *)__kmp_fast_allocate(
180 thread,
sizeof(kmp_dephash_entry_t));
182 entry = (kmp_dephash_entry_t *)__kmp_thread_malloc(
183 thread,
sizeof(kmp_dephash_entry_t));
187 entry->last_out = NULL;
189 entry->last_out = __kmp_node_ref(h->last_all);
190 entry->last_set = NULL;
191 entry->prev_set = NULL;
192 entry->last_flag = 0;
193 entry->mtx_lock = NULL;
194 entry->next_in_bucket = h->buckets[bucket];
195 h->buckets[bucket] = entry;
197 if (entry->next_in_bucket)
203static kmp_depnode_list_t *__kmp_add_node(kmp_info_t *thread,
204 kmp_depnode_list_t *list,
205 kmp_depnode_t *node) {
206 kmp_depnode_list_t *new_head;
209 new_head = (kmp_depnode_list_t *)__kmp_fast_allocate(
210 thread,
sizeof(kmp_depnode_list_t));
212 new_head = (kmp_depnode_list_t *)__kmp_thread_malloc(
213 thread,
sizeof(kmp_depnode_list_t));
216 new_head->node = __kmp_node_ref(node);
217 new_head->next = list;
222static inline void __kmp_track_dependence(kmp_int32 gtid, kmp_depnode_t *source,
224 kmp_task_t *sink_task) {
226 kmp_taskdata_t *task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
227 kmp_taskdata_t *task_sink = KMP_TASK_TO_TASKDATA(sink_task);
228 if (source->dn.task && sink_task) {
231 KMP_ASSERT(task_source->is_taskgraph == task_sink->is_taskgraph);
233 if (task_sink->is_taskgraph &&
234 __kmp_tdg_is_recording(task_sink->tdg->tdg_status)) {
235 kmp_node_info_t *source_info =
236 &task_sink->tdg->record_map[task_source->td_tdg_task_id];
238 for (
int i = 0; i < source_info->nsuccessors; i++) {
239 if (source_info->successors[i] == task_sink->td_tdg_task_id) {
245 if (source_info->nsuccessors >= source_info->successors_size) {
246 kmp_uint old_size = source_info->successors_size;
247 source_info->successors_size = 2 * source_info->successors_size;
248 kmp_int32 *old_succ_ids = source_info->successors;
249 kmp_int32 *new_succ_ids = (kmp_int32 *)__kmp_allocate(
250 source_info->successors_size *
sizeof(kmp_int32));
251 KMP_MEMCPY(new_succ_ids, old_succ_ids, old_size *
sizeof(kmp_int32));
252 source_info->successors = new_succ_ids;
253 __kmp_free(old_succ_ids);
256 source_info->successors[source_info->nsuccessors] =
257 task_sink->td_tdg_task_id;
258 source_info->nsuccessors++;
260 kmp_node_info_t *sink_info =
261 &(task_sink->tdg->record_map[task_sink->td_tdg_task_id]);
262 sink_info->npredecessors++;
266#ifdef KMP_SUPPORT_GRAPH_OUTPUT
267 kmp_taskdata_t *task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
270 kmp_taskdata_t *task_sink = KMP_TASK_TO_TASKDATA(sink_task);
272 __kmp_printf(
"%d(%s) -> %d(%s)\n", source->dn.id,
273 task_source->td_ident->psource, sink->dn.id,
274 task_sink->td_ident->psource);
276#if OMPT_SUPPORT && OMPT_OPTIONAL
280 if (ompt_enabled.ompt_callback_task_dependence) {
281 kmp_taskdata_t *task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
282 ompt_data_t *sink_data;
284 sink_data = &(KMP_TASK_TO_TASKDATA(sink_task)->ompt_task_info.task_data);
286 sink_data = &__kmp_threads[gtid]->th.ompt_thread_info.task_data;
288 ompt_callbacks.ompt_callback(ompt_callback_task_dependence)(
289 &(task_source->ompt_task_info.task_data), sink_data);
294kmp_base_depnode_t *__kmpc_task_get_depnode(kmp_task_t *task) {
295 kmp_taskdata_t *td = KMP_TASK_TO_TASKDATA(task);
296 return td->td_depnode ? &(td->td_depnode->dn) : NULL;
299kmp_depnode_list_t *__kmpc_task_get_successors(kmp_task_t *task) {
300 kmp_taskdata_t *td = KMP_TASK_TO_TASKDATA(task);
301 return td->td_depnode->dn.successors;
304static inline kmp_int32
305__kmp_depnode_link_successor(kmp_int32 gtid, kmp_info_t *thread,
306 kmp_task_t *task, kmp_depnode_t *node,
307 kmp_depnode_list_t *plist) {
310 kmp_int32 npredecessors = 0;
312 for (kmp_depnode_list_t *p = plist; p; p = p->next) {
313 kmp_depnode_t *dep = p->node;
315 kmp_tdg_status tdg_status = KMP_TDG_NONE;
317 kmp_taskdata_t *td = KMP_TASK_TO_TASKDATA(task);
318 if (td->is_taskgraph)
319 tdg_status = KMP_TASK_TO_TASKDATA(task)->tdg->tdg_status;
320 if (__kmp_tdg_is_recording(tdg_status))
321 __kmp_track_dependence(gtid, dep, node, task);
325 KMP_ACQUIRE_DEPNODE(gtid, dep);
327 if (!dep->dn.successors || dep->dn.successors->node != node) {
329 if (!(__kmp_tdg_is_recording(tdg_status)) && task)
331 __kmp_track_dependence(gtid, dep, node, task);
332 dep->dn.successors = __kmp_add_node(thread, dep->dn.successors, node);
333 KA_TRACE(40, (
"__kmp_process_deps: T#%d adding dependence from %p to "
335 gtid, KMP_TASK_TO_TASKDATA(dep->dn.task),
336 KMP_TASK_TO_TASKDATA(task)));
340 KMP_RELEASE_DEPNODE(gtid, dep);
343 return npredecessors;
347static inline kmp_int32 __kmp_depnode_link_successor(kmp_int32 gtid,
350 kmp_depnode_t *source,
351 kmp_depnode_t *sink) {
354 kmp_int32 npredecessors = 0;
356 kmp_tdg_status tdg_status = KMP_TDG_NONE;
357 kmp_taskdata_t *td = KMP_TASK_TO_TASKDATA(task);
359 if (td->is_taskgraph)
360 tdg_status = KMP_TASK_TO_TASKDATA(task)->tdg->tdg_status;
361 if (__kmp_tdg_is_recording(tdg_status) && sink->dn.task)
362 __kmp_track_dependence(gtid, sink, source, task);
367 KMP_ACQUIRE_DEPNODE(gtid, sink);
369 if (!sink->dn.successors || sink->dn.successors->node != source) {
371 if (!(__kmp_tdg_is_recording(tdg_status)) && task)
373 __kmp_track_dependence(gtid, sink, source, task);
374 sink->dn.successors = __kmp_add_node(thread, sink->dn.successors, source);
375 KA_TRACE(40, (
"__kmp_process_deps: T#%d adding dependence from %p to "
377 gtid, KMP_TASK_TO_TASKDATA(sink->dn.task),
378 KMP_TASK_TO_TASKDATA(task)));
380 if (__kmp_tdg_is_recording(tdg_status)) {
381 kmp_taskdata_t *tdd = KMP_TASK_TO_TASKDATA(sink->dn.task);
382 if (tdd->is_taskgraph) {
383 if (tdd->td_flags.onced)
395 KMP_RELEASE_DEPNODE(gtid, sink);
397 return npredecessors;
400static inline kmp_int32
401__kmp_process_dep_all(kmp_int32 gtid, kmp_depnode_t *node, kmp_dephash_t *h,
402 bool dep_barrier, kmp_task_t *task) {
403 KA_TRACE(30, (
"__kmp_process_dep_all: T#%d processing dep_all, "
404 "dep_barrier = %d\n",
406 kmp_info_t *thread = __kmp_threads[gtid];
407 kmp_int32 npredecessors = 0;
411 __kmp_depnode_link_successor(gtid, thread, task, node, h->last_all);
412 __kmp_node_deref(thread, h->last_all);
414 h->last_all = __kmp_node_ref(node);
423 for (
size_t i = 0; i < h->size; i++) {
424 kmp_dephash_entry_t *info = h->buckets[i];
427 for (; info; info = info->next_in_bucket) {
429 kmp_depnode_t *last_out = info->last_out;
430 kmp_depnode_list_t *last_set = info->last_set;
431 kmp_depnode_list_t *prev_set = info->prev_set;
434 __kmp_depnode_link_successor(gtid, thread, task, node, last_set);
435 __kmp_depnode_list_free(thread, last_set);
436 __kmp_depnode_list_free(thread, prev_set);
437 info->last_set = NULL;
438 info->prev_set = NULL;
442 __kmp_depnode_link_successor(gtid, thread, task, node, last_out);
444 __kmp_node_deref(thread, last_out);
446 info->last_out = __kmp_node_ref(node);
448 info->last_out = NULL;
452 KA_TRACE(30, (
"__kmp_process_dep_all: T#%d found %d predecessors\n", gtid,
454 return npredecessors;
457template <
bool filter>
458static inline kmp_int32
459__kmp_process_deps(kmp_int32 gtid, kmp_depnode_t *node, kmp_dephash_t **hash,
460 bool dep_barrier, kmp_int32 ndeps,
461 kmp_depend_info_t *dep_list, kmp_task_t *task) {
462 KA_TRACE(30, (
"__kmp_process_deps<%d>: T#%d processing %d dependences : "
463 "dep_barrier = %d\n",
464 filter, gtid, ndeps, dep_barrier));
466 kmp_info_t *thread = __kmp_threads[gtid];
467 kmp_int32 npredecessors = 0;
468 for (kmp_int32 i = 0; i < ndeps; i++) {
469 const kmp_depend_info_t *dep = &dep_list[i];
471 if (filter && dep->base_addr == 0)
474 kmp_dephash_entry_t *info =
475 __kmp_dephash_find(thread, hash, dep->base_addr);
476 kmp_depnode_t *last_out = info->last_out;
477 kmp_depnode_list_t *last_set = info->last_set;
478 kmp_depnode_list_t *prev_set = info->prev_set;
480 if (dep->flags.out) {
483 __kmp_depnode_link_successor(gtid, thread, task, node, last_set);
484 __kmp_depnode_list_free(thread, last_set);
485 __kmp_depnode_list_free(thread, prev_set);
486 info->last_set = NULL;
487 info->prev_set = NULL;
491 __kmp_depnode_link_successor(gtid, thread, task, node, last_out);
493 __kmp_node_deref(thread, last_out);
495 info->last_out = __kmp_node_ref(node);
500 info->last_out = NULL;
503 if (info->last_flag == 0 || info->last_flag == dep->flag) {
507 __kmp_depnode_link_successor(gtid, thread, task, node, last_out);
510 __kmp_depnode_link_successor(gtid, thread, task, node, prev_set);
513 __kmp_node_deref(thread, last_out);
514 info->last_out = NULL;
515 __kmp_depnode_list_free(thread, prev_set);
516 info->prev_set = NULL;
521 __kmp_depnode_link_successor(gtid, thread, task, node, last_set);
523 __kmp_node_deref(thread, last_out);
524 info->last_out = NULL;
526 __kmp_depnode_list_free(thread, prev_set);
529 info->prev_set = last_set;
531 info->prev_set = NULL;
534 info->last_set = NULL;
539 info->last_flag = dep->flag;
540 info->last_set = __kmp_add_node(thread, info->last_set, node);
543 if (dep->flag == KMP_DEP_MTX) {
544 if (info->mtx_lock == NULL) {
545 info->mtx_lock = (kmp_lock_t *)__kmp_allocate(
sizeof(kmp_lock_t));
546 __kmp_init_lock(info->mtx_lock);
548 KMP_DEBUG_ASSERT(node->dn.mtx_num_locks < MAX_MTX_DEPS);
551 for (m = 0; m < MAX_MTX_DEPS; ++m) {
553 if (node->dn.mtx_locks[m] < info->mtx_lock) {
554 KMP_DEBUG_ASSERT(!node->dn.mtx_locks[node->dn.mtx_num_locks]);
555 for (
int n = node->dn.mtx_num_locks; n > m; --n) {
557 KMP_DEBUG_ASSERT(node->dn.mtx_locks[n - 1] != NULL);
558 node->dn.mtx_locks[n] = node->dn.mtx_locks[n - 1];
560 node->dn.mtx_locks[m] = info->mtx_lock;
564 KMP_DEBUG_ASSERT(m < MAX_MTX_DEPS);
565 node->dn.mtx_num_locks++;
569 KA_TRACE(30, (
"__kmp_process_deps<%d>: T#%d found %d predecessors\n", filter,
570 gtid, npredecessors));
571 return npredecessors;
574#define NO_DEP_BARRIER (false)
575#define DEP_BARRIER (true)
578static bool __kmp_check_deps(kmp_int32 gtid, kmp_depnode_t *node,
579 kmp_task_t *task, kmp_dephash_t **hash,
580 bool dep_barrier, kmp_int32 ndeps,
581 kmp_depend_info_t *dep_list,
582 kmp_int32 ndeps_noalias,
583 kmp_depend_info_t *noalias_dep_list) {
584 int i, n_mtxs = 0, dep_all = 0;
586 kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task);
588 KA_TRACE(20, (
"__kmp_check_deps: T#%d checking dependences for task %p : %d "
589 "possibly aliased dependences, %d non-aliased dependences : "
590 "dep_barrier=%d .\n",
591 gtid, taskdata, ndeps, ndeps_noalias, dep_barrier));
595 for (i = 0; i < ndeps; i++) {
596 if (dep_list[i].base_addr != 0 &&
597 dep_list[i].base_addr != (kmp_intptr_t)KMP_SIZE_T_MAX) {
599 dep_list[i].flag == KMP_DEP_IN || dep_list[i].flag == KMP_DEP_OUT ||
600 dep_list[i].flag == KMP_DEP_INOUT ||
601 dep_list[i].flag == KMP_DEP_MTX || dep_list[i].flag == KMP_DEP_SET);
602 for (
int j = i + 1; j < ndeps; j++) {
603 if (dep_list[i].base_addr == dep_list[j].base_addr) {
604 if (dep_list[i].flag != dep_list[j].flag) {
606 dep_list[i].flag = KMP_DEP_OUT;
608 dep_list[j].base_addr = 0;
611 if (dep_list[i].flag == KMP_DEP_MTX) {
613 if (n_mtxs < MAX_MTX_DEPS && task != NULL) {
616 dep_list[i].flag = KMP_DEP_OUT;
619 }
else if (dep_list[i].flag == KMP_DEP_ALL ||
620 dep_list[i].base_addr == (kmp_intptr_t)KMP_SIZE_T_MAX) {
633 node->dn.npredecessors = -1;
640 npredecessors = __kmp_process_deps<true>(gtid, node, hash, dep_barrier,
641 ndeps, dep_list, task);
642 npredecessors += __kmp_process_deps<false>(
643 gtid, node, hash, dep_barrier, ndeps_noalias, noalias_dep_list, task);
645 npredecessors = __kmp_process_dep_all(gtid, node, *hash, dep_barrier, task);
648 node->dn.task = task;
658 node->dn.npredecessors.fetch_add(npredecessors) + npredecessors;
660 KA_TRACE(20, (
"__kmp_check_deps: T#%d found %d predecessors for task %p \n",
661 gtid, npredecessors, taskdata));
665 return npredecessors > 0 ? true :
false;
685 kmp_task_t *new_task, kmp_int32 ndeps,
686 kmp_depend_info_t *dep_list,
687 kmp_int32 ndeps_noalias,
688 kmp_depend_info_t *noalias_dep_list) {
690 kmp_taskdata_t *new_taskdata = KMP_TASK_TO_TASKDATA(new_task);
691 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(enter): T#%d loc=%p task=%p\n", gtid,
692 loc_ref, new_taskdata));
693 __kmp_assert_valid_gtid(gtid);
694 kmp_info_t *thread = __kmp_threads[gtid];
695 kmp_taskdata_t *current_task = thread->th.th_current_task;
699 if (new_taskdata->is_taskgraph &&
700 __kmp_tdg_is_recording(new_taskdata->tdg->tdg_status)) {
701 kmp_tdg_info_t *tdg = new_taskdata->tdg;
703 if (new_taskdata->td_tdg_task_id >= tdg->map_size) {
704 __kmp_acquire_bootstrap_lock(&tdg->graph_lock);
705 if (new_taskdata->td_tdg_task_id >= tdg->map_size) {
706 kmp_uint old_size = tdg->map_size;
707 kmp_uint new_size = old_size * 2;
708 kmp_node_info_t *old_record = tdg->record_map;
709 kmp_node_info_t *new_record = (kmp_node_info_t *)__kmp_allocate(
710 new_size *
sizeof(kmp_node_info_t));
711 KMP_MEMCPY(new_record, tdg->record_map,
712 old_size *
sizeof(kmp_node_info_t));
713 tdg->record_map = new_record;
715 __kmp_free(old_record);
717 for (kmp_int i = old_size; i < new_size; i++) {
718 kmp_int32 *successorsList = (kmp_int32 *)__kmp_allocate(
719 __kmp_successors_size *
sizeof(kmp_int32));
720 new_record[i].task =
nullptr;
721 new_record[i].successors = successorsList;
722 new_record[i].nsuccessors = 0;
723 new_record[i].npredecessors = 0;
724 new_record[i].successors_size = __kmp_successors_size;
725 KMP_ATOMIC_ST_REL(&new_record[i].npredecessors_counter, 0);
729 tdg->map_size = new_size;
731 __kmp_release_bootstrap_lock(&tdg->graph_lock);
733 tdg->record_map[new_taskdata->td_tdg_task_id].task = new_task;
734 tdg->record_map[new_taskdata->td_tdg_task_id].parent_task =
735 new_taskdata->td_parent;
736 KMP_ATOMIC_INC(&tdg->num_tasks);
740 if (ompt_enabled.enabled) {
741 if (!current_task->ompt_task_info.frame.enter_frame.ptr)
742 current_task->ompt_task_info.frame.enter_frame.ptr =
743 OMPT_GET_FRAME_ADDRESS(0);
744 if (ompt_enabled.ompt_callback_task_create) {
745 ompt_callbacks.ompt_callback(ompt_callback_task_create)(
746 &(current_task->ompt_task_info.task_data),
747 &(current_task->ompt_task_info.frame),
748 &(new_taskdata->ompt_task_info.task_data),
749 TASK_TYPE_DETAILS_FORMAT(new_taskdata), 1,
750 OMPT_LOAD_OR_GET_RETURN_ADDRESS(gtid));
753 new_taskdata->ompt_task_info.frame.enter_frame.ptr =
754 OMPT_GET_FRAME_ADDRESS(0);
759 if (ndeps + ndeps_noalias > 0 && ompt_enabled.ompt_callback_dependences) {
762 int ompt_ndeps = ndeps + ndeps_noalias;
763 ompt_dependence_t *ompt_deps = (ompt_dependence_t *)KMP_OMPT_DEPS_ALLOC(
764 thread, (ndeps + ndeps_noalias) *
sizeof(ompt_dependence_t));
766 KMP_ASSERT(ompt_deps != NULL);
768 for (i = 0; i < ndeps; i++) {
769 ompt_deps[i].variable.ptr = (
void *)dep_list[i].base_addr;
770 if (dep_list[i].base_addr == (kmp_intptr_t)KMP_SIZE_T_MAX)
771 ompt_deps[i].dependence_type = ompt_dependence_type_out_all_memory;
772 else if (dep_list[i].flags.in && dep_list[i].flags.out)
773 ompt_deps[i].dependence_type = ompt_dependence_type_inout;
774 else if (dep_list[i].flags.out)
775 ompt_deps[i].dependence_type = ompt_dependence_type_out;
776 else if (dep_list[i].flags.in)
777 ompt_deps[i].dependence_type = ompt_dependence_type_in;
778 else if (dep_list[i].flags.mtx)
779 ompt_deps[i].dependence_type = ompt_dependence_type_mutexinoutset;
780 else if (dep_list[i].flags.set)
781 ompt_deps[i].dependence_type = ompt_dependence_type_inoutset;
782 else if (dep_list[i].flags.all)
783 ompt_deps[i].dependence_type = ompt_dependence_type_out_all_memory;
785 for (i = 0; i < ndeps_noalias; i++) {
786 ompt_deps[ndeps + i].variable.ptr = (
void *)noalias_dep_list[i].base_addr;
787 if (noalias_dep_list[i].base_addr == (kmp_intptr_t)KMP_SIZE_T_MAX)
788 ompt_deps[ndeps + i].dependence_type =
789 ompt_dependence_type_out_all_memory;
790 else if (noalias_dep_list[i].flags.in && noalias_dep_list[i].flags.out)
791 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_inout;
792 else if (noalias_dep_list[i].flags.out)
793 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_out;
794 else if (noalias_dep_list[i].flags.in)
795 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_in;
796 else if (noalias_dep_list[i].flags.mtx)
797 ompt_deps[ndeps + i].dependence_type =
798 ompt_dependence_type_mutexinoutset;
799 else if (noalias_dep_list[i].flags.set)
800 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_inoutset;
801 else if (noalias_dep_list[i].flags.all)
802 ompt_deps[ndeps + i].dependence_type =
803 ompt_dependence_type_out_all_memory;
805 ompt_callbacks.ompt_callback(ompt_callback_dependences)(
806 &(new_taskdata->ompt_task_info.task_data), ompt_deps, ompt_ndeps);
809 KMP_OMPT_DEPS_FREE(thread, ompt_deps);
814 bool serial = current_task->td_flags.team_serial ||
815 current_task->td_flags.tasking_ser ||
816 current_task->td_flags.final;
817 kmp_task_team_t *task_team = thread->th.th_task_team;
819 !(task_team && (task_team->tt.tt_found_proxy_tasks ||
820 task_team->tt.tt_hidden_helper_task_encountered));
822 if (!serial && (ndeps > 0 || ndeps_noalias > 0)) {
824 if (current_task->td_dephash == NULL)
825 current_task->td_dephash = __kmp_dephash_create(thread, current_task);
828 kmp_depnode_t *node =
829 (kmp_depnode_t *)__kmp_fast_allocate(thread,
sizeof(kmp_depnode_t));
831 kmp_depnode_t *node =
832 (kmp_depnode_t *)__kmp_thread_malloc(thread,
sizeof(kmp_depnode_t));
835 __kmp_init_node(node,
false);
836 new_taskdata->td_depnode = node;
838 if (__kmp_check_deps(gtid, node, new_task, ¤t_task->td_dephash,
839 NO_DEP_BARRIER, ndeps, dep_list, ndeps_noalias,
841 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d task had blocking "
843 "loc=%p task=%p, return: TASK_CURRENT_NOT_QUEUED\n",
844 gtid, loc_ref, new_taskdata));
846 if (ompt_enabled.enabled) {
847 current_task->ompt_task_info.frame.enter_frame = ompt_data_none;
850 return TASK_CURRENT_NOT_QUEUED;
853 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d ignored dependences "
854 "for task (serialized) loc=%p task=%p\n",
855 gtid, loc_ref, new_taskdata));
858 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d task had no blocking "
860 "loc=%p task=%p, transferring to __kmp_omp_task\n",
861 gtid, loc_ref, new_taskdata));
863 kmp_int32 ret = __kmp_omp_task(gtid, new_task,
true);
865 if (ompt_enabled.enabled) {
866 current_task->ompt_task_info.frame.enter_frame = ompt_data_none;
873void __ompt_taskwait_dep_finish(kmp_taskdata_t *current_task,
874 ompt_data_t *taskwait_task_data) {
875 if (ompt_enabled.ompt_callback_task_schedule) {
876 ompt_callbacks.ompt_callback(ompt_callback_task_schedule)(
877 taskwait_task_data, ompt_taskwait_complete, NULL);
879 current_task->ompt_task_info.frame.enter_frame.ptr = NULL;
880 *taskwait_task_data = ompt_data_none;
896 kmp_depend_info_t *dep_list, kmp_int32 ndeps_noalias,
897 kmp_depend_info_t *noalias_dep_list) {
898 __kmpc_omp_taskwait_deps_51(loc_ref, gtid, ndeps, dep_list, ndeps_noalias,
899 noalias_dep_list,
false);
907void __kmpc_omp_taskwait_deps_51(
ident_t *loc_ref, kmp_int32 gtid,
908 kmp_int32 ndeps, kmp_depend_info_t *dep_list,
909 kmp_int32 ndeps_noalias,
910 kmp_depend_info_t *noalias_dep_list,
911 kmp_int32 has_no_wait) {
912 KA_TRACE(10, (
"__kmpc_omp_taskwait_deps(enter): T#%d loc=%p nowait#%d\n",
913 gtid, loc_ref, has_no_wait));
914 if (ndeps == 0 && ndeps_noalias == 0) {
915 KA_TRACE(10, (
"__kmpc_omp_taskwait_deps(exit): T#%d has no dependences to "
916 "wait upon : loc=%p\n",
920 __kmp_assert_valid_gtid(gtid);
921 kmp_info_t *thread = __kmp_threads[gtid];
922 kmp_taskdata_t *current_task = thread->th.th_current_task;
930 ompt_data_t *taskwait_task_data = &thread->th.ompt_thread_info.task_data;
931 KMP_ASSERT(taskwait_task_data->ptr == NULL);
932 if (ompt_enabled.enabled) {
933 if (!current_task->ompt_task_info.frame.enter_frame.ptr)
934 current_task->ompt_task_info.frame.enter_frame.ptr =
935 OMPT_GET_FRAME_ADDRESS(0);
936 if (ompt_enabled.ompt_callback_task_create) {
937 ompt_callbacks.ompt_callback(ompt_callback_task_create)(
938 &(current_task->ompt_task_info.task_data),
939 &(current_task->ompt_task_info.frame), taskwait_task_data,
940 ompt_task_taskwait | ompt_task_undeferred | ompt_task_mergeable, 1,
941 OMPT_LOAD_OR_GET_RETURN_ADDRESS(gtid));
947 if (ndeps + ndeps_noalias > 0 && ompt_enabled.ompt_callback_dependences) {
950 int ompt_ndeps = ndeps + ndeps_noalias;
951 ompt_dependence_t *ompt_deps = (ompt_dependence_t *)KMP_OMPT_DEPS_ALLOC(
952 thread, (ndeps + ndeps_noalias) *
sizeof(ompt_dependence_t));
954 KMP_ASSERT(ompt_deps != NULL);
956 for (i = 0; i < ndeps; i++) {
957 ompt_deps[i].variable.ptr = (
void *)dep_list[i].base_addr;
958 if (dep_list[i].flags.in && dep_list[i].flags.out)
959 ompt_deps[i].dependence_type = ompt_dependence_type_inout;
960 else if (dep_list[i].flags.out)
961 ompt_deps[i].dependence_type = ompt_dependence_type_out;
962 else if (dep_list[i].flags.in)
963 ompt_deps[i].dependence_type = ompt_dependence_type_in;
964 else if (dep_list[i].flags.mtx)
965 ompt_deps[ndeps + i].dependence_type =
966 ompt_dependence_type_mutexinoutset;
967 else if (dep_list[i].flags.set)
968 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_inoutset;
970 for (i = 0; i < ndeps_noalias; i++) {
971 ompt_deps[ndeps + i].variable.ptr = (
void *)noalias_dep_list[i].base_addr;
972 if (noalias_dep_list[i].flags.in && noalias_dep_list[i].flags.out)
973 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_inout;
974 else if (noalias_dep_list[i].flags.out)
975 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_out;
976 else if (noalias_dep_list[i].flags.in)
977 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_in;
978 else if (noalias_dep_list[i].flags.mtx)
979 ompt_deps[ndeps + i].dependence_type =
980 ompt_dependence_type_mutexinoutset;
981 else if (noalias_dep_list[i].flags.set)
982 ompt_deps[ndeps + i].dependence_type = ompt_dependence_type_inoutset;
984 ompt_callbacks.ompt_callback(ompt_callback_dependences)(
985 taskwait_task_data, ompt_deps, ompt_ndeps);
988 KMP_OMPT_DEPS_FREE(thread, ompt_deps);
997 bool ignore = current_task->td_flags.team_serial ||
998 current_task->td_flags.tasking_ser ||
999 current_task->td_flags.final;
1001 ignore && thread->th.th_task_team != NULL &&
1002 thread->th.th_task_team->tt.tt_found_proxy_tasks == FALSE &&
1003 thread->th.th_task_team->tt.tt_hidden_helper_task_encountered == FALSE;
1004 ignore = ignore || current_task->td_dephash == NULL;
1007 KA_TRACE(10, (
"__kmpc_omp_taskwait_deps(exit): T#%d has no blocking "
1008 "dependences : loc=%p\n",
1011 __ompt_taskwait_dep_finish(current_task, taskwait_task_data);
1016 kmp_depnode_t node = {0};
1017 __kmp_init_node(&node,
true);
1019 if (!__kmp_check_deps(gtid, &node, NULL, ¤t_task->td_dephash,
1020 DEP_BARRIER, ndeps, dep_list, ndeps_noalias,
1021 noalias_dep_list)) {
1022 KA_TRACE(10, (
"__kmpc_omp_taskwait_deps(exit): T#%d has no blocking "
1023 "dependences : loc=%p\n",
1026 __ompt_taskwait_dep_finish(current_task, taskwait_task_data);
1032 while ((nrefs = node.dn.nrefs) > 3) {
1033 KMP_DEBUG_ASSERT((nrefs & 1) == 1);
1036 KMP_DEBUG_ASSERT(nrefs == 3);
1041 int thread_finished = FALSE;
1042 kmp_flag_32<false, false> flag(
1043 (std::atomic<kmp_uint32> *)&node.dn.npredecessors, 0U);
1044 while (node.dn.npredecessors > 0) {
1045 flag.execute_tasks(thread, gtid, FALSE,
1046 &thread_finished USE_ITT_BUILD_ARG(NULL),
1047 __kmp_task_stealing_constraint);
1056 while ((nrefs = node.dn.nrefs) > 3) {
1057 KMP_DEBUG_ASSERT((nrefs & 1) == 1);
1060 KMP_DEBUG_ASSERT(nrefs == 3);
1063 __ompt_taskwait_dep_finish(current_task, taskwait_task_data);
1065 KA_TRACE(10, (
"__kmpc_omp_taskwait_deps(exit): T#%d finished waiting : loc=%p\
kmp_int32 __kmpc_omp_task_with_deps(ident_t *loc_ref, kmp_int32 gtid, kmp_task_t *new_task, kmp_int32 ndeps, kmp_depend_info_t *dep_list, kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list)
void __kmpc_omp_wait_deps(ident_t *loc_ref, kmp_int32 gtid, kmp_int32 ndeps, kmp_depend_info_t *dep_list, kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list)