/* * kmp_taskdeps.cpp */ //===----------------------------------------------------------------------===// // // The LLVM Compiler Infrastructure // // This file is dual licensed under the MIT and the University of Illinois Open // Source Licenses. See LICENSE.txt for details. // //===----------------------------------------------------------------------===// //#define KMP_SUPPORT_GRAPH_OUTPUT 1 #include "kmp.h" #include "kmp_io.h" #include "kmp_wait_release.h" #if OMPT_SUPPORT #include "ompt-specific.h" #endif #if OMP_40_ENABLED // TODO: Improve memory allocation? keep a list of pre-allocated structures? // allocate in blocks? re-use list finished list entries? // TODO: don't use atomic ref counters for stack-allocated nodes. // TODO: find an alternate to atomic refs for heap-allocated nodes? // TODO: Finish graph output support // TODO: kmp_lock_t seems a tad to big (and heavy weight) for this. Check other // runtime locks // TODO: Any ITT support needed? #ifdef KMP_SUPPORT_GRAPH_OUTPUT static kmp_int32 kmp_node_id_seed = 0; #endif static void __kmp_init_node(kmp_depnode_t *node) { node->dn.task = NULL; // set to null initially, it will point to the right // task once dependences have been processed node->dn.successors = NULL; __kmp_init_lock(&node->dn.lock); node->dn.nrefs = 1; // init creates the first reference to the node #ifdef KMP_SUPPORT_GRAPH_OUTPUT node->dn.id = KMP_TEST_THEN_INC32(&kmp_node_id_seed); #endif } static inline kmp_depnode_t *__kmp_node_ref(kmp_depnode_t *node) { KMP_TEST_THEN_INC32(CCAST(kmp_int32 *, &node->dn.nrefs)); return node; } static inline void __kmp_node_deref(kmp_info_t *thread, kmp_depnode_t *node) { if (!node) return; kmp_int32 n = KMP_TEST_THEN_DEC32(CCAST(kmp_int32 *, &node->dn.nrefs)) - 1; if (n == 0) { KMP_ASSERT(node->dn.nrefs == 0); #if USE_FAST_MEMORY __kmp_fast_free(thread, node); #else __kmp_thread_free(thread, node); #endif } } #define KMP_ACQUIRE_DEPNODE(gtid, n) __kmp_acquire_lock(&(n)->dn.lock, (gtid)) #define KMP_RELEASE_DEPNODE(gtid, n) __kmp_release_lock(&(n)->dn.lock, (gtid)) static void __kmp_depnode_list_free(kmp_info_t *thread, kmp_depnode_list *list); enum { KMP_DEPHASH_OTHER_SIZE = 97, KMP_DEPHASH_MASTER_SIZE = 997 }; static inline kmp_int32 __kmp_dephash_hash(kmp_intptr_t addr, size_t hsize) { // TODO alternate to try: set = (((Addr64)(addrUsefulBits * 9.618)) % // m_num_sets ); return ((addr >> 6) ^ (addr >> 2)) % hsize; } static kmp_dephash_t *__kmp_dephash_create(kmp_info_t *thread, kmp_taskdata_t *current_task) { kmp_dephash_t *h; size_t h_size; if (current_task->td_flags.tasktype == TASK_IMPLICIT) h_size = KMP_DEPHASH_MASTER_SIZE; else h_size = KMP_DEPHASH_OTHER_SIZE; kmp_int32 size = h_size * sizeof(kmp_dephash_entry_t *) + sizeof(kmp_dephash_t); #if USE_FAST_MEMORY h = (kmp_dephash_t *)__kmp_fast_allocate(thread, size); #else h = (kmp_dephash_t *)__kmp_thread_malloc(thread, size); #endif h->size = h_size; #ifdef KMP_DEBUG h->nelements = 0; h->nconflicts = 0; #endif h->buckets = (kmp_dephash_entry **)(h + 1); for (size_t i = 0; i < h_size; i++) h->buckets[i] = 0; return h; } void __kmp_dephash_free_entries(kmp_info_t *thread, kmp_dephash_t *h) { for (size_t i = 0; i < h->size; i++) { if (h->buckets[i]) { kmp_dephash_entry_t *next; for (kmp_dephash_entry_t *entry = h->buckets[i]; entry; entry = next) { next = entry->next_in_bucket; __kmp_depnode_list_free(thread, entry->last_ins); __kmp_node_deref(thread, entry->last_out); #if USE_FAST_MEMORY __kmp_fast_free(thread, entry); #else __kmp_thread_free(thread, entry); #endif } h->buckets[i] = 0; } } } void __kmp_dephash_free(kmp_info_t *thread, kmp_dephash_t *h) { __kmp_dephash_free_entries(thread, h); #if USE_FAST_MEMORY __kmp_fast_free(thread, h); #else __kmp_thread_free(thread, h); #endif } static kmp_dephash_entry * __kmp_dephash_find(kmp_info_t *thread, kmp_dephash_t *h, kmp_intptr_t addr) { kmp_int32 bucket = __kmp_dephash_hash(addr, h->size); kmp_dephash_entry_t *entry; for (entry = h->buckets[bucket]; entry; entry = entry->next_in_bucket) if (entry->addr == addr) break; if (entry == NULL) { // create entry. This is only done by one thread so no locking required #if USE_FAST_MEMORY entry = (kmp_dephash_entry_t *)__kmp_fast_allocate( thread, sizeof(kmp_dephash_entry_t)); #else entry = (kmp_dephash_entry_t *)__kmp_thread_malloc( thread, sizeof(kmp_dephash_entry_t)); #endif entry->addr = addr; entry->last_out = NULL; entry->last_ins = NULL; entry->next_in_bucket = h->buckets[bucket]; h->buckets[bucket] = entry; #ifdef KMP_DEBUG h->nelements++; if (entry->next_in_bucket) h->nconflicts++; #endif } return entry; } static kmp_depnode_list_t *__kmp_add_node(kmp_info_t *thread, kmp_depnode_list_t *list, kmp_depnode_t *node) { kmp_depnode_list_t *new_head; #if USE_FAST_MEMORY new_head = (kmp_depnode_list_t *)__kmp_fast_allocate( thread, sizeof(kmp_depnode_list_t)); #else new_head = (kmp_depnode_list_t *)__kmp_thread_malloc( thread, sizeof(kmp_depnode_list_t)); #endif new_head->node = __kmp_node_ref(node); new_head->next = list; return new_head; } static void __kmp_depnode_list_free(kmp_info_t *thread, kmp_depnode_list *list) { kmp_depnode_list *next; for (; list; list = next) { next = list->next; __kmp_node_deref(thread, list->node); #if USE_FAST_MEMORY __kmp_fast_free(thread, list); #else __kmp_thread_free(thread, list); #endif } } static inline void __kmp_track_dependence(kmp_depnode_t *source, kmp_depnode_t *sink, kmp_task_t *sink_task) { #ifdef KMP_SUPPORT_GRAPH_OUTPUT kmp_taskdata_t *task_source = KMP_TASK_TO_TASKDATA(source->dn.task); // do not use sink->dn.task as that is only filled after the dependencies // are already processed! kmp_taskdata_t *task_sink = KMP_TASK_TO_TASKDATA(sink_task); __kmp_printf("%d(%s) -> %d(%s)\n", source->dn.id, task_source->td_ident->psource, sink->dn.id, task_sink->td_ident->psource); #endif #if OMPT_SUPPORT && OMPT_OPTIONAL /* OMPT tracks dependences between task (a=source, b=sink) in which task a blocks the execution of b through the ompt_new_dependence_callback */ if (ompt_enabled.ompt_callback_task_dependence) { kmp_taskdata_t *task_source = KMP_TASK_TO_TASKDATA(source->dn.task); kmp_taskdata_t *task_sink = KMP_TASK_TO_TASKDATA(sink_task); ompt_callbacks.ompt_callback(ompt_callback_task_dependence)( &(task_source->ompt_task_info.task_data), &(task_sink->ompt_task_info.task_data)); } #endif /* OMPT_SUPPORT && OMPT_OPTIONAL */ } template static inline kmp_int32 __kmp_process_deps(kmp_int32 gtid, kmp_depnode_t *node, kmp_dephash_t *hash, bool dep_barrier, kmp_int32 ndeps, kmp_depend_info_t *dep_list, kmp_task_t *task) { KA_TRACE(30, ("__kmp_process_deps<%d>: T#%d processing %d dependencies : " "dep_barrier = %d\n", filter, gtid, ndeps, dep_barrier)); kmp_info_t *thread = __kmp_threads[gtid]; kmp_int32 npredecessors = 0; for (kmp_int32 i = 0; i < ndeps; i++) { const kmp_depend_info_t *dep = &dep_list[i]; KMP_DEBUG_ASSERT(dep->flags.in); if (filter && dep->base_addr == 0) continue; // skip filtered entries kmp_dephash_entry_t *info = __kmp_dephash_find(thread, hash, dep->base_addr); kmp_depnode_t *last_out = info->last_out; if (dep->flags.out && info->last_ins) { for (kmp_depnode_list_t *p = info->last_ins; p; p = p->next) { kmp_depnode_t *indep = p->node; if (indep->dn.task) { KMP_ACQUIRE_DEPNODE(gtid, indep); if (indep->dn.task) { __kmp_track_dependence(indep, node, task); indep->dn.successors = __kmp_add_node(thread, indep->dn.successors, node); KA_TRACE(40, ("__kmp_process_deps<%d>: T#%d adding dependence from " "%p to %p\n", filter, gtid, KMP_TASK_TO_TASKDATA(indep->dn.task), KMP_TASK_TO_TASKDATA(task))); npredecessors++; } KMP_RELEASE_DEPNODE(gtid, indep); } } __kmp_depnode_list_free(thread, info->last_ins); info->last_ins = NULL; } else if (last_out && last_out->dn.task) { KMP_ACQUIRE_DEPNODE(gtid, last_out); if (last_out->dn.task) { __kmp_track_dependence(last_out, node, task); last_out->dn.successors = __kmp_add_node(thread, last_out->dn.successors, node); KA_TRACE( 40, ("__kmp_process_deps<%d>: T#%d adding dependence from %p to %p\n", filter, gtid, KMP_TASK_TO_TASKDATA(last_out->dn.task), KMP_TASK_TO_TASKDATA(task))); npredecessors++; } KMP_RELEASE_DEPNODE(gtid, last_out); } if (dep_barrier) { // if this is a sync point in the serial sequence, then the previous // outputs are guaranteed to be completed after // the execution of this task so the previous output nodes can be cleared. __kmp_node_deref(thread, last_out); info->last_out = NULL; } else { if (dep->flags.out) { __kmp_node_deref(thread, last_out); info->last_out = __kmp_node_ref(node); } else info->last_ins = __kmp_add_node(thread, info->last_ins, node); } } KA_TRACE(30, ("__kmp_process_deps<%d>: T#%d found %d predecessors\n", filter, gtid, npredecessors)); return npredecessors; } #define NO_DEP_BARRIER (false) #define DEP_BARRIER (true) // returns true if the task has any outstanding dependence static bool __kmp_check_deps(kmp_int32 gtid, kmp_depnode_t *node, kmp_task_t *task, kmp_dephash_t *hash, bool dep_barrier, kmp_int32 ndeps, kmp_depend_info_t *dep_list, kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list) { int i; #if KMP_DEBUG kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task); #endif 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)); // Filter deps in dep_list // TODO: Different algorithm for large dep_list ( > 10 ? ) for (i = 0; i < ndeps; i++) { if (dep_list[i].base_addr != 0) for (int j = i + 1; j < ndeps; j++) if (dep_list[i].base_addr == dep_list[j].base_addr) { dep_list[i].flags.in |= dep_list[j].flags.in; dep_list[i].flags.out |= dep_list[j].flags.out; dep_list[j].base_addr = 0; // Mark j element as void } } // doesn't need to be atomic as no other thread is going to be accessing this // node just yet. // npredecessors is set -1 to ensure that none of the releasing tasks queues // this task before we have finished processing all the dependencies node->dn.npredecessors = -1; // used to pack all npredecessors additions into a single atomic operation at // the end int npredecessors; npredecessors = __kmp_process_deps(gtid, node, hash, dep_barrier, ndeps, dep_list, task); npredecessors += __kmp_process_deps( gtid, node, hash, dep_barrier, ndeps_noalias, noalias_dep_list, task); node->dn.task = task; KMP_MB(); // Account for our initial fake value npredecessors++; // Update predecessors and obtain current value to check if there are still // any outstandig dependences (some tasks may have finished while we processed // the dependences) npredecessors = KMP_TEST_THEN_ADD32(CCAST(kmp_int32 *, &node->dn.npredecessors), npredecessors) + npredecessors; KA_TRACE(20, ("__kmp_check_deps: T#%d found %d predecessors for task %p \n", gtid, npredecessors, taskdata)); // beyond this point the task could be queued (and executed) by a releasing // task... return npredecessors > 0 ? true : false; } void __kmp_release_deps(kmp_int32 gtid, kmp_taskdata_t *task) { kmp_info_t *thread = __kmp_threads[gtid]; kmp_depnode_t *node = task->td_depnode; if (task->td_dephash) { KA_TRACE( 40, ("__kmp_release_deps: T#%d freeing dependencies hash of task %p.\n", gtid, task)); __kmp_dephash_free(thread, task->td_dephash); task->td_dephash = NULL; } if (!node) return; KA_TRACE(20, ("__kmp_release_deps: T#%d notifying successors of task %p.\n", gtid, task)); KMP_ACQUIRE_DEPNODE(gtid, node); node->dn.task = NULL; // mark this task as finished, so no new dependencies are generated KMP_RELEASE_DEPNODE(gtid, node); kmp_depnode_list_t *next; for (kmp_depnode_list_t *p = node->dn.successors; p; p = next) { kmp_depnode_t *successor = p->node; kmp_int32 npredecessors = KMP_TEST_THEN_DEC32(CCAST(kmp_int32 *, &successor->dn.npredecessors)) - 1; // successor task can be NULL for wait_depends or because deps are still // being processed if (npredecessors == 0) { KMP_MB(); if (successor->dn.task) { KA_TRACE(20, ("__kmp_release_deps: T#%d successor %p of %p scheduled " "for execution.\n", gtid, successor->dn.task, task)); __kmp_omp_task(gtid, successor->dn.task, false); } } next = p->next; __kmp_node_deref(thread, p->node); #if USE_FAST_MEMORY __kmp_fast_free(thread, p); #else __kmp_thread_free(thread, p); #endif } __kmp_node_deref(thread, node); KA_TRACE( 20, ("__kmp_release_deps: T#%d all successors of %p notified of completion\n", gtid, task)); } /*! @ingroup TASKING @param loc_ref location of the original task directive @param gtid Global Thread ID of encountering thread @param new_task task thunk allocated by __kmp_omp_task_alloc() for the ''new task'' @param ndeps Number of depend items with possible aliasing @param dep_list List of depend items with possible aliasing @param ndeps_noalias Number of depend items with no aliasing @param noalias_dep_list List of depend items with no aliasing @return Returns either TASK_CURRENT_NOT_QUEUED if the current task was not suspendend and queued, or TASK_CURRENT_QUEUED if it was suspended and queued Schedule a non-thread-switchable task with dependences for execution */ 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) { kmp_taskdata_t *new_taskdata = KMP_TASK_TO_TASKDATA(new_task); KA_TRACE(10, ("__kmpc_omp_task_with_deps(enter): T#%d loc=%p task=%p\n", gtid, loc_ref, new_taskdata)); kmp_info_t *thread = __kmp_threads[gtid]; kmp_taskdata_t *current_task = thread->th.th_current_task; #if OMPT_SUPPORT OMPT_STORE_RETURN_ADDRESS(gtid); if (ompt_enabled.enabled) { if (ompt_enabled.ompt_callback_task_create) { kmp_taskdata_t *parent = new_taskdata->td_parent; ompt_data_t task_data = ompt_data_none; ompt_callbacks.ompt_callback(ompt_callback_task_create)( parent ? &(parent->ompt_task_info.task_data) : &task_data, parent ? &(parent->ompt_task_info.frame) : NULL, &(new_taskdata->ompt_task_info.task_data), ompt_task_explicit | TASK_TYPE_DETAILS_FORMAT(new_taskdata), 1, OMPT_LOAD_RETURN_ADDRESS(gtid)); } new_taskdata->ompt_task_info.frame.enter_frame = OMPT_GET_FRAME_ADDRESS(0); } #if OMPT_OPTIONAL /* OMPT grab all dependences if requested by the tool */ if (ndeps + ndeps_noalias > 0 && ompt_enabled.ompt_callback_task_dependences) { kmp_int32 i; new_taskdata->ompt_task_info.ndeps = ndeps + ndeps_noalias; new_taskdata->ompt_task_info.deps = (ompt_task_dependence_t *)KMP_OMPT_DEPS_ALLOC( thread, (ndeps + ndeps_noalias) * sizeof(ompt_task_dependence_t)); KMP_ASSERT(new_taskdata->ompt_task_info.deps != NULL); for (i = 0; i < ndeps; i++) { new_taskdata->ompt_task_info.deps[i].variable_addr = (void *)dep_list[i].base_addr; if (dep_list[i].flags.in && dep_list[i].flags.out) new_taskdata->ompt_task_info.deps[i].dependence_flags = ompt_task_dependence_type_inout; else if (dep_list[i].flags.out) new_taskdata->ompt_task_info.deps[i].dependence_flags = ompt_task_dependence_type_out; else if (dep_list[i].flags.in) new_taskdata->ompt_task_info.deps[i].dependence_flags = ompt_task_dependence_type_in; } for (i = 0; i < ndeps_noalias; i++) { new_taskdata->ompt_task_info.deps[ndeps + i].variable_addr = (void *)noalias_dep_list[i].base_addr; if (noalias_dep_list[i].flags.in && noalias_dep_list[i].flags.out) new_taskdata->ompt_task_info.deps[ndeps + i].dependence_flags = ompt_task_dependence_type_inout; else if (noalias_dep_list[i].flags.out) new_taskdata->ompt_task_info.deps[ndeps + i].dependence_flags = ompt_task_dependence_type_out; else if (noalias_dep_list[i].flags.in) new_taskdata->ompt_task_info.deps[ndeps + i].dependence_flags = ompt_task_dependence_type_in; } ompt_callbacks.ompt_callback(ompt_callback_task_dependences)( &(new_taskdata->ompt_task_info.task_data), new_taskdata->ompt_task_info.deps, new_taskdata->ompt_task_info.ndeps); /* We can now free the allocated memory for the dependencies */ /* For OMPD we might want to delay the free until task_end */ KMP_OMPT_DEPS_FREE(thread, new_taskdata->ompt_task_info.deps); new_taskdata->ompt_task_info.deps = NULL; new_taskdata->ompt_task_info.ndeps = 0; } #endif /* OMPT_OPTIONAL */ #endif /* OMPT_SUPPORT */ bool serial = current_task->td_flags.team_serial || current_task->td_flags.tasking_ser || current_task->td_flags.final; #if OMP_45_ENABLED kmp_task_team_t *task_team = thread->th.th_task_team; serial = serial && !(task_team && task_team->tt.tt_found_proxy_tasks); #endif if (!serial && (ndeps > 0 || ndeps_noalias > 0)) { /* if no dependencies have been tracked yet, create the dependence hash */ if (current_task->td_dephash == NULL) current_task->td_dephash = __kmp_dephash_create(thread, current_task); #if USE_FAST_MEMORY kmp_depnode_t *node = (kmp_depnode_t *)__kmp_fast_allocate(thread, sizeof(kmp_depnode_t)); #else kmp_depnode_t *node = (kmp_depnode_t *)__kmp_thread_malloc(thread, sizeof(kmp_depnode_t)); #endif __kmp_init_node(node); new_taskdata->td_depnode = node; if (__kmp_check_deps(gtid, node, new_task, current_task->td_dephash, NO_DEP_BARRIER, ndeps, dep_list, ndeps_noalias, noalias_dep_list)) { KA_TRACE(10, ("__kmpc_omp_task_with_deps(exit): T#%d task had blocking " "dependencies: " "loc=%p task=%p, return: TASK_CURRENT_NOT_QUEUED\n", gtid, loc_ref, new_taskdata)); return TASK_CURRENT_NOT_QUEUED; } } else { KA_TRACE(10, ("__kmpc_omp_task_with_deps(exit): T#%d ignored dependencies " "for task (serialized)" "loc=%p task=%p\n", gtid, loc_ref, new_taskdata)); } KA_TRACE(10, ("__kmpc_omp_task_with_deps(exit): T#%d task had no blocking " "dependencies : " "loc=%p task=%p, transferring to __kmpc_omp_task\n", gtid, loc_ref, new_taskdata)); return __kmp_omp_task(gtid, new_task, true); } /*! @ingroup TASKING @param loc_ref location of the original task directive @param gtid Global Thread ID of encountering thread @param ndeps Number of depend items with possible aliasing @param dep_list List of depend items with possible aliasing @param ndeps_noalias Number of depend items with no aliasing @param noalias_dep_list List of depend items with no aliasing Blocks the current task until all specifies dependencies have been fulfilled. */ 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) { KA_TRACE(10, ("__kmpc_omp_wait_deps(enter): T#%d loc=%p\n", gtid, loc_ref)); if (ndeps == 0 && ndeps_noalias == 0) { KA_TRACE(10, ("__kmpc_omp_wait_deps(exit): T#%d has no dependencies to " "wait upon : loc=%p\n", gtid, loc_ref)); return; } kmp_info_t *thread = __kmp_threads[gtid]; kmp_taskdata_t *current_task = thread->th.th_current_task; // We can return immediately as: // - dependences are not computed in serial teams (except with proxy tasks) // - if the dephash is not yet created it means we have nothing to wait for bool ignore = current_task->td_flags.team_serial || current_task->td_flags.tasking_ser || current_task->td_flags.final; #if OMP_45_ENABLED ignore = ignore && thread->th.th_task_team != NULL && thread->th.th_task_team->tt.tt_found_proxy_tasks == FALSE; #endif ignore = ignore || current_task->td_dephash == NULL; if (ignore) { KA_TRACE(10, ("__kmpc_omp_wait_deps(exit): T#%d has no blocking " "dependencies : loc=%p\n", gtid, loc_ref)); return; } kmp_depnode_t node; __kmp_init_node(&node); if (!__kmp_check_deps(gtid, &node, NULL, current_task->td_dephash, DEP_BARRIER, ndeps, dep_list, ndeps_noalias, noalias_dep_list)) { KA_TRACE(10, ("__kmpc_omp_wait_deps(exit): T#%d has no blocking " "dependencies : loc=%p\n", gtid, loc_ref)); return; } int thread_finished = FALSE; kmp_flag_32 flag((volatile kmp_uint32 *)&(node.dn.npredecessors), 0U); while (node.dn.npredecessors > 0) { flag.execute_tasks(thread, gtid, FALSE, &thread_finished, #if USE_ITT_BUILD NULL, #endif __kmp_task_stealing_constraint); } KA_TRACE(10, ("__kmpc_omp_wait_deps(exit): T#%d finished waiting : loc=%p\n", gtid, loc_ref)); } #endif /* OMP_40_ENABLED */