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;
100 kmp_int32 size = h_size *
sizeof(kmp_dephash_entry_t) +
sizeof(kmp_dephash_t);
103 h = (kmp_dephash_t *) __kmp_fast_allocate( thread, size );
105 h = (kmp_dephash_t *) __kmp_thread_malloc( thread, size );
113 h->buckets = (kmp_dephash_entry **)(h+1);
115 for (
size_t i = 0; i < h_size; i++ )
122 __kmp_dephash_free ( kmp_info_t *thread, kmp_dephash_t *h )
124 for (
size_t i=0; i < h->size; i++ ) {
125 if ( h->buckets[i] ) {
126 kmp_dephash_entry_t *next;
127 for ( kmp_dephash_entry_t *entry = h->buckets[i]; entry; entry = next ) {
128 next = entry->next_in_bucket;
129 __kmp_depnode_list_free(thread,entry->last_ins);
130 __kmp_node_deref(thread,entry->last_out);
132 __kmp_fast_free(thread,entry);
134 __kmp_thread_free(thread,entry);
140 __kmp_fast_free(thread,h);
142 __kmp_thread_free(thread,h);
146 static kmp_dephash_entry *
147 __kmp_dephash_find ( kmp_info_t *thread, kmp_dephash_t *h, kmp_intptr_t addr )
149 kmp_int32 bucket = __kmp_dephash_hash(addr,h->size);
151 kmp_dephash_entry_t *entry;
152 for ( entry = h->buckets[bucket]; entry; entry = entry->next_in_bucket )
153 if ( entry->addr == addr )
break;
155 if ( entry == NULL ) {
158 entry = (kmp_dephash_entry_t *) __kmp_fast_allocate( thread,
sizeof(kmp_dephash_entry_t) );
160 entry = (kmp_dephash_entry_t *) __kmp_thread_malloc( thread,
sizeof(kmp_dephash_entry_t) );
163 entry->last_out = NULL;
164 entry->last_ins = NULL;
165 entry->next_in_bucket = h->buckets[bucket];
166 h->buckets[bucket] = entry;
169 if ( entry->next_in_bucket ) h->nconflicts++;
175 static kmp_depnode_list_t *
176 __kmp_add_node ( kmp_info_t *thread, kmp_depnode_list_t *list, kmp_depnode_t *node )
178 kmp_depnode_list_t *new_head;
181 new_head = (kmp_depnode_list_t *) __kmp_fast_allocate(thread,
sizeof(kmp_depnode_list_t));
183 new_head = (kmp_depnode_list_t *) __kmp_thread_malloc(thread,
sizeof(kmp_depnode_list_t));
186 new_head->node = __kmp_node_ref(node);
187 new_head->next = list;
193 __kmp_depnode_list_free ( kmp_info_t *thread, kmp_depnode_list *list )
195 kmp_depnode_list *next;
197 for ( ; list ; list = next ) {
200 __kmp_node_deref(thread,list->node);
202 __kmp_fast_free(thread,list);
204 __kmp_thread_free(thread,list);
210 __kmp_track_dependence ( kmp_depnode_t *source, kmp_depnode_t *sink,
211 kmp_task_t *sink_task )
213 #ifdef KMP_SUPPORT_GRAPH_OUTPUT 214 kmp_taskdata_t * task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
215 kmp_taskdata_t * task_sink = KMP_TASK_TO_TASKDATA(sink->dn.task);
217 __kmp_printf(
"%d(%s) -> %d(%s)\n", source->dn.id, task_source->td_ident->psource, sink->dn.id, task_sink->td_ident->psource);
219 #if OMPT_SUPPORT && OMPT_TRACE 223 ompt_callbacks.ompt_callback(ompt_event_task_dependence_pair))
225 kmp_taskdata_t * task_source = KMP_TASK_TO_TASKDATA(source->dn.task);
226 kmp_taskdata_t * task_sink = KMP_TASK_TO_TASKDATA(sink_task);
228 ompt_callbacks.ompt_callback(ompt_event_task_dependence_pair)(
229 task_source->ompt_task_info.task_id,
230 task_sink->ompt_task_info.task_id);
235 template<
bool filter >
236 static inline kmp_int32
237 __kmp_process_deps ( kmp_int32 gtid, kmp_depnode_t *node, kmp_dephash_t *hash,
238 bool dep_barrier,kmp_int32 ndeps, kmp_depend_info_t *dep_list,
241 KA_TRACE(30, (
"__kmp_process_deps<%d>: T#%d processing %d depencies : dep_barrier = %d\n", filter, gtid, ndeps, dep_barrier ) );
243 kmp_info_t *thread = __kmp_threads[ gtid ];
244 kmp_int32 npredecessors=0;
245 for ( kmp_int32 i = 0; i < ndeps ; i++ ) {
246 const kmp_depend_info_t * dep = &dep_list[i];
248 KMP_DEBUG_ASSERT(dep->flags.in);
250 if ( filter && dep->base_addr == 0 )
continue;
252 kmp_dephash_entry_t *info = __kmp_dephash_find(thread,hash,dep->base_addr);
253 kmp_depnode_t *last_out = info->last_out;
255 if ( dep->flags.out && info->last_ins ) {
256 for ( kmp_depnode_list_t * p = info->last_ins; p; p = p->next ) {
257 kmp_depnode_t * indep = p->node;
258 if ( indep->dn.task ) {
259 KMP_ACQUIRE_DEPNODE(gtid,indep);
260 if ( indep->dn.task ) {
261 __kmp_track_dependence(indep,node,task);
262 indep->dn.successors = __kmp_add_node(thread, indep->dn.successors, node);
263 KA_TRACE(40,(
"__kmp_process_deps<%d>: T#%d adding dependence from %p to %p\n",
264 filter,gtid, KMP_TASK_TO_TASKDATA(indep->dn.task), KMP_TASK_TO_TASKDATA(node->dn.task)));
267 KMP_RELEASE_DEPNODE(gtid,indep);
271 __kmp_depnode_list_free(thread,info->last_ins);
272 info->last_ins = NULL;
274 }
else if ( last_out && last_out->dn.task ) {
275 KMP_ACQUIRE_DEPNODE(gtid,last_out);
276 if ( last_out->dn.task ) {
277 __kmp_track_dependence(last_out,node,task);
278 last_out->dn.successors = __kmp_add_node(thread, last_out->dn.successors, node);
279 KA_TRACE(40,(
"__kmp_process_deps<%d>: T#%d adding dependence from %p to %p\n",
280 filter,gtid, KMP_TASK_TO_TASKDATA(last_out->dn.task), KMP_TASK_TO_TASKDATA(node->dn.task)));
284 KMP_RELEASE_DEPNODE(gtid,last_out);
290 __kmp_node_deref(thread,last_out);
291 info->last_out = NULL;
293 if ( dep->flags.out ) {
294 __kmp_node_deref(thread,last_out);
295 info->last_out = __kmp_node_ref(node);
297 info->last_ins = __kmp_add_node(thread, info->last_ins, node);
302 KA_TRACE(30, (
"__kmp_process_deps<%d>: T#%d found %d predecessors\n", filter, gtid, npredecessors ) );
304 return npredecessors;
307 #define NO_DEP_BARRIER (false) 308 #define DEP_BARRIER (true) 312 __kmp_check_deps ( kmp_int32 gtid, kmp_depnode_t *node, kmp_task_t *task, kmp_dephash_t *hash,
bool dep_barrier,
313 kmp_int32 ndeps, kmp_depend_info_t *dep_list,
314 kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list )
319 kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(task);
321 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 ) );
325 for ( i = 0; i < ndeps; i ++ ) {
326 if ( dep_list[i].base_addr != 0 )
327 for (
int j = i+1; j < ndeps; j++ )
328 if ( dep_list[i].base_addr == dep_list[j].base_addr ) {
329 dep_list[i].flags.in |= dep_list[j].flags.in;
330 dep_list[i].flags.out |= dep_list[j].flags.out;
331 dep_list[j].base_addr = 0;
337 node->dn.npredecessors = -1;
342 npredecessors = __kmp_process_deps<true>(gtid, node, hash, dep_barrier,
343 ndeps, dep_list, task);
344 npredecessors += __kmp_process_deps<false>(gtid, node, hash, dep_barrier,
345 ndeps_noalias, noalias_dep_list, task);
347 node->dn.task = task;
354 npredecessors = KMP_TEST_THEN_ADD32(&node->dn.npredecessors, npredecessors) + npredecessors;
356 KA_TRACE(20, (
"__kmp_check_deps: T#%d found %d predecessors for task %p \n", gtid, npredecessors, taskdata ) );
359 return npredecessors > 0 ?
true :
false;
363 __kmp_release_deps ( kmp_int32 gtid, kmp_taskdata_t *task )
365 kmp_info_t *thread = __kmp_threads[ gtid ];
366 kmp_depnode_t *node = task->td_depnode;
368 if ( task->td_dephash ) {
369 KA_TRACE(40, (
"__kmp_realease_deps: T#%d freeing dependencies hash of task %p.\n", gtid, task ) );
370 __kmp_dephash_free(thread,task->td_dephash);
375 KA_TRACE(20, (
"__kmp_realease_deps: T#%d notifying succesors of task %p.\n", gtid, task ) );
377 KMP_ACQUIRE_DEPNODE(gtid,node);
378 node->dn.task = NULL;
379 KMP_RELEASE_DEPNODE(gtid,node);
381 kmp_depnode_list_t *next;
382 for ( kmp_depnode_list_t *p = node->dn.successors; p; p = next ) {
383 kmp_depnode_t *successor = p->node;
384 kmp_int32 npredecessors = KMP_TEST_THEN_DEC32(&successor->dn.npredecessors) - 1;
387 if ( npredecessors == 0 ) {
389 if ( successor->dn.task ) {
390 KA_TRACE(20, (
"__kmp_realease_deps: T#%d successor %p of %p scheduled for execution.\n", gtid, successor->dn.task, task ) );
391 __kmp_omp_task(gtid,successor->dn.task,
false);
396 __kmp_node_deref(thread,p->node);
398 __kmp_fast_free(thread,p);
400 __kmp_thread_free(thread,p);
404 __kmp_node_deref(thread,node);
406 KA_TRACE(20, (
"__kmp_realease_deps: T#%d all successors of %p notified of completation\n", gtid, task ) );
425 kmp_int32 ndeps, kmp_depend_info_t *dep_list,
426 kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list )
429 kmp_taskdata_t * new_taskdata = KMP_TASK_TO_TASKDATA(new_task);
430 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(enter): T#%d loc=%p task=%p\n",
431 gtid, loc_ref, new_taskdata ) );
433 kmp_info_t *thread = __kmp_threads[ gtid ];
434 kmp_taskdata_t * current_task = thread->th.th_current_task;
436 #if OMPT_SUPPORT && OMPT_TRACE 438 if (ompt_enabled && ndeps+ndeps_noalias > 0 &&
439 ompt_callbacks.ompt_callback(ompt_event_task_dependences))
443 new_taskdata->ompt_task_info.ndeps = ndeps+ndeps_noalias;
444 new_taskdata->ompt_task_info.deps = (ompt_task_dependence_t *)
445 KMP_OMPT_DEPS_ALLOC(thread,
446 (ndeps+ndeps_noalias)*
sizeof(ompt_task_dependence_t));
448 KMP_ASSERT(new_taskdata->ompt_task_info.deps != NULL);
450 for (i = 0; i < ndeps; i++)
452 new_taskdata->ompt_task_info.deps[i].variable_addr =
453 (
void*) dep_list[i].base_addr;
454 if (dep_list[i].flags.in && dep_list[i].flags.out)
455 new_taskdata->ompt_task_info.deps[i].dependence_flags =
456 ompt_task_dependence_type_inout;
457 else if (dep_list[i].flags.out)
458 new_taskdata->ompt_task_info.deps[i].dependence_flags =
459 ompt_task_dependence_type_out;
460 else if (dep_list[i].flags.in)
461 new_taskdata->ompt_task_info.deps[i].dependence_flags =
462 ompt_task_dependence_type_in;
464 for (i = 0; i < ndeps_noalias; i++)
466 new_taskdata->ompt_task_info.deps[ndeps+i].variable_addr =
467 (
void*) noalias_dep_list[i].base_addr;
468 if (noalias_dep_list[i].flags.in && noalias_dep_list[i].flags.out)
469 new_taskdata->ompt_task_info.deps[ndeps+i].dependence_flags =
470 ompt_task_dependence_type_inout;
471 else if (noalias_dep_list[i].flags.out)
472 new_taskdata->ompt_task_info.deps[ndeps+i].dependence_flags =
473 ompt_task_dependence_type_out;
474 else if (noalias_dep_list[i].flags.in)
475 new_taskdata->ompt_task_info.deps[ndeps+i].dependence_flags =
476 ompt_task_dependence_type_in;
481 bool serial = current_task->td_flags.team_serial || current_task->td_flags.tasking_ser || current_task->td_flags.final;
483 serial = serial && !(new_taskdata->td_flags.proxy == TASK_PROXY);
486 if ( !serial && ( ndeps > 0 || ndeps_noalias > 0 )) {
488 if ( current_task->td_dephash == NULL )
489 current_task->td_dephash = __kmp_dephash_create(thread, current_task);
492 kmp_depnode_t *node = (kmp_depnode_t *) __kmp_fast_allocate(thread,
sizeof(kmp_depnode_t));
494 kmp_depnode_t *node = (kmp_depnode_t *) __kmp_thread_malloc(thread,
sizeof(kmp_depnode_t));
497 __kmp_init_node(node);
498 new_taskdata->td_depnode = node;
500 if ( __kmp_check_deps( gtid, node, new_task, current_task->td_dephash, NO_DEP_BARRIER,
501 ndeps, dep_list, ndeps_noalias,noalias_dep_list ) ) {
502 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d task had blocking dependencies: " 503 "loc=%p task=%p, return: TASK_CURRENT_NOT_QUEUED\n", gtid, loc_ref,
505 return TASK_CURRENT_NOT_QUEUED;
509 kmp_task_team_t * task_team = thread->th.th_task_team;
510 if ( task_team && task_team->tt.tt_found_proxy_tasks )
514 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d ignored dependencies for task (serialized)" 515 "loc=%p task=%p\n", gtid, loc_ref, new_taskdata ) );
518 KA_TRACE(10, (
"__kmpc_omp_task_with_deps(exit): T#%d task had no blocking dependencies : " 519 "loc=%p task=%p, transferring to __kmpc_omp_task\n", gtid, loc_ref,
522 return __kmpc_omp_task(loc_ref,gtid,new_task);
538 kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list )
540 KA_TRACE(10, (
"__kmpc_omp_wait_deps(enter): T#%d loc=%p\n", gtid, loc_ref) );
542 if ( ndeps == 0 && ndeps_noalias == 0 ) {
543 KA_TRACE(10, (
"__kmpc_omp_wait_deps(exit): T#%d has no dependencies to wait upon : loc=%p\n", gtid, loc_ref) );
547 kmp_info_t *thread = __kmp_threads[ gtid ];
548 kmp_taskdata_t * current_task = thread->th.th_current_task;
553 bool ignore = current_task->td_flags.team_serial || current_task->td_flags.tasking_ser || current_task->td_flags.final;
555 ignore = ignore && thread->th.th_task_team != NULL && thread->th.th_task_team->tt.tt_found_proxy_tasks == FALSE;
557 ignore = ignore || current_task->td_dephash == NULL;
560 KA_TRACE(10, (
"__kmpc_omp_wait_deps(exit): T#%d has no blocking dependencies : loc=%p\n", gtid, loc_ref) );
565 __kmp_init_node(&node);
567 if (!__kmp_check_deps( gtid, &node, NULL, current_task->td_dephash, DEP_BARRIER,
568 ndeps, dep_list, ndeps_noalias, noalias_dep_list )) {
569 KA_TRACE(10, (
"__kmpc_omp_wait_deps(exit): T#%d has no blocking dependencies : loc=%p\n", gtid, loc_ref) );
573 int thread_finished = FALSE;
574 kmp_flag_32 flag((
volatile kmp_uint32 *)&(node.dn.npredecessors), 0U);
575 while ( node.dn.npredecessors > 0 ) {
576 flag.execute_tasks(thread, gtid, FALSE, &thread_finished,
580 __kmp_task_stealing_constraint );
583 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)