blob: 1b4e86940ad5655a21a82e14ac5e2965555d9bc0 [file] [log] [blame]
Jim Cownie5e8470a2013-09-27 10:38:44 +00001/*
2 * kmp_taskdeps.cpp
Jim Cownie5e8470a2013-09-27 10:38:44 +00003 */
4
5
6//===----------------------------------------------------------------------===//
7//
8// The LLVM Compiler Infrastructure
9//
10// This file is dual licensed under the MIT and the University of Illinois Open
11// Source Licenses. See LICENSE.txt for details.
12//
13//===----------------------------------------------------------------------===//
14
15
16//#define KMP_SUPPORT_GRAPH_OUTPUT 1
17
18#include "kmp.h"
19#include "kmp_io.h"
Jim Cownie4cc4bb42014-10-07 16:25:50 +000020#include "kmp_wait_release.h"
Jim Cownie5e8470a2013-09-27 10:38:44 +000021
22#if OMP_40_ENABLED
23
Jonathan Peyton30419822017-05-12 18:01:32 +000024// TODO: Improve memory allocation? keep a list of pre-allocated structures?
25// allocate in blocks? re-use list finished list entries?
26// TODO: don't use atomic ref counters for stack-allocated nodes.
27// TODO: find an alternate to atomic refs for heap-allocated nodes?
28// TODO: Finish graph output support
29// TODO: kmp_lock_t seems a tad to big (and heavy weight) for this. Check other
30// runtime locks
31// TODO: Any ITT support needed?
Jim Cownie5e8470a2013-09-27 10:38:44 +000032
33#ifdef KMP_SUPPORT_GRAPH_OUTPUT
34static kmp_int32 kmp_node_id_seed = 0;
35#endif
36
Jonathan Peyton30419822017-05-12 18:01:32 +000037static void __kmp_init_node(kmp_depnode_t *node) {
38 node->dn.task = NULL; // set to null initially, it will point to the right
39 // task once dependences have been processed
40 node->dn.successors = NULL;
41 __kmp_init_lock(&node->dn.lock);
42 node->dn.nrefs = 1; // init creates the first reference to the node
Jim Cownie5e8470a2013-09-27 10:38:44 +000043#ifdef KMP_SUPPORT_GRAPH_OUTPUT
Jonathan Peyton30419822017-05-12 18:01:32 +000044 node->dn.id = KMP_TEST_THEN_INC32(&kmp_node_id_seed);
Jim Cownie5e8470a2013-09-27 10:38:44 +000045#endif
46}
47
Jonathan Peyton30419822017-05-12 18:01:32 +000048static inline kmp_depnode_t *__kmp_node_ref(kmp_depnode_t *node) {
49 KMP_TEST_THEN_INC32(&node->dn.nrefs);
50 return node;
Jim Cownie5e8470a2013-09-27 10:38:44 +000051}
52
Jonathan Peyton30419822017-05-12 18:01:32 +000053static inline void __kmp_node_deref(kmp_info_t *thread, kmp_depnode_t *node) {
54 if (!node)
55 return;
Jim Cownie5e8470a2013-09-27 10:38:44 +000056
Jonathan Peyton30419822017-05-12 18:01:32 +000057 kmp_int32 n = KMP_TEST_THEN_DEC32(&node->dn.nrefs) - 1;
58 if (n == 0) {
59 KMP_ASSERT(node->dn.nrefs == 0);
Jim Cownie5e8470a2013-09-27 10:38:44 +000060#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +000061 __kmp_fast_free(thread, node);
Jim Cownie5e8470a2013-09-27 10:38:44 +000062#else
Jonathan Peyton30419822017-05-12 18:01:32 +000063 __kmp_thread_free(thread, node);
Jim Cownie5e8470a2013-09-27 10:38:44 +000064#endif
Jonathan Peyton30419822017-05-12 18:01:32 +000065 }
Jim Cownie5e8470a2013-09-27 10:38:44 +000066}
67
Jonathan Peyton30419822017-05-12 18:01:32 +000068#define KMP_ACQUIRE_DEPNODE(gtid, n) __kmp_acquire_lock(&(n)->dn.lock, (gtid))
69#define KMP_RELEASE_DEPNODE(gtid, n) __kmp_release_lock(&(n)->dn.lock, (gtid))
Jim Cownie5e8470a2013-09-27 10:38:44 +000070
Jonathan Peyton30419822017-05-12 18:01:32 +000071static void __kmp_depnode_list_free(kmp_info_t *thread, kmp_depnode_list *list);
Jim Cownie5e8470a2013-09-27 10:38:44 +000072
Jonathan Peyton30419822017-05-12 18:01:32 +000073enum { KMP_DEPHASH_OTHER_SIZE = 97, KMP_DEPHASH_MASTER_SIZE = 997 };
Jim Cownie5e8470a2013-09-27 10:38:44 +000074
Jonathan Peyton30419822017-05-12 18:01:32 +000075static inline kmp_int32 __kmp_dephash_hash(kmp_intptr_t addr, size_t hsize) {
76 // TODO alternate to try: set = (((Addr64)(addrUsefulBits * 9.618)) %
77 // m_num_sets );
78 return ((addr >> 6) ^ (addr >> 2)) % hsize;
Jim Cownie5e8470a2013-09-27 10:38:44 +000079}
80
Jonathan Peyton30419822017-05-12 18:01:32 +000081static kmp_dephash_t *__kmp_dephash_create(kmp_info_t *thread,
82 kmp_taskdata_t *current_task) {
83 kmp_dephash_t *h;
Jim Cownie4cc4bb42014-10-07 16:25:50 +000084
Jonathan Peyton30419822017-05-12 18:01:32 +000085 size_t h_size;
Jonathan Peyton7d454512016-01-28 23:10:44 +000086
Jonathan Peyton30419822017-05-12 18:01:32 +000087 if (current_task->td_flags.tasktype == TASK_IMPLICIT)
88 h_size = KMP_DEPHASH_MASTER_SIZE;
89 else
90 h_size = KMP_DEPHASH_OTHER_SIZE;
Jonathan Peyton7d454512016-01-28 23:10:44 +000091
Jonathan Peyton30419822017-05-12 18:01:32 +000092 kmp_int32 size =
93 h_size * sizeof(kmp_dephash_entry_t *) + sizeof(kmp_dephash_t);
Jim Cownie4cc4bb42014-10-07 16:25:50 +000094
Jim Cownie5e8470a2013-09-27 10:38:44 +000095#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +000096 h = (kmp_dephash_t *)__kmp_fast_allocate(thread, size);
Jim Cownie5e8470a2013-09-27 10:38:44 +000097#else
Jonathan Peyton30419822017-05-12 18:01:32 +000098 h = (kmp_dephash_t *)__kmp_thread_malloc(thread, size);
Jim Cownie5e8470a2013-09-27 10:38:44 +000099#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000100 h->size = h_size;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000101
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000102#ifdef KMP_DEBUG
Jonathan Peyton30419822017-05-12 18:01:32 +0000103 h->nelements = 0;
104 h->nconflicts = 0;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000105#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000106 h->buckets = (kmp_dephash_entry **)(h + 1);
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000107
Jonathan Peyton30419822017-05-12 18:01:32 +0000108 for (size_t i = 0; i < h_size; i++)
109 h->buckets[i] = 0;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000110
Jonathan Peyton30419822017-05-12 18:01:32 +0000111 return h;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000112}
113
Jonathan Peyton30419822017-05-12 18:01:32 +0000114void __kmp_dephash_free_entries(kmp_info_t *thread, kmp_dephash_t *h) {
115 for (size_t i = 0; i < h->size; i++) {
116 if (h->buckets[i]) {
117 kmp_dephash_entry_t *next;
118 for (kmp_dephash_entry_t *entry = h->buckets[i]; entry; entry = next) {
119 next = entry->next_in_bucket;
120 __kmp_depnode_list_free(thread, entry->last_ins);
121 __kmp_node_deref(thread, entry->last_out);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000122#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +0000123 __kmp_fast_free(thread, entry);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000124#else
Jonathan Peyton30419822017-05-12 18:01:32 +0000125 __kmp_thread_free(thread, entry);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000126#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000127 }
128 h->buckets[i] = 0;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000129 }
Jonathan Peyton30419822017-05-12 18:01:32 +0000130 }
Andrey Churbanovdf0d75e2016-10-27 11:43:07 +0000131}
132
Jonathan Peyton30419822017-05-12 18:01:32 +0000133void __kmp_dephash_free(kmp_info_t *thread, kmp_dephash_t *h) {
134 __kmp_dephash_free_entries(thread, h);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000135#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +0000136 __kmp_fast_free(thread, h);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000137#else
Jonathan Peyton30419822017-05-12 18:01:32 +0000138 __kmp_thread_free(thread, h);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000139#endif
140}
141
142static kmp_dephash_entry *
Jonathan Peyton30419822017-05-12 18:01:32 +0000143__kmp_dephash_find(kmp_info_t *thread, kmp_dephash_t *h, kmp_intptr_t addr) {
144 kmp_int32 bucket = __kmp_dephash_hash(addr, h->size);
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000145
Jonathan Peyton30419822017-05-12 18:01:32 +0000146 kmp_dephash_entry_t *entry;
147 for (entry = h->buckets[bucket]; entry; entry = entry->next_in_bucket)
148 if (entry->addr == addr)
149 break;
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000150
Jonathan Peyton30419822017-05-12 18:01:32 +0000151 if (entry == NULL) {
152// create entry. This is only done by one thread so no locking required
Jim Cownie5e8470a2013-09-27 10:38:44 +0000153#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +0000154 entry = (kmp_dephash_entry_t *)__kmp_fast_allocate(
155 thread, sizeof(kmp_dephash_entry_t));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000156#else
Jonathan Peyton30419822017-05-12 18:01:32 +0000157 entry = (kmp_dephash_entry_t *)__kmp_thread_malloc(
158 thread, sizeof(kmp_dephash_entry_t));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000159#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000160 entry->addr = addr;
161 entry->last_out = NULL;
162 entry->last_ins = NULL;
163 entry->next_in_bucket = h->buckets[bucket];
164 h->buckets[bucket] = entry;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000165#ifdef KMP_DEBUG
Jonathan Peyton30419822017-05-12 18:01:32 +0000166 h->nelements++;
167 if (entry->next_in_bucket)
168 h->nconflicts++;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000169#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000170 }
171 return entry;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000172}
173
Jonathan Peyton30419822017-05-12 18:01:32 +0000174static kmp_depnode_list_t *__kmp_add_node(kmp_info_t *thread,
175 kmp_depnode_list_t *list,
176 kmp_depnode_t *node) {
177 kmp_depnode_list_t *new_head;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000178
179#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +0000180 new_head = (kmp_depnode_list_t *)__kmp_fast_allocate(
181 thread, sizeof(kmp_depnode_list_t));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000182#else
Jonathan Peyton30419822017-05-12 18:01:32 +0000183 new_head = (kmp_depnode_list_t *)__kmp_thread_malloc(
184 thread, sizeof(kmp_depnode_list_t));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000185#endif
186
Jonathan Peyton30419822017-05-12 18:01:32 +0000187 new_head->node = __kmp_node_ref(node);
188 new_head->next = list;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000189
Jonathan Peyton30419822017-05-12 18:01:32 +0000190 return new_head;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000191}
192
Jonathan Peyton30419822017-05-12 18:01:32 +0000193static void __kmp_depnode_list_free(kmp_info_t *thread,
194 kmp_depnode_list *list) {
195 kmp_depnode_list *next;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000196
Jonathan Peyton30419822017-05-12 18:01:32 +0000197 for (; list; list = next) {
198 next = list->next;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000199
Jonathan Peyton30419822017-05-12 18:01:32 +0000200 __kmp_node_deref(thread, list->node);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000201#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +0000202 __kmp_fast_free(thread, list);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000203#else
Jonathan Peyton30419822017-05-12 18:01:32 +0000204 __kmp_thread_free(thread, list);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000205#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000206 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000207}
208
Jonathan Peyton30419822017-05-12 18:01:32 +0000209static inline void __kmp_track_dependence(kmp_depnode_t *source,
210 kmp_depnode_t *sink,
211 kmp_task_t *sink_task) {
Jim Cownie5e8470a2013-09-27 10:38:44 +0000212#ifdef KMP_SUPPORT_GRAPH_OUTPUT
Jonathan Peyton30419822017-05-12 18:01:32 +0000213 kmp_taskdata_t *task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
214 // do not use sink->dn.task as that is only filled after the dependencies
215 // are already processed!
216 kmp_taskdata_t *task_sink = KMP_TASK_TO_TASKDATA(sink_task);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000217
Jonathan Peyton30419822017-05-12 18:01:32 +0000218 __kmp_printf("%d(%s) -> %d(%s)\n", source->dn.id,
219 task_source->td_ident->psource, sink->dn.id,
220 task_sink->td_ident->psource);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000221#endif
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000222#if OMPT_SUPPORT && OMPT_TRACE
Jonathan Peyton30419822017-05-12 18:01:32 +0000223 // OMPT tracks dependences between task (a=source, b=sink) in which
224 // task a blocks the execution of b through the ompt_new_dependence_callback
225 if (ompt_enabled &&
226 ompt_callbacks.ompt_callback(ompt_event_task_dependence_pair)) {
227 kmp_taskdata_t *task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
228 kmp_taskdata_t *task_sink = KMP_TASK_TO_TASKDATA(sink_task);
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000229
Jonathan Peyton30419822017-05-12 18:01:32 +0000230 ompt_callbacks.ompt_callback(ompt_event_task_dependence_pair)(
231 task_source->ompt_task_info.task_id, task_sink->ompt_task_info.task_id);
232 }
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000233#endif /* OMPT_SUPPORT && OMPT_TRACE */
Jim Cownie5e8470a2013-09-27 10:38:44 +0000234}
235
Jonathan Peyton30419822017-05-12 18:01:32 +0000236template <bool filter>
Jim Cownie5e8470a2013-09-27 10:38:44 +0000237static inline kmp_int32
Jonathan Peyton30419822017-05-12 18:01:32 +0000238__kmp_process_deps(kmp_int32 gtid, kmp_depnode_t *node, kmp_dephash_t *hash,
239 bool dep_barrier, kmp_int32 ndeps,
240 kmp_depend_info_t *dep_list, kmp_task_t *task) {
241 KA_TRACE(30, ("__kmp_process_deps<%d>: T#%d processing %d dependencies : "
242 "dep_barrier = %d\n",
243 filter, gtid, ndeps, dep_barrier));
Jonathan Peyton61118492016-05-20 19:03:38 +0000244
Jonathan Peyton30419822017-05-12 18:01:32 +0000245 kmp_info_t *thread = __kmp_threads[gtid];
246 kmp_int32 npredecessors = 0;
247 for (kmp_int32 i = 0; i < ndeps; i++) {
248 const kmp_depend_info_t *dep = &dep_list[i];
Jim Cownie5e8470a2013-09-27 10:38:44 +0000249
Jonathan Peyton30419822017-05-12 18:01:32 +0000250 KMP_DEBUG_ASSERT(dep->flags.in);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000251
Jonathan Peyton30419822017-05-12 18:01:32 +0000252 if (filter && dep->base_addr == 0)
253 continue; // skip filtered entries
Jim Cownie5e8470a2013-09-27 10:38:44 +0000254
Jonathan Peyton30419822017-05-12 18:01:32 +0000255 kmp_dephash_entry_t *info =
256 __kmp_dephash_find(thread, hash, dep->base_addr);
257 kmp_depnode_t *last_out = info->last_out;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000258
Jonathan Peyton30419822017-05-12 18:01:32 +0000259 if (dep->flags.out && info->last_ins) {
260 for (kmp_depnode_list_t *p = info->last_ins; p; p = p->next) {
261 kmp_depnode_t *indep = p->node;
262 if (indep->dn.task) {
263 KMP_ACQUIRE_DEPNODE(gtid, indep);
264 if (indep->dn.task) {
265 __kmp_track_dependence(indep, node, task);
266 indep->dn.successors =
267 __kmp_add_node(thread, indep->dn.successors, node);
268 KA_TRACE(40, ("__kmp_process_deps<%d>: T#%d adding dependence from "
269 "%p to %p\n",
270 filter, gtid, KMP_TASK_TO_TASKDATA(indep->dn.task),
271 KMP_TASK_TO_TASKDATA(task)));
272 npredecessors++;
273 }
274 KMP_RELEASE_DEPNODE(gtid, indep);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000275 }
Jonathan Peyton30419822017-05-12 18:01:32 +0000276 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000277
Jonathan Peyton30419822017-05-12 18:01:32 +0000278 __kmp_depnode_list_free(thread, info->last_ins);
279 info->last_ins = NULL;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000280
Jonathan Peyton30419822017-05-12 18:01:32 +0000281 } else if (last_out && last_out->dn.task) {
282 KMP_ACQUIRE_DEPNODE(gtid, last_out);
283 if (last_out->dn.task) {
284 __kmp_track_dependence(last_out, node, task);
285 last_out->dn.successors =
286 __kmp_add_node(thread, last_out->dn.successors, node);
287 KA_TRACE(
288 40,
289 ("__kmp_process_deps<%d>: T#%d adding dependence from %p to %p\n",
290 filter, gtid, KMP_TASK_TO_TASKDATA(last_out->dn.task),
291 KMP_TASK_TO_TASKDATA(task)));
292
293 npredecessors++;
294 }
295 KMP_RELEASE_DEPNODE(gtid, last_out);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000296 }
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000297
Jonathan Peyton30419822017-05-12 18:01:32 +0000298 if (dep_barrier) {
299 // if this is a sync point in the serial sequence, then the previous
300 // outputs are guaranteed to be completed after
301 // the execution of this task so the previous output nodes can be cleared.
302 __kmp_node_deref(thread, last_out);
303 info->last_out = NULL;
304 } else {
305 if (dep->flags.out) {
306 __kmp_node_deref(thread, last_out);
307 info->last_out = __kmp_node_ref(node);
308 } else
309 info->last_ins = __kmp_add_node(thread, info->last_ins, node);
310 }
311 }
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000312
Jonathan Peyton30419822017-05-12 18:01:32 +0000313 KA_TRACE(30, ("__kmp_process_deps<%d>: T#%d found %d predecessors\n", filter,
314 gtid, npredecessors));
315
316 return npredecessors;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000317}
318
319#define NO_DEP_BARRIER (false)
320#define DEP_BARRIER (true)
321
322// returns true if the task has any outstanding dependence
Jonathan Peyton30419822017-05-12 18:01:32 +0000323static bool __kmp_check_deps(kmp_int32 gtid, kmp_depnode_t *node,
324 kmp_task_t *task, kmp_dephash_t *hash,
325 bool dep_barrier, kmp_int32 ndeps,
326 kmp_depend_info_t *dep_list,
327 kmp_int32 ndeps_noalias,
328 kmp_depend_info_t *noalias_dep_list) {
329 int i;
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000330
Jonathan Peytond2eb3c72015-08-26 20:02:21 +0000331#if KMP_DEBUG
Jonathan Peyton30419822017-05-12 18:01:32 +0000332 kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task);
Jonathan Peytond2eb3c72015-08-26 20:02:21 +0000333#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000334 KA_TRACE(20, ("__kmp_check_deps: T#%d checking dependencies for task %p : %d "
335 "possibly aliased dependencies, %d non-aliased depedencies : "
336 "dep_barrier=%d .\n",
337 gtid, taskdata, ndeps, ndeps_noalias, dep_barrier));
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000338
Jonathan Peyton30419822017-05-12 18:01:32 +0000339 // Filter deps in dep_list
340 // TODO: Different algorithm for large dep_list ( > 10 ? )
341 for (i = 0; i < ndeps; i++) {
342 if (dep_list[i].base_addr != 0)
343 for (int j = i + 1; j < ndeps; j++)
344 if (dep_list[i].base_addr == dep_list[j].base_addr) {
345 dep_list[i].flags.in |= dep_list[j].flags.in;
346 dep_list[i].flags.out |= dep_list[j].flags.out;
347 dep_list[j].base_addr = 0; // Mark j element as void
348 }
349 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000350
Jonathan Peyton30419822017-05-12 18:01:32 +0000351 // doesn't need to be atomic as no other thread is going to be accessing this
352 // node just yet.
353 // npredecessors is set -1 to ensure that none of the releasing tasks queues
354 // this task before we have finished processing all the dependencies
355 node->dn.npredecessors = -1;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000356
Jonathan Peyton30419822017-05-12 18:01:32 +0000357 // used to pack all npredecessors additions into a single atomic operation at
358 // the end
359 int npredecessors;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000360
Jonathan Peyton30419822017-05-12 18:01:32 +0000361 npredecessors = __kmp_process_deps<true>(gtid, node, hash, dep_barrier, ndeps,
362 dep_list, task);
363 npredecessors += __kmp_process_deps<false>(
364 gtid, node, hash, dep_barrier, ndeps_noalias, noalias_dep_list, task);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000365
Jonathan Peyton30419822017-05-12 18:01:32 +0000366 node->dn.task = task;
367 KMP_MB();
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000368
Jonathan Peyton30419822017-05-12 18:01:32 +0000369 // Account for our initial fake value
370 npredecessors++;
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000371
Jonathan Peyton30419822017-05-12 18:01:32 +0000372 // Update predecessors and obtain current value to check if there are still
373 // any outstandig dependences (some tasks may have finished while we processed
374 // the dependences)
375 npredecessors = KMP_TEST_THEN_ADD32(&node->dn.npredecessors, npredecessors) +
376 npredecessors;
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000377
Jonathan Peyton30419822017-05-12 18:01:32 +0000378 KA_TRACE(20, ("__kmp_check_deps: T#%d found %d predecessors for task %p \n",
379 gtid, npredecessors, taskdata));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000380
Jonathan Peyton30419822017-05-12 18:01:32 +0000381 // beyond this point the task could be queued (and executed) by a releasing
382 // task...
383 return npredecessors > 0 ? true : false;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000384}
385
Jonathan Peyton30419822017-05-12 18:01:32 +0000386void __kmp_release_deps(kmp_int32 gtid, kmp_taskdata_t *task) {
387 kmp_info_t *thread = __kmp_threads[gtid];
388 kmp_depnode_t *node = task->td_depnode;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000389
Jonathan Peyton30419822017-05-12 18:01:32 +0000390 if (task->td_dephash) {
391 KA_TRACE(
392 40, ("__kmp_release_deps: T#%d freeing dependencies hash of task %p.\n",
393 gtid, task));
394 __kmp_dephash_free(thread, task->td_dephash);
395 task->td_dephash = NULL;
396 }
397
398 if (!node)
399 return;
400
401 KA_TRACE(20, ("__kmp_release_deps: T#%d notifying successors of task %p.\n",
402 gtid, task));
403
404 KMP_ACQUIRE_DEPNODE(gtid, node);
405 node->dn.task =
406 NULL; // mark this task as finished, so no new dependencies are generated
407 KMP_RELEASE_DEPNODE(gtid, node);
408
409 kmp_depnode_list_t *next;
410 for (kmp_depnode_list_t *p = node->dn.successors; p; p = next) {
411 kmp_depnode_t *successor = p->node;
412 kmp_int32 npredecessors =
413 KMP_TEST_THEN_DEC32(&successor->dn.npredecessors) - 1;
414
415 // successor task can be NULL for wait_depends or because deps are still
416 // being processed
417 if (npredecessors == 0) {
418 KMP_MB();
419 if (successor->dn.task) {
420 KA_TRACE(20, ("__kmp_release_deps: T#%d successor %p of %p scheduled "
421 "for execution.\n",
422 gtid, successor->dn.task, task));
423 __kmp_omp_task(gtid, successor->dn.task, false);
424 }
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000425 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000426
Jonathan Peyton30419822017-05-12 18:01:32 +0000427 next = p->next;
428 __kmp_node_deref(thread, p->node);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000429#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +0000430 __kmp_fast_free(thread, p);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000431#else
Jonathan Peyton30419822017-05-12 18:01:32 +0000432 __kmp_thread_free(thread, p);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000433#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000434 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000435
Jonathan Peyton30419822017-05-12 18:01:32 +0000436 __kmp_node_deref(thread, node);
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000437
Jonathan Peyton30419822017-05-12 18:01:32 +0000438 KA_TRACE(
439 20,
440 ("__kmp_release_deps: T#%d all successors of %p notified of completion\n",
441 gtid, task));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000442}
443
444/*!
445@ingroup TASKING
446@param loc_ref location of the original task directive
447@param gtid Global Thread ID of encountering thread
Jonathan Peyton30419822017-05-12 18:01:32 +0000448@param new_task task thunk allocated by __kmp_omp_task_alloc() for the ''new
449task''
Jim Cownie5e8470a2013-09-27 10:38:44 +0000450@param ndeps Number of depend items with possible aliasing
451@param dep_list List of depend items with possible aliasing
452@param ndeps_noalias Number of depend items with no aliasing
453@param noalias_dep_list List of depend items with no aliasing
454
Jonathan Peyton30419822017-05-12 18:01:32 +0000455@return Returns either TASK_CURRENT_NOT_QUEUED if the current task was not
456suspendend and queued, or TASK_CURRENT_QUEUED if it was suspended and queued
Jim Cownie5e8470a2013-09-27 10:38:44 +0000457
458Schedule a non-thread-switchable task with dependences for execution
459*/
Jonathan Peyton30419822017-05-12 18:01:32 +0000460kmp_int32 __kmpc_omp_task_with_deps(ident_t *loc_ref, kmp_int32 gtid,
461 kmp_task_t *new_task, kmp_int32 ndeps,
462 kmp_depend_info_t *dep_list,
463 kmp_int32 ndeps_noalias,
464 kmp_depend_info_t *noalias_dep_list) {
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000465
Jonathan Peyton30419822017-05-12 18:01:32 +0000466 kmp_taskdata_t *new_taskdata = KMP_TASK_TO_TASKDATA(new_task);
467 KA_TRACE(10, ("__kmpc_omp_task_with_deps(enter): T#%d loc=%p task=%p\n", gtid,
468 loc_ref, new_taskdata));
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000469
Jonathan Peyton30419822017-05-12 18:01:32 +0000470 kmp_info_t *thread = __kmp_threads[gtid];
471 kmp_taskdata_t *current_task = thread->th.th_current_task;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000472
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000473#if OMPT_SUPPORT && OMPT_TRACE
Jonathan Peyton30419822017-05-12 18:01:32 +0000474 /* OMPT grab all dependences if requested by the tool */
475 if (ompt_enabled && ndeps + ndeps_noalias > 0 &&
476 ompt_callbacks.ompt_callback(ompt_event_task_dependences)) {
477 kmp_int32 i;
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000478
Jonathan Peyton30419822017-05-12 18:01:32 +0000479 new_taskdata->ompt_task_info.ndeps = ndeps + ndeps_noalias;
480 new_taskdata->ompt_task_info.deps =
481 (ompt_task_dependence_t *)KMP_OMPT_DEPS_ALLOC(
482 thread, (ndeps + ndeps_noalias) * sizeof(ompt_task_dependence_t));
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000483
Jonathan Peyton30419822017-05-12 18:01:32 +0000484 KMP_ASSERT(new_taskdata->ompt_task_info.deps != NULL);
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000485
Jonathan Peyton30419822017-05-12 18:01:32 +0000486 for (i = 0; i < ndeps; i++) {
487 new_taskdata->ompt_task_info.deps[i].variable_addr =
488 (void *)dep_list[i].base_addr;
489 if (dep_list[i].flags.in && dep_list[i].flags.out)
490 new_taskdata->ompt_task_info.deps[i].dependence_flags =
491 ompt_task_dependence_type_inout;
492 else if (dep_list[i].flags.out)
493 new_taskdata->ompt_task_info.deps[i].dependence_flags =
494 ompt_task_dependence_type_out;
495 else if (dep_list[i].flags.in)
496 new_taskdata->ompt_task_info.deps[i].dependence_flags =
497 ompt_task_dependence_type_in;
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000498 }
Jonathan Peyton30419822017-05-12 18:01:32 +0000499 for (i = 0; i < ndeps_noalias; i++) {
500 new_taskdata->ompt_task_info.deps[ndeps + i].variable_addr =
501 (void *)noalias_dep_list[i].base_addr;
502 if (noalias_dep_list[i].flags.in && noalias_dep_list[i].flags.out)
503 new_taskdata->ompt_task_info.deps[ndeps + i].dependence_flags =
504 ompt_task_dependence_type_inout;
505 else if (noalias_dep_list[i].flags.out)
506 new_taskdata->ompt_task_info.deps[ndeps + i].dependence_flags =
507 ompt_task_dependence_type_out;
508 else if (noalias_dep_list[i].flags.in)
509 new_taskdata->ompt_task_info.deps[ndeps + i].dependence_flags =
510 ompt_task_dependence_type_in;
511 }
512 }
Jonas Hahnfeld39b68622016-01-28 10:39:52 +0000513#endif /* OMPT_SUPPORT && OMPT_TRACE */
514
Jonathan Peyton30419822017-05-12 18:01:32 +0000515 bool serial = current_task->td_flags.team_serial ||
516 current_task->td_flags.tasking_ser ||
517 current_task->td_flags.final;
Jonathan Peytondf6818b2016-06-14 17:57:47 +0000518#if OMP_45_ENABLED
Jonathan Peyton30419822017-05-12 18:01:32 +0000519 kmp_task_team_t *task_team = thread->th.th_task_team;
520 serial = serial && !(task_team && task_team->tt.tt_found_proxy_tasks);
Andrey Churbanov535b6fa2015-05-07 17:41:51 +0000521#endif
Jim Cownie5e8470a2013-09-27 10:38:44 +0000522
Jonathan Peyton30419822017-05-12 18:01:32 +0000523 if (!serial && (ndeps > 0 || ndeps_noalias > 0)) {
524 /* if no dependencies have been tracked yet, create the dependence hash */
525 if (current_task->td_dephash == NULL)
526 current_task->td_dephash = __kmp_dephash_create(thread, current_task);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000527
528#if USE_FAST_MEMORY
Jonathan Peyton30419822017-05-12 18:01:32 +0000529 kmp_depnode_t *node =
530 (kmp_depnode_t *)__kmp_fast_allocate(thread, sizeof(kmp_depnode_t));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000531#else
Jonathan Peyton30419822017-05-12 18:01:32 +0000532 kmp_depnode_t *node =
533 (kmp_depnode_t *)__kmp_thread_malloc(thread, sizeof(kmp_depnode_t));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000534#endif
535
Jonathan Peyton30419822017-05-12 18:01:32 +0000536 __kmp_init_node(node);
537 new_taskdata->td_depnode = node;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000538
Jonathan Peyton30419822017-05-12 18:01:32 +0000539 if (__kmp_check_deps(gtid, node, new_task, current_task->td_dephash,
540 NO_DEP_BARRIER, ndeps, dep_list, ndeps_noalias,
541 noalias_dep_list)) {
542 KA_TRACE(10, ("__kmpc_omp_task_with_deps(exit): T#%d task had blocking "
543 "dependencies: "
544 "loc=%p task=%p, return: TASK_CURRENT_NOT_QUEUED\n",
545 gtid, loc_ref, new_taskdata));
546 return TASK_CURRENT_NOT_QUEUED;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000547 }
Jonathan Peyton30419822017-05-12 18:01:32 +0000548 } else {
549 KA_TRACE(10, ("__kmpc_omp_task_with_deps(exit): T#%d ignored dependencies "
550 "for task (serialized)"
551 "loc=%p task=%p\n",
552 gtid, loc_ref, new_taskdata));
553 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000554
Jonathan Peyton30419822017-05-12 18:01:32 +0000555 KA_TRACE(10, ("__kmpc_omp_task_with_deps(exit): T#%d task had no blocking "
556 "dependencies : "
557 "loc=%p task=%p, transferring to __kmpc_omp_task\n",
558 gtid, loc_ref, new_taskdata));
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000559
Jonathan Peyton30419822017-05-12 18:01:32 +0000560 return __kmpc_omp_task(loc_ref, gtid, new_task);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000561}
562
563/*!
564@ingroup TASKING
565@param loc_ref location of the original task directive
566@param gtid Global Thread ID of encountering thread
567@param ndeps Number of depend items with possible aliasing
568@param dep_list List of depend items with possible aliasing
569@param ndeps_noalias Number of depend items with no aliasing
570@param noalias_dep_list List of depend items with no aliasing
571
572Blocks the current task until all specifies dependencies have been fulfilled.
573*/
Jonathan Peyton30419822017-05-12 18:01:32 +0000574void __kmpc_omp_wait_deps(ident_t *loc_ref, kmp_int32 gtid, kmp_int32 ndeps,
575 kmp_depend_info_t *dep_list, kmp_int32 ndeps_noalias,
576 kmp_depend_info_t *noalias_dep_list) {
577 KA_TRACE(10, ("__kmpc_omp_wait_deps(enter): T#%d loc=%p\n", gtid, loc_ref));
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000578
Jonathan Peyton30419822017-05-12 18:01:32 +0000579 if (ndeps == 0 && ndeps_noalias == 0) {
580 KA_TRACE(10, ("__kmpc_omp_wait_deps(exit): T#%d has no dependencies to "
581 "wait upon : loc=%p\n",
582 gtid, loc_ref));
583 return;
584 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000585
Jonathan Peyton30419822017-05-12 18:01:32 +0000586 kmp_info_t *thread = __kmp_threads[gtid];
587 kmp_taskdata_t *current_task = thread->th.th_current_task;
Jim Cownie5e8470a2013-09-27 10:38:44 +0000588
Jonathan Peyton30419822017-05-12 18:01:32 +0000589 // We can return immediately as:
590 // - dependences are not computed in serial teams (except with proxy tasks)
591 // - if the dephash is not yet created it means we have nothing to wait for
592 bool ignore = current_task->td_flags.team_serial ||
593 current_task->td_flags.tasking_ser ||
594 current_task->td_flags.final;
Jonathan Peytondf6818b2016-06-14 17:57:47 +0000595#if OMP_45_ENABLED
Jonathan Peyton30419822017-05-12 18:01:32 +0000596 ignore = ignore && thread->th.th_task_team != NULL &&
597 thread->th.th_task_team->tt.tt_found_proxy_tasks == FALSE;
Andrey Churbanov535b6fa2015-05-07 17:41:51 +0000598#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000599 ignore = ignore || current_task->td_dephash == NULL;
Andrey Churbanov535b6fa2015-05-07 17:41:51 +0000600
Jonathan Peyton30419822017-05-12 18:01:32 +0000601 if (ignore) {
602 KA_TRACE(10, ("__kmpc_omp_wait_deps(exit): T#%d has no blocking "
603 "dependencies : loc=%p\n",
604 gtid, loc_ref));
605 return;
606 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000607
Jonathan Peyton30419822017-05-12 18:01:32 +0000608 kmp_depnode_t node;
609 __kmp_init_node(&node);
Jim Cownie5e8470a2013-09-27 10:38:44 +0000610
Jonathan Peyton30419822017-05-12 18:01:32 +0000611 if (!__kmp_check_deps(gtid, &node, NULL, current_task->td_dephash,
612 DEP_BARRIER, ndeps, dep_list, ndeps_noalias,
613 noalias_dep_list)) {
614 KA_TRACE(10, ("__kmpc_omp_wait_deps(exit): T#%d has no blocking "
615 "dependencies : loc=%p\n",
616 gtid, loc_ref));
617 return;
618 }
Jim Cownie5e8470a2013-09-27 10:38:44 +0000619
Jonathan Peyton30419822017-05-12 18:01:32 +0000620 int thread_finished = FALSE;
621 kmp_flag_32 flag((volatile kmp_uint32 *)&(node.dn.npredecessors), 0U);
622 while (node.dn.npredecessors > 0) {
623 flag.execute_tasks(thread, gtid, FALSE, &thread_finished,
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000624#if USE_ITT_BUILD
Jonathan Peyton30419822017-05-12 18:01:32 +0000625 NULL,
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000626#endif
Jonathan Peyton30419822017-05-12 18:01:32 +0000627 __kmp_task_stealing_constraint);
628 }
Jim Cownie4cc4bb42014-10-07 16:25:50 +0000629
Jonathan Peyton30419822017-05-12 18:01:32 +0000630 KA_TRACE(10, ("__kmpc_omp_wait_deps(exit): T#%d finished waiting : loc=%p\n",
631 gtid, loc_ref));
Jim Cownie5e8470a2013-09-27 10:38:44 +0000632}
633
634#endif /* OMP_40_ENABLED */