20 #include "kmp_wait_release.h" 31 #ifdef KMP_SUPPORT_GRAPH_OUTPUT 32 static kmp_int32 kmp_node_id_seed = 0;
36 __kmp_init_node ( kmp_depnode_t *node )
39 node->dn.successors = NULL;
40 __kmp_init_lock(&node->dn.lock);
42 #ifdef KMP_SUPPORT_GRAPH_OUTPUT 43 node->dn.id = KMP_TEST_THEN_INC32(&kmp_node_id_seed);
47 static inline kmp_depnode_t *
48 __kmp_node_ref ( kmp_depnode_t *node )
50 KMP_TEST_THEN_INC32(&node->dn.nrefs);
55 __kmp_node_deref ( kmp_info_t *thread, kmp_depnode_t *node )
59 kmp_int32 n = KMP_TEST_THEN_DEC32(&node->dn.nrefs) - 1;
61 KMP_ASSERT(node->dn.nrefs == 0);
63 __kmp_fast_free(thread,node);
65 __kmp_thread_free(thread,node);
70 #define KMP_ACQUIRE_DEPNODE(gtid,n) __kmp_acquire_lock(&(n)->dn.lock,(gtid)) 71 #define KMP_RELEASE_DEPNODE(gtid,n) __kmp_release_lock(&(n)->dn.lock,(gtid)) 74 __kmp_depnode_list_free ( kmp_info_t *thread, kmp_depnode_list *list );
77 KMP_DEPHASH_OTHER_SIZE = 97,
78 KMP_DEPHASH_MASTER_SIZE = 997
81 static inline kmp_int32
82 __kmp_dephash_hash ( kmp_intptr_t addr,
size_t hsize )
85 return ((addr >> 6) ^ (addr >> 2)) % hsize;
88 static kmp_dephash_t *
89 __kmp_dephash_create ( kmp_info_t *thread, kmp_taskdata_t *current_task )
95 if ( current_task->td_flags.tasktype == TASK_IMPLICIT )
96 h_size = KMP_DEPHASH_MASTER_SIZE;
98 h_size = KMP_DEPHASH_OTHER_SIZE;
101 h_size *
sizeof(kmp_dephash_entry_t *) +
sizeof(kmp_dephash_t);
104 h = (kmp_dephash_t *) __kmp_fast_allocate( thread, size );
106 h = (kmp_dephash_t *) __kmp_thread_malloc( thread, size );
114 h->buckets = (kmp_dephash_entry **)(h+1);
116 for (
size_t i = 0; i < h_size; i++ )
123 __kmp_dephash_free_entries(kmp_info_t *thread, kmp_dephash_t *h)
125 for (
size_t i = 0; i < h->size; i++) {
127 kmp_dephash_entry_t *next;
128 for (kmp_dephash_entry_t *entry = h->buckets[i]; entry; entry = next) {
129 next = entry->next_in_bucket;
130 __kmp_depnode_list_free(thread,entry->last_ins);
131 __kmp_node_deref(thread,entry->last_out);
133 __kmp_fast_free(thread,entry);
135 __kmp_thread_free(thread,entry);
144 __kmp_dephash_free(kmp_info_t *thread, kmp_dephash_t *h)
146 __kmp_dephash_free_entries(thread, h);
148 __kmp_fast_free(thread,h);
150 __kmp_thread_free(thread,h);
154 static kmp_dephash_entry *
155 __kmp_dephash_find ( kmp_info_t *thread, kmp_dephash_t *h, kmp_intptr_t addr )
157 kmp_int32 bucket = __kmp_dephash_hash(addr,h->size);
159 kmp_dephash_entry_t *entry;
160 for ( entry = h->buckets[bucket]; entry; entry = entry->next_in_bucket )
161 if ( entry->addr == addr )
break;
163 if ( entry == NULL ) {
166 entry = (kmp_dephash_entry_t *) __kmp_fast_allocate( thread,
sizeof(kmp_dephash_entry_t) );
168 entry = (kmp_dephash_entry_t *) __kmp_thread_malloc( thread,
sizeof(kmp_dephash_entry_t) );
171 entry->last_out = NULL;
172 entry->last_ins = NULL;
173 entry->next_in_bucket = h->buckets[bucket];
174 h->buckets[bucket] = entry;
177 if ( entry->next_in_bucket ) h->nconflicts++;
183 static kmp_depnode_list_t *
184 __kmp_add_node ( kmp_info_t *thread, kmp_depnode_list_t *list, kmp_depnode_t *node )
186 kmp_depnode_list_t *new_head;
189 new_head = (kmp_depnode_list_t *) __kmp_fast_allocate(thread,
sizeof(kmp_depnode_list_t));
191 new_head = (kmp_depnode_list_t *) __kmp_thread_malloc(thread,
sizeof(kmp_depnode_list_t));
194 new_head->node = __kmp_node_ref(node);
195 new_head->next = list;
201 __kmp_depnode_list_free ( kmp_info_t *thread, kmp_depnode_list *list )
203 kmp_depnode_list *next;
205 for ( ; list ; list = next ) {
208 __kmp_node_deref(thread,list->node);
210 __kmp_fast_free(thread,list);
212 __kmp_thread_free(thread,list);
218 __kmp_track_dependence ( kmp_depnode_t *source, kmp_depnode_t *sink,
219 kmp_task_t *sink_task )
221 #ifdef KMP_SUPPORT_GRAPH_OUTPUT 222 kmp_taskdata_t * task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
223 kmp_taskdata_t * task_sink = KMP_TASK_TO_TASKDATA(sink->dn.task);
225 __kmp_printf(
"%d(%s) -> %d(%s)\n", source->dn.id, task_source->td_ident->psource, sink->dn.id, task_sink->td_ident->psource);
227 #if OMPT_SUPPORT && OMPT_TRACE 231 ompt_callbacks.ompt_callback(ompt_event_task_dependence_pair))
233 kmp_taskdata_t * task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
234 kmp_taskdata_t * task_sink = KMP_TASK_TO_TASKDATA(sink_task);
236 ompt_callbacks.ompt_callback(ompt_event_task_dependence_pair)(
237 task_source->ompt_task_info.task_id,
238 task_sink->ompt_task_info.task_id);
243 template<
bool filter >
244 static inline kmp_int32
245 __kmp_process_deps ( kmp_int32 gtid, kmp_depnode_t *node, kmp_dephash_t *hash,
246 bool dep_barrier,kmp_int32 ndeps, kmp_depend_info_t *dep_list,
249 KA_TRACE(30, (
"__kmp_process_deps<%d>: T#%d processing %d depencies : dep_barrier = %d\n", filter, gtid, ndeps, dep_barrier ) );
251 kmp_info_t *thread = __kmp_threads[ gtid ];
252 kmp_int32 npredecessors=0;
253 for ( kmp_int32 i = 0; i < ndeps ; i++ ) {
254 const kmp_depend_info_t * dep = &dep_list[i];
256 KMP_DEBUG_ASSERT(dep->flags.in);
258 if ( filter && dep->base_addr == 0 )
continue;
260 kmp_dephash_entry_t *info = __kmp_dephash_find(thread,hash,dep->base_addr);
261 kmp_depnode_t *last_out = info->last_out;
263 if ( dep->flags.out && info->last_ins ) {
264 for ( kmp_depnode_list_t * p = info->last_ins; p; p = p->next ) {
265 kmp_depnode_t * indep = p->node;
266 if ( indep->dn.task ) {
267 KMP_ACQUIRE_DEPNODE(gtid,indep);
268 if ( indep->dn.task ) {
269 __kmp_track_dependence(indep,node,task);
270 indep->dn.successors = __kmp_add_node(thread, indep->dn.successors, node);
271 KA_TRACE(40,(
"__kmp_process_deps<%d>: T#%d adding dependence from %p to %p\n",
272 filter,gtid, KMP_TASK_TO_TASKDATA(indep->dn.task), KMP_TASK_TO_TASKDATA(node->dn.task)));
275 KMP_RELEASE_DEPNODE(gtid,indep);
279 __kmp_depnode_list_free(thread,info->last_ins);
280 info->last_ins = NULL;
282 }
else if ( last_out && last_out->dn.task ) {
283 KMP_ACQUIRE_DEPNODE(gtid,last_out);
284 if ( last_out->dn.task ) {
285 __kmp_track_dependence(last_out,node,task);
286 last_out->dn.successors = __kmp_add_node(thread, last_out->dn.successors, node);
287 KA_TRACE(40,(
"__kmp_process_deps<%d>: T#%d adding dependence from %p to %p\n",
288 filter,gtid, KMP_TASK_TO_TASKDATA(last_out->dn.task), KMP_TASK_TO_TASKDATA(node->dn.task)));
292 KMP_RELEASE_DEPNODE(gtid,last_out);
298 __kmp_node_deref(thread,last_out);
299 info->last_out = NULL;
301 if ( dep->flags.out ) {
302 __kmp_node_deref(thread,last_out);
303 info->last_out = __kmp_node_ref(node);
305 info->last_ins = __kmp_add_node(thread, info->last_ins, node);
310 KA_TRACE(30, (
"__kmp_process_deps<%d>: T#%d found %d predecessors\n", filter, gtid, npredecessors ) );
312 return npredecessors;
315 #define NO_DEP_BARRIER (false) 316 #define DEP_BARRIER (true) 320 __kmp_check_deps ( kmp_int32 gtid, kmp_depnode_t *node, kmp_task_t *task, kmp_dephash_t *hash,
bool dep_barrier,
321 kmp_int32 ndeps, kmp_depend_info_t *dep_list,
322 kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list )
327 kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(task);
329 KA_TRACE(20, (
"__kmp_check_deps: T#%d checking dependencies for task %p : %d possibly aliased dependencies, %d non-aliased depedencies : dep_barrier=%d .\n", gtid, taskdata, ndeps, ndeps_noalias, dep_barrier ) );
333 for ( i = 0; i < ndeps; i ++ ) {
334 if ( dep_list[i].base_addr != 0 )
335 for (
int j = i+1; j < ndeps; j++ )
336 if ( dep_list[i].base_addr == dep_list[j].base_addr ) {
337 dep_list[i].flags.in |= dep_list[j].flags.in;
338 dep_list[i].flags.out |= dep_list[j].flags.out;
339 dep_list[j].base_addr = 0;
345 node->dn.npredecessors = -1;
350 npredecessors = __kmp_process_deps<true>(gtid, node, hash, dep_barrier,
351 ndeps, dep_list, task);
352 npredecessors += __kmp_process_deps<false>(gtid, node, hash, dep_barrier,
353 ndeps_noalias, noalias_dep_list, task);
355 node->dn.task = task;
362 npredecessors = KMP_TEST_THEN_ADD32(&node->dn.npredecessors, npredecessors) + npredecessors;
364 KA_TRACE(20, (
"__kmp_check_deps: T#%d found %d predecessors for task %p \n", gtid, npredecessors, taskdata ) );
367 return npredecessors > 0 ?
true :
false;
371 __kmp_release_deps ( kmp_int32 gtid, kmp_taskdata_t *task )
373 kmp_info_t *thread = __kmp_threads[ gtid ];
374 kmp_depnode_t *node = task->td_depnode;
376 if ( task->td_dephash ) {
377 KA_TRACE(40, (
"__kmp_realease_deps: T#%d freeing dependencies hash of task %p.\n", gtid, task ) );
378 __kmp_dephash_free(thread,task->td_dephash);
383 KA_TRACE(20, (
"__kmp_realease_deps: T#%d notifying succesors of task %p.\n", gtid, task ) );
385 KMP_ACQUIRE_DEPNODE(gtid,node);
386 node->dn.task = NULL;
387 KMP_RELEASE_DEPNODE(gtid,node);
389 kmp_depnode_list_t *next;
390 for ( kmp_depnode_list_t *p = node->dn.successors; p; p = next ) {
391 kmp_depnode_t *successor = p->node;
392 kmp_int32 npredecessors = KMP_TEST_THEN_DEC32(&successor->dn.npredecessors) - 1;
395 if ( npredecessors == 0 ) {
397 if ( successor->dn.task ) {
398 KA_TRACE(20, (
"__kmp_realease_deps: T#%d successor %p of %p scheduled for execution.\n", gtid, successor->dn.task, task ) );
399 __kmp_omp_task(gtid,successor->dn.task,
false);
404 __kmp_node_deref(thread,p->node);
406 __kmp_fast_free(thread,p);
408 __kmp_thread_free(thread,p);
412 __kmp_node_deref(thread,node);
414 KA_TRACE(20, (
"__kmp_realease_deps: T#%d all successors of %p notified of completation\n", gtid, task ) );
433 kmp_int32 ndeps, kmp_depend_info_t *dep_list,
434 kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list )
437 kmp_taskdata_t * new_taskdata = KMP_TASK_TO_TASKDATA(new_task);
438 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(enter): T#%d loc=%p task=%p\n",
439 gtid, loc_ref, new_taskdata ) );
441 kmp_info_t *thread = __kmp_threads[ gtid ];
442 kmp_taskdata_t * current_task = thread->th.th_current_task;
444 #if OMPT_SUPPORT && OMPT_TRACE 446 if (ompt_enabled && ndeps+ndeps_noalias > 0 &&
447 ompt_callbacks.ompt_callback(ompt_event_task_dependences))
451 new_taskdata->ompt_task_info.ndeps = ndeps+ndeps_noalias;
452 new_taskdata->ompt_task_info.deps = (ompt_task_dependence_t *)
453 KMP_OMPT_DEPS_ALLOC(thread,
454 (ndeps+ndeps_noalias)*
sizeof(ompt_task_dependence_t));
456 KMP_ASSERT(new_taskdata->ompt_task_info.deps != NULL);
458 for (i = 0; i < ndeps; i++)
460 new_taskdata->ompt_task_info.deps[i].variable_addr =
461 (
void*) dep_list[i].base_addr;
462 if (dep_list[i].flags.in && dep_list[i].flags.out)
463 new_taskdata->ompt_task_info.deps[i].dependence_flags =
464 ompt_task_dependence_type_inout;
465 else if (dep_list[i].flags.out)
466 new_taskdata->ompt_task_info.deps[i].dependence_flags =
467 ompt_task_dependence_type_out;
468 else if (dep_list[i].flags.in)
469 new_taskdata->ompt_task_info.deps[i].dependence_flags =
470 ompt_task_dependence_type_in;
472 for (i = 0; i < ndeps_noalias; i++)
474 new_taskdata->ompt_task_info.deps[ndeps+i].variable_addr =
475 (
void*) noalias_dep_list[i].base_addr;
476 if (noalias_dep_list[i].flags.in && noalias_dep_list[i].flags.out)
477 new_taskdata->ompt_task_info.deps[ndeps+i].dependence_flags =
478 ompt_task_dependence_type_inout;
479 else if (noalias_dep_list[i].flags.out)
480 new_taskdata->ompt_task_info.deps[ndeps+i].dependence_flags =
481 ompt_task_dependence_type_out;
482 else if (noalias_dep_list[i].flags.in)
483 new_taskdata->ompt_task_info.deps[ndeps+i].dependence_flags =
484 ompt_task_dependence_type_in;
489 bool serial = current_task->td_flags.team_serial || current_task->td_flags.tasking_ser || current_task->td_flags.final;
491 serial = serial && !(new_taskdata->td_flags.proxy == TASK_PROXY);
494 if ( !serial && ( ndeps > 0 || ndeps_noalias > 0 )) {
496 if ( current_task->td_dephash == NULL )
497 current_task->td_dephash = __kmp_dephash_create(thread, current_task);
500 kmp_depnode_t *node = (kmp_depnode_t *) __kmp_fast_allocate(thread,
sizeof(kmp_depnode_t));
502 kmp_depnode_t *node = (kmp_depnode_t *) __kmp_thread_malloc(thread,
sizeof(kmp_depnode_t));
505 __kmp_init_node(node);
506 new_taskdata->td_depnode = node;
508 if ( __kmp_check_deps( gtid, node, new_task, current_task->td_dephash, NO_DEP_BARRIER,
509 ndeps, dep_list, ndeps_noalias,noalias_dep_list ) ) {
510 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d task had blocking dependencies: " 511 "loc=%p task=%p, return: TASK_CURRENT_NOT_QUEUED\n", gtid, loc_ref,
513 return TASK_CURRENT_NOT_QUEUED;
517 kmp_task_team_t * task_team = thread->th.th_task_team;
518 if ( task_team && task_team->tt.tt_found_proxy_tasks )
522 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d ignored dependencies for task (serialized)" 523 "loc=%p task=%p\n", gtid, loc_ref, new_taskdata ) );
526 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d task had no blocking dependencies : " 527 "loc=%p task=%p, transferring to __kmpc_omp_task\n", gtid, loc_ref,
530 return __kmpc_omp_task(loc_ref,gtid,new_task);
546 kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list )
548 KA_TRACE(10, (
"__kmpc_omp_wait_deps(enter): T#%d loc=%p\n", gtid, loc_ref) );
550 if ( ndeps == 0 && ndeps_noalias == 0 ) {
551 KA_TRACE(10, (
"__kmpc_omp_wait_deps(exit): T#%d has no dependencies to wait upon : loc=%p\n", gtid, loc_ref) );
555 kmp_info_t *thread = __kmp_threads[ gtid ];
556 kmp_taskdata_t * current_task = thread->th.th_current_task;
561 bool ignore = current_task->td_flags.team_serial || current_task->td_flags.tasking_ser || current_task->td_flags.final;
563 ignore = ignore && thread->th.th_task_team != NULL && thread->th.th_task_team->tt.tt_found_proxy_tasks == FALSE;
565 ignore = ignore || current_task->td_dephash == NULL;
568 KA_TRACE(10, (
"__kmpc_omp_wait_deps(exit): T#%d has no blocking dependencies : loc=%p\n", gtid, loc_ref) );
573 __kmp_init_node(&node);
575 if (!__kmp_check_deps( gtid, &node, NULL, current_task->td_dephash, DEP_BARRIER,
576 ndeps, dep_list, ndeps_noalias, noalias_dep_list )) {
577 KA_TRACE(10, (
"__kmpc_omp_wait_deps(exit): T#%d has no blocking dependencies : loc=%p\n", gtid, loc_ref) );
581 int thread_finished = FALSE;
582 kmp_flag_32 flag((
volatile kmp_uint32 *)&(node.dn.npredecessors), 0U);
583 while ( node.dn.npredecessors > 0 ) {
584 flag.execute_tasks(thread, gtid, FALSE, &thread_finished,
588 __kmp_task_stealing_constraint );
591 KA_TRACE(10, (
"__kmpc_omp_wait_deps(exit): T#%d finished waiting : loc=%p\n", gtid, loc_ref) );
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)
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)