17cc577a4SJonathan Peyton /* 2de4749b7SJonathan Peyton * kmp_tasking.cpp -- OpenMP 3.0 tasking support. 37cc577a4SJonathan Peyton */ 47cc577a4SJonathan Peyton 57cc577a4SJonathan Peyton 67cc577a4SJonathan Peyton //===----------------------------------------------------------------------===// 77cc577a4SJonathan Peyton // 87cc577a4SJonathan Peyton // The LLVM Compiler Infrastructure 97cc577a4SJonathan Peyton // 107cc577a4SJonathan Peyton // This file is dual licensed under the MIT and the University of Illinois Open 117cc577a4SJonathan Peyton // Source Licenses. See LICENSE.txt for details. 127cc577a4SJonathan Peyton // 137cc577a4SJonathan Peyton //===----------------------------------------------------------------------===// 147cc577a4SJonathan Peyton 157cc577a4SJonathan Peyton 167cc577a4SJonathan Peyton #include "kmp.h" 177cc577a4SJonathan Peyton #include "kmp_i18n.h" 187cc577a4SJonathan Peyton #include "kmp_itt.h" 197cc577a4SJonathan Peyton #include "kmp_stats.h" 203041982dSJonathan Peyton #include "kmp_wait_release.h" 217cc577a4SJonathan Peyton 227cc577a4SJonathan Peyton #if OMPT_SUPPORT 237cc577a4SJonathan Peyton #include "ompt-specific.h" 247cc577a4SJonathan Peyton #endif 257cc577a4SJonathan Peyton 267cc577a4SJonathan Peyton #include "tsan_annotations.h" 277cc577a4SJonathan Peyton 287cc577a4SJonathan Peyton /* forward declaration */ 293041982dSJonathan Peyton static void __kmp_enable_tasking(kmp_task_team_t *task_team, 303041982dSJonathan Peyton kmp_info_t *this_thr); 313041982dSJonathan Peyton static void __kmp_alloc_task_deque(kmp_info_t *thread, 323041982dSJonathan Peyton kmp_thread_data_t *thread_data); 333041982dSJonathan Peyton static int __kmp_realloc_task_threads_data(kmp_info_t *thread, 343041982dSJonathan Peyton kmp_task_team_t *task_team); 357cc577a4SJonathan Peyton 367cc577a4SJonathan Peyton #ifdef OMP_45_ENABLED 377cc577a4SJonathan Peyton static void __kmp_bottom_half_finish_proxy(kmp_int32 gtid, kmp_task_t *ptask); 387cc577a4SJonathan Peyton #endif 397cc577a4SJonathan Peyton 407cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK 417cc577a4SJonathan Peyton 427cc577a4SJonathan Peyton // __kmp_trace_task_stack: print the tied tasks from the task stack in order 437cc577a4SJonathan Peyton // from top do bottom 447cc577a4SJonathan Peyton // 457cc577a4SJonathan Peyton // gtid: global thread identifier for thread containing stack 467cc577a4SJonathan Peyton // thread_data: thread data for task team thread containing stack 477cc577a4SJonathan Peyton // threshold: value above which the trace statement triggers 487cc577a4SJonathan Peyton // location: string identifying call site of this function (for trace) 493041982dSJonathan Peyton static void __kmp_trace_task_stack(kmp_int32 gtid, 503041982dSJonathan Peyton kmp_thread_data_t *thread_data, 513041982dSJonathan Peyton int threshold, char *location) { 527cc577a4SJonathan Peyton kmp_task_stack_t *task_stack = &thread_data->td.td_susp_tied_tasks; 537cc577a4SJonathan Peyton kmp_taskdata_t **stack_top = task_stack->ts_top; 547cc577a4SJonathan Peyton kmp_int32 entries = task_stack->ts_entries; 557cc577a4SJonathan Peyton kmp_taskdata_t *tied_task; 567cc577a4SJonathan Peyton 573041982dSJonathan Peyton KA_TRACE( 583041982dSJonathan Peyton threshold, 593041982dSJonathan Peyton ("__kmp_trace_task_stack(start): location = %s, gtid = %d, entries = %d, " 607cc577a4SJonathan Peyton "first_block = %p, stack_top = %p \n", 617cc577a4SJonathan Peyton location, gtid, entries, task_stack->ts_first_block, stack_top)); 627cc577a4SJonathan Peyton 637cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(stack_top != NULL); 647cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(entries > 0); 657cc577a4SJonathan Peyton 663041982dSJonathan Peyton while (entries != 0) { 677cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(stack_top != &task_stack->ts_first_block.sb_block[0]); 687cc577a4SJonathan Peyton // fix up ts_top if we need to pop from previous block 693041982dSJonathan Peyton if (entries & TASK_STACK_INDEX_MASK == 0) { 707cc577a4SJonathan Peyton kmp_stack_block_t *stack_block = (kmp_stack_block_t *)(stack_top); 717cc577a4SJonathan Peyton 727cc577a4SJonathan Peyton stack_block = stack_block->sb_prev; 737cc577a4SJonathan Peyton stack_top = &stack_block->sb_block[TASK_STACK_BLOCK_SIZE]; 747cc577a4SJonathan Peyton } 757cc577a4SJonathan Peyton 767cc577a4SJonathan Peyton // finish bookkeeping 777cc577a4SJonathan Peyton stack_top--; 787cc577a4SJonathan Peyton entries--; 797cc577a4SJonathan Peyton 807cc577a4SJonathan Peyton tied_task = *stack_top; 817cc577a4SJonathan Peyton 827cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(tied_task != NULL); 837cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(tied_task->td_flags.tasktype == TASK_TIED); 847cc577a4SJonathan Peyton 853041982dSJonathan Peyton KA_TRACE(threshold, 863041982dSJonathan Peyton ("__kmp_trace_task_stack(%s): gtid=%d, entry=%d, " 877cc577a4SJonathan Peyton "stack_top=%p, tied_task=%p\n", 887cc577a4SJonathan Peyton location, gtid, entries, stack_top, tied_task)); 897cc577a4SJonathan Peyton } 907cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(stack_top == &task_stack->ts_first_block.sb_block[0]); 917cc577a4SJonathan Peyton 923041982dSJonathan Peyton KA_TRACE(threshold, 933041982dSJonathan Peyton ("__kmp_trace_task_stack(exit): location = %s, gtid = %d\n", 947cc577a4SJonathan Peyton location, gtid)); 957cc577a4SJonathan Peyton } 967cc577a4SJonathan Peyton 977cc577a4SJonathan Peyton // __kmp_init_task_stack: initialize the task stack for the first time 987cc577a4SJonathan Peyton // after a thread_data structure is created. 997cc577a4SJonathan Peyton // It should not be necessary to do this again (assuming the stack works). 1007cc577a4SJonathan Peyton // 1017cc577a4SJonathan Peyton // gtid: global thread identifier of calling thread 1027cc577a4SJonathan Peyton // thread_data: thread data for task team thread containing stack 1033041982dSJonathan Peyton static void __kmp_init_task_stack(kmp_int32 gtid, 1043041982dSJonathan Peyton kmp_thread_data_t *thread_data) { 1057cc577a4SJonathan Peyton kmp_task_stack_t *task_stack = &thread_data->td.td_susp_tied_tasks; 1067cc577a4SJonathan Peyton kmp_stack_block_t *first_block; 1077cc577a4SJonathan Peyton 1087cc577a4SJonathan Peyton // set up the first block of the stack 1097cc577a4SJonathan Peyton first_block = &task_stack->ts_first_block; 1107cc577a4SJonathan Peyton task_stack->ts_top = (kmp_taskdata_t **)first_block; 1113041982dSJonathan Peyton memset((void *)first_block, '\0', 1123041982dSJonathan Peyton TASK_STACK_BLOCK_SIZE * sizeof(kmp_taskdata_t *)); 1137cc577a4SJonathan Peyton 1147cc577a4SJonathan Peyton // initialize the stack to be empty 1157cc577a4SJonathan Peyton task_stack->ts_entries = TASK_STACK_EMPTY; 1167cc577a4SJonathan Peyton first_block->sb_next = NULL; 1177cc577a4SJonathan Peyton first_block->sb_prev = NULL; 1187cc577a4SJonathan Peyton } 1197cc577a4SJonathan Peyton 1207cc577a4SJonathan Peyton // __kmp_free_task_stack: free the task stack when thread_data is destroyed. 1217cc577a4SJonathan Peyton // 1227cc577a4SJonathan Peyton // gtid: global thread identifier for calling thread 1237cc577a4SJonathan Peyton // thread_data: thread info for thread containing stack 1243041982dSJonathan Peyton static void __kmp_free_task_stack(kmp_int32 gtid, 1253041982dSJonathan Peyton kmp_thread_data_t *thread_data) { 1267cc577a4SJonathan Peyton kmp_task_stack_t *task_stack = &thread_data->td.td_susp_tied_tasks; 1277cc577a4SJonathan Peyton kmp_stack_block_t *stack_block = &task_stack->ts_first_block; 1287cc577a4SJonathan Peyton 1297cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_stack->ts_entries == TASK_STACK_EMPTY); 1307cc577a4SJonathan Peyton // free from the second block of the stack 1317cc577a4SJonathan Peyton while (stack_block != NULL) { 1327cc577a4SJonathan Peyton kmp_stack_block_t *next_block = (stack_block) ? stack_block->sb_next : NULL; 1337cc577a4SJonathan Peyton 1347cc577a4SJonathan Peyton stack_block->sb_next = NULL; 1357cc577a4SJonathan Peyton stack_block->sb_prev = NULL; 1367cc577a4SJonathan Peyton if (stack_block != &task_stack->ts_first_block) { 1373041982dSJonathan Peyton __kmp_thread_free(thread, 1383041982dSJonathan Peyton stack_block); // free the block, if not the first 1397cc577a4SJonathan Peyton } 1407cc577a4SJonathan Peyton stack_block = next_block; 1417cc577a4SJonathan Peyton } 1427cc577a4SJonathan Peyton // initialize the stack to be empty 1437cc577a4SJonathan Peyton task_stack->ts_entries = 0; 1447cc577a4SJonathan Peyton task_stack->ts_top = NULL; 1457cc577a4SJonathan Peyton } 1467cc577a4SJonathan Peyton 1477cc577a4SJonathan Peyton // __kmp_push_task_stack: Push the tied task onto the task stack. 1487cc577a4SJonathan Peyton // Grow the stack if necessary by allocating another block. 1497cc577a4SJonathan Peyton // 1507cc577a4SJonathan Peyton // gtid: global thread identifier for calling thread 1517cc577a4SJonathan Peyton // thread: thread info for thread containing stack 1527cc577a4SJonathan Peyton // tied_task: the task to push on the stack 1533041982dSJonathan Peyton static void __kmp_push_task_stack(kmp_int32 gtid, kmp_info_t *thread, 1543041982dSJonathan Peyton kmp_taskdata_t *tied_task) { 1557cc577a4SJonathan Peyton // GEH - need to consider what to do if tt_threads_data not allocated yet 1563041982dSJonathan Peyton kmp_thread_data_t *thread_data = 1573041982dSJonathan Peyton &thread->th.th_task_team->tt.tt_threads_data[__kmp_tid_from_gtid(gtid)]; 1587cc577a4SJonathan Peyton kmp_task_stack_t *task_stack = &thread_data->td.td_susp_tied_tasks; 1597cc577a4SJonathan Peyton 1607cc577a4SJonathan Peyton if (tied_task->td_flags.team_serial || tied_task->td_flags.tasking_ser) { 1617cc577a4SJonathan Peyton return; // Don't push anything on stack if team or team tasks are serialized 1627cc577a4SJonathan Peyton } 1637cc577a4SJonathan Peyton 1647cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(tied_task->td_flags.tasktype == TASK_TIED); 1657cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_stack->ts_top != NULL); 1667cc577a4SJonathan Peyton 1673041982dSJonathan Peyton KA_TRACE(20, 1683041982dSJonathan Peyton ("__kmp_push_task_stack(enter): GTID: %d; THREAD: %p; TASK: %p\n", 1697cc577a4SJonathan Peyton gtid, thread, tied_task)); 1707cc577a4SJonathan Peyton // Store entry 1717cc577a4SJonathan Peyton *(task_stack->ts_top) = tied_task; 1727cc577a4SJonathan Peyton 1737cc577a4SJonathan Peyton // Do bookkeeping for next push 1747cc577a4SJonathan Peyton task_stack->ts_top++; 1757cc577a4SJonathan Peyton task_stack->ts_entries++; 1767cc577a4SJonathan Peyton 1773041982dSJonathan Peyton if (task_stack->ts_entries & TASK_STACK_INDEX_MASK == 0) { 1787cc577a4SJonathan Peyton // Find beginning of this task block 1797cc577a4SJonathan Peyton kmp_stack_block_t *stack_block = 1807cc577a4SJonathan Peyton (kmp_stack_block_t *)(task_stack->ts_top - TASK_STACK_BLOCK_SIZE); 1817cc577a4SJonathan Peyton 1827cc577a4SJonathan Peyton // Check if we already have a block 1833041982dSJonathan Peyton if (stack_block->sb_next != 1843041982dSJonathan Peyton NULL) { // reset ts_top to beginning of next block 1857cc577a4SJonathan Peyton task_stack->ts_top = &stack_block->sb_next->sb_block[0]; 1863041982dSJonathan Peyton } else { // Alloc new block and link it up 1873041982dSJonathan Peyton kmp_stack_block_t *new_block = (kmp_stack_block_t *)__kmp_thread_calloc( 1883041982dSJonathan Peyton thread, sizeof(kmp_stack_block_t)); 1897cc577a4SJonathan Peyton 1907cc577a4SJonathan Peyton task_stack->ts_top = &new_block->sb_block[0]; 1917cc577a4SJonathan Peyton stack_block->sb_next = new_block; 1927cc577a4SJonathan Peyton new_block->sb_prev = stack_block; 1937cc577a4SJonathan Peyton new_block->sb_next = NULL; 1947cc577a4SJonathan Peyton 1953041982dSJonathan Peyton KA_TRACE( 1963041982dSJonathan Peyton 30, 1973041982dSJonathan Peyton ("__kmp_push_task_stack(): GTID: %d; TASK: %p; Alloc new block: %p\n", 1987cc577a4SJonathan Peyton gtid, tied_task, new_block)); 1997cc577a4SJonathan Peyton } 2007cc577a4SJonathan Peyton } 2013041982dSJonathan Peyton KA_TRACE(20, ("__kmp_push_task_stack(exit): GTID: %d; TASK: %p\n", gtid, 2023041982dSJonathan Peyton tied_task)); 2037cc577a4SJonathan Peyton } 2047cc577a4SJonathan Peyton 2057cc577a4SJonathan Peyton // __kmp_pop_task_stack: Pop the tied task from the task stack. Don't return 2067cc577a4SJonathan Peyton // the task, just check to make sure it matches the ending task passed in. 2077cc577a4SJonathan Peyton // 2087cc577a4SJonathan Peyton // gtid: global thread identifier for the calling thread 2097cc577a4SJonathan Peyton // thread: thread info structure containing stack 2107cc577a4SJonathan Peyton // tied_task: the task popped off the stack 2117cc577a4SJonathan Peyton // ending_task: the task that is ending (should match popped task) 2123041982dSJonathan Peyton static void __kmp_pop_task_stack(kmp_int32 gtid, kmp_info_t *thread, 2133041982dSJonathan Peyton kmp_taskdata_t *ending_task) { 2147cc577a4SJonathan Peyton // GEH - need to consider what to do if tt_threads_data not allocated yet 2153041982dSJonathan Peyton kmp_thread_data_t *thread_data = 2163041982dSJonathan Peyton &thread->th.th_task_team->tt_threads_data[__kmp_tid_from_gtid(gtid)]; 2177cc577a4SJonathan Peyton kmp_task_stack_t *task_stack = &thread_data->td.td_susp_tied_tasks; 2187cc577a4SJonathan Peyton kmp_taskdata_t *tied_task; 2197cc577a4SJonathan Peyton 2207cc577a4SJonathan Peyton if (ending_task->td_flags.team_serial || ending_task->td_flags.tasking_ser) { 2213041982dSJonathan Peyton // Don't pop anything from stack if team or team tasks are serialized 2223041982dSJonathan Peyton return; 2237cc577a4SJonathan Peyton } 2247cc577a4SJonathan Peyton 2257cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_stack->ts_top != NULL); 2267cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_stack->ts_entries > 0); 2277cc577a4SJonathan Peyton 2283041982dSJonathan Peyton KA_TRACE(20, ("__kmp_pop_task_stack(enter): GTID: %d; THREAD: %p\n", gtid, 2293041982dSJonathan Peyton thread)); 2307cc577a4SJonathan Peyton 2317cc577a4SJonathan Peyton // fix up ts_top if we need to pop from previous block 2323041982dSJonathan Peyton if (task_stack->ts_entries & TASK_STACK_INDEX_MASK == 0) { 2333041982dSJonathan Peyton kmp_stack_block_t *stack_block = (kmp_stack_block_t *)(task_stack->ts_top); 2347cc577a4SJonathan Peyton 2357cc577a4SJonathan Peyton stack_block = stack_block->sb_prev; 2367cc577a4SJonathan Peyton task_stack->ts_top = &stack_block->sb_block[TASK_STACK_BLOCK_SIZE]; 2377cc577a4SJonathan Peyton } 2387cc577a4SJonathan Peyton 2397cc577a4SJonathan Peyton // finish bookkeeping 2407cc577a4SJonathan Peyton task_stack->ts_top--; 2417cc577a4SJonathan Peyton task_stack->ts_entries--; 2427cc577a4SJonathan Peyton 2437cc577a4SJonathan Peyton tied_task = *(task_stack->ts_top); 2447cc577a4SJonathan Peyton 2457cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(tied_task != NULL); 2467cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(tied_task->td_flags.tasktype == TASK_TIED); 2477cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(tied_task == ending_task); // If we built the stack correctly 2487cc577a4SJonathan Peyton 2493041982dSJonathan Peyton KA_TRACE(20, ("__kmp_pop_task_stack(exit): GTID: %d; TASK: %p\n", gtid, 2503041982dSJonathan Peyton tied_task)); 2517cc577a4SJonathan Peyton return; 2527cc577a4SJonathan Peyton } 2537cc577a4SJonathan Peyton #endif /* BUILD_TIED_TASK_STACK */ 2547cc577a4SJonathan Peyton 2557cc577a4SJonathan Peyton // __kmp_push_task: Add a task to the thread's deque 2563041982dSJonathan Peyton static kmp_int32 __kmp_push_task(kmp_int32 gtid, kmp_task_t *task) { 2577cc577a4SJonathan Peyton kmp_info_t *thread = __kmp_threads[gtid]; 2587cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task); 2597cc577a4SJonathan Peyton kmp_task_team_t *task_team = thread->th.th_task_team; 2607cc577a4SJonathan Peyton kmp_int32 tid = __kmp_tid_from_gtid(gtid); 2617cc577a4SJonathan Peyton kmp_thread_data_t *thread_data; 2627cc577a4SJonathan Peyton 2633041982dSJonathan Peyton KA_TRACE(20, 2643041982dSJonathan Peyton ("__kmp_push_task: T#%d trying to push task %p.\n", gtid, taskdata)); 2657cc577a4SJonathan Peyton 2667cc577a4SJonathan Peyton if (taskdata->td_flags.tiedness == TASK_UNTIED) { 2673041982dSJonathan Peyton // untied task needs to increment counter so that the task structure is not 2683041982dSJonathan Peyton // freed prematurely 2697cc577a4SJonathan Peyton kmp_int32 counter = 1 + KMP_TEST_THEN_INC32(&taskdata->td_untied_count); 2703041982dSJonathan Peyton KA_TRACE( 2713041982dSJonathan Peyton 20, 2723041982dSJonathan Peyton ("__kmp_push_task: T#%d untied_count (%d) incremented for task %p\n", 2737cc577a4SJonathan Peyton gtid, counter, taskdata)); 2747cc577a4SJonathan Peyton } 2757cc577a4SJonathan Peyton 2767cc577a4SJonathan Peyton // The first check avoids building task_team thread data if serialized 2777cc577a4SJonathan Peyton if (taskdata->td_flags.task_serial) { 2783041982dSJonathan Peyton KA_TRACE(20, ("__kmp_push_task: T#%d team serialized; returning " 2793041982dSJonathan Peyton "TASK_NOT_PUSHED for task %p\n", 2807cc577a4SJonathan Peyton gtid, taskdata)); 2817cc577a4SJonathan Peyton return TASK_NOT_PUSHED; 2827cc577a4SJonathan Peyton } 2837cc577a4SJonathan Peyton 2843041982dSJonathan Peyton // Now that serialized tasks have returned, we can assume that we are not in 2853041982dSJonathan Peyton // immediate exec mode 2867cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(__kmp_tasking_mode != tskm_immediate_exec); 2877cc577a4SJonathan Peyton if (!KMP_TASKING_ENABLED(task_team)) { 2887cc577a4SJonathan Peyton __kmp_enable_tasking(task_team, thread); 2897cc577a4SJonathan Peyton } 2907cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(TCR_4(task_team->tt.tt_found_tasks) == TRUE); 2917cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(TCR_PTR(task_team->tt.tt_threads_data) != NULL); 2927cc577a4SJonathan Peyton 2937cc577a4SJonathan Peyton // Find tasking deque specific to encountering thread 2947cc577a4SJonathan Peyton thread_data = &task_team->tt.tt_threads_data[tid]; 2957cc577a4SJonathan Peyton 2967cc577a4SJonathan Peyton // No lock needed since only owner can allocate 2977cc577a4SJonathan Peyton if (thread_data->td.td_deque == NULL) { 2987cc577a4SJonathan Peyton __kmp_alloc_task_deque(thread, thread_data); 2997cc577a4SJonathan Peyton } 3007cc577a4SJonathan Peyton 3017cc577a4SJonathan Peyton // Check if deque is full 3023041982dSJonathan Peyton if (TCR_4(thread_data->td.td_deque_ntasks) >= 3033041982dSJonathan Peyton TASK_DEQUE_SIZE(thread_data->td)) { 3043041982dSJonathan Peyton KA_TRACE(20, ("__kmp_push_task: T#%d deque is full; returning " 3053041982dSJonathan Peyton "TASK_NOT_PUSHED for task %p\n", 3067cc577a4SJonathan Peyton gtid, taskdata)); 3077cc577a4SJonathan Peyton return TASK_NOT_PUSHED; 3087cc577a4SJonathan Peyton } 3097cc577a4SJonathan Peyton 3107cc577a4SJonathan Peyton // Lock the deque for the task push operation 3117cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&thread_data->td.td_deque_lock); 3127cc577a4SJonathan Peyton 3137cc577a4SJonathan Peyton #if OMP_45_ENABLED 3147cc577a4SJonathan Peyton // Need to recheck as we can get a proxy task from a thread outside of OpenMP 3153041982dSJonathan Peyton if (TCR_4(thread_data->td.td_deque_ntasks) >= 3163041982dSJonathan Peyton TASK_DEQUE_SIZE(thread_data->td)) { 3177cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&thread_data->td.td_deque_lock); 3183041982dSJonathan Peyton KA_TRACE(20, ("__kmp_push_task: T#%d deque is full on 2nd check; returning " 3193041982dSJonathan Peyton "TASK_NOT_PUSHED for task %p\n", 3207cc577a4SJonathan Peyton gtid, taskdata)); 3217cc577a4SJonathan Peyton return TASK_NOT_PUSHED; 3227cc577a4SJonathan Peyton } 3237cc577a4SJonathan Peyton #else 3247cc577a4SJonathan Peyton // Must have room since no thread can add tasks but calling thread 3253041982dSJonathan Peyton KMP_DEBUG_ASSERT(TCR_4(thread_data->td.td_deque_ntasks) < 3263041982dSJonathan Peyton TASK_DEQUE_SIZE(thread_data->td)); 3277cc577a4SJonathan Peyton #endif 3287cc577a4SJonathan Peyton 3293041982dSJonathan Peyton thread_data->td.td_deque[thread_data->td.td_deque_tail] = 3303041982dSJonathan Peyton taskdata; // Push taskdata 3317cc577a4SJonathan Peyton // Wrap index. 3323041982dSJonathan Peyton thread_data->td.td_deque_tail = 3333041982dSJonathan Peyton (thread_data->td.td_deque_tail + 1) & TASK_DEQUE_MASK(thread_data->td); 3343041982dSJonathan Peyton TCW_4(thread_data->td.td_deque_ntasks, 3353041982dSJonathan Peyton TCR_4(thread_data->td.td_deque_ntasks) + 1); // Adjust task count 3367cc577a4SJonathan Peyton 3377cc577a4SJonathan Peyton KA_TRACE(20, ("__kmp_push_task: T#%d returning TASK_SUCCESSFULLY_PUSHED: " 3387cc577a4SJonathan Peyton "task=%p ntasks=%d head=%u tail=%u\n", 3397cc577a4SJonathan Peyton gtid, taskdata, thread_data->td.td_deque_ntasks, 3407cc577a4SJonathan Peyton thread_data->td.td_deque_head, thread_data->td.td_deque_tail)); 3417cc577a4SJonathan Peyton 3427cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&thread_data->td.td_deque_lock); 3437cc577a4SJonathan Peyton 3447cc577a4SJonathan Peyton return TASK_SUCCESSFULLY_PUSHED; 3457cc577a4SJonathan Peyton } 3467cc577a4SJonathan Peyton 3473041982dSJonathan Peyton // __kmp_pop_current_task_from_thread: set up current task from called thread 3483041982dSJonathan Peyton // when team ends 3493041982dSJonathan Peyton // 3507cc577a4SJonathan Peyton // this_thr: thread structure to set current_task in. 3513041982dSJonathan Peyton void __kmp_pop_current_task_from_thread(kmp_info_t *this_thr) { 3523041982dSJonathan Peyton KF_TRACE(10, ("__kmp_pop_current_task_from_thread(enter): T#%d " 3533041982dSJonathan Peyton "this_thread=%p, curtask=%p, " 3547cc577a4SJonathan Peyton "curtask_parent=%p\n", 3557cc577a4SJonathan Peyton 0, this_thr, this_thr->th.th_current_task, 3567cc577a4SJonathan Peyton this_thr->th.th_current_task->td_parent)); 3577cc577a4SJonathan Peyton 3587cc577a4SJonathan Peyton this_thr->th.th_current_task = this_thr->th.th_current_task->td_parent; 3597cc577a4SJonathan Peyton 3603041982dSJonathan Peyton KF_TRACE(10, ("__kmp_pop_current_task_from_thread(exit): T#%d " 3613041982dSJonathan Peyton "this_thread=%p, curtask=%p, " 3627cc577a4SJonathan Peyton "curtask_parent=%p\n", 3637cc577a4SJonathan Peyton 0, this_thr, this_thr->th.th_current_task, 3647cc577a4SJonathan Peyton this_thr->th.th_current_task->td_parent)); 3657cc577a4SJonathan Peyton } 3667cc577a4SJonathan Peyton 3673041982dSJonathan Peyton // __kmp_push_current_task_to_thread: set up current task in called thread for a 3683041982dSJonathan Peyton // new team 3693041982dSJonathan Peyton // 3707cc577a4SJonathan Peyton // this_thr: thread structure to set up 3717cc577a4SJonathan Peyton // team: team for implicit task data 3727cc577a4SJonathan Peyton // tid: thread within team to set up 3733041982dSJonathan Peyton void __kmp_push_current_task_to_thread(kmp_info_t *this_thr, kmp_team_t *team, 3743041982dSJonathan Peyton int tid) { 3753041982dSJonathan Peyton // current task of the thread is a parent of the new just created implicit 3763041982dSJonathan Peyton // tasks of new team 3773041982dSJonathan Peyton KF_TRACE(10, ("__kmp_push_current_task_to_thread(enter): T#%d this_thread=%p " 3783041982dSJonathan Peyton "curtask=%p " 3797cc577a4SJonathan Peyton "parent_task=%p\n", 3807cc577a4SJonathan Peyton tid, this_thr, this_thr->th.th_current_task, 3817cc577a4SJonathan Peyton team->t.t_implicit_task_taskdata[tid].td_parent)); 3827cc577a4SJonathan Peyton 3837cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(this_thr != NULL); 3847cc577a4SJonathan Peyton 3857cc577a4SJonathan Peyton if (tid == 0) { 3867cc577a4SJonathan Peyton if (this_thr->th.th_current_task != &team->t.t_implicit_task_taskdata[0]) { 3873041982dSJonathan Peyton team->t.t_implicit_task_taskdata[0].td_parent = 3883041982dSJonathan Peyton this_thr->th.th_current_task; 3897cc577a4SJonathan Peyton this_thr->th.th_current_task = &team->t.t_implicit_task_taskdata[0]; 3907cc577a4SJonathan Peyton } 3917cc577a4SJonathan Peyton } else { 3923041982dSJonathan Peyton team->t.t_implicit_task_taskdata[tid].td_parent = 3933041982dSJonathan Peyton team->t.t_implicit_task_taskdata[0].td_parent; 3947cc577a4SJonathan Peyton this_thr->th.th_current_task = &team->t.t_implicit_task_taskdata[tid]; 3957cc577a4SJonathan Peyton } 3967cc577a4SJonathan Peyton 3973041982dSJonathan Peyton KF_TRACE(10, ("__kmp_push_current_task_to_thread(exit): T#%d this_thread=%p " 3983041982dSJonathan Peyton "curtask=%p " 3997cc577a4SJonathan Peyton "parent_task=%p\n", 4007cc577a4SJonathan Peyton tid, this_thr, this_thr->th.th_current_task, 4017cc577a4SJonathan Peyton team->t.t_implicit_task_taskdata[tid].td_parent)); 4027cc577a4SJonathan Peyton } 4037cc577a4SJonathan Peyton 4047cc577a4SJonathan Peyton // __kmp_task_start: bookkeeping for a task starting execution 4053041982dSJonathan Peyton // 4067cc577a4SJonathan Peyton // GTID: global thread id of calling thread 4077cc577a4SJonathan Peyton // task: task starting execution 4087cc577a4SJonathan Peyton // current_task: task suspending 4093041982dSJonathan Peyton static void __kmp_task_start(kmp_int32 gtid, kmp_task_t *task, 4103041982dSJonathan Peyton kmp_taskdata_t *current_task) { 4117cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task); 4127cc577a4SJonathan Peyton kmp_info_t *thread = __kmp_threads[gtid]; 4137cc577a4SJonathan Peyton 4143041982dSJonathan Peyton KA_TRACE(10, 4153041982dSJonathan Peyton ("__kmp_task_start(enter): T#%d starting task %p: current_task=%p\n", 4167cc577a4SJonathan Peyton gtid, taskdata, current_task)); 4177cc577a4SJonathan Peyton 4187cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.tasktype == TASK_EXPLICIT); 4197cc577a4SJonathan Peyton 4207cc577a4SJonathan Peyton // mark currently executing task as suspended 4217cc577a4SJonathan Peyton // TODO: GEH - make sure root team implicit task is initialized properly. 4227cc577a4SJonathan Peyton // KMP_DEBUG_ASSERT( current_task -> td_flags.executing == 1 ); 4237cc577a4SJonathan Peyton current_task->td_flags.executing = 0; 4247cc577a4SJonathan Peyton 4257cc577a4SJonathan Peyton // Add task to stack if tied 4267cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK 4273041982dSJonathan Peyton if (taskdata->td_flags.tiedness == TASK_TIED) { 4287cc577a4SJonathan Peyton __kmp_push_task_stack(gtid, thread, taskdata); 4297cc577a4SJonathan Peyton } 4307cc577a4SJonathan Peyton #endif /* BUILD_TIED_TASK_STACK */ 4317cc577a4SJonathan Peyton 4327cc577a4SJonathan Peyton // mark starting task as executing and as current task 4337cc577a4SJonathan Peyton thread->th.th_current_task = taskdata; 4347cc577a4SJonathan Peyton 4353041982dSJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.started == 0 || 4363041982dSJonathan Peyton taskdata->td_flags.tiedness == TASK_UNTIED); 4373041982dSJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.executing == 0 || 4383041982dSJonathan Peyton taskdata->td_flags.tiedness == TASK_UNTIED); 4397cc577a4SJonathan Peyton taskdata->td_flags.started = 1; 4407cc577a4SJonathan Peyton taskdata->td_flags.executing = 1; 4417cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.complete == 0); 4427cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.freed == 0); 4437cc577a4SJonathan Peyton 4447cc577a4SJonathan Peyton // GEH TODO: shouldn't we pass some sort of location identifier here? 4457cc577a4SJonathan Peyton // APT: yes, we will pass location here. 4467cc577a4SJonathan Peyton // need to store current thread state (in a thread or taskdata structure) 4477cc577a4SJonathan Peyton // before setting work_state, otherwise wrong state is set after end of task 4487cc577a4SJonathan Peyton 4493041982dSJonathan Peyton KA_TRACE(10, ("__kmp_task_start(exit): T#%d task=%p\n", gtid, taskdata)); 4507cc577a4SJonathan Peyton 4517cc577a4SJonathan Peyton #if OMPT_SUPPORT 4523041982dSJonathan Peyton if (ompt_enabled && ompt_callbacks.ompt_callback(ompt_event_task_begin)) { 4537cc577a4SJonathan Peyton kmp_taskdata_t *parent = taskdata->td_parent; 4547cc577a4SJonathan Peyton ompt_callbacks.ompt_callback(ompt_event_task_begin)( 4557cc577a4SJonathan Peyton parent ? parent->ompt_task_info.task_id : ompt_task_id_none, 4567cc577a4SJonathan Peyton parent ? &(parent->ompt_task_info.frame) : NULL, 4573041982dSJonathan Peyton taskdata->ompt_task_info.task_id, taskdata->ompt_task_info.function); 4587cc577a4SJonathan Peyton } 4597cc577a4SJonathan Peyton #endif 4607cc577a4SJonathan Peyton #if OMP_40_ENABLED && OMPT_SUPPORT && OMPT_TRACE 4617cc577a4SJonathan Peyton /* OMPT emit all dependences if requested by the tool */ 4627cc577a4SJonathan Peyton if (ompt_enabled && taskdata->ompt_task_info.ndeps > 0 && 4633041982dSJonathan Peyton ompt_callbacks.ompt_callback(ompt_event_task_dependences)) { 4647cc577a4SJonathan Peyton ompt_callbacks.ompt_callback(ompt_event_task_dependences)( 4653041982dSJonathan Peyton taskdata->ompt_task_info.task_id, taskdata->ompt_task_info.deps, 4663041982dSJonathan Peyton taskdata->ompt_task_info.ndeps); 4677cc577a4SJonathan Peyton /* We can now free the allocated memory for the dependencies */ 4687cc577a4SJonathan Peyton KMP_OMPT_DEPS_FREE(thread, taskdata->ompt_task_info.deps); 4697cc577a4SJonathan Peyton taskdata->ompt_task_info.deps = NULL; 4707cc577a4SJonathan Peyton taskdata->ompt_task_info.ndeps = 0; 4717cc577a4SJonathan Peyton } 4727cc577a4SJonathan Peyton #endif /* OMP_40_ENABLED && OMPT_SUPPORT && OMPT_TRACE */ 4737cc577a4SJonathan Peyton 4747cc577a4SJonathan Peyton return; 4757cc577a4SJonathan Peyton } 4767cc577a4SJonathan Peyton 4773041982dSJonathan Peyton // __kmpc_omp_task_begin_if0: report that a given serialized task has started 4783041982dSJonathan Peyton // execution 4793041982dSJonathan Peyton // 4807cc577a4SJonathan Peyton // loc_ref: source location information; points to beginning of task block. 4817cc577a4SJonathan Peyton // gtid: global thread number. 4827cc577a4SJonathan Peyton // task: task thunk for the started task. 4833041982dSJonathan Peyton void __kmpc_omp_task_begin_if0(ident_t *loc_ref, kmp_int32 gtid, 4843041982dSJonathan Peyton kmp_task_t *task) { 4857cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task); 4867cc577a4SJonathan Peyton kmp_taskdata_t *current_task = __kmp_threads[gtid]->th.th_current_task; 4877cc577a4SJonathan Peyton 4883041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_begin_if0(enter): T#%d loc=%p task=%p " 4893041982dSJonathan Peyton "current_task=%p\n", 4907cc577a4SJonathan Peyton gtid, loc_ref, taskdata, current_task)); 4917cc577a4SJonathan Peyton 4927cc577a4SJonathan Peyton if (taskdata->td_flags.tiedness == TASK_UNTIED) { 4933041982dSJonathan Peyton // untied task needs to increment counter so that the task structure is not 4943041982dSJonathan Peyton // freed prematurely 4957cc577a4SJonathan Peyton kmp_int32 counter = 1 + KMP_TEST_THEN_INC32(&taskdata->td_untied_count); 4963041982dSJonathan Peyton KA_TRACE(20, ("__kmpc_omp_task_begin_if0: T#%d untied_count (%d) " 4973041982dSJonathan Peyton "incremented for task %p\n", 4987cc577a4SJonathan Peyton gtid, counter, taskdata)); 4997cc577a4SJonathan Peyton } 5007cc577a4SJonathan Peyton 5013041982dSJonathan Peyton taskdata->td_flags.task_serial = 5023041982dSJonathan Peyton 1; // Execute this task immediately, not deferred. 5037cc577a4SJonathan Peyton __kmp_task_start(gtid, task, current_task); 5047cc577a4SJonathan Peyton 5053041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_begin_if0(exit): T#%d loc=%p task=%p,\n", gtid, 5063041982dSJonathan Peyton loc_ref, taskdata)); 5077cc577a4SJonathan Peyton 5087cc577a4SJonathan Peyton return; 5097cc577a4SJonathan Peyton } 5107cc577a4SJonathan Peyton 5117cc577a4SJonathan Peyton #ifdef TASK_UNUSED 5127cc577a4SJonathan Peyton // __kmpc_omp_task_begin: report that a given task has started execution 5137cc577a4SJonathan Peyton // NEVER GENERATED BY COMPILER, DEPRECATED!!! 5143041982dSJonathan Peyton void __kmpc_omp_task_begin(ident_t *loc_ref, kmp_int32 gtid, kmp_task_t *task) { 5157cc577a4SJonathan Peyton kmp_taskdata_t *current_task = __kmp_threads[gtid]->th.th_current_task; 5167cc577a4SJonathan Peyton 5173041982dSJonathan Peyton KA_TRACE( 5183041982dSJonathan Peyton 10, 5193041982dSJonathan Peyton ("__kmpc_omp_task_begin(enter): T#%d loc=%p task=%p current_task=%p\n", 5207cc577a4SJonathan Peyton gtid, loc_ref, KMP_TASK_TO_TASKDATA(task), current_task)); 5217cc577a4SJonathan Peyton 5227cc577a4SJonathan Peyton __kmp_task_start(gtid, task, current_task); 5237cc577a4SJonathan Peyton 5243041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_begin(exit): T#%d loc=%p task=%p,\n", gtid, 5253041982dSJonathan Peyton loc_ref, KMP_TASK_TO_TASKDATA(task))); 5267cc577a4SJonathan Peyton return; 5277cc577a4SJonathan Peyton } 5287cc577a4SJonathan Peyton #endif // TASK_UNUSED 5297cc577a4SJonathan Peyton 5307cc577a4SJonathan Peyton // __kmp_free_task: free the current task space and the space for shareds 5313041982dSJonathan Peyton // 5327cc577a4SJonathan Peyton // gtid: Global thread ID of calling thread 5337cc577a4SJonathan Peyton // taskdata: task to free 5347cc577a4SJonathan Peyton // thread: thread data structure of caller 5353041982dSJonathan Peyton static void __kmp_free_task(kmp_int32 gtid, kmp_taskdata_t *taskdata, 5363041982dSJonathan Peyton kmp_info_t *thread) { 5373041982dSJonathan Peyton KA_TRACE(30, ("__kmp_free_task: T#%d freeing data from task %p\n", gtid, 5383041982dSJonathan Peyton taskdata)); 5397cc577a4SJonathan Peyton 5407cc577a4SJonathan Peyton // Check to make sure all flags and counters have the correct values 5417cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.tasktype == TASK_EXPLICIT); 5427cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.executing == 0); 5437cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.complete == 1); 5447cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.freed == 0); 5453041982dSJonathan Peyton KMP_DEBUG_ASSERT(TCR_4(taskdata->td_allocated_child_tasks) == 0 || 5463041982dSJonathan Peyton taskdata->td_flags.task_serial == 1); 5477cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(TCR_4(taskdata->td_incomplete_child_tasks) == 0); 5487cc577a4SJonathan Peyton 5497cc577a4SJonathan Peyton taskdata->td_flags.freed = 1; 5507cc577a4SJonathan Peyton ANNOTATE_HAPPENS_BEFORE(taskdata); 5517cc577a4SJonathan Peyton // deallocate the taskdata and shared variable blocks associated with this task 5527cc577a4SJonathan Peyton #if USE_FAST_MEMORY 5537cc577a4SJonathan Peyton __kmp_fast_free(thread, taskdata); 5547cc577a4SJonathan Peyton #else /* ! USE_FAST_MEMORY */ 5557cc577a4SJonathan Peyton __kmp_thread_free(thread, taskdata); 5567cc577a4SJonathan Peyton #endif 5577cc577a4SJonathan Peyton 5583041982dSJonathan Peyton KA_TRACE(20, ("__kmp_free_task: T#%d freed task %p\n", gtid, taskdata)); 5597cc577a4SJonathan Peyton } 5607cc577a4SJonathan Peyton 5613041982dSJonathan Peyton // __kmp_free_task_and_ancestors: free the current task and ancestors without 5623041982dSJonathan Peyton // children 5637cc577a4SJonathan Peyton // 5647cc577a4SJonathan Peyton // gtid: Global thread ID of calling thread 5657cc577a4SJonathan Peyton // taskdata: task to free 5667cc577a4SJonathan Peyton // thread: thread data structure of caller 5673041982dSJonathan Peyton static void __kmp_free_task_and_ancestors(kmp_int32 gtid, 5683041982dSJonathan Peyton kmp_taskdata_t *taskdata, 5693041982dSJonathan Peyton kmp_info_t *thread) { 5707cc577a4SJonathan Peyton #if OMP_45_ENABLED 5717cc577a4SJonathan Peyton // Proxy tasks must always be allowed to free their parents 5727cc577a4SJonathan Peyton // because they can be run in background even in serial mode. 5733041982dSJonathan Peyton kmp_int32 team_serial = 5743041982dSJonathan Peyton (taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser) && 5753041982dSJonathan Peyton !taskdata->td_flags.proxy; 5767cc577a4SJonathan Peyton #else 5773041982dSJonathan Peyton kmp_int32 team_serial = 5783041982dSJonathan Peyton taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser; 5797cc577a4SJonathan Peyton #endif 5807cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.tasktype == TASK_EXPLICIT); 5817cc577a4SJonathan Peyton 5823041982dSJonathan Peyton kmp_int32 children = 5833041982dSJonathan Peyton KMP_TEST_THEN_DEC32((kmp_int32 *)(&taskdata->td_allocated_child_tasks)) - 5843041982dSJonathan Peyton 1; 5857cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(children >= 0); 5867cc577a4SJonathan Peyton 5877cc577a4SJonathan Peyton // Now, go up the ancestor tree to see if any ancestors can now be freed. 5883041982dSJonathan Peyton while (children == 0) { 5897cc577a4SJonathan Peyton kmp_taskdata_t *parent_taskdata = taskdata->td_parent; 5907cc577a4SJonathan Peyton 5917cc577a4SJonathan Peyton KA_TRACE(20, ("__kmp_free_task_and_ancestors(enter): T#%d task %p complete " 5923041982dSJonathan Peyton "and freeing itself\n", 5933041982dSJonathan Peyton gtid, taskdata)); 5947cc577a4SJonathan Peyton 5957cc577a4SJonathan Peyton // --- Deallocate my ancestor task --- 5967cc577a4SJonathan Peyton __kmp_free_task(gtid, taskdata, thread); 5977cc577a4SJonathan Peyton 5987cc577a4SJonathan Peyton taskdata = parent_taskdata; 5997cc577a4SJonathan Peyton 6003041982dSJonathan Peyton // Stop checking ancestors at implicit task instead of walking up ancestor 6013041982dSJonathan Peyton // tree to avoid premature deallocation of ancestors. 6027cc577a4SJonathan Peyton if (team_serial || taskdata->td_flags.tasktype == TASK_IMPLICIT) 6037cc577a4SJonathan Peyton return; 6047cc577a4SJonathan Peyton 6057cc577a4SJonathan Peyton // Predecrement simulated by "- 1" calculation 6063041982dSJonathan Peyton children = KMP_TEST_THEN_DEC32( 6073041982dSJonathan Peyton (kmp_int32 *)(&taskdata->td_allocated_child_tasks)) - 6083041982dSJonathan Peyton 1; 6097cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(children >= 0); 6107cc577a4SJonathan Peyton } 6117cc577a4SJonathan Peyton 6123041982dSJonathan Peyton KA_TRACE( 6133041982dSJonathan Peyton 20, ("__kmp_free_task_and_ancestors(exit): T#%d task %p has %d children; " 6143041982dSJonathan Peyton "not freeing it yet\n", 6153041982dSJonathan Peyton gtid, taskdata, children)); 6167cc577a4SJonathan Peyton } 6177cc577a4SJonathan Peyton 6187cc577a4SJonathan Peyton // __kmp_task_finish: bookkeeping to do when a task finishes execution 6193041982dSJonathan Peyton // 6207cc577a4SJonathan Peyton // gtid: global thread ID for calling thread 6217cc577a4SJonathan Peyton // task: task to be finished 6227cc577a4SJonathan Peyton // resumed_task: task to be resumed. (may be NULL if task is serialized) 6233041982dSJonathan Peyton static void __kmp_task_finish(kmp_int32 gtid, kmp_task_t *task, 6243041982dSJonathan Peyton kmp_taskdata_t *resumed_task) { 6257cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task); 6267cc577a4SJonathan Peyton kmp_info_t *thread = __kmp_threads[gtid]; 6273041982dSJonathan Peyton kmp_task_team_t *task_team = 6283041982dSJonathan Peyton thread->th.th_task_team; // might be NULL for serial teams... 6297cc577a4SJonathan Peyton kmp_int32 children = 0; 6307cc577a4SJonathan Peyton 6317cc577a4SJonathan Peyton #if OMPT_SUPPORT 6323041982dSJonathan Peyton if (ompt_enabled && ompt_callbacks.ompt_callback(ompt_event_task_end)) { 6337cc577a4SJonathan Peyton kmp_taskdata_t *parent = taskdata->td_parent; 6347cc577a4SJonathan Peyton ompt_callbacks.ompt_callback(ompt_event_task_end)( 6357cc577a4SJonathan Peyton taskdata->ompt_task_info.task_id); 6367cc577a4SJonathan Peyton } 6377cc577a4SJonathan Peyton #endif 6387cc577a4SJonathan Peyton 6393041982dSJonathan Peyton KA_TRACE(10, ("__kmp_task_finish(enter): T#%d finishing task %p and resuming " 6403041982dSJonathan Peyton "task %p\n", 6417cc577a4SJonathan Peyton gtid, taskdata, resumed_task)); 6427cc577a4SJonathan Peyton 6437cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.tasktype == TASK_EXPLICIT); 6447cc577a4SJonathan Peyton 6457cc577a4SJonathan Peyton // Pop task from stack if tied 6467cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK 6473041982dSJonathan Peyton if (taskdata->td_flags.tiedness == TASK_TIED) { 6487cc577a4SJonathan Peyton __kmp_pop_task_stack(gtid, thread, taskdata); 6497cc577a4SJonathan Peyton } 6507cc577a4SJonathan Peyton #endif /* BUILD_TIED_TASK_STACK */ 6517cc577a4SJonathan Peyton 6527cc577a4SJonathan Peyton if (taskdata->td_flags.tiedness == TASK_UNTIED) { 6533041982dSJonathan Peyton // untied task needs to check the counter so that the task structure is not 6543041982dSJonathan Peyton // freed prematurely 6557cc577a4SJonathan Peyton kmp_int32 counter = KMP_TEST_THEN_DEC32(&taskdata->td_untied_count) - 1; 6563041982dSJonathan Peyton KA_TRACE( 6573041982dSJonathan Peyton 20, 6583041982dSJonathan Peyton ("__kmp_task_finish: T#%d untied_count (%d) decremented for task %p\n", 6597cc577a4SJonathan Peyton gtid, counter, taskdata)); 6607cc577a4SJonathan Peyton if (counter > 0) { 6613041982dSJonathan Peyton // untied task is not done, to be continued possibly by other thread, do 6623041982dSJonathan Peyton // not free it now 6637cc577a4SJonathan Peyton if (resumed_task == NULL) { 6647cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.task_serial); 6653041982dSJonathan Peyton resumed_task = taskdata->td_parent; // In a serialized task, the resumed 6663041982dSJonathan Peyton // task is the parent 6677cc577a4SJonathan Peyton } 6687cc577a4SJonathan Peyton thread->th.th_current_task = resumed_task; // restore current_task 6697cc577a4SJonathan Peyton resumed_task->td_flags.executing = 1; // resume previous task 6703041982dSJonathan Peyton KA_TRACE(10, ("__kmp_task_finish(exit): T#%d partially done task %p, " 6713041982dSJonathan Peyton "resuming task %p\n", 6727cc577a4SJonathan Peyton gtid, taskdata, resumed_task)); 6737cc577a4SJonathan Peyton return; 6747cc577a4SJonathan Peyton } 6757cc577a4SJonathan Peyton } 6767cc577a4SJonathan Peyton 6777cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.complete == 0); 6787cc577a4SJonathan Peyton taskdata->td_flags.complete = 1; // mark the task as completed 6797cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.started == 1); 6807cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.freed == 0); 6817cc577a4SJonathan Peyton 6823041982dSJonathan Peyton // Only need to keep track of count if team parallel and tasking not 6833041982dSJonathan Peyton // serialized 6847cc577a4SJonathan Peyton if (!(taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser)) { 6857cc577a4SJonathan Peyton // Predecrement simulated by "- 1" calculation 6863041982dSJonathan Peyton children = 6873041982dSJonathan Peyton KMP_TEST_THEN_DEC32( 6883041982dSJonathan Peyton (kmp_int32 *)(&taskdata->td_parent->td_incomplete_child_tasks)) - 6893041982dSJonathan Peyton 1; 6907cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(children >= 0); 6917cc577a4SJonathan Peyton #if OMP_40_ENABLED 6927cc577a4SJonathan Peyton if (taskdata->td_taskgroup) 6937cc577a4SJonathan Peyton KMP_TEST_THEN_DEC32((kmp_int32 *)(&taskdata->td_taskgroup->count)); 6947cc577a4SJonathan Peyton #if OMP_45_ENABLED 6957cc577a4SJonathan Peyton } 6967cc577a4SJonathan Peyton // if we found proxy tasks there could exist a dependency chain 6977cc577a4SJonathan Peyton // with the proxy task as origin 6983041982dSJonathan Peyton if (!(taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser) || 6993041982dSJonathan Peyton (task_team && task_team->tt.tt_found_proxy_tasks)) { 7007cc577a4SJonathan Peyton #endif 7017cc577a4SJonathan Peyton __kmp_release_deps(gtid, taskdata); 7027cc577a4SJonathan Peyton #endif 7037cc577a4SJonathan Peyton } 7047cc577a4SJonathan Peyton 7053041982dSJonathan Peyton // td_flags.executing must be marked as 0 after __kmp_release_deps has been 7063041982dSJonathan Peyton // called. Othertwise, if a task is executed immediately from the release_deps 7073041982dSJonathan Peyton // code, the flag will be reset to 1 again by this same function 7087cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.executing == 1); 7097cc577a4SJonathan Peyton taskdata->td_flags.executing = 0; // suspend the finishing task 7107cc577a4SJonathan Peyton 7113041982dSJonathan Peyton KA_TRACE( 7123041982dSJonathan Peyton 20, ("__kmp_task_finish: T#%d finished task %p, %d incomplete children\n", 7137cc577a4SJonathan Peyton gtid, taskdata, children)); 7147cc577a4SJonathan Peyton 7157cc577a4SJonathan Peyton #if OMP_40_ENABLED 7167cc577a4SJonathan Peyton /* If the tasks' destructor thunk flag has been set, we need to invoke the 7173041982dSJonathan Peyton destructor thunk that has been generated by the compiler. The code is 7183041982dSJonathan Peyton placed here, since at this point other tasks might have been released 7197cc577a4SJonathan Peyton hence overlapping the destructor invokations with some other work in the 7203041982dSJonathan Peyton released tasks. The OpenMP spec is not specific on when the destructors 7213041982dSJonathan Peyton are invoked, so we should be free to choose. */ 7227cc577a4SJonathan Peyton if (taskdata->td_flags.destructors_thunk) { 7237cc577a4SJonathan Peyton kmp_routine_entry_t destr_thunk = task->data1.destructors; 7247cc577a4SJonathan Peyton KMP_ASSERT(destr_thunk); 7257cc577a4SJonathan Peyton destr_thunk(gtid, task); 7267cc577a4SJonathan Peyton } 7277cc577a4SJonathan Peyton #endif // OMP_40_ENABLED 7287cc577a4SJonathan Peyton 7297cc577a4SJonathan Peyton // bookkeeping for resuming task: 7307cc577a4SJonathan Peyton // GEH - note tasking_ser => task_serial 7313041982dSJonathan Peyton KMP_DEBUG_ASSERT( 7323041982dSJonathan Peyton (taskdata->td_flags.tasking_ser || taskdata->td_flags.task_serial) == 7337cc577a4SJonathan Peyton taskdata->td_flags.task_serial); 7343041982dSJonathan Peyton if (taskdata->td_flags.task_serial) { 7357cc577a4SJonathan Peyton if (resumed_task == NULL) { 7363041982dSJonathan Peyton resumed_task = taskdata->td_parent; // In a serialized task, the resumed 7373041982dSJonathan Peyton // task is the parent 7383041982dSJonathan Peyton } else 7397cc577a4SJonathan Peyton #if OMP_45_ENABLED 7407cc577a4SJonathan Peyton if (!(task_team && task_team->tt.tt_found_proxy_tasks)) 7417cc577a4SJonathan Peyton #endif 7427cc577a4SJonathan Peyton { 7437cc577a4SJonathan Peyton // verify resumed task passed in points to parent 7447cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(resumed_task == taskdata->td_parent); 7457cc577a4SJonathan Peyton } 7463041982dSJonathan Peyton } else { 7473041982dSJonathan Peyton KMP_DEBUG_ASSERT(resumed_task != 7483041982dSJonathan Peyton NULL); // verify that resumed task is passed as arguemnt 7497cc577a4SJonathan Peyton } 7507cc577a4SJonathan Peyton 7517cc577a4SJonathan Peyton // Free this task and then ancestor tasks if they have no children. 7527cc577a4SJonathan Peyton // Restore th_current_task first as suggested by John: 7537cc577a4SJonathan Peyton // johnmc: if an asynchronous inquiry peers into the runtime system 7547cc577a4SJonathan Peyton // it doesn't see the freed task as the current task. 7557cc577a4SJonathan Peyton thread->th.th_current_task = resumed_task; 7567cc577a4SJonathan Peyton __kmp_free_task_and_ancestors(gtid, taskdata, thread); 7577cc577a4SJonathan Peyton 7587cc577a4SJonathan Peyton // TODO: GEH - make sure root team implicit task is initialized properly. 7597cc577a4SJonathan Peyton // KMP_DEBUG_ASSERT( resumed_task->td_flags.executing == 0 ); 7607cc577a4SJonathan Peyton resumed_task->td_flags.executing = 1; // resume previous task 7617cc577a4SJonathan Peyton 7623041982dSJonathan Peyton KA_TRACE( 7633041982dSJonathan Peyton 10, ("__kmp_task_finish(exit): T#%d finished task %p, resuming task %p\n", 7647cc577a4SJonathan Peyton gtid, taskdata, resumed_task)); 7657cc577a4SJonathan Peyton 7667cc577a4SJonathan Peyton return; 7677cc577a4SJonathan Peyton } 7687cc577a4SJonathan Peyton 7697cc577a4SJonathan Peyton // __kmpc_omp_task_complete_if0: report that a task has completed execution 7703041982dSJonathan Peyton // 7717cc577a4SJonathan Peyton // loc_ref: source location information; points to end of task block. 7727cc577a4SJonathan Peyton // gtid: global thread number. 7737cc577a4SJonathan Peyton // task: task thunk for the completed task. 7743041982dSJonathan Peyton void __kmpc_omp_task_complete_if0(ident_t *loc_ref, kmp_int32 gtid, 7753041982dSJonathan Peyton kmp_task_t *task) { 7767cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_complete_if0(enter): T#%d loc=%p task=%p\n", 7777cc577a4SJonathan Peyton gtid, loc_ref, KMP_TASK_TO_TASKDATA(task))); 7783041982dSJonathan Peyton // this routine will provide task to resume 7793041982dSJonathan Peyton __kmp_task_finish(gtid, task, NULL); 7807cc577a4SJonathan Peyton 7817cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_complete_if0(exit): T#%d loc=%p task=%p\n", 7827cc577a4SJonathan Peyton gtid, loc_ref, KMP_TASK_TO_TASKDATA(task))); 7837cc577a4SJonathan Peyton return; 7847cc577a4SJonathan Peyton } 7857cc577a4SJonathan Peyton 7867cc577a4SJonathan Peyton #ifdef TASK_UNUSED 7877cc577a4SJonathan Peyton // __kmpc_omp_task_complete: report that a task has completed execution 7887cc577a4SJonathan Peyton // NEVER GENERATED BY COMPILER, DEPRECATED!!! 7893041982dSJonathan Peyton void __kmpc_omp_task_complete(ident_t *loc_ref, kmp_int32 gtid, 7903041982dSJonathan Peyton kmp_task_t *task) { 7913041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_complete(enter): T#%d loc=%p task=%p\n", gtid, 7923041982dSJonathan Peyton loc_ref, KMP_TASK_TO_TASKDATA(task))); 7937cc577a4SJonathan Peyton 7947cc577a4SJonathan Peyton __kmp_task_finish(gtid, task, NULL); // Not sure how to find task to resume 7957cc577a4SJonathan Peyton 7963041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_complete(exit): T#%d loc=%p task=%p\n", gtid, 7973041982dSJonathan Peyton loc_ref, KMP_TASK_TO_TASKDATA(task))); 7987cc577a4SJonathan Peyton return; 7997cc577a4SJonathan Peyton } 8007cc577a4SJonathan Peyton #endif // TASK_UNUSED 8017cc577a4SJonathan Peyton 8027cc577a4SJonathan Peyton #if OMPT_SUPPORT 8033041982dSJonathan Peyton // __kmp_task_init_ompt: Initialize OMPT fields maintained by a task. This will 8043041982dSJonathan Peyton // only be called after ompt_tool, so we already know whether ompt is enabled 8053041982dSJonathan Peyton // or not. 8063041982dSJonathan Peyton static inline void __kmp_task_init_ompt(kmp_taskdata_t *task, int tid, 8073041982dSJonathan Peyton void *function) { 8087cc577a4SJonathan Peyton if (ompt_enabled) { 8097cc577a4SJonathan Peyton task->ompt_task_info.task_id = __ompt_task_id_new(tid); 8107cc577a4SJonathan Peyton task->ompt_task_info.function = function; 8117cc577a4SJonathan Peyton task->ompt_task_info.frame.exit_runtime_frame = NULL; 8127cc577a4SJonathan Peyton task->ompt_task_info.frame.reenter_runtime_frame = NULL; 8137cc577a4SJonathan Peyton #if OMP_40_ENABLED 8147cc577a4SJonathan Peyton task->ompt_task_info.ndeps = 0; 8157cc577a4SJonathan Peyton task->ompt_task_info.deps = NULL; 8167cc577a4SJonathan Peyton #endif /* OMP_40_ENABLED */ 8177cc577a4SJonathan Peyton } 8187cc577a4SJonathan Peyton } 8197cc577a4SJonathan Peyton #endif 8207cc577a4SJonathan Peyton 8213041982dSJonathan Peyton // __kmp_init_implicit_task: Initialize the appropriate fields in the implicit 8223041982dSJonathan Peyton // task for a given thread 8237cc577a4SJonathan Peyton // 8247cc577a4SJonathan Peyton // loc_ref: reference to source location of parallel region 8257cc577a4SJonathan Peyton // this_thr: thread data structure corresponding to implicit task 8267cc577a4SJonathan Peyton // team: team for this_thr 8277cc577a4SJonathan Peyton // tid: thread id of given thread within team 8287cc577a4SJonathan Peyton // set_curr_task: TRUE if need to push current task to thread 8293041982dSJonathan Peyton // NOTE: Routine does not set up the implicit task ICVS. This is assumed to 8303041982dSJonathan Peyton // have already been done elsewhere. 8317cc577a4SJonathan Peyton // TODO: Get better loc_ref. Value passed in may be NULL 8323041982dSJonathan Peyton void __kmp_init_implicit_task(ident_t *loc_ref, kmp_info_t *this_thr, 8333041982dSJonathan Peyton kmp_team_t *team, int tid, int set_curr_task) { 8347cc577a4SJonathan Peyton kmp_taskdata_t *task = &team->t.t_implicit_task_taskdata[tid]; 8357cc577a4SJonathan Peyton 8363041982dSJonathan Peyton KF_TRACE( 8373041982dSJonathan Peyton 10, 8383041982dSJonathan Peyton ("__kmp_init_implicit_task(enter): T#:%d team=%p task=%p, reinit=%s\n", 8397cc577a4SJonathan Peyton tid, team, task, set_curr_task ? "TRUE" : "FALSE")); 8407cc577a4SJonathan Peyton 8417cc577a4SJonathan Peyton task->td_task_id = KMP_GEN_TASK_ID(); 8427cc577a4SJonathan Peyton task->td_team = team; 8433041982dSJonathan Peyton // task->td_parent = NULL; // fix for CQ230101 (broken parent task info 8443041982dSJonathan Peyton // in debugger) 8457cc577a4SJonathan Peyton task->td_ident = loc_ref; 8467cc577a4SJonathan Peyton task->td_taskwait_ident = NULL; 8477cc577a4SJonathan Peyton task->td_taskwait_counter = 0; 8487cc577a4SJonathan Peyton task->td_taskwait_thread = 0; 8497cc577a4SJonathan Peyton 8507cc577a4SJonathan Peyton task->td_flags.tiedness = TASK_TIED; 8517cc577a4SJonathan Peyton task->td_flags.tasktype = TASK_IMPLICIT; 8527cc577a4SJonathan Peyton #if OMP_45_ENABLED 8537cc577a4SJonathan Peyton task->td_flags.proxy = TASK_FULL; 8547cc577a4SJonathan Peyton #endif 8557cc577a4SJonathan Peyton 8567cc577a4SJonathan Peyton // All implicit tasks are executed immediately, not deferred 8577cc577a4SJonathan Peyton task->td_flags.task_serial = 1; 8587cc577a4SJonathan Peyton task->td_flags.tasking_ser = (__kmp_tasking_mode == tskm_immediate_exec); 8597cc577a4SJonathan Peyton task->td_flags.team_serial = (team->t.t_serialized) ? 1 : 0; 8607cc577a4SJonathan Peyton 8617cc577a4SJonathan Peyton task->td_flags.started = 1; 8627cc577a4SJonathan Peyton task->td_flags.executing = 1; 8637cc577a4SJonathan Peyton task->td_flags.complete = 0; 8647cc577a4SJonathan Peyton task->td_flags.freed = 0; 8657cc577a4SJonathan Peyton 8667cc577a4SJonathan Peyton #if OMP_40_ENABLED 8677cc577a4SJonathan Peyton task->td_depnode = NULL; 8687cc577a4SJonathan Peyton #endif 8697cc577a4SJonathan Peyton 8703041982dSJonathan Peyton if (set_curr_task) { // only do this init first time thread is created 8717cc577a4SJonathan Peyton task->td_incomplete_child_tasks = 0; 872*642688b6SJonathan Peyton // Not used: don't need to deallocate implicit task 873*642688b6SJonathan Peyton task->td_allocated_child_tasks = 0; 8747cc577a4SJonathan Peyton #if OMP_40_ENABLED 8757cc577a4SJonathan Peyton task->td_taskgroup = NULL; // An implicit task does not have taskgroup 8767cc577a4SJonathan Peyton task->td_dephash = NULL; 8777cc577a4SJonathan Peyton #endif 8787cc577a4SJonathan Peyton __kmp_push_current_task_to_thread(this_thr, team, tid); 8797cc577a4SJonathan Peyton } else { 8807cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task->td_incomplete_child_tasks == 0); 8817cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task->td_allocated_child_tasks == 0); 8827cc577a4SJonathan Peyton } 8837cc577a4SJonathan Peyton 8847cc577a4SJonathan Peyton #if OMPT_SUPPORT 8857cc577a4SJonathan Peyton __kmp_task_init_ompt(task, tid, NULL); 8867cc577a4SJonathan Peyton #endif 8877cc577a4SJonathan Peyton 8883041982dSJonathan Peyton KF_TRACE(10, ("__kmp_init_implicit_task(exit): T#:%d team=%p task=%p\n", tid, 8893041982dSJonathan Peyton team, task)); 8907cc577a4SJonathan Peyton } 8917cc577a4SJonathan Peyton 8923041982dSJonathan Peyton // __kmp_finish_implicit_task: Release resources associated to implicit tasks 8933041982dSJonathan Peyton // at the end of parallel regions. Some resources are kept for reuse in the next 8943041982dSJonathan Peyton // parallel region. 8957cc577a4SJonathan Peyton // 8963041982dSJonathan Peyton // thread: thread data structure corresponding to implicit task 8973041982dSJonathan Peyton void __kmp_finish_implicit_task(kmp_info_t *thread) { 8987cc577a4SJonathan Peyton kmp_taskdata_t *task = thread->th.th_current_task; 8997cc577a4SJonathan Peyton if (task->td_dephash) 9007cc577a4SJonathan Peyton __kmp_dephash_free_entries(thread, task->td_dephash); 9017cc577a4SJonathan Peyton } 9027cc577a4SJonathan Peyton 9033041982dSJonathan Peyton // __kmp_free_implicit_task: Release resources associated to implicit tasks 9043041982dSJonathan Peyton // when these are destroyed regions 9057cc577a4SJonathan Peyton // 9063041982dSJonathan Peyton // thread: thread data structure corresponding to implicit task 9073041982dSJonathan Peyton void __kmp_free_implicit_task(kmp_info_t *thread) { 9087cc577a4SJonathan Peyton kmp_taskdata_t *task = thread->th.th_current_task; 9097cc577a4SJonathan Peyton if (task->td_dephash) 9107cc577a4SJonathan Peyton __kmp_dephash_free(thread, task->td_dephash); 9117cc577a4SJonathan Peyton task->td_dephash = NULL; 9127cc577a4SJonathan Peyton } 9137cc577a4SJonathan Peyton 9143041982dSJonathan Peyton // Round up a size to a power of two specified by val: Used to insert padding 9153041982dSJonathan Peyton // between structures co-allocated using a single malloc() call 9163041982dSJonathan Peyton static size_t __kmp_round_up_to_val(size_t size, size_t val) { 9177cc577a4SJonathan Peyton if (size & (val - 1)) { 9187cc577a4SJonathan Peyton size &= ~(val - 1); 9197cc577a4SJonathan Peyton if (size <= KMP_SIZE_T_MAX - val) { 9207cc577a4SJonathan Peyton size += val; // Round up if there is no overflow. 9217cc577a4SJonathan Peyton }; // if 9227cc577a4SJonathan Peyton }; // if 9237cc577a4SJonathan Peyton return size; 9247cc577a4SJonathan Peyton } // __kmp_round_up_to_va 9257cc577a4SJonathan Peyton 9267cc577a4SJonathan Peyton // __kmp_task_alloc: Allocate the taskdata and task data structures for a task 9277cc577a4SJonathan Peyton // 9287cc577a4SJonathan Peyton // loc_ref: source location information 9297cc577a4SJonathan Peyton // gtid: global thread number. 9303041982dSJonathan Peyton // flags: include tiedness & task type (explicit vs. implicit) of the ''new'' 9313041982dSJonathan Peyton // task encountered. Converted from kmp_int32 to kmp_tasking_flags_t in routine. 9323041982dSJonathan Peyton // sizeof_kmp_task_t: Size in bytes of kmp_task_t data structure including 9333041982dSJonathan Peyton // private vars accessed in task. 9343041982dSJonathan Peyton // sizeof_shareds: Size in bytes of array of pointers to shared vars accessed 9353041982dSJonathan Peyton // in task. 9367cc577a4SJonathan Peyton // task_entry: Pointer to task code entry point generated by compiler. 9377cc577a4SJonathan Peyton // returns: a pointer to the allocated kmp_task_t structure (task). 9383041982dSJonathan Peyton kmp_task_t *__kmp_task_alloc(ident_t *loc_ref, kmp_int32 gtid, 9393041982dSJonathan Peyton kmp_tasking_flags_t *flags, 9407cc577a4SJonathan Peyton size_t sizeof_kmp_task_t, size_t sizeof_shareds, 9413041982dSJonathan Peyton kmp_routine_entry_t task_entry) { 9427cc577a4SJonathan Peyton kmp_task_t *task; 9437cc577a4SJonathan Peyton kmp_taskdata_t *taskdata; 9447cc577a4SJonathan Peyton kmp_info_t *thread = __kmp_threads[gtid]; 9457cc577a4SJonathan Peyton kmp_team_t *team = thread->th.th_team; 9467cc577a4SJonathan Peyton kmp_taskdata_t *parent_task = thread->th.th_current_task; 9477cc577a4SJonathan Peyton size_t shareds_offset; 9487cc577a4SJonathan Peyton 9497cc577a4SJonathan Peyton KA_TRACE(10, ("__kmp_task_alloc(enter): T#%d loc=%p, flags=(0x%x) " 9507cc577a4SJonathan Peyton "sizeof_task=%ld sizeof_shared=%ld entry=%p\n", 9517cc577a4SJonathan Peyton gtid, loc_ref, *((kmp_int32 *)flags), sizeof_kmp_task_t, 9527cc577a4SJonathan Peyton sizeof_shareds, task_entry)); 9537cc577a4SJonathan Peyton 9547cc577a4SJonathan Peyton if (parent_task->td_flags.final) { 9557cc577a4SJonathan Peyton if (flags->merged_if0) { 9567cc577a4SJonathan Peyton } 9577cc577a4SJonathan Peyton flags->final = 1; 9587cc577a4SJonathan Peyton } 9597cc577a4SJonathan Peyton 9607cc577a4SJonathan Peyton #if OMP_45_ENABLED 9617cc577a4SJonathan Peyton if (flags->proxy == TASK_PROXY) { 9627cc577a4SJonathan Peyton flags->tiedness = TASK_UNTIED; 9637cc577a4SJonathan Peyton flags->merged_if0 = 1; 9647cc577a4SJonathan Peyton 9653041982dSJonathan Peyton /* are we running in a sequential parallel or tskm_immediate_exec... we need 9663041982dSJonathan Peyton tasking support enabled */ 9677cc577a4SJonathan Peyton if ((thread->th.th_task_team) == NULL) { 9687cc577a4SJonathan Peyton /* This should only happen if the team is serialized 9693041982dSJonathan Peyton setup a task team and propagate it to the thread */ 9707cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(team->t.t_serialized); 9713041982dSJonathan Peyton KA_TRACE(30, 9723041982dSJonathan Peyton ("T#%d creating task team in __kmp_task_alloc for proxy task\n", 9733041982dSJonathan Peyton gtid)); 9743041982dSJonathan Peyton __kmp_task_team_setup( 9753041982dSJonathan Peyton thread, team, 9763041982dSJonathan Peyton 1); // 1 indicates setup the current team regardless of nthreads 9777cc577a4SJonathan Peyton thread->th.th_task_team = team->t.t_task_team[thread->th.th_task_state]; 9787cc577a4SJonathan Peyton } 9797cc577a4SJonathan Peyton kmp_task_team_t *task_team = thread->th.th_task_team; 9807cc577a4SJonathan Peyton 9817cc577a4SJonathan Peyton /* tasking must be enabled now as the task might not be pushed */ 9827cc577a4SJonathan Peyton if (!KMP_TASKING_ENABLED(task_team)) { 9833041982dSJonathan Peyton KA_TRACE( 9843041982dSJonathan Peyton 30, 9853041982dSJonathan Peyton ("T#%d enabling tasking in __kmp_task_alloc for proxy task\n", gtid)); 9867cc577a4SJonathan Peyton __kmp_enable_tasking(task_team, thread); 9877cc577a4SJonathan Peyton kmp_int32 tid = thread->th.th_info.ds.ds_tid; 9887cc577a4SJonathan Peyton kmp_thread_data_t *thread_data = &task_team->tt.tt_threads_data[tid]; 9897cc577a4SJonathan Peyton // No lock needed since only owner can allocate 9907cc577a4SJonathan Peyton if (thread_data->td.td_deque == NULL) { 9917cc577a4SJonathan Peyton __kmp_alloc_task_deque(thread, thread_data); 9927cc577a4SJonathan Peyton } 9937cc577a4SJonathan Peyton } 9947cc577a4SJonathan Peyton 9957cc577a4SJonathan Peyton if (task_team->tt.tt_found_proxy_tasks == FALSE) 9967cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_found_proxy_tasks, TRUE); 9977cc577a4SJonathan Peyton } 9987cc577a4SJonathan Peyton #endif 9997cc577a4SJonathan Peyton 10007cc577a4SJonathan Peyton // Calculate shared structure offset including padding after kmp_task_t struct 10017cc577a4SJonathan Peyton // to align pointers in shared struct 10027cc577a4SJonathan Peyton shareds_offset = sizeof(kmp_taskdata_t) + sizeof_kmp_task_t; 10037cc577a4SJonathan Peyton shareds_offset = __kmp_round_up_to_val(shareds_offset, sizeof(void *)); 10047cc577a4SJonathan Peyton 10057cc577a4SJonathan Peyton // Allocate a kmp_taskdata_t block and a kmp_task_t block. 10063041982dSJonathan Peyton KA_TRACE(30, ("__kmp_task_alloc: T#%d First malloc size: %ld\n", gtid, 10073041982dSJonathan Peyton shareds_offset)); 10083041982dSJonathan Peyton KA_TRACE(30, ("__kmp_task_alloc: T#%d Second malloc size: %ld\n", gtid, 10093041982dSJonathan Peyton sizeof_shareds)); 10107cc577a4SJonathan Peyton 10117cc577a4SJonathan Peyton // Avoid double allocation here by combining shareds with taskdata 10127cc577a4SJonathan Peyton #if USE_FAST_MEMORY 10133041982dSJonathan Peyton taskdata = (kmp_taskdata_t *)__kmp_fast_allocate(thread, shareds_offset + 10143041982dSJonathan Peyton sizeof_shareds); 10157cc577a4SJonathan Peyton #else /* ! USE_FAST_MEMORY */ 10163041982dSJonathan Peyton taskdata = (kmp_taskdata_t *)__kmp_thread_malloc(thread, shareds_offset + 10173041982dSJonathan Peyton sizeof_shareds); 10187cc577a4SJonathan Peyton #endif /* USE_FAST_MEMORY */ 10197cc577a4SJonathan Peyton ANNOTATE_HAPPENS_AFTER(taskdata); 10207cc577a4SJonathan Peyton 10217cc577a4SJonathan Peyton task = KMP_TASKDATA_TO_TASK(taskdata); 10227cc577a4SJonathan Peyton 10237cc577a4SJonathan Peyton // Make sure task & taskdata are aligned appropriately 10247cc577a4SJonathan Peyton #if KMP_ARCH_X86 || KMP_ARCH_PPC64 || !KMP_HAVE_QUAD 10257cc577a4SJonathan Peyton KMP_DEBUG_ASSERT((((kmp_uintptr_t)taskdata) & (sizeof(double) - 1)) == 0); 10267cc577a4SJonathan Peyton KMP_DEBUG_ASSERT((((kmp_uintptr_t)task) & (sizeof(double) - 1)) == 0); 10277cc577a4SJonathan Peyton #else 10287cc577a4SJonathan Peyton KMP_DEBUG_ASSERT((((kmp_uintptr_t)taskdata) & (sizeof(_Quad) - 1)) == 0); 10297cc577a4SJonathan Peyton KMP_DEBUG_ASSERT((((kmp_uintptr_t)task) & (sizeof(_Quad) - 1)) == 0); 10307cc577a4SJonathan Peyton #endif 10317cc577a4SJonathan Peyton if (sizeof_shareds > 0) { 10327cc577a4SJonathan Peyton // Avoid double allocation here by combining shareds with taskdata 10337cc577a4SJonathan Peyton task->shareds = &((char *)taskdata)[shareds_offset]; 10347cc577a4SJonathan Peyton // Make sure shareds struct is aligned to pointer size 10353041982dSJonathan Peyton KMP_DEBUG_ASSERT((((kmp_uintptr_t)task->shareds) & (sizeof(void *) - 1)) == 10363041982dSJonathan Peyton 0); 10377cc577a4SJonathan Peyton } else { 10387cc577a4SJonathan Peyton task->shareds = NULL; 10397cc577a4SJonathan Peyton } 10407cc577a4SJonathan Peyton task->routine = task_entry; 10417cc577a4SJonathan Peyton task->part_id = 0; // AC: Always start with 0 part id 10427cc577a4SJonathan Peyton 10437cc577a4SJonathan Peyton taskdata->td_task_id = KMP_GEN_TASK_ID(); 10447cc577a4SJonathan Peyton taskdata->td_team = team; 10457cc577a4SJonathan Peyton taskdata->td_alloc_thread = thread; 10467cc577a4SJonathan Peyton taskdata->td_parent = parent_task; 10477cc577a4SJonathan Peyton taskdata->td_level = parent_task->td_level + 1; // increment nesting level 10487cc577a4SJonathan Peyton taskdata->td_untied_count = 0; 10497cc577a4SJonathan Peyton taskdata->td_ident = loc_ref; 10507cc577a4SJonathan Peyton taskdata->td_taskwait_ident = NULL; 10517cc577a4SJonathan Peyton taskdata->td_taskwait_counter = 0; 10527cc577a4SJonathan Peyton taskdata->td_taskwait_thread = 0; 10537cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_parent != NULL); 10547cc577a4SJonathan Peyton #if OMP_45_ENABLED 10557cc577a4SJonathan Peyton // avoid copying icvs for proxy tasks 10567cc577a4SJonathan Peyton if (flags->proxy == TASK_FULL) 10577cc577a4SJonathan Peyton #endif 10587cc577a4SJonathan Peyton copy_icvs(&taskdata->td_icvs, &taskdata->td_parent->td_icvs); 10597cc577a4SJonathan Peyton 10607cc577a4SJonathan Peyton taskdata->td_flags.tiedness = flags->tiedness; 10617cc577a4SJonathan Peyton taskdata->td_flags.final = flags->final; 10627cc577a4SJonathan Peyton taskdata->td_flags.merged_if0 = flags->merged_if0; 10637cc577a4SJonathan Peyton #if OMP_40_ENABLED 10647cc577a4SJonathan Peyton taskdata->td_flags.destructors_thunk = flags->destructors_thunk; 10657cc577a4SJonathan Peyton #endif // OMP_40_ENABLED 10667cc577a4SJonathan Peyton #if OMP_45_ENABLED 10677cc577a4SJonathan Peyton taskdata->td_flags.proxy = flags->proxy; 10687cc577a4SJonathan Peyton taskdata->td_task_team = thread->th.th_task_team; 10697cc577a4SJonathan Peyton taskdata->td_size_alloc = shareds_offset + sizeof_shareds; 10707cc577a4SJonathan Peyton #endif 10717cc577a4SJonathan Peyton taskdata->td_flags.tasktype = TASK_EXPLICIT; 10727cc577a4SJonathan Peyton 10737cc577a4SJonathan Peyton // GEH - TODO: fix this to copy parent task's value of tasking_ser flag 10747cc577a4SJonathan Peyton taskdata->td_flags.tasking_ser = (__kmp_tasking_mode == tskm_immediate_exec); 10757cc577a4SJonathan Peyton 10767cc577a4SJonathan Peyton // GEH - TODO: fix this to copy parent task's value of team_serial flag 10777cc577a4SJonathan Peyton taskdata->td_flags.team_serial = (team->t.t_serialized) ? 1 : 0; 10787cc577a4SJonathan Peyton 10793041982dSJonathan Peyton // GEH - Note we serialize the task if the team is serialized to make sure 10803041982dSJonathan Peyton // implicit parallel region tasks are not left until program termination to 10813041982dSJonathan Peyton // execute. Also, it helps locality to execute immediately. 10823041982dSJonathan Peyton 10833041982dSJonathan Peyton taskdata->td_flags.task_serial = 10843041982dSJonathan Peyton (parent_task->td_flags.final || taskdata->td_flags.team_serial || 10853041982dSJonathan Peyton taskdata->td_flags.tasking_ser); 10867cc577a4SJonathan Peyton 10877cc577a4SJonathan Peyton taskdata->td_flags.started = 0; 10887cc577a4SJonathan Peyton taskdata->td_flags.executing = 0; 10897cc577a4SJonathan Peyton taskdata->td_flags.complete = 0; 10907cc577a4SJonathan Peyton taskdata->td_flags.freed = 0; 10917cc577a4SJonathan Peyton 10927cc577a4SJonathan Peyton taskdata->td_flags.native = flags->native; 10937cc577a4SJonathan Peyton 10947cc577a4SJonathan Peyton taskdata->td_incomplete_child_tasks = 0; 10953041982dSJonathan Peyton taskdata->td_allocated_child_tasks = 1; // start at one because counts current 10963041982dSJonathan Peyton // task and children 10977cc577a4SJonathan Peyton #if OMP_40_ENABLED 10983041982dSJonathan Peyton taskdata->td_taskgroup = 10993041982dSJonathan Peyton parent_task->td_taskgroup; // task inherits taskgroup from the parent task 11007cc577a4SJonathan Peyton taskdata->td_dephash = NULL; 11017cc577a4SJonathan Peyton taskdata->td_depnode = NULL; 11027cc577a4SJonathan Peyton #endif 11037cc577a4SJonathan Peyton 11043041982dSJonathan Peyton // Only need to keep track of child task counts if team parallel and tasking not 11053041982dSJonathan Peyton // serialized or if it is a proxy task 11067cc577a4SJonathan Peyton #if OMP_45_ENABLED 11073041982dSJonathan Peyton if (flags->proxy == TASK_PROXY || 11083041982dSJonathan Peyton !(taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser)) 11097cc577a4SJonathan Peyton #else 11107cc577a4SJonathan Peyton if (!(taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser)) 11117cc577a4SJonathan Peyton #endif 11127cc577a4SJonathan Peyton { 11137cc577a4SJonathan Peyton KMP_TEST_THEN_INC32((kmp_int32 *)(&parent_task->td_incomplete_child_tasks)); 11147cc577a4SJonathan Peyton #if OMP_40_ENABLED 11157cc577a4SJonathan Peyton if (parent_task->td_taskgroup) 11167cc577a4SJonathan Peyton KMP_TEST_THEN_INC32((kmp_int32 *)(&parent_task->td_taskgroup->count)); 11177cc577a4SJonathan Peyton #endif 11183041982dSJonathan Peyton // Only need to keep track of allocated child tasks for explicit tasks since 11193041982dSJonathan Peyton // implicit not deallocated 11207cc577a4SJonathan Peyton if (taskdata->td_parent->td_flags.tasktype == TASK_EXPLICIT) { 11213041982dSJonathan Peyton KMP_TEST_THEN_INC32( 11223041982dSJonathan Peyton (kmp_int32 *)(&taskdata->td_parent->td_allocated_child_tasks)); 11237cc577a4SJonathan Peyton } 11247cc577a4SJonathan Peyton } 11257cc577a4SJonathan Peyton 11267cc577a4SJonathan Peyton KA_TRACE(20, ("__kmp_task_alloc(exit): T#%d created task %p parent=%p\n", 11277cc577a4SJonathan Peyton gtid, taskdata, taskdata->td_parent)); 11287cc577a4SJonathan Peyton ANNOTATE_HAPPENS_BEFORE(task); 11297cc577a4SJonathan Peyton 11307cc577a4SJonathan Peyton #if OMPT_SUPPORT 11317cc577a4SJonathan Peyton __kmp_task_init_ompt(taskdata, gtid, (void *)task_entry); 11327cc577a4SJonathan Peyton #endif 11337cc577a4SJonathan Peyton 11347cc577a4SJonathan Peyton return task; 11357cc577a4SJonathan Peyton } 11367cc577a4SJonathan Peyton 11373041982dSJonathan Peyton kmp_task_t *__kmpc_omp_task_alloc(ident_t *loc_ref, kmp_int32 gtid, 11383041982dSJonathan Peyton kmp_int32 flags, size_t sizeof_kmp_task_t, 11393041982dSJonathan Peyton size_t sizeof_shareds, 11403041982dSJonathan Peyton kmp_routine_entry_t task_entry) { 11417cc577a4SJonathan Peyton kmp_task_t *retval; 11427cc577a4SJonathan Peyton kmp_tasking_flags_t *input_flags = (kmp_tasking_flags_t *)&flags; 11437cc577a4SJonathan Peyton 11447cc577a4SJonathan Peyton input_flags->native = FALSE; 11457cc577a4SJonathan Peyton // __kmp_task_alloc() sets up all other runtime flags 11467cc577a4SJonathan Peyton 11477cc577a4SJonathan Peyton #if OMP_45_ENABLED 11487cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_alloc(enter): T#%d loc=%p, flags=(%s %s) " 11497cc577a4SJonathan Peyton "sizeof_task=%ld sizeof_shared=%ld entry=%p\n", 11507cc577a4SJonathan Peyton gtid, loc_ref, input_flags->tiedness ? "tied " : "untied", 11513041982dSJonathan Peyton input_flags->proxy ? "proxy" : "", sizeof_kmp_task_t, 11523041982dSJonathan Peyton sizeof_shareds, task_entry)); 11537cc577a4SJonathan Peyton #else 11547cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_alloc(enter): T#%d loc=%p, flags=(%s) " 11557cc577a4SJonathan Peyton "sizeof_task=%ld sizeof_shared=%ld entry=%p\n", 11567cc577a4SJonathan Peyton gtid, loc_ref, input_flags->tiedness ? "tied " : "untied", 11577cc577a4SJonathan Peyton sizeof_kmp_task_t, sizeof_shareds, task_entry)); 11587cc577a4SJonathan Peyton #endif 11597cc577a4SJonathan Peyton 11607cc577a4SJonathan Peyton retval = __kmp_task_alloc(loc_ref, gtid, input_flags, sizeof_kmp_task_t, 11617cc577a4SJonathan Peyton sizeof_shareds, task_entry); 11627cc577a4SJonathan Peyton 11637cc577a4SJonathan Peyton KA_TRACE(20, ("__kmpc_omp_task_alloc(exit): T#%d retval %p\n", gtid, retval)); 11647cc577a4SJonathan Peyton 11657cc577a4SJonathan Peyton return retval; 11667cc577a4SJonathan Peyton } 11677cc577a4SJonathan Peyton 11687cc577a4SJonathan Peyton // __kmp_invoke_task: invoke the specified task 11697cc577a4SJonathan Peyton // 11707cc577a4SJonathan Peyton // gtid: global thread ID of caller 11717cc577a4SJonathan Peyton // task: the task to invoke 11727cc577a4SJonathan Peyton // current_task: the task to resume after task invokation 11733041982dSJonathan Peyton static void __kmp_invoke_task(kmp_int32 gtid, kmp_task_t *task, 11743041982dSJonathan Peyton kmp_taskdata_t *current_task) { 11757cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task); 11767cc577a4SJonathan Peyton kmp_uint64 cur_time; 11777cc577a4SJonathan Peyton #if OMP_40_ENABLED 11787cc577a4SJonathan Peyton int discard = 0 /* false */; 11797cc577a4SJonathan Peyton #endif 11803041982dSJonathan Peyton KA_TRACE( 11813041982dSJonathan Peyton 30, ("__kmp_invoke_task(enter): T#%d invoking task %p, current_task=%p\n", 11827cc577a4SJonathan Peyton gtid, taskdata, current_task)); 11837cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task); 11847cc577a4SJonathan Peyton #if OMP_45_ENABLED 11857cc577a4SJonathan Peyton if (taskdata->td_flags.proxy == TASK_PROXY && 11863041982dSJonathan Peyton taskdata->td_flags.complete == 1) { 11877cc577a4SJonathan Peyton // This is a proxy task that was already completed but it needs to run 11887cc577a4SJonathan Peyton // its bottom-half finish 11893041982dSJonathan Peyton KA_TRACE( 11903041982dSJonathan Peyton 30, 11913041982dSJonathan Peyton ("__kmp_invoke_task: T#%d running bottom finish for proxy task %p\n", 11927cc577a4SJonathan Peyton gtid, taskdata)); 11937cc577a4SJonathan Peyton 11947cc577a4SJonathan Peyton __kmp_bottom_half_finish_proxy(gtid, task); 11957cc577a4SJonathan Peyton 11963041982dSJonathan Peyton KA_TRACE(30, ("__kmp_invoke_task(exit): T#%d completed bottom finish for " 11973041982dSJonathan Peyton "proxy task %p, resuming task %p\n", 11983041982dSJonathan Peyton gtid, taskdata, current_task)); 11997cc577a4SJonathan Peyton 12007cc577a4SJonathan Peyton return; 12017cc577a4SJonathan Peyton } 12027cc577a4SJonathan Peyton #endif 12037cc577a4SJonathan Peyton 12047cc577a4SJonathan Peyton #if USE_ITT_BUILD && USE_ITT_NOTIFY 12057cc577a4SJonathan Peyton if (__kmp_forkjoin_frames_mode == 3) { 12063041982dSJonathan Peyton // Get the current time stamp to measure task execution time to correct 12073041982dSJonathan Peyton // barrier imbalance time 12087cc577a4SJonathan Peyton cur_time = __itt_get_timestamp(); 12097cc577a4SJonathan Peyton } 12107cc577a4SJonathan Peyton #endif 12117cc577a4SJonathan Peyton 12127cc577a4SJonathan Peyton #if OMP_45_ENABLED 12137cc577a4SJonathan Peyton // Proxy tasks are not handled by the runtime 12147cc577a4SJonathan Peyton if (taskdata->td_flags.proxy != TASK_PROXY) { 12157cc577a4SJonathan Peyton #endif 12167cc577a4SJonathan Peyton ANNOTATE_HAPPENS_AFTER(task); 12177cc577a4SJonathan Peyton __kmp_task_start(gtid, task, current_task); 12187cc577a4SJonathan Peyton #if OMP_45_ENABLED 12197cc577a4SJonathan Peyton } 12207cc577a4SJonathan Peyton #endif 12217cc577a4SJonathan Peyton 12227cc577a4SJonathan Peyton #if OMPT_SUPPORT 12237cc577a4SJonathan Peyton ompt_thread_info_t oldInfo; 12247cc577a4SJonathan Peyton kmp_info_t *thread; 12257cc577a4SJonathan Peyton if (ompt_enabled) { 12267cc577a4SJonathan Peyton // Store the threads states and restore them after the task 12277cc577a4SJonathan Peyton thread = __kmp_threads[gtid]; 12287cc577a4SJonathan Peyton oldInfo = thread->th.ompt_thread_info; 12297cc577a4SJonathan Peyton thread->th.ompt_thread_info.wait_id = 0; 12307cc577a4SJonathan Peyton thread->th.ompt_thread_info.state = ompt_state_work_parallel; 12313041982dSJonathan Peyton taskdata->ompt_task_info.frame.exit_runtime_frame = 12323041982dSJonathan Peyton __builtin_frame_address(0); 12337cc577a4SJonathan Peyton } 12347cc577a4SJonathan Peyton #endif 12357cc577a4SJonathan Peyton 12367cc577a4SJonathan Peyton #if OMP_40_ENABLED 12377cc577a4SJonathan Peyton // TODO: cancel tasks if the parallel region has also been cancelled 12387cc577a4SJonathan Peyton // TODO: check if this sequence can be hoisted above __kmp_task_start 12397cc577a4SJonathan Peyton // if cancellation has been enabled for this run ... 12407cc577a4SJonathan Peyton if (__kmp_omp_cancellation) { 12417cc577a4SJonathan Peyton kmp_info_t *this_thr = __kmp_threads[gtid]; 12427cc577a4SJonathan Peyton kmp_team_t *this_team = this_thr->th.th_team; 12437cc577a4SJonathan Peyton kmp_taskgroup_t *taskgroup = taskdata->td_taskgroup; 12443041982dSJonathan Peyton if ((taskgroup && taskgroup->cancel_request) || 12453041982dSJonathan Peyton (this_team->t.t_cancel_request == cancel_parallel)) { 12467cc577a4SJonathan Peyton KMP_COUNT_BLOCK(TASK_cancelled); 12477cc577a4SJonathan Peyton // this task belongs to a task group and we need to cancel it 12487cc577a4SJonathan Peyton discard = 1 /* true */; 12497cc577a4SJonathan Peyton } 12507cc577a4SJonathan Peyton } 12517cc577a4SJonathan Peyton 12527cc577a4SJonathan Peyton // Invoke the task routine and pass in relevant data. 12537cc577a4SJonathan Peyton // Thunks generated by gcc take a different argument list. 12547cc577a4SJonathan Peyton if (!discard) { 12557cc577a4SJonathan Peyton #if KMP_STATS_ENABLED 12567cc577a4SJonathan Peyton KMP_COUNT_BLOCK(TASK_executed); 12577cc577a4SJonathan Peyton switch (KMP_GET_THREAD_STATE()) { 12583041982dSJonathan Peyton case FORK_JOIN_BARRIER: 12593041982dSJonathan Peyton KMP_PUSH_PARTITIONED_TIMER(OMP_task_join_bar); 12603041982dSJonathan Peyton break; 12613041982dSJonathan Peyton case PLAIN_BARRIER: 12623041982dSJonathan Peyton KMP_PUSH_PARTITIONED_TIMER(OMP_task_plain_bar); 12633041982dSJonathan Peyton break; 12643041982dSJonathan Peyton case TASKYIELD: 12653041982dSJonathan Peyton KMP_PUSH_PARTITIONED_TIMER(OMP_task_taskyield); 12663041982dSJonathan Peyton break; 12673041982dSJonathan Peyton case TASKWAIT: 12683041982dSJonathan Peyton KMP_PUSH_PARTITIONED_TIMER(OMP_task_taskwait); 12693041982dSJonathan Peyton break; 12703041982dSJonathan Peyton case TASKGROUP: 12713041982dSJonathan Peyton KMP_PUSH_PARTITIONED_TIMER(OMP_task_taskgroup); 12723041982dSJonathan Peyton break; 12733041982dSJonathan Peyton default: 12743041982dSJonathan Peyton KMP_PUSH_PARTITIONED_TIMER(OMP_task_immediate); 12753041982dSJonathan Peyton break; 12767cc577a4SJonathan Peyton } 12777cc577a4SJonathan Peyton #endif // KMP_STATS_ENABLED 12787cc577a4SJonathan Peyton #endif // OMP_40_ENABLED 12797cc577a4SJonathan Peyton 12807cc577a4SJonathan Peyton #if OMPT_SUPPORT && OMPT_TRACE 12817cc577a4SJonathan Peyton /* let OMPT know that we're about to run this task */ 12823041982dSJonathan Peyton if (ompt_enabled && ompt_callbacks.ompt_callback(ompt_event_task_switch)) { 12837cc577a4SJonathan Peyton ompt_callbacks.ompt_callback(ompt_event_task_switch)( 12847cc577a4SJonathan Peyton current_task->ompt_task_info.task_id, 12857cc577a4SJonathan Peyton taskdata->ompt_task_info.task_id); 12867cc577a4SJonathan Peyton } 12877cc577a4SJonathan Peyton #endif 12887cc577a4SJonathan Peyton 12897cc577a4SJonathan Peyton #ifdef KMP_GOMP_COMPAT 12907cc577a4SJonathan Peyton if (taskdata->td_flags.native) { 12917cc577a4SJonathan Peyton ((void (*)(void *))(*(task->routine)))(task->shareds); 12923041982dSJonathan Peyton } else 12937cc577a4SJonathan Peyton #endif /* KMP_GOMP_COMPAT */ 12947cc577a4SJonathan Peyton { 12957cc577a4SJonathan Peyton (*(task->routine))(gtid, task); 12967cc577a4SJonathan Peyton } 12977cc577a4SJonathan Peyton KMP_POP_PARTITIONED_TIMER(); 12987cc577a4SJonathan Peyton 12997cc577a4SJonathan Peyton #if OMPT_SUPPORT && OMPT_TRACE 13007cc577a4SJonathan Peyton /* let OMPT know that we're returning to the callee task */ 13013041982dSJonathan Peyton if (ompt_enabled && ompt_callbacks.ompt_callback(ompt_event_task_switch)) { 13027cc577a4SJonathan Peyton ompt_callbacks.ompt_callback(ompt_event_task_switch)( 13037cc577a4SJonathan Peyton taskdata->ompt_task_info.task_id, 13047cc577a4SJonathan Peyton current_task->ompt_task_info.task_id); 13057cc577a4SJonathan Peyton } 13067cc577a4SJonathan Peyton #endif 13077cc577a4SJonathan Peyton 13087cc577a4SJonathan Peyton #if OMP_40_ENABLED 13097cc577a4SJonathan Peyton } 13107cc577a4SJonathan Peyton #endif // OMP_40_ENABLED 13117cc577a4SJonathan Peyton 13127cc577a4SJonathan Peyton #if OMPT_SUPPORT 13137cc577a4SJonathan Peyton if (ompt_enabled) { 13147cc577a4SJonathan Peyton thread->th.ompt_thread_info = oldInfo; 13157cc577a4SJonathan Peyton taskdata->ompt_task_info.frame.exit_runtime_frame = NULL; 13167cc577a4SJonathan Peyton } 13177cc577a4SJonathan Peyton #endif 13187cc577a4SJonathan Peyton 13197cc577a4SJonathan Peyton #if OMP_45_ENABLED 13207cc577a4SJonathan Peyton // Proxy tasks are not handled by the runtime 13217cc577a4SJonathan Peyton if (taskdata->td_flags.proxy != TASK_PROXY) { 13227cc577a4SJonathan Peyton #endif 13237cc577a4SJonathan Peyton ANNOTATE_HAPPENS_BEFORE(taskdata->td_parent); 13247cc577a4SJonathan Peyton __kmp_task_finish(gtid, task, current_task); 13257cc577a4SJonathan Peyton #if OMP_45_ENABLED 13267cc577a4SJonathan Peyton } 13277cc577a4SJonathan Peyton #endif 13287cc577a4SJonathan Peyton 13297cc577a4SJonathan Peyton #if USE_ITT_BUILD && USE_ITT_NOTIFY 13307cc577a4SJonathan Peyton // Barrier imbalance - correct arrive time after the task finished 13317cc577a4SJonathan Peyton if (__kmp_forkjoin_frames_mode == 3) { 13327cc577a4SJonathan Peyton kmp_info_t *this_thr = __kmp_threads[gtid]; 13337cc577a4SJonathan Peyton if (this_thr->th.th_bar_arrive_time) { 13347cc577a4SJonathan Peyton this_thr->th.th_bar_arrive_time += (__itt_get_timestamp() - cur_time); 13357cc577a4SJonathan Peyton } 13367cc577a4SJonathan Peyton } 13377cc577a4SJonathan Peyton #endif 13383041982dSJonathan Peyton KA_TRACE( 13393041982dSJonathan Peyton 30, 13403041982dSJonathan Peyton ("__kmp_invoke_task(exit): T#%d completed task %p, resuming task %p\n", 13417cc577a4SJonathan Peyton gtid, taskdata, current_task)); 13427cc577a4SJonathan Peyton return; 13437cc577a4SJonathan Peyton } 13447cc577a4SJonathan Peyton 13457cc577a4SJonathan Peyton // __kmpc_omp_task_parts: Schedule a thread-switchable task for execution 13467cc577a4SJonathan Peyton // 13477cc577a4SJonathan Peyton // loc_ref: location of original task pragma (ignored) 13487cc577a4SJonathan Peyton // gtid: Global Thread ID of encountering thread 13497cc577a4SJonathan Peyton // new_task: task thunk allocated by __kmp_omp_task_alloc() for the ''new task'' 13507cc577a4SJonathan Peyton // Returns: 13513041982dSJonathan Peyton // TASK_CURRENT_NOT_QUEUED (0) if did not suspend and queue current task to 13523041982dSJonathan Peyton // be resumed later. 13533041982dSJonathan Peyton // TASK_CURRENT_QUEUED (1) if suspended and queued the current task to be 13543041982dSJonathan Peyton // resumed later. 13553041982dSJonathan Peyton kmp_int32 __kmpc_omp_task_parts(ident_t *loc_ref, kmp_int32 gtid, 13563041982dSJonathan Peyton kmp_task_t *new_task) { 13577cc577a4SJonathan Peyton kmp_taskdata_t *new_taskdata = KMP_TASK_TO_TASKDATA(new_task); 13587cc577a4SJonathan Peyton 13593041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task_parts(enter): T#%d loc=%p task=%p\n", gtid, 13603041982dSJonathan Peyton loc_ref, new_taskdata)); 13617cc577a4SJonathan Peyton 13623041982dSJonathan Peyton /* Should we execute the new task or queue it? For now, let's just always try 13633041982dSJonathan Peyton to queue it. If the queue fills up, then we'll execute it. */ 13647cc577a4SJonathan Peyton 13657cc577a4SJonathan Peyton if (__kmp_push_task(gtid, new_task) == TASK_NOT_PUSHED) // if cannot defer 13667cc577a4SJonathan Peyton { // Execute this task immediately 13677cc577a4SJonathan Peyton kmp_taskdata_t *current_task = __kmp_threads[gtid]->th.th_current_task; 13687cc577a4SJonathan Peyton new_taskdata->td_flags.task_serial = 1; 13697cc577a4SJonathan Peyton __kmp_invoke_task(gtid, new_task, current_task); 13707cc577a4SJonathan Peyton } 13717cc577a4SJonathan Peyton 13723041982dSJonathan Peyton KA_TRACE( 13733041982dSJonathan Peyton 10, 13743041982dSJonathan Peyton ("__kmpc_omp_task_parts(exit): T#%d returning TASK_CURRENT_NOT_QUEUED: " 13753041982dSJonathan Peyton "loc=%p task=%p, return: TASK_CURRENT_NOT_QUEUED\n", 13763041982dSJonathan Peyton gtid, loc_ref, new_taskdata)); 13777cc577a4SJonathan Peyton 13787cc577a4SJonathan Peyton ANNOTATE_HAPPENS_BEFORE(new_task); 13797cc577a4SJonathan Peyton return TASK_CURRENT_NOT_QUEUED; 13807cc577a4SJonathan Peyton } 13817cc577a4SJonathan Peyton 13827cc577a4SJonathan Peyton // __kmp_omp_task: Schedule a non-thread-switchable task for execution 13833041982dSJonathan Peyton // 13847cc577a4SJonathan Peyton // gtid: Global Thread ID of encountering thread 13857cc577a4SJonathan Peyton // new_task:non-thread-switchable task thunk allocated by __kmp_omp_task_alloc() 13863041982dSJonathan Peyton // serialize_immediate: if TRUE then if the task is executed immediately its 13873041982dSJonathan Peyton // execution will be serialized 13883041982dSJonathan Peyton // Returns: 13893041982dSJonathan Peyton // TASK_CURRENT_NOT_QUEUED (0) if did not suspend and queue current task to 13903041982dSJonathan Peyton // be resumed later. 13913041982dSJonathan Peyton // TASK_CURRENT_QUEUED (1) if suspended and queued the current task to be 13923041982dSJonathan Peyton // resumed later. 13933041982dSJonathan Peyton kmp_int32 __kmp_omp_task(kmp_int32 gtid, kmp_task_t *new_task, 13943041982dSJonathan Peyton bool serialize_immediate) { 13957cc577a4SJonathan Peyton kmp_taskdata_t *new_taskdata = KMP_TASK_TO_TASKDATA(new_task); 13967cc577a4SJonathan Peyton 13977cc577a4SJonathan Peyton #if OMPT_SUPPORT 13987cc577a4SJonathan Peyton if (ompt_enabled) { 13997cc577a4SJonathan Peyton new_taskdata->ompt_task_info.frame.reenter_runtime_frame = 14007cc577a4SJonathan Peyton __builtin_frame_address(1); 14017cc577a4SJonathan Peyton } 14027cc577a4SJonathan Peyton #endif 14037cc577a4SJonathan Peyton 14047cc577a4SJonathan Peyton /* Should we execute the new task or queue it? For now, let's just always try to 14057cc577a4SJonathan Peyton queue it. If the queue fills up, then we'll execute it. */ 14067cc577a4SJonathan Peyton #if OMP_45_ENABLED 14073041982dSJonathan Peyton if (new_taskdata->td_flags.proxy == TASK_PROXY || 14083041982dSJonathan Peyton __kmp_push_task(gtid, new_task) == TASK_NOT_PUSHED) // if cannot defer 14097cc577a4SJonathan Peyton #else 14107cc577a4SJonathan Peyton if (__kmp_push_task(gtid, new_task) == TASK_NOT_PUSHED) // if cannot defer 14117cc577a4SJonathan Peyton #endif 14127cc577a4SJonathan Peyton { // Execute this task immediately 14137cc577a4SJonathan Peyton kmp_taskdata_t *current_task = __kmp_threads[gtid]->th.th_current_task; 14147cc577a4SJonathan Peyton if (serialize_immediate) 14157cc577a4SJonathan Peyton new_taskdata->td_flags.task_serial = 1; 14167cc577a4SJonathan Peyton __kmp_invoke_task(gtid, new_task, current_task); 14177cc577a4SJonathan Peyton } 14187cc577a4SJonathan Peyton 14197cc577a4SJonathan Peyton #if OMPT_SUPPORT 14207cc577a4SJonathan Peyton if (ompt_enabled) { 14217cc577a4SJonathan Peyton new_taskdata->ompt_task_info.frame.reenter_runtime_frame = NULL; 14227cc577a4SJonathan Peyton } 14237cc577a4SJonathan Peyton #endif 14247cc577a4SJonathan Peyton 14257cc577a4SJonathan Peyton ANNOTATE_HAPPENS_BEFORE(new_task); 14267cc577a4SJonathan Peyton return TASK_CURRENT_NOT_QUEUED; 14277cc577a4SJonathan Peyton } 14287cc577a4SJonathan Peyton 14293041982dSJonathan Peyton // __kmpc_omp_task: Wrapper around __kmp_omp_task to schedule a 14303041982dSJonathan Peyton // non-thread-switchable task from the parent thread only! 14313041982dSJonathan Peyton // 14327cc577a4SJonathan Peyton // loc_ref: location of original task pragma (ignored) 14337cc577a4SJonathan Peyton // gtid: Global Thread ID of encountering thread 14343041982dSJonathan Peyton // new_task: non-thread-switchable task thunk allocated by 14353041982dSJonathan Peyton // __kmp_omp_task_alloc() 14363041982dSJonathan Peyton // Returns: 14373041982dSJonathan Peyton // TASK_CURRENT_NOT_QUEUED (0) if did not suspend and queue current task to 14383041982dSJonathan Peyton // be resumed later. 14393041982dSJonathan Peyton // TASK_CURRENT_QUEUED (1) if suspended and queued the current task to be 14403041982dSJonathan Peyton // resumed later. 14413041982dSJonathan Peyton kmp_int32 __kmpc_omp_task(ident_t *loc_ref, kmp_int32 gtid, 14423041982dSJonathan Peyton kmp_task_t *new_task) { 14437cc577a4SJonathan Peyton kmp_int32 res; 14447cc577a4SJonathan Peyton KMP_SET_THREAD_STATE_BLOCK(EXPLICIT_TASK); 14457cc577a4SJonathan Peyton 14467cc577a4SJonathan Peyton #if KMP_DEBUG 14477cc577a4SJonathan Peyton kmp_taskdata_t *new_taskdata = KMP_TASK_TO_TASKDATA(new_task); 14487cc577a4SJonathan Peyton #endif 14493041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task(enter): T#%d loc=%p task=%p\n", gtid, loc_ref, 14503041982dSJonathan Peyton new_taskdata)); 14517cc577a4SJonathan Peyton 14527cc577a4SJonathan Peyton res = __kmp_omp_task(gtid, new_task, true); 14537cc577a4SJonathan Peyton 14543041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_omp_task(exit): T#%d returning " 14553041982dSJonathan Peyton "TASK_CURRENT_NOT_QUEUED: loc=%p task=%p\n", 14567cc577a4SJonathan Peyton gtid, loc_ref, new_taskdata)); 14577cc577a4SJonathan Peyton return res; 14587cc577a4SJonathan Peyton } 14597cc577a4SJonathan Peyton 14603041982dSJonathan Peyton // __kmpc_omp_taskwait: Wait until all tasks generated by the current task are 14613041982dSJonathan Peyton // complete 14623041982dSJonathan Peyton kmp_int32 __kmpc_omp_taskwait(ident_t *loc_ref, kmp_int32 gtid) { 14637cc577a4SJonathan Peyton kmp_taskdata_t *taskdata; 14647cc577a4SJonathan Peyton kmp_info_t *thread; 14657cc577a4SJonathan Peyton int thread_finished = FALSE; 14667cc577a4SJonathan Peyton KMP_SET_THREAD_STATE_BLOCK(TASKWAIT); 14677cc577a4SJonathan Peyton 14687cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_omp_taskwait(enter): T#%d loc=%p\n", gtid, loc_ref)); 14697cc577a4SJonathan Peyton 14707cc577a4SJonathan Peyton if (__kmp_tasking_mode != tskm_immediate_exec) { 14717cc577a4SJonathan Peyton thread = __kmp_threads[gtid]; 14727cc577a4SJonathan Peyton taskdata = thread->th.th_current_task; 14737cc577a4SJonathan Peyton #if OMPT_SUPPORT && OMPT_TRACE 14747cc577a4SJonathan Peyton ompt_task_id_t my_task_id; 14757cc577a4SJonathan Peyton ompt_parallel_id_t my_parallel_id; 14767cc577a4SJonathan Peyton 14777cc577a4SJonathan Peyton if (ompt_enabled) { 14787cc577a4SJonathan Peyton kmp_team_t *team = thread->th.th_team; 14797cc577a4SJonathan Peyton my_task_id = taskdata->ompt_task_info.task_id; 14807cc577a4SJonathan Peyton my_parallel_id = team->t.ompt_team_info.parallel_id; 14817cc577a4SJonathan Peyton 14823041982dSJonathan Peyton taskdata->ompt_task_info.frame.reenter_runtime_frame = 14833041982dSJonathan Peyton __builtin_frame_address(1); 14847cc577a4SJonathan Peyton if (ompt_callbacks.ompt_callback(ompt_event_taskwait_begin)) { 14853041982dSJonathan Peyton ompt_callbacks.ompt_callback(ompt_event_taskwait_begin)(my_parallel_id, 14863041982dSJonathan Peyton my_task_id); 14877cc577a4SJonathan Peyton } 14887cc577a4SJonathan Peyton } 14897cc577a4SJonathan Peyton #endif 14907cc577a4SJonathan Peyton 14913041982dSJonathan Peyton // Debugger: The taskwait is active. Store location and thread encountered the 14923041982dSJonathan Peyton // taskwait. 14937cc577a4SJonathan Peyton #if USE_ITT_BUILD 14947cc577a4SJonathan Peyton // Note: These values are used by ITT events as well. 14957cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 14967cc577a4SJonathan Peyton taskdata->td_taskwait_counter += 1; 14977cc577a4SJonathan Peyton taskdata->td_taskwait_ident = loc_ref; 14987cc577a4SJonathan Peyton taskdata->td_taskwait_thread = gtid + 1; 14997cc577a4SJonathan Peyton 15007cc577a4SJonathan Peyton #if USE_ITT_BUILD 15017cc577a4SJonathan Peyton void *itt_sync_obj = __kmp_itt_taskwait_object(gtid); 15027cc577a4SJonathan Peyton if (itt_sync_obj != NULL) 15037cc577a4SJonathan Peyton __kmp_itt_taskwait_starting(gtid, itt_sync_obj); 15047cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 15057cc577a4SJonathan Peyton 15063041982dSJonathan Peyton bool must_wait = 15073041982dSJonathan Peyton !taskdata->td_flags.team_serial && !taskdata->td_flags.final; 15087cc577a4SJonathan Peyton 15097cc577a4SJonathan Peyton #if OMP_45_ENABLED 15103041982dSJonathan Peyton must_wait = must_wait || (thread->th.th_task_team != NULL && 15113041982dSJonathan Peyton thread->th.th_task_team->tt.tt_found_proxy_tasks); 15127cc577a4SJonathan Peyton #endif 15133041982dSJonathan Peyton if (must_wait) { 15147cc577a4SJonathan Peyton kmp_flag_32 flag(&(taskdata->td_incomplete_child_tasks), 0U); 15157cc577a4SJonathan Peyton while (TCR_4(taskdata->td_incomplete_child_tasks) != 0) { 15163041982dSJonathan Peyton flag.execute_tasks(thread, gtid, FALSE, 15173041982dSJonathan Peyton &thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), 15183041982dSJonathan Peyton __kmp_task_stealing_constraint); 15197cc577a4SJonathan Peyton } 15207cc577a4SJonathan Peyton } 15217cc577a4SJonathan Peyton #if USE_ITT_BUILD 15227cc577a4SJonathan Peyton if (itt_sync_obj != NULL) 15237cc577a4SJonathan Peyton __kmp_itt_taskwait_finished(gtid, itt_sync_obj); 15247cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 15257cc577a4SJonathan Peyton 15263041982dSJonathan Peyton // Debugger: The taskwait is completed. Location remains, but thread is 15273041982dSJonathan Peyton // negated. 15287cc577a4SJonathan Peyton taskdata->td_taskwait_thread = -taskdata->td_taskwait_thread; 15297cc577a4SJonathan Peyton 15307cc577a4SJonathan Peyton #if OMPT_SUPPORT && OMPT_TRACE 15317cc577a4SJonathan Peyton if (ompt_enabled) { 15327cc577a4SJonathan Peyton if (ompt_callbacks.ompt_callback(ompt_event_taskwait_end)) { 15333041982dSJonathan Peyton ompt_callbacks.ompt_callback(ompt_event_taskwait_end)(my_parallel_id, 15343041982dSJonathan Peyton my_task_id); 15357cc577a4SJonathan Peyton } 15367cc577a4SJonathan Peyton taskdata->ompt_task_info.frame.reenter_runtime_frame = NULL; 15377cc577a4SJonathan Peyton } 15387cc577a4SJonathan Peyton #endif 15397cc577a4SJonathan Peyton ANNOTATE_HAPPENS_AFTER(taskdata); 15407cc577a4SJonathan Peyton } 15417cc577a4SJonathan Peyton 15427cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_omp_taskwait(exit): T#%d task %p finished waiting, " 15433041982dSJonathan Peyton "returning TASK_CURRENT_NOT_QUEUED\n", 15443041982dSJonathan Peyton gtid, taskdata)); 15457cc577a4SJonathan Peyton 15467cc577a4SJonathan Peyton return TASK_CURRENT_NOT_QUEUED; 15477cc577a4SJonathan Peyton } 15487cc577a4SJonathan Peyton 15497cc577a4SJonathan Peyton // __kmpc_omp_taskyield: switch to a different task 15503041982dSJonathan Peyton kmp_int32 __kmpc_omp_taskyield(ident_t *loc_ref, kmp_int32 gtid, int end_part) { 15517cc577a4SJonathan Peyton kmp_taskdata_t *taskdata; 15527cc577a4SJonathan Peyton kmp_info_t *thread; 15537cc577a4SJonathan Peyton int thread_finished = FALSE; 15547cc577a4SJonathan Peyton 15557cc577a4SJonathan Peyton KMP_COUNT_BLOCK(OMP_TASKYIELD); 15567cc577a4SJonathan Peyton KMP_SET_THREAD_STATE_BLOCK(TASKYIELD); 15577cc577a4SJonathan Peyton 15587cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_omp_taskyield(enter): T#%d loc=%p end_part = %d\n", 15597cc577a4SJonathan Peyton gtid, loc_ref, end_part)); 15607cc577a4SJonathan Peyton 15617cc577a4SJonathan Peyton if (__kmp_tasking_mode != tskm_immediate_exec && __kmp_init_parallel) { 15627cc577a4SJonathan Peyton thread = __kmp_threads[gtid]; 15637cc577a4SJonathan Peyton taskdata = thread->th.th_current_task; 15647cc577a4SJonathan Peyton // Should we model this as a task wait or not? 15653041982dSJonathan Peyton // Debugger: The taskwait is active. Store location and thread encountered the 15663041982dSJonathan Peyton // taskwait. 15677cc577a4SJonathan Peyton #if USE_ITT_BUILD 15687cc577a4SJonathan Peyton // Note: These values are used by ITT events as well. 15697cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 15707cc577a4SJonathan Peyton taskdata->td_taskwait_counter += 1; 15717cc577a4SJonathan Peyton taskdata->td_taskwait_ident = loc_ref; 15727cc577a4SJonathan Peyton taskdata->td_taskwait_thread = gtid + 1; 15737cc577a4SJonathan Peyton 15747cc577a4SJonathan Peyton #if USE_ITT_BUILD 15757cc577a4SJonathan Peyton void *itt_sync_obj = __kmp_itt_taskwait_object(gtid); 15767cc577a4SJonathan Peyton if (itt_sync_obj != NULL) 15777cc577a4SJonathan Peyton __kmp_itt_taskwait_starting(gtid, itt_sync_obj); 15787cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 15797cc577a4SJonathan Peyton if (!taskdata->td_flags.team_serial) { 15807cc577a4SJonathan Peyton kmp_task_team_t *task_team = thread->th.th_task_team; 15817cc577a4SJonathan Peyton if (task_team != NULL) { 15827cc577a4SJonathan Peyton if (KMP_TASKING_ENABLED(task_team)) { 15833041982dSJonathan Peyton __kmp_execute_tasks_32( 15843041982dSJonathan Peyton thread, gtid, NULL, FALSE, 15853041982dSJonathan Peyton &thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), 15863041982dSJonathan Peyton __kmp_task_stealing_constraint); 15877cc577a4SJonathan Peyton } 15887cc577a4SJonathan Peyton } 15897cc577a4SJonathan Peyton } 15907cc577a4SJonathan Peyton #if USE_ITT_BUILD 15917cc577a4SJonathan Peyton if (itt_sync_obj != NULL) 15927cc577a4SJonathan Peyton __kmp_itt_taskwait_finished(gtid, itt_sync_obj); 15937cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 15947cc577a4SJonathan Peyton 15953041982dSJonathan Peyton // Debugger: The taskwait is completed. Location remains, but thread is 15963041982dSJonathan Peyton // negated. 15977cc577a4SJonathan Peyton taskdata->td_taskwait_thread = -taskdata->td_taskwait_thread; 15987cc577a4SJonathan Peyton } 15997cc577a4SJonathan Peyton 16007cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_omp_taskyield(exit): T#%d task %p resuming, " 16013041982dSJonathan Peyton "returning TASK_CURRENT_NOT_QUEUED\n", 16023041982dSJonathan Peyton gtid, taskdata)); 16037cc577a4SJonathan Peyton 16047cc577a4SJonathan Peyton return TASK_CURRENT_NOT_QUEUED; 16057cc577a4SJonathan Peyton } 16067cc577a4SJonathan Peyton 160772ba2109SAndrey Churbanov // TODO: change to OMP_50_ENABLED, need to change build tools for this to work 160872ba2109SAndrey Churbanov #if OMP_45_ENABLED 160972ba2109SAndrey Churbanov // Task Reduction implementation 161072ba2109SAndrey Churbanov 161172ba2109SAndrey Churbanov typedef struct kmp_task_red_flags { 161272ba2109SAndrey Churbanov unsigned lazy_priv : 1; // hint: (1) use lazy allocation (big objects) 161372ba2109SAndrey Churbanov unsigned reserved31 : 31; 161472ba2109SAndrey Churbanov } kmp_task_red_flags_t; 161572ba2109SAndrey Churbanov 161672ba2109SAndrey Churbanov // internal structure for reduction data item related info 161772ba2109SAndrey Churbanov typedef struct kmp_task_red_data { 161872ba2109SAndrey Churbanov void *reduce_shar; // shared reduction item 161972ba2109SAndrey Churbanov size_t reduce_size; // size of data item 162072ba2109SAndrey Churbanov void *reduce_priv; // thread specific data 162172ba2109SAndrey Churbanov void *reduce_pend; // end of private data for comparison op 162272ba2109SAndrey Churbanov void *reduce_init; // data initialization routine 162372ba2109SAndrey Churbanov void *reduce_fini; // data finalization routine 162472ba2109SAndrey Churbanov void *reduce_comb; // data combiner routine 162572ba2109SAndrey Churbanov kmp_task_red_flags_t flags; // flags for additional info from compiler 162672ba2109SAndrey Churbanov } kmp_task_red_data_t; 162772ba2109SAndrey Churbanov 162872ba2109SAndrey Churbanov // structure sent us by compiler - one per reduction item 162972ba2109SAndrey Churbanov typedef struct kmp_task_red_input { 163072ba2109SAndrey Churbanov void *reduce_shar; // shared reduction item 163172ba2109SAndrey Churbanov size_t reduce_size; // size of data item 163272ba2109SAndrey Churbanov void *reduce_init; // data initialization routine 163372ba2109SAndrey Churbanov void *reduce_fini; // data finalization routine 163472ba2109SAndrey Churbanov void *reduce_comb; // data combiner routine 163572ba2109SAndrey Churbanov kmp_task_red_flags_t flags; // flags for additional info from compiler 163672ba2109SAndrey Churbanov } kmp_task_red_input_t; 163772ba2109SAndrey Churbanov 163872ba2109SAndrey Churbanov /*! 163972ba2109SAndrey Churbanov @ingroup TASKING 164072ba2109SAndrey Churbanov @param gtid Global thread ID 164172ba2109SAndrey Churbanov @param num Number of data items to reduce 164272ba2109SAndrey Churbanov @param data Array of data for reduction 164372ba2109SAndrey Churbanov @return The taskgroup identifier 164472ba2109SAndrey Churbanov 164572ba2109SAndrey Churbanov Initialize task reduction for the taskgroup. 164672ba2109SAndrey Churbanov */ 16473041982dSJonathan Peyton void *__kmpc_task_reduction_init(int gtid, int num, void *data) { 164872ba2109SAndrey Churbanov kmp_info_t *thread = __kmp_threads[gtid]; 164972ba2109SAndrey Churbanov kmp_taskgroup_t *tg = thread->th.th_current_task->td_taskgroup; 165072ba2109SAndrey Churbanov kmp_int32 nth = thread->th.th_team_nproc; 165172ba2109SAndrey Churbanov kmp_task_red_input_t *input = (kmp_task_red_input_t *)data; 165272ba2109SAndrey Churbanov kmp_task_red_data_t *arr; 165372ba2109SAndrey Churbanov 165472ba2109SAndrey Churbanov // check input data just in case 165572ba2109SAndrey Churbanov KMP_ASSERT(tg != NULL); 165672ba2109SAndrey Churbanov KMP_ASSERT(data != NULL); 165772ba2109SAndrey Churbanov KMP_ASSERT(num > 0); 165872ba2109SAndrey Churbanov if (nth == 1) { 165972ba2109SAndrey Churbanov KA_TRACE(10, ("__kmpc_task_reduction_init: T#%d, tg %p, exiting nth=1\n", 166072ba2109SAndrey Churbanov gtid, tg)); 166172ba2109SAndrey Churbanov return (void *)tg; 166272ba2109SAndrey Churbanov } 166372ba2109SAndrey Churbanov KA_TRACE(10, ("__kmpc_task_reduction_init: T#%d, taskgroup %p, #items %d\n", 166472ba2109SAndrey Churbanov gtid, tg, num)); 16653041982dSJonathan Peyton arr = (kmp_task_red_data_t *)__kmp_thread_malloc( 16663041982dSJonathan Peyton thread, num * sizeof(kmp_task_red_data_t)); 166772ba2109SAndrey Churbanov for (int i = 0; i < num; ++i) { 166872ba2109SAndrey Churbanov void (*f_init)(void *) = (void (*)(void *))(input[i].reduce_init); 166972ba2109SAndrey Churbanov size_t size = input[i].reduce_size - 1; 167072ba2109SAndrey Churbanov // round the size up to cache line per thread-specific item 167172ba2109SAndrey Churbanov size += CACHE_LINE - size % CACHE_LINE; 167272ba2109SAndrey Churbanov KMP_ASSERT(input[i].reduce_comb != NULL); // combiner is mandatory 167372ba2109SAndrey Churbanov arr[i].reduce_shar = input[i].reduce_shar; 167472ba2109SAndrey Churbanov arr[i].reduce_size = size; 167572ba2109SAndrey Churbanov arr[i].reduce_init = input[i].reduce_init; 167672ba2109SAndrey Churbanov arr[i].reduce_fini = input[i].reduce_fini; 167772ba2109SAndrey Churbanov arr[i].reduce_comb = input[i].reduce_comb; 167872ba2109SAndrey Churbanov arr[i].flags = input[i].flags; 167972ba2109SAndrey Churbanov if (!input[i].flags.lazy_priv) { 168072ba2109SAndrey Churbanov // allocate cache-line aligned block and fill it with zeros 168172ba2109SAndrey Churbanov arr[i].reduce_priv = __kmp_allocate(nth * size); 168272ba2109SAndrey Churbanov arr[i].reduce_pend = (char *)(arr[i].reduce_priv) + nth * size; 168372ba2109SAndrey Churbanov if (f_init != NULL) { 168472ba2109SAndrey Churbanov // initialize thread-specific items 168572ba2109SAndrey Churbanov for (int j = 0; j < nth; ++j) { 168672ba2109SAndrey Churbanov f_init((char *)(arr[i].reduce_priv) + j * size); 168772ba2109SAndrey Churbanov } 168872ba2109SAndrey Churbanov } 168972ba2109SAndrey Churbanov } else { 169072ba2109SAndrey Churbanov // only allocate space for pointers now, 169172ba2109SAndrey Churbanov // objects will be lazily allocated/initialized once requested 169272ba2109SAndrey Churbanov arr[i].reduce_priv = __kmp_allocate(nth * sizeof(void *)); 169372ba2109SAndrey Churbanov } 169472ba2109SAndrey Churbanov } 169572ba2109SAndrey Churbanov tg->reduce_data = (void *)arr; 169672ba2109SAndrey Churbanov tg->reduce_num_data = num; 169772ba2109SAndrey Churbanov return (void *)tg; 169872ba2109SAndrey Churbanov } 169972ba2109SAndrey Churbanov 170072ba2109SAndrey Churbanov /*! 170172ba2109SAndrey Churbanov @ingroup TASKING 170272ba2109SAndrey Churbanov @param gtid Global thread ID 170372ba2109SAndrey Churbanov @param tskgrp The taskgroup ID (optional) 170472ba2109SAndrey Churbanov @param data Shared location of the item 170572ba2109SAndrey Churbanov @return The pointer to per-thread data 170672ba2109SAndrey Churbanov 170772ba2109SAndrey Churbanov Get thread-specific location of data item 170872ba2109SAndrey Churbanov */ 17093041982dSJonathan Peyton void *__kmpc_task_reduction_get_th_data(int gtid, void *tskgrp, void *data) { 171072ba2109SAndrey Churbanov kmp_info_t *thread = __kmp_threads[gtid]; 171172ba2109SAndrey Churbanov kmp_int32 nth = thread->th.th_team_nproc; 171272ba2109SAndrey Churbanov if (nth == 1) 171372ba2109SAndrey Churbanov return data; // nothing to do 171472ba2109SAndrey Churbanov 171572ba2109SAndrey Churbanov kmp_taskgroup_t *tg = (kmp_taskgroup_t *)tskgrp; 171672ba2109SAndrey Churbanov if (tg == NULL) 171772ba2109SAndrey Churbanov tg = thread->th.th_current_task->td_taskgroup; 171872ba2109SAndrey Churbanov KMP_ASSERT(tg != NULL); 171972ba2109SAndrey Churbanov kmp_task_red_data_t *arr = (kmp_task_red_data_t *)(tg->reduce_data); 172072ba2109SAndrey Churbanov kmp_int32 num = tg->reduce_num_data; 172172ba2109SAndrey Churbanov kmp_int32 tid = thread->th.th_info.ds.ds_tid; 172272ba2109SAndrey Churbanov 172372ba2109SAndrey Churbanov KMP_ASSERT(data != NULL); 172472ba2109SAndrey Churbanov while (tg != NULL) { 172572ba2109SAndrey Churbanov for (int i = 0; i < num; ++i) { 172672ba2109SAndrey Churbanov if (!arr[i].flags.lazy_priv) { 172772ba2109SAndrey Churbanov if (data == arr[i].reduce_shar || 172872ba2109SAndrey Churbanov (data >= arr[i].reduce_priv && data < arr[i].reduce_pend)) 172972ba2109SAndrey Churbanov return (char *)(arr[i].reduce_priv) + tid * arr[i].reduce_size; 173072ba2109SAndrey Churbanov } else { 173172ba2109SAndrey Churbanov // check shared location first 173272ba2109SAndrey Churbanov void **p_priv = (void **)(arr[i].reduce_priv); 173372ba2109SAndrey Churbanov if (data == arr[i].reduce_shar) 173472ba2109SAndrey Churbanov goto found; 173572ba2109SAndrey Churbanov // check if we get some thread specific location as parameter 173672ba2109SAndrey Churbanov for (int j = 0; j < nth; ++j) 173772ba2109SAndrey Churbanov if (data == p_priv[j]) 173872ba2109SAndrey Churbanov goto found; 173972ba2109SAndrey Churbanov continue; // not found, continue search 174072ba2109SAndrey Churbanov found: 174172ba2109SAndrey Churbanov if (p_priv[tid] == NULL) { 174272ba2109SAndrey Churbanov // allocate thread specific object lazily 174372ba2109SAndrey Churbanov void (*f_init)(void *) = (void (*)(void *))(arr[i].reduce_init); 174472ba2109SAndrey Churbanov p_priv[tid] = __kmp_allocate(arr[i].reduce_size); 174572ba2109SAndrey Churbanov if (f_init != NULL) { 174672ba2109SAndrey Churbanov f_init(p_priv[tid]); 174772ba2109SAndrey Churbanov } 174872ba2109SAndrey Churbanov } 174972ba2109SAndrey Churbanov return p_priv[tid]; 175072ba2109SAndrey Churbanov } 175172ba2109SAndrey Churbanov } 175272ba2109SAndrey Churbanov tg = tg->parent; 175372ba2109SAndrey Churbanov arr = (kmp_task_red_data_t *)(tg->reduce_data); 175472ba2109SAndrey Churbanov num = tg->reduce_num_data; 175572ba2109SAndrey Churbanov } 175672ba2109SAndrey Churbanov KMP_ASSERT2(0, "Unknown task reduction item"); 175772ba2109SAndrey Churbanov return NULL; // ERROR, this line never executed 175872ba2109SAndrey Churbanov } 175972ba2109SAndrey Churbanov 176072ba2109SAndrey Churbanov // Finalize task reduction. 176172ba2109SAndrey Churbanov // Called from __kmpc_end_taskgroup() 17623041982dSJonathan Peyton static void __kmp_task_reduction_fini(kmp_info_t *th, kmp_taskgroup_t *tg) { 176372ba2109SAndrey Churbanov kmp_int32 nth = th->th.th_team_nproc; 176472ba2109SAndrey Churbanov KMP_DEBUG_ASSERT(nth > 1); // should not be called if nth == 1 176572ba2109SAndrey Churbanov kmp_task_red_data_t *arr = (kmp_task_red_data_t *)tg->reduce_data; 176672ba2109SAndrey Churbanov kmp_int32 num = tg->reduce_num_data; 176772ba2109SAndrey Churbanov for (int i = 0; i < num; ++i) { 176872ba2109SAndrey Churbanov void *sh_data = arr[i].reduce_shar; 176972ba2109SAndrey Churbanov void (*f_fini)(void *) = (void (*)(void *))(arr[i].reduce_fini); 17703041982dSJonathan Peyton void (*f_comb)(void *, void *) = 17713041982dSJonathan Peyton (void (*)(void *, void *))(arr[i].reduce_comb); 177272ba2109SAndrey Churbanov if (!arr[i].flags.lazy_priv) { 177372ba2109SAndrey Churbanov void *pr_data = arr[i].reduce_priv; 177472ba2109SAndrey Churbanov size_t size = arr[i].reduce_size; 177572ba2109SAndrey Churbanov for (int j = 0; j < nth; ++j) { 177672ba2109SAndrey Churbanov void *priv_data = (char *)pr_data + j * size; 177772ba2109SAndrey Churbanov f_comb(sh_data, priv_data); // combine results 177872ba2109SAndrey Churbanov if (f_fini) 177972ba2109SAndrey Churbanov f_fini(priv_data); // finalize if needed 178072ba2109SAndrey Churbanov } 178172ba2109SAndrey Churbanov } else { 178272ba2109SAndrey Churbanov void **pr_data = (void **)(arr[i].reduce_priv); 178372ba2109SAndrey Churbanov for (int j = 0; j < nth; ++j) { 178472ba2109SAndrey Churbanov if (pr_data[j] != NULL) { 178572ba2109SAndrey Churbanov f_comb(sh_data, pr_data[j]); // combine results 178672ba2109SAndrey Churbanov if (f_fini) 178772ba2109SAndrey Churbanov f_fini(pr_data[j]); // finalize if needed 178872ba2109SAndrey Churbanov __kmp_free(pr_data[j]); 178972ba2109SAndrey Churbanov } 179072ba2109SAndrey Churbanov } 179172ba2109SAndrey Churbanov } 179272ba2109SAndrey Churbanov __kmp_free(arr[i].reduce_priv); 179372ba2109SAndrey Churbanov } 179472ba2109SAndrey Churbanov __kmp_thread_free(th, arr); 179572ba2109SAndrey Churbanov tg->reduce_data = NULL; 179672ba2109SAndrey Churbanov tg->reduce_num_data = 0; 179772ba2109SAndrey Churbanov } 179872ba2109SAndrey Churbanov #endif 17997cc577a4SJonathan Peyton 18007cc577a4SJonathan Peyton #if OMP_40_ENABLED 18017cc577a4SJonathan Peyton // __kmpc_taskgroup: Start a new taskgroup 18023041982dSJonathan Peyton void __kmpc_taskgroup(ident_t *loc, int gtid) { 18037cc577a4SJonathan Peyton kmp_info_t *thread = __kmp_threads[gtid]; 18047cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = thread->th.th_current_task; 18057cc577a4SJonathan Peyton kmp_taskgroup_t *tg_new = 18067cc577a4SJonathan Peyton (kmp_taskgroup_t *)__kmp_thread_malloc(thread, sizeof(kmp_taskgroup_t)); 18077cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_taskgroup: T#%d loc=%p group=%p\n", gtid, loc, tg_new)); 18087cc577a4SJonathan Peyton tg_new->count = 0; 18097cc577a4SJonathan Peyton tg_new->cancel_request = cancel_noreq; 18107cc577a4SJonathan Peyton tg_new->parent = taskdata->td_taskgroup; 181172ba2109SAndrey Churbanov // TODO: change to OMP_50_ENABLED, need to change build tools for this to work 181272ba2109SAndrey Churbanov #if OMP_45_ENABLED 181372ba2109SAndrey Churbanov tg_new->reduce_data = NULL; 181472ba2109SAndrey Churbanov tg_new->reduce_num_data = 0; 181572ba2109SAndrey Churbanov #endif 18167cc577a4SJonathan Peyton taskdata->td_taskgroup = tg_new; 18177cc577a4SJonathan Peyton } 18187cc577a4SJonathan Peyton 18197cc577a4SJonathan Peyton // __kmpc_end_taskgroup: Wait until all tasks generated by the current task 18207cc577a4SJonathan Peyton // and its descendants are complete 18213041982dSJonathan Peyton void __kmpc_end_taskgroup(ident_t *loc, int gtid) { 18227cc577a4SJonathan Peyton kmp_info_t *thread = __kmp_threads[gtid]; 18237cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = thread->th.th_current_task; 18247cc577a4SJonathan Peyton kmp_taskgroup_t *taskgroup = taskdata->td_taskgroup; 18257cc577a4SJonathan Peyton int thread_finished = FALSE; 18267cc577a4SJonathan Peyton 18277cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_end_taskgroup(enter): T#%d loc=%p\n", gtid, loc)); 18287cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskgroup != NULL); 18297cc577a4SJonathan Peyton KMP_SET_THREAD_STATE_BLOCK(TASKGROUP); 18307cc577a4SJonathan Peyton 18317cc577a4SJonathan Peyton if (__kmp_tasking_mode != tskm_immediate_exec) { 18327cc577a4SJonathan Peyton #if USE_ITT_BUILD 18333041982dSJonathan Peyton // For ITT the taskgroup wait is similar to taskwait until we need to 18343041982dSJonathan Peyton // distinguish them 18357cc577a4SJonathan Peyton void *itt_sync_obj = __kmp_itt_taskwait_object(gtid); 18367cc577a4SJonathan Peyton if (itt_sync_obj != NULL) 18377cc577a4SJonathan Peyton __kmp_itt_taskwait_starting(gtid, itt_sync_obj); 18387cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 18397cc577a4SJonathan Peyton 18407cc577a4SJonathan Peyton #if OMP_45_ENABLED 18413041982dSJonathan Peyton if (!taskdata->td_flags.team_serial || 18423041982dSJonathan Peyton (thread->th.th_task_team != NULL && 18433041982dSJonathan Peyton thread->th.th_task_team->tt.tt_found_proxy_tasks)) 18447cc577a4SJonathan Peyton #else 18457cc577a4SJonathan Peyton if (!taskdata->td_flags.team_serial) 18467cc577a4SJonathan Peyton #endif 18477cc577a4SJonathan Peyton { 18487cc577a4SJonathan Peyton kmp_flag_32 flag(&(taskgroup->count), 0U); 18497cc577a4SJonathan Peyton while (TCR_4(taskgroup->count) != 0) { 18503041982dSJonathan Peyton flag.execute_tasks(thread, gtid, FALSE, 18513041982dSJonathan Peyton &thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), 18523041982dSJonathan Peyton __kmp_task_stealing_constraint); 18537cc577a4SJonathan Peyton } 18547cc577a4SJonathan Peyton } 18557cc577a4SJonathan Peyton 18567cc577a4SJonathan Peyton #if USE_ITT_BUILD 18577cc577a4SJonathan Peyton if (itt_sync_obj != NULL) 18587cc577a4SJonathan Peyton __kmp_itt_taskwait_finished(gtid, itt_sync_obj); 18597cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 18607cc577a4SJonathan Peyton } 18617cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskgroup->count == 0); 18627cc577a4SJonathan Peyton 186372ba2109SAndrey Churbanov // TODO: change to OMP_50_ENABLED, need to change build tools for this to work 186472ba2109SAndrey Churbanov #if OMP_45_ENABLED 186572ba2109SAndrey Churbanov if (taskgroup->reduce_data != NULL) // need to reduce? 186672ba2109SAndrey Churbanov __kmp_task_reduction_fini(thread, taskgroup); 186772ba2109SAndrey Churbanov #endif 18687cc577a4SJonathan Peyton // Restore parent taskgroup for the current task 18697cc577a4SJonathan Peyton taskdata->td_taskgroup = taskgroup->parent; 18707cc577a4SJonathan Peyton __kmp_thread_free(thread, taskgroup); 18717cc577a4SJonathan Peyton 18723041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_end_taskgroup(exit): T#%d task %p finished waiting\n", 18733041982dSJonathan Peyton gtid, taskdata)); 18747cc577a4SJonathan Peyton ANNOTATE_HAPPENS_AFTER(taskdata); 18757cc577a4SJonathan Peyton } 18767cc577a4SJonathan Peyton #endif 18777cc577a4SJonathan Peyton 18787cc577a4SJonathan Peyton // __kmp_remove_my_task: remove a task from my own deque 18793041982dSJonathan Peyton static kmp_task_t *__kmp_remove_my_task(kmp_info_t *thread, kmp_int32 gtid, 18803041982dSJonathan Peyton kmp_task_team_t *task_team, 18813041982dSJonathan Peyton kmp_int32 is_constrained) { 18827cc577a4SJonathan Peyton kmp_task_t *task; 18837cc577a4SJonathan Peyton kmp_taskdata_t *taskdata; 18847cc577a4SJonathan Peyton kmp_thread_data_t *thread_data; 18857cc577a4SJonathan Peyton kmp_uint32 tail; 18867cc577a4SJonathan Peyton 18877cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(__kmp_tasking_mode != tskm_immediate_exec); 18883041982dSJonathan Peyton KMP_DEBUG_ASSERT(task_team->tt.tt_threads_data != 18893041982dSJonathan Peyton NULL); // Caller should check this condition 18907cc577a4SJonathan Peyton 18917cc577a4SJonathan Peyton thread_data = &task_team->tt.tt_threads_data[__kmp_tid_from_gtid(gtid)]; 18927cc577a4SJonathan Peyton 18937cc577a4SJonathan Peyton KA_TRACE(10, ("__kmp_remove_my_task(enter): T#%d ntasks=%d head=%u tail=%u\n", 18943041982dSJonathan Peyton gtid, thread_data->td.td_deque_ntasks, 18953041982dSJonathan Peyton thread_data->td.td_deque_head, thread_data->td.td_deque_tail)); 18967cc577a4SJonathan Peyton 18977cc577a4SJonathan Peyton if (TCR_4(thread_data->td.td_deque_ntasks) == 0) { 18983041982dSJonathan Peyton KA_TRACE(10, 18993041982dSJonathan Peyton ("__kmp_remove_my_task(exit #1): T#%d No tasks to remove: " 19003041982dSJonathan Peyton "ntasks=%d head=%u tail=%u\n", 19013041982dSJonathan Peyton gtid, thread_data->td.td_deque_ntasks, 19023041982dSJonathan Peyton thread_data->td.td_deque_head, thread_data->td.td_deque_tail)); 19037cc577a4SJonathan Peyton return NULL; 19047cc577a4SJonathan Peyton } 19057cc577a4SJonathan Peyton 19067cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&thread_data->td.td_deque_lock); 19077cc577a4SJonathan Peyton 19087cc577a4SJonathan Peyton if (TCR_4(thread_data->td.td_deque_ntasks) == 0) { 19097cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&thread_data->td.td_deque_lock); 19103041982dSJonathan Peyton KA_TRACE(10, 19113041982dSJonathan Peyton ("__kmp_remove_my_task(exit #2): T#%d No tasks to remove: " 19123041982dSJonathan Peyton "ntasks=%d head=%u tail=%u\n", 19133041982dSJonathan Peyton gtid, thread_data->td.td_deque_ntasks, 19143041982dSJonathan Peyton thread_data->td.td_deque_head, thread_data->td.td_deque_tail)); 19157cc577a4SJonathan Peyton return NULL; 19167cc577a4SJonathan Peyton } 19177cc577a4SJonathan Peyton 19183041982dSJonathan Peyton tail = (thread_data->td.td_deque_tail - 1) & 19193041982dSJonathan Peyton TASK_DEQUE_MASK(thread_data->td); // Wrap index. 19207cc577a4SJonathan Peyton taskdata = thread_data->td.td_deque[tail]; 19217cc577a4SJonathan Peyton 19227cc577a4SJonathan Peyton if (is_constrained && (taskdata->td_flags.tiedness == TASK_TIED)) { 19237cc577a4SJonathan Peyton // we need to check if the candidate obeys task scheduling constraint: 19247cc577a4SJonathan Peyton // only child of current task can be scheduled 19257cc577a4SJonathan Peyton kmp_taskdata_t *current = thread->th.th_current_task; 19267cc577a4SJonathan Peyton kmp_int32 level = current->td_level; 19277cc577a4SJonathan Peyton kmp_taskdata_t *parent = taskdata->td_parent; 19287cc577a4SJonathan Peyton while (parent != current && parent->td_level > level) { 19293041982dSJonathan Peyton parent = parent->td_parent; // check generation up to the level of the 19303041982dSJonathan Peyton // current task 19317cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(parent != NULL); 19327cc577a4SJonathan Peyton } 19337cc577a4SJonathan Peyton if (parent != current) { 19343041982dSJonathan Peyton // If the tail task is not a child, then no other child can appear in the 19353041982dSJonathan Peyton // deque. 19367cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&thread_data->td.td_deque_lock); 19373041982dSJonathan Peyton KA_TRACE(10, 19383041982dSJonathan Peyton ("__kmp_remove_my_task(exit #2): T#%d No tasks to remove: " 19393041982dSJonathan Peyton "ntasks=%d head=%u tail=%u\n", 19403041982dSJonathan Peyton gtid, thread_data->td.td_deque_ntasks, 19413041982dSJonathan Peyton thread_data->td.td_deque_head, thread_data->td.td_deque_tail)); 19427cc577a4SJonathan Peyton return NULL; 19437cc577a4SJonathan Peyton } 19447cc577a4SJonathan Peyton } 19457cc577a4SJonathan Peyton 19467cc577a4SJonathan Peyton thread_data->td.td_deque_tail = tail; 19477cc577a4SJonathan Peyton TCW_4(thread_data->td.td_deque_ntasks, thread_data->td.td_deque_ntasks - 1); 19487cc577a4SJonathan Peyton 19497cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&thread_data->td.td_deque_lock); 19507cc577a4SJonathan Peyton 19513041982dSJonathan Peyton KA_TRACE(10, ("__kmp_remove_my_task(exit #2): T#%d task %p removed: " 19523041982dSJonathan Peyton "ntasks=%d head=%u tail=%u\n", 19533041982dSJonathan Peyton gtid, taskdata, thread_data->td.td_deque_ntasks, 19543041982dSJonathan Peyton thread_data->td.td_deque_head, thread_data->td.td_deque_tail)); 19557cc577a4SJonathan Peyton 19567cc577a4SJonathan Peyton task = KMP_TASKDATA_TO_TASK(taskdata); 19577cc577a4SJonathan Peyton return task; 19587cc577a4SJonathan Peyton } 19597cc577a4SJonathan Peyton 19607cc577a4SJonathan Peyton // __kmp_steal_task: remove a task from another thread's deque 19617cc577a4SJonathan Peyton // Assume that calling thread has already checked existence of 19627cc577a4SJonathan Peyton // task_team thread_data before calling this routine. 19637cc577a4SJonathan Peyton static kmp_task_t * 19647cc577a4SJonathan Peyton __kmp_steal_task(kmp_info_t *victim, kmp_int32 gtid, kmp_task_team_t *task_team, 19657cc577a4SJonathan Peyton volatile kmp_uint32 *unfinished_threads, int *thread_finished, 19667cc577a4SJonathan Peyton kmp_int32 is_constrained) 19677cc577a4SJonathan Peyton { 19687cc577a4SJonathan Peyton kmp_task_t *task; 19697cc577a4SJonathan Peyton kmp_taskdata_t *taskdata; 19707cc577a4SJonathan Peyton kmp_thread_data_t *victim_td, *threads_data; 19717cc577a4SJonathan Peyton kmp_int32 victim_tid; 19727cc577a4SJonathan Peyton 19737cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(__kmp_tasking_mode != tskm_immediate_exec); 19747cc577a4SJonathan Peyton 19757cc577a4SJonathan Peyton threads_data = task_team->tt.tt_threads_data; 19767cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(threads_data != NULL); // Caller should check this condition 19777cc577a4SJonathan Peyton 19787cc577a4SJonathan Peyton victim_tid = victim->th.th_info.ds.ds_tid; 19797cc577a4SJonathan Peyton victim_td = &threads_data[victim_tid]; 19807cc577a4SJonathan Peyton 19813041982dSJonathan Peyton KA_TRACE(10, ("__kmp_steal_task(enter): T#%d try to steal from T#%d: " 19823041982dSJonathan Peyton "task_team=%p ntasks=%d " 19837cc577a4SJonathan Peyton "head=%u tail=%u\n", 19843041982dSJonathan Peyton gtid, __kmp_gtid_from_thread(victim), task_team, 19853041982dSJonathan Peyton victim_td->td.td_deque_ntasks, victim_td->td.td_deque_head, 19863041982dSJonathan Peyton victim_td->td.td_deque_tail)); 19877cc577a4SJonathan Peyton 19883041982dSJonathan Peyton if ((TCR_4(victim_td->td.td_deque_ntasks) == 19893041982dSJonathan Peyton 0) || // Caller should not check this condition 19903041982dSJonathan Peyton (TCR_PTR(victim->th.th_task_team) != 19913041982dSJonathan Peyton task_team)) // GEH: why would this happen? 19927cc577a4SJonathan Peyton { 19933041982dSJonathan Peyton KA_TRACE(10, ("__kmp_steal_task(exit #1): T#%d could not steal from T#%d: " 19943041982dSJonathan Peyton "task_team=%p " 19957cc577a4SJonathan Peyton "ntasks=%d head=%u tail=%u\n", 19963041982dSJonathan Peyton gtid, __kmp_gtid_from_thread(victim), task_team, 19973041982dSJonathan Peyton victim_td->td.td_deque_ntasks, victim_td->td.td_deque_head, 19983041982dSJonathan Peyton victim_td->td.td_deque_tail)); 19997cc577a4SJonathan Peyton return NULL; 20007cc577a4SJonathan Peyton } 20017cc577a4SJonathan Peyton 20027cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&victim_td->td.td_deque_lock); 20037cc577a4SJonathan Peyton 20047cc577a4SJonathan Peyton // Check again after we acquire the lock 20057cc577a4SJonathan Peyton if ((TCR_4(victim_td->td.td_deque_ntasks) == 0) || 20063041982dSJonathan Peyton (TCR_PTR(victim->th.th_task_team) != 20073041982dSJonathan Peyton task_team)) // GEH: why would this happen? 20087cc577a4SJonathan Peyton { 20097cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&victim_td->td.td_deque_lock); 20103041982dSJonathan Peyton KA_TRACE(10, ("__kmp_steal_task(exit #2): T#%d could not steal from T#%d: " 20113041982dSJonathan Peyton "task_team=%p " 20127cc577a4SJonathan Peyton "ntasks=%d head=%u tail=%u\n", 20133041982dSJonathan Peyton gtid, __kmp_gtid_from_thread(victim), task_team, 20143041982dSJonathan Peyton victim_td->td.td_deque_ntasks, victim_td->td.td_deque_head, 20153041982dSJonathan Peyton victim_td->td.td_deque_tail)); 20167cc577a4SJonathan Peyton return NULL; 20177cc577a4SJonathan Peyton } 20187cc577a4SJonathan Peyton 20197cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(victim_td->td.td_deque != NULL); 20207cc577a4SJonathan Peyton 20217cc577a4SJonathan Peyton taskdata = victim_td->td.td_deque[victim_td->td.td_deque_head]; 20227cc577a4SJonathan Peyton if (is_constrained) { 20237cc577a4SJonathan Peyton // we need to check if the candidate obeys task scheduling constraint: 20247cc577a4SJonathan Peyton // only descendant of current task can be scheduled 20257cc577a4SJonathan Peyton kmp_taskdata_t *current = __kmp_threads[gtid]->th.th_current_task; 20267cc577a4SJonathan Peyton kmp_int32 level = current->td_level; 20277cc577a4SJonathan Peyton kmp_taskdata_t *parent = taskdata->td_parent; 20287cc577a4SJonathan Peyton while (parent != current && parent->td_level > level) { 20293041982dSJonathan Peyton parent = parent->td_parent; // check generation up to the level of the 20303041982dSJonathan Peyton // current task 20317cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(parent != NULL); 20327cc577a4SJonathan Peyton } 20337cc577a4SJonathan Peyton if (parent != current) { 20347cc577a4SJonathan Peyton // If the head task is not a descendant of the current task then do not 20357cc577a4SJonathan Peyton // steal it. No other task in victim's deque can be a descendant of the 20367cc577a4SJonathan Peyton // current task. 20377cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&victim_td->td.td_deque_lock); 20383041982dSJonathan Peyton KA_TRACE(10, ("__kmp_steal_task(exit #2): T#%d could not steal from " 20393041982dSJonathan Peyton "T#%d: task_team=%p " 20407cc577a4SJonathan Peyton "ntasks=%d head=%u tail=%u\n", 20413041982dSJonathan Peyton gtid, 20423041982dSJonathan Peyton __kmp_gtid_from_thread(threads_data[victim_tid].td.td_thr), 20437cc577a4SJonathan Peyton task_team, victim_td->td.td_deque_ntasks, 20447cc577a4SJonathan Peyton victim_td->td.td_deque_head, victim_td->td.td_deque_tail)); 20457cc577a4SJonathan Peyton return NULL; 20467cc577a4SJonathan Peyton } 20477cc577a4SJonathan Peyton } 20487cc577a4SJonathan Peyton // Bump head pointer and Wrap. 20493041982dSJonathan Peyton victim_td->td.td_deque_head = 20503041982dSJonathan Peyton (victim_td->td.td_deque_head + 1) & TASK_DEQUE_MASK(victim_td->td); 20517cc577a4SJonathan Peyton if (*thread_finished) { 20523041982dSJonathan Peyton // We need to un-mark this victim as a finished victim. This must be done 20533041982dSJonathan Peyton // before releasing the lock, or else other threads (starting with the 20543041982dSJonathan Peyton // master victim) might be prematurely released from the barrier!!! 20557cc577a4SJonathan Peyton kmp_uint32 count; 20567cc577a4SJonathan Peyton 20577cc577a4SJonathan Peyton count = KMP_TEST_THEN_INC32((kmp_int32 *)unfinished_threads); 20587cc577a4SJonathan Peyton 20593041982dSJonathan Peyton KA_TRACE( 20603041982dSJonathan Peyton 20, 20613041982dSJonathan Peyton ("__kmp_steal_task: T#%d inc unfinished_threads to %d: task_team=%p\n", 20627cc577a4SJonathan Peyton gtid, count + 1, task_team)); 20637cc577a4SJonathan Peyton 20647cc577a4SJonathan Peyton *thread_finished = FALSE; 20657cc577a4SJonathan Peyton } 20663041982dSJonathan Peyton TCW_4(victim_td->td.td_deque_ntasks, 20673041982dSJonathan Peyton TCR_4(victim_td->td.td_deque_ntasks) - 1); 20683041982dSJonathan Peyton 20697cc577a4SJonathan Peyton 20707cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&victim_td->td.td_deque_lock); 20717cc577a4SJonathan Peyton 20727cc577a4SJonathan Peyton KMP_COUNT_BLOCK(TASK_stolen); 20733041982dSJonathan Peyton KA_TRACE( 20743041982dSJonathan Peyton 10, 20753041982dSJonathan Peyton ("__kmp_steal_task(exit #3): T#%d stole task %p from T#%d: task_team=%p " 20767cc577a4SJonathan Peyton "ntasks=%d head=%u tail=%u\n", 20777cc577a4SJonathan Peyton gtid, taskdata, __kmp_gtid_from_thread(victim), task_team, 20787cc577a4SJonathan Peyton victim_td->td.td_deque_ntasks, victim_td->td.td_deque_head, 20797cc577a4SJonathan Peyton victim_td->td.td_deque_tail)); 20807cc577a4SJonathan Peyton 20817cc577a4SJonathan Peyton task = KMP_TASKDATA_TO_TASK(taskdata); 20827cc577a4SJonathan Peyton return task; 20837cc577a4SJonathan Peyton } 20847cc577a4SJonathan Peyton 20857cc577a4SJonathan Peyton 20863041982dSJonathan Peyton // __kmp_execute_tasks_template: Choose and execute tasks until either the 20873041982dSJonathan Peyton // condition is statisfied (return true) or there are none left (return false). 20883041982dSJonathan Peyton // 20897cc577a4SJonathan Peyton // final_spin is TRUE if this is the spin at the release barrier. 20907cc577a4SJonathan Peyton // thread_finished indicates whether the thread is finished executing all 20917cc577a4SJonathan Peyton // the tasks it has on its deque, and is at the release barrier. 20927cc577a4SJonathan Peyton // spinner is the location on which to spin. 20937cc577a4SJonathan Peyton // spinner == NULL means only execute a single task and return. 20947cc577a4SJonathan Peyton // checker is the value to check to terminate the spin. 20957cc577a4SJonathan Peyton template <class C> 20963041982dSJonathan Peyton static inline int __kmp_execute_tasks_template( 20973041982dSJonathan Peyton kmp_info_t *thread, kmp_int32 gtid, C *flag, int final_spin, 20983041982dSJonathan Peyton int *thread_finished USE_ITT_BUILD_ARG(void *itt_sync_obj), 20993041982dSJonathan Peyton kmp_int32 is_constrained) { 21007cc577a4SJonathan Peyton kmp_task_team_t *task_team = thread->th.th_task_team; 21017cc577a4SJonathan Peyton kmp_thread_data_t *threads_data; 21027cc577a4SJonathan Peyton kmp_task_t *task; 21037cc577a4SJonathan Peyton kmp_info_t *other_thread; 21047cc577a4SJonathan Peyton kmp_taskdata_t *current_task = thread->th.th_current_task; 21057cc577a4SJonathan Peyton volatile kmp_uint32 *unfinished_threads; 21063041982dSJonathan Peyton kmp_int32 nthreads, victim = -2, use_own_tasks = 1, new_victim = 0, 21073041982dSJonathan Peyton tid = thread->th.th_info.ds.ds_tid; 21087cc577a4SJonathan Peyton 21097cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(__kmp_tasking_mode != tskm_immediate_exec); 21107cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(thread == __kmp_threads[gtid]); 21117cc577a4SJonathan Peyton 21123041982dSJonathan Peyton if (task_team == NULL) 21133041982dSJonathan Peyton return FALSE; 21147cc577a4SJonathan Peyton 21153041982dSJonathan Peyton KA_TRACE(15, ("__kmp_execute_tasks_template(enter): T#%d final_spin=%d " 21163041982dSJonathan Peyton "*thread_finished=%d\n", 21177cc577a4SJonathan Peyton gtid, final_spin, *thread_finished)); 21187cc577a4SJonathan Peyton 2119581490e7SAndrey Churbanov thread->th.th_reap_state = KMP_NOT_SAFE_TO_REAP; 21207cc577a4SJonathan Peyton threads_data = (kmp_thread_data_t *)TCR_PTR(task_team->tt.tt_threads_data); 21217cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(threads_data != NULL); 21227cc577a4SJonathan Peyton 21237cc577a4SJonathan Peyton nthreads = task_team->tt.tt_nproc; 21247cc577a4SJonathan Peyton unfinished_threads = &(task_team->tt.tt_unfinished_threads); 21257cc577a4SJonathan Peyton #if OMP_45_ENABLED 21267cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(nthreads > 1 || task_team->tt.tt_found_proxy_tasks); 21277cc577a4SJonathan Peyton #else 21287cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(nthreads > 1); 21297cc577a4SJonathan Peyton #endif 21307cc577a4SJonathan Peyton KMP_DEBUG_ASSERT((int)(TCR_4(*unfinished_threads)) >= 0); 21317cc577a4SJonathan Peyton 21323041982dSJonathan Peyton while (1) { // Outer loop keeps trying to find tasks in case of single thread 21333041982dSJonathan Peyton // getting tasks from target constructs 21347cc577a4SJonathan Peyton while (1) { // Inner loop to find a task and execute it 21357cc577a4SJonathan Peyton task = NULL; 21367cc577a4SJonathan Peyton if (use_own_tasks) { // check on own queue first 21377cc577a4SJonathan Peyton task = __kmp_remove_my_task(thread, gtid, task_team, is_constrained); 21387cc577a4SJonathan Peyton } 21397cc577a4SJonathan Peyton if ((task == NULL) && (nthreads > 1)) { // Steal a task 21407cc577a4SJonathan Peyton int asleep = 1; 21417cc577a4SJonathan Peyton use_own_tasks = 0; 21427cc577a4SJonathan Peyton // Try to steal from the last place I stole from successfully. 21437cc577a4SJonathan Peyton if (victim == -2) { // haven't stolen anything yet 21447cc577a4SJonathan Peyton victim = threads_data[tid].td.td_deque_last_stolen; 21453041982dSJonathan Peyton if (victim != 21463041982dSJonathan Peyton -1) // if we have a last stolen from victim, get the thread 21477cc577a4SJonathan Peyton other_thread = threads_data[victim].td.td_thr; 21487cc577a4SJonathan Peyton } 21497cc577a4SJonathan Peyton if (victim != -1) { // found last victim 21507cc577a4SJonathan Peyton asleep = 0; 21513041982dSJonathan Peyton } else if (!new_victim) { // no recent steals and we haven't already 21523041982dSJonathan Peyton // used a new victim; select a random thread 21537cc577a4SJonathan Peyton do { // Find a different thread to steal work from. 21543041982dSJonathan Peyton // Pick a random thread. Initial plan was to cycle through all the 21553041982dSJonathan Peyton // threads, and only return if we tried to steal from every thread, 21563041982dSJonathan Peyton // and failed. Arch says that's not such a great idea. 21577cc577a4SJonathan Peyton victim = __kmp_get_random(thread) % (nthreads - 1); 21587cc577a4SJonathan Peyton if (victim >= tid) { 21597cc577a4SJonathan Peyton ++victim; // Adjusts random distribution to exclude self 21607cc577a4SJonathan Peyton } 21617cc577a4SJonathan Peyton // Found a potential victim 21627cc577a4SJonathan Peyton other_thread = threads_data[victim].td.td_thr; 21633041982dSJonathan Peyton // There is a slight chance that __kmp_enable_tasking() did not wake 21643041982dSJonathan Peyton // up all threads waiting at the barrier. If victim is sleeping, 21653041982dSJonathan Peyton // then wake it up. Since we were going to pay the cache miss 21663041982dSJonathan Peyton // penalty for referencing another thread's kmp_info_t struct 21673041982dSJonathan Peyton // anyway, 21683041982dSJonathan Peyton // the check shouldn't cost too much performance at this point. In 21693041982dSJonathan Peyton // extra barrier mode, tasks do not sleep at the separate tasking 21703041982dSJonathan Peyton // barrier, so this isn't a problem. 21717cc577a4SJonathan Peyton asleep = 0; 21727cc577a4SJonathan Peyton if ((__kmp_tasking_mode == tskm_task_teams) && 21737cc577a4SJonathan Peyton (__kmp_dflt_blocktime != KMP_MAX_BLOCKTIME) && 21747cc577a4SJonathan Peyton (TCR_PTR(other_thread->th.th_sleep_loc) != NULL)) { 21757cc577a4SJonathan Peyton asleep = 1; 21763041982dSJonathan Peyton __kmp_null_resume_wrapper(__kmp_gtid_from_thread(other_thread), 21773041982dSJonathan Peyton other_thread->th.th_sleep_loc); 21783041982dSJonathan Peyton // A sleeping thread should not have any tasks on it's queue. 21793041982dSJonathan Peyton // There is a slight possibility that it resumes, steals a task 21803041982dSJonathan Peyton // from another thread, which spawns more tasks, all in the time 21813041982dSJonathan Peyton // that it takes this thread to check => don't write an assertion 21823041982dSJonathan Peyton // that the victim's queue is empty. Try stealing from a 21833041982dSJonathan Peyton // different thread. 21847cc577a4SJonathan Peyton } 21857cc577a4SJonathan Peyton } while (asleep); 21867cc577a4SJonathan Peyton } 21877cc577a4SJonathan Peyton 21887cc577a4SJonathan Peyton if (!asleep) { 21897cc577a4SJonathan Peyton // We have a victim to try to steal from 21903041982dSJonathan Peyton task = __kmp_steal_task(other_thread, gtid, task_team, 21913041982dSJonathan Peyton unfinished_threads, thread_finished, 21923041982dSJonathan Peyton is_constrained); 21937cc577a4SJonathan Peyton } 21947cc577a4SJonathan Peyton if (task != NULL) { // set last stolen to victim 21957cc577a4SJonathan Peyton if (threads_data[tid].td.td_deque_last_stolen != victim) { 21967cc577a4SJonathan Peyton threads_data[tid].td.td_deque_last_stolen = victim; 21973041982dSJonathan Peyton // The pre-refactored code did not try more than 1 successful new 21983041982dSJonathan Peyton // vicitm, unless the last one generated more local tasks; 21993041982dSJonathan Peyton // new_victim keeps track of this 22007cc577a4SJonathan Peyton new_victim = 1; 22017cc577a4SJonathan Peyton } 22023041982dSJonathan Peyton } else { // No tasks found; unset last_stolen 22037cc577a4SJonathan Peyton KMP_CHECK_UPDATE(threads_data[tid].td.td_deque_last_stolen, -1); 22047cc577a4SJonathan Peyton victim = -2; // no successful victim found 22057cc577a4SJonathan Peyton } 22067cc577a4SJonathan Peyton } 22077cc577a4SJonathan Peyton 22087cc577a4SJonathan Peyton if (task == NULL) // break out of tasking loop 22097cc577a4SJonathan Peyton break; 22107cc577a4SJonathan Peyton 22117cc577a4SJonathan Peyton // Found a task; execute it 22127cc577a4SJonathan Peyton #if USE_ITT_BUILD && USE_ITT_NOTIFY 22137cc577a4SJonathan Peyton if (__itt_sync_create_ptr || KMP_ITT_DEBUG) { 22143041982dSJonathan Peyton if (itt_sync_obj == NULL) { // we are at fork barrier where we could not 22153041982dSJonathan Peyton // get the object reliably 22167cc577a4SJonathan Peyton itt_sync_obj = __kmp_itt_barrier_object(gtid, bs_forkjoin_barrier); 22177cc577a4SJonathan Peyton } 22187cc577a4SJonathan Peyton __kmp_itt_task_starting(itt_sync_obj); 22197cc577a4SJonathan Peyton } 22207cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD && USE_ITT_NOTIFY */ 22217cc577a4SJonathan Peyton __kmp_invoke_task(gtid, task, current_task); 22227cc577a4SJonathan Peyton #if USE_ITT_BUILD 22233041982dSJonathan Peyton if (itt_sync_obj != NULL) 22243041982dSJonathan Peyton __kmp_itt_task_finished(itt_sync_obj); 22257cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 22263041982dSJonathan Peyton // If this thread is only partway through the barrier and the condition is 22273041982dSJonathan Peyton // met, then return now, so that the barrier gather/release pattern can 22283041982dSJonathan Peyton // proceed. If this thread is in the last spin loop in the barrier, 22293041982dSJonathan Peyton // waiting to be released, we know that the termination condition will not 22303041982dSJonathan Peyton // be satisified, so don't waste any cycles checking it. 22317cc577a4SJonathan Peyton if (flag == NULL || (!final_spin && flag->done_check())) { 22323041982dSJonathan Peyton KA_TRACE( 22333041982dSJonathan Peyton 15, 22343041982dSJonathan Peyton ("__kmp_execute_tasks_template: T#%d spin condition satisfied\n", 22353041982dSJonathan Peyton gtid)); 22367cc577a4SJonathan Peyton return TRUE; 22377cc577a4SJonathan Peyton } 22387cc577a4SJonathan Peyton if (thread->th.th_task_team == NULL) { 22397cc577a4SJonathan Peyton break; 22407cc577a4SJonathan Peyton } 22413041982dSJonathan Peyton // Yield before executing next task 22423041982dSJonathan Peyton KMP_YIELD(__kmp_library == library_throughput); 22433041982dSJonathan Peyton // If execution of a stolen task results in more tasks being placed on our 22443041982dSJonathan Peyton // run queue, reset use_own_tasks 22457cc577a4SJonathan Peyton if (!use_own_tasks && TCR_4(threads_data[tid].td.td_deque_ntasks) != 0) { 22463041982dSJonathan Peyton KA_TRACE(20, ("__kmp_execute_tasks_template: T#%d stolen task spawned " 22473041982dSJonathan Peyton "other tasks, restart\n", 22483041982dSJonathan Peyton gtid)); 22497cc577a4SJonathan Peyton use_own_tasks = 1; 22507cc577a4SJonathan Peyton new_victim = 0; 22517cc577a4SJonathan Peyton } 22527cc577a4SJonathan Peyton } 22537cc577a4SJonathan Peyton 22543041982dSJonathan Peyton // The task source has been exhausted. If in final spin loop of barrier, check 22553041982dSJonathan Peyton // if termination condition is satisfied. 22567cc577a4SJonathan Peyton #if OMP_45_ENABLED 22573041982dSJonathan Peyton // The work queue may be empty but there might be proxy tasks still 22583041982dSJonathan Peyton // executing 22597cc577a4SJonathan Peyton if (final_spin && TCR_4(current_task->td_incomplete_child_tasks) == 0) 22607cc577a4SJonathan Peyton #else 22617cc577a4SJonathan Peyton if (final_spin) 22627cc577a4SJonathan Peyton #endif 22637cc577a4SJonathan Peyton { 22643041982dSJonathan Peyton // First, decrement the #unfinished threads, if that has not already been 22653041982dSJonathan Peyton // done. This decrement might be to the spin location, and result in the 22663041982dSJonathan Peyton // termination condition being satisfied. 22677cc577a4SJonathan Peyton if (!*thread_finished) { 22687cc577a4SJonathan Peyton kmp_uint32 count; 22697cc577a4SJonathan Peyton 22707cc577a4SJonathan Peyton count = KMP_TEST_THEN_DEC32((kmp_int32 *)unfinished_threads) - 1; 22713041982dSJonathan Peyton KA_TRACE(20, ("__kmp_execute_tasks_template: T#%d dec " 22723041982dSJonathan Peyton "unfinished_threads to %d task_team=%p\n", 22737cc577a4SJonathan Peyton gtid, count, task_team)); 22747cc577a4SJonathan Peyton *thread_finished = TRUE; 22757cc577a4SJonathan Peyton } 22767cc577a4SJonathan Peyton 22777cc577a4SJonathan Peyton // It is now unsafe to reference thread->th.th_team !!! 22783041982dSJonathan Peyton // Decrementing task_team->tt.tt_unfinished_threads can allow the master 22793041982dSJonathan Peyton // thread to pass through the barrier, where it might reset each thread's 22803041982dSJonathan Peyton // th.th_team field for the next parallel region. If we can steal more 22813041982dSJonathan Peyton // work, we know that this has not happened yet. 22827cc577a4SJonathan Peyton if (flag != NULL && flag->done_check()) { 22833041982dSJonathan Peyton KA_TRACE( 22843041982dSJonathan Peyton 15, 22853041982dSJonathan Peyton ("__kmp_execute_tasks_template: T#%d spin condition satisfied\n", 22863041982dSJonathan Peyton gtid)); 22877cc577a4SJonathan Peyton return TRUE; 22887cc577a4SJonathan Peyton } 22897cc577a4SJonathan Peyton } 22907cc577a4SJonathan Peyton 22913041982dSJonathan Peyton // If this thread's task team is NULL, master has recognized that there are 22923041982dSJonathan Peyton // no more tasks; bail out 22937cc577a4SJonathan Peyton if (thread->th.th_task_team == NULL) { 22943041982dSJonathan Peyton KA_TRACE(15, 22953041982dSJonathan Peyton ("__kmp_execute_tasks_template: T#%d no more tasks\n", gtid)); 22967cc577a4SJonathan Peyton return FALSE; 22977cc577a4SJonathan Peyton } 22987cc577a4SJonathan Peyton 22997cc577a4SJonathan Peyton #if OMP_45_ENABLED 23003041982dSJonathan Peyton // We could be getting tasks from target constructs; if this is the only 23013041982dSJonathan Peyton // thread, keep trying to execute tasks from own queue 23027cc577a4SJonathan Peyton if (nthreads == 1) 23037cc577a4SJonathan Peyton use_own_tasks = 1; 23047cc577a4SJonathan Peyton else 23057cc577a4SJonathan Peyton #endif 23067cc577a4SJonathan Peyton { 23073041982dSJonathan Peyton KA_TRACE(15, 23083041982dSJonathan Peyton ("__kmp_execute_tasks_template: T#%d can't find work\n", gtid)); 23097cc577a4SJonathan Peyton return FALSE; 23107cc577a4SJonathan Peyton } 23117cc577a4SJonathan Peyton } 23127cc577a4SJonathan Peyton } 23137cc577a4SJonathan Peyton 23143041982dSJonathan Peyton int __kmp_execute_tasks_32( 23153041982dSJonathan Peyton kmp_info_t *thread, kmp_int32 gtid, kmp_flag_32 *flag, int final_spin, 23163041982dSJonathan Peyton int *thread_finished USE_ITT_BUILD_ARG(void *itt_sync_obj), 23173041982dSJonathan Peyton kmp_int32 is_constrained) { 23183041982dSJonathan Peyton return __kmp_execute_tasks_template( 23193041982dSJonathan Peyton thread, gtid, flag, final_spin, 23203041982dSJonathan Peyton thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained); 23217cc577a4SJonathan Peyton } 23227cc577a4SJonathan Peyton 23233041982dSJonathan Peyton int __kmp_execute_tasks_64( 23243041982dSJonathan Peyton kmp_info_t *thread, kmp_int32 gtid, kmp_flag_64 *flag, int final_spin, 23253041982dSJonathan Peyton int *thread_finished USE_ITT_BUILD_ARG(void *itt_sync_obj), 23263041982dSJonathan Peyton kmp_int32 is_constrained) { 23273041982dSJonathan Peyton return __kmp_execute_tasks_template( 23283041982dSJonathan Peyton thread, gtid, flag, final_spin, 23293041982dSJonathan Peyton thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained); 23307cc577a4SJonathan Peyton } 23317cc577a4SJonathan Peyton 23323041982dSJonathan Peyton int __kmp_execute_tasks_oncore( 23333041982dSJonathan Peyton kmp_info_t *thread, kmp_int32 gtid, kmp_flag_oncore *flag, int final_spin, 23343041982dSJonathan Peyton int *thread_finished USE_ITT_BUILD_ARG(void *itt_sync_obj), 23353041982dSJonathan Peyton kmp_int32 is_constrained) { 23363041982dSJonathan Peyton return __kmp_execute_tasks_template( 23373041982dSJonathan Peyton thread, gtid, flag, final_spin, 23383041982dSJonathan Peyton thread_finished USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained); 23397cc577a4SJonathan Peyton } 23407cc577a4SJonathan Peyton 23417cc577a4SJonathan Peyton // __kmp_enable_tasking: Allocate task team and resume threads sleeping at the 23427cc577a4SJonathan Peyton // next barrier so they can assist in executing enqueued tasks. 23437cc577a4SJonathan Peyton // First thread in allocates the task team atomically. 23443041982dSJonathan Peyton static void __kmp_enable_tasking(kmp_task_team_t *task_team, 23453041982dSJonathan Peyton kmp_info_t *this_thr) { 23467cc577a4SJonathan Peyton kmp_thread_data_t *threads_data; 23477cc577a4SJonathan Peyton int nthreads, i, is_init_thread; 23487cc577a4SJonathan Peyton 23497cc577a4SJonathan Peyton KA_TRACE(10, ("__kmp_enable_tasking(enter): T#%d\n", 23507cc577a4SJonathan Peyton __kmp_gtid_from_thread(this_thr))); 23517cc577a4SJonathan Peyton 23527cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_team != NULL); 23537cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(this_thr->th.th_team != NULL); 23547cc577a4SJonathan Peyton 23557cc577a4SJonathan Peyton nthreads = task_team->tt.tt_nproc; 23567cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(nthreads > 0); 23577cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(nthreads == this_thr->th.th_team->t.t_nproc); 23587cc577a4SJonathan Peyton 23597cc577a4SJonathan Peyton // Allocate or increase the size of threads_data if necessary 23607cc577a4SJonathan Peyton is_init_thread = __kmp_realloc_task_threads_data(this_thr, task_team); 23617cc577a4SJonathan Peyton 23627cc577a4SJonathan Peyton if (!is_init_thread) { 23637cc577a4SJonathan Peyton // Some other thread already set up the array. 23643041982dSJonathan Peyton KA_TRACE( 23653041982dSJonathan Peyton 20, 23663041982dSJonathan Peyton ("__kmp_enable_tasking(exit): T#%d: threads array already set up.\n", 23677cc577a4SJonathan Peyton __kmp_gtid_from_thread(this_thr))); 23687cc577a4SJonathan Peyton return; 23697cc577a4SJonathan Peyton } 23707cc577a4SJonathan Peyton threads_data = (kmp_thread_data_t *)TCR_PTR(task_team->tt.tt_threads_data); 23717cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(threads_data != NULL); 23727cc577a4SJonathan Peyton 23737cc577a4SJonathan Peyton if ((__kmp_tasking_mode == tskm_task_teams) && 23743041982dSJonathan Peyton (__kmp_dflt_blocktime != KMP_MAX_BLOCKTIME)) { 23757cc577a4SJonathan Peyton // Release any threads sleeping at the barrier, so that they can steal 23767cc577a4SJonathan Peyton // tasks and execute them. In extra barrier mode, tasks do not sleep 23777cc577a4SJonathan Peyton // at the separate tasking barrier, so this isn't a problem. 23787cc577a4SJonathan Peyton for (i = 0; i < nthreads; i++) { 23797cc577a4SJonathan Peyton volatile void *sleep_loc; 23807cc577a4SJonathan Peyton kmp_info_t *thread = threads_data[i].td.td_thr; 23817cc577a4SJonathan Peyton 23827cc577a4SJonathan Peyton if (i == this_thr->th.th_info.ds.ds_tid) { 23837cc577a4SJonathan Peyton continue; 23847cc577a4SJonathan Peyton } 23857cc577a4SJonathan Peyton // Since we haven't locked the thread's suspend mutex lock at this 23867cc577a4SJonathan Peyton // point, there is a small window where a thread might be putting 23877cc577a4SJonathan Peyton // itself to sleep, but hasn't set the th_sleep_loc field yet. 23887cc577a4SJonathan Peyton // To work around this, __kmp_execute_tasks_template() periodically checks 23893041982dSJonathan Peyton // see if other threads are sleeping (using the same random mechanism that 23903041982dSJonathan Peyton // is used for task stealing) and awakens them if they are. 23913041982dSJonathan Peyton if ((sleep_loc = TCR_PTR(thread->th.th_sleep_loc)) != NULL) { 23927cc577a4SJonathan Peyton KF_TRACE(50, ("__kmp_enable_tasking: T#%d waking up thread T#%d\n", 23937cc577a4SJonathan Peyton __kmp_gtid_from_thread(this_thr), 23947cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread))); 23957cc577a4SJonathan Peyton __kmp_null_resume_wrapper(__kmp_gtid_from_thread(thread), sleep_loc); 23963041982dSJonathan Peyton } else { 23977cc577a4SJonathan Peyton KF_TRACE(50, ("__kmp_enable_tasking: T#%d don't wake up thread T#%d\n", 23987cc577a4SJonathan Peyton __kmp_gtid_from_thread(this_thr), 23997cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread))); 24007cc577a4SJonathan Peyton } 24017cc577a4SJonathan Peyton } 24027cc577a4SJonathan Peyton } 24037cc577a4SJonathan Peyton 24047cc577a4SJonathan Peyton KA_TRACE(10, ("__kmp_enable_tasking(exit): T#%d\n", 24057cc577a4SJonathan Peyton __kmp_gtid_from_thread(this_thr))); 24067cc577a4SJonathan Peyton } 24077cc577a4SJonathan Peyton 24087cc577a4SJonathan Peyton /* // TODO: Check the comment consistency 24097cc577a4SJonathan Peyton * Utility routines for "task teams". A task team (kmp_task_t) is kind of 24107cc577a4SJonathan Peyton * like a shadow of the kmp_team_t data struct, with a different lifetime. 24117cc577a4SJonathan Peyton * After a child * thread checks into a barrier and calls __kmp_release() from 24127cc577a4SJonathan Peyton * the particular variant of __kmp_<barrier_kind>_barrier_gather(), it can no 24137cc577a4SJonathan Peyton * longer assume that the kmp_team_t structure is intact (at any moment, the 24147cc577a4SJonathan Peyton * master thread may exit the barrier code and free the team data structure, 24157cc577a4SJonathan Peyton * and return the threads to the thread pool). 24167cc577a4SJonathan Peyton * 24177cc577a4SJonathan Peyton * This does not work with the the tasking code, as the thread is still 24187cc577a4SJonathan Peyton * expected to participate in the execution of any tasks that may have been 24197cc577a4SJonathan Peyton * spawned my a member of the team, and the thread still needs access to all 24207cc577a4SJonathan Peyton * to each thread in the team, so that it can steal work from it. 24217cc577a4SJonathan Peyton * 24227cc577a4SJonathan Peyton * Enter the existence of the kmp_task_team_t struct. It employs a reference 24237cc577a4SJonathan Peyton * counting mechanims, and is allocated by the master thread before calling 24247cc577a4SJonathan Peyton * __kmp_<barrier_kind>_release, and then is release by the last thread to 24257cc577a4SJonathan Peyton * exit __kmp_<barrier_kind>_release at the next barrier. I.e. the lifetimes 24267cc577a4SJonathan Peyton * of the kmp_task_team_t structs for consecutive barriers can overlap 24277cc577a4SJonathan Peyton * (and will, unless the master thread is the last thread to exit the barrier 24287cc577a4SJonathan Peyton * release phase, which is not typical). 24297cc577a4SJonathan Peyton * 24307cc577a4SJonathan Peyton * The existence of such a struct is useful outside the context of tasking, 24317cc577a4SJonathan Peyton * but for now, I'm trying to keep it specific to the OMP_30_ENABLED macro, 24327cc577a4SJonathan Peyton * so that any performance differences show up when comparing the 2.5 vs. 3.0 24337cc577a4SJonathan Peyton * libraries. 24347cc577a4SJonathan Peyton * 24357cc577a4SJonathan Peyton * We currently use the existence of the threads array as an indicator that 24367cc577a4SJonathan Peyton * tasks were spawned since the last barrier. If the structure is to be 24377cc577a4SJonathan Peyton * useful outside the context of tasking, then this will have to change, but 24387cc577a4SJonathan Peyton * not settting the field minimizes the performance impact of tasking on 24397cc577a4SJonathan Peyton * barriers, when no explicit tasks were spawned (pushed, actually). 24407cc577a4SJonathan Peyton */ 24417cc577a4SJonathan Peyton 24423041982dSJonathan Peyton static kmp_task_team_t *__kmp_free_task_teams = 24433041982dSJonathan Peyton NULL; // Free list for task_team data structures 24447cc577a4SJonathan Peyton // Lock for task team data structures 24453041982dSJonathan Peyton static kmp_bootstrap_lock_t __kmp_task_team_lock = 24463041982dSJonathan Peyton KMP_BOOTSTRAP_LOCK_INITIALIZER(__kmp_task_team_lock); 24477cc577a4SJonathan Peyton 24487cc577a4SJonathan Peyton // __kmp_alloc_task_deque: 24497cc577a4SJonathan Peyton // Allocates a task deque for a particular thread, and initialize the necessary 24507cc577a4SJonathan Peyton // data structures relating to the deque. This only happens once per thread 24513041982dSJonathan Peyton // per task team since task teams are recycled. No lock is needed during 24523041982dSJonathan Peyton // allocation since each thread allocates its own deque. 24533041982dSJonathan Peyton static void __kmp_alloc_task_deque(kmp_info_t *thread, 24543041982dSJonathan Peyton kmp_thread_data_t *thread_data) { 24557cc577a4SJonathan Peyton __kmp_init_bootstrap_lock(&thread_data->td.td_deque_lock); 24567cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(thread_data->td.td_deque == NULL); 24577cc577a4SJonathan Peyton 24587cc577a4SJonathan Peyton // Initialize last stolen task field to "none" 24597cc577a4SJonathan Peyton thread_data->td.td_deque_last_stolen = -1; 24607cc577a4SJonathan Peyton 24617cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(TCR_4(thread_data->td.td_deque_ntasks) == 0); 24627cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(thread_data->td.td_deque_head == 0); 24637cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(thread_data->td.td_deque_tail == 0); 24647cc577a4SJonathan Peyton 24653041982dSJonathan Peyton KE_TRACE( 24663041982dSJonathan Peyton 10, 24673041982dSJonathan Peyton ("__kmp_alloc_task_deque: T#%d allocating deque[%d] for thread_data %p\n", 24687cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread), INITIAL_TASK_DEQUE_SIZE, thread_data)); 24697cc577a4SJonathan Peyton // Allocate space for task deque, and zero the deque 24707cc577a4SJonathan Peyton // Cannot use __kmp_thread_calloc() because threads not around for 24717cc577a4SJonathan Peyton // kmp_reap_task_team( ). 24723041982dSJonathan Peyton thread_data->td.td_deque = (kmp_taskdata_t **)__kmp_allocate( 24733041982dSJonathan Peyton INITIAL_TASK_DEQUE_SIZE * sizeof(kmp_taskdata_t *)); 24747cc577a4SJonathan Peyton thread_data->td.td_deque_size = INITIAL_TASK_DEQUE_SIZE; 24757cc577a4SJonathan Peyton } 24767cc577a4SJonathan Peyton 24777cc577a4SJonathan Peyton // __kmp_realloc_task_deque: 24783041982dSJonathan Peyton // Re-allocates a task deque for a particular thread, copies the content from 24793041982dSJonathan Peyton // the old deque and adjusts the necessary data structures relating to the 24803041982dSJonathan Peyton // deque. This operation must be done with a the deque_lock being held 24813041982dSJonathan Peyton static void __kmp_realloc_task_deque(kmp_info_t *thread, 24823041982dSJonathan Peyton kmp_thread_data_t *thread_data) { 24837cc577a4SJonathan Peyton kmp_int32 size = TASK_DEQUE_SIZE(thread_data->td); 24847cc577a4SJonathan Peyton kmp_int32 new_size = 2 * size; 24857cc577a4SJonathan Peyton 24863041982dSJonathan Peyton KE_TRACE(10, ("__kmp_realloc_task_deque: T#%d reallocating deque[from %d to " 24873041982dSJonathan Peyton "%d] for thread_data %p\n", 24887cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread), size, new_size, thread_data)); 24897cc577a4SJonathan Peyton 24903041982dSJonathan Peyton kmp_taskdata_t **new_deque = 24913041982dSJonathan Peyton (kmp_taskdata_t **)__kmp_allocate(new_size * sizeof(kmp_taskdata_t *)); 24927cc577a4SJonathan Peyton 24937cc577a4SJonathan Peyton int i, j; 24943041982dSJonathan Peyton for (i = thread_data->td.td_deque_head, j = 0; j < size; 24953041982dSJonathan Peyton i = (i + 1) & TASK_DEQUE_MASK(thread_data->td), j++) 24967cc577a4SJonathan Peyton new_deque[j] = thread_data->td.td_deque[i]; 24977cc577a4SJonathan Peyton 24987cc577a4SJonathan Peyton __kmp_free(thread_data->td.td_deque); 24997cc577a4SJonathan Peyton 25007cc577a4SJonathan Peyton thread_data->td.td_deque_head = 0; 25017cc577a4SJonathan Peyton thread_data->td.td_deque_tail = size; 25027cc577a4SJonathan Peyton thread_data->td.td_deque = new_deque; 25037cc577a4SJonathan Peyton thread_data->td.td_deque_size = new_size; 25047cc577a4SJonathan Peyton } 25057cc577a4SJonathan Peyton 25067cc577a4SJonathan Peyton // __kmp_free_task_deque: 25073041982dSJonathan Peyton // Deallocates a task deque for a particular thread. Happens at library 25083041982dSJonathan Peyton // deallocation so don't need to reset all thread data fields. 25093041982dSJonathan Peyton static void __kmp_free_task_deque(kmp_thread_data_t *thread_data) { 25107cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&thread_data->td.td_deque_lock); 25117cc577a4SJonathan Peyton 25127cc577a4SJonathan Peyton if (thread_data->td.td_deque != NULL) { 25137cc577a4SJonathan Peyton TCW_4(thread_data->td.td_deque_ntasks, 0); 25147cc577a4SJonathan Peyton __kmp_free(thread_data->td.td_deque); 25157cc577a4SJonathan Peyton thread_data->td.td_deque = NULL; 25167cc577a4SJonathan Peyton } 25177cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&thread_data->td.td_deque_lock); 25187cc577a4SJonathan Peyton 25197cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK 25207cc577a4SJonathan Peyton // GEH: Figure out what to do here for td_susp_tied_tasks 25217cc577a4SJonathan Peyton if (thread_data->td.td_susp_tied_tasks.ts_entries != TASK_STACK_EMPTY) { 25227cc577a4SJonathan Peyton __kmp_free_task_stack(__kmp_thread_from_gtid(gtid), thread_data); 25237cc577a4SJonathan Peyton } 25247cc577a4SJonathan Peyton #endif // BUILD_TIED_TASK_STACK 25257cc577a4SJonathan Peyton } 25267cc577a4SJonathan Peyton 25277cc577a4SJonathan Peyton // __kmp_realloc_task_threads_data: 25283041982dSJonathan Peyton // Allocates a threads_data array for a task team, either by allocating an 25293041982dSJonathan Peyton // initial array or enlarging an existing array. Only the first thread to get 25303041982dSJonathan Peyton // the lock allocs or enlarges the array and re-initializes the array eleemnts. 25317cc577a4SJonathan Peyton // That thread returns "TRUE", the rest return "FALSE". 25327cc577a4SJonathan Peyton // Assumes that the new array size is given by task_team -> tt.tt_nproc. 25337cc577a4SJonathan Peyton // The current size is given by task_team -> tt.tt_max_threads. 25343041982dSJonathan Peyton static int __kmp_realloc_task_threads_data(kmp_info_t *thread, 25353041982dSJonathan Peyton kmp_task_team_t *task_team) { 25367cc577a4SJonathan Peyton kmp_thread_data_t **threads_data_p; 25377cc577a4SJonathan Peyton kmp_int32 nthreads, maxthreads; 25387cc577a4SJonathan Peyton int is_init_thread = FALSE; 25397cc577a4SJonathan Peyton 25407cc577a4SJonathan Peyton if (TCR_4(task_team->tt.tt_found_tasks)) { 25417cc577a4SJonathan Peyton // Already reallocated and initialized. 25427cc577a4SJonathan Peyton return FALSE; 25437cc577a4SJonathan Peyton } 25447cc577a4SJonathan Peyton 25457cc577a4SJonathan Peyton threads_data_p = &task_team->tt.tt_threads_data; 25467cc577a4SJonathan Peyton nthreads = task_team->tt.tt_nproc; 25477cc577a4SJonathan Peyton maxthreads = task_team->tt.tt_max_threads; 25487cc577a4SJonathan Peyton 25493041982dSJonathan Peyton // All threads must lock when they encounter the first task of the implicit 25503041982dSJonathan Peyton // task region to make sure threads_data fields are (re)initialized before 25513041982dSJonathan Peyton // used. 25527cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&task_team->tt.tt_threads_lock); 25537cc577a4SJonathan Peyton 25547cc577a4SJonathan Peyton if (!TCR_4(task_team->tt.tt_found_tasks)) { 25557cc577a4SJonathan Peyton // first thread to enable tasking 25567cc577a4SJonathan Peyton kmp_team_t *team = thread->th.th_team; 25577cc577a4SJonathan Peyton int i; 25587cc577a4SJonathan Peyton 25597cc577a4SJonathan Peyton is_init_thread = TRUE; 25607cc577a4SJonathan Peyton if (maxthreads < nthreads) { 25617cc577a4SJonathan Peyton 25627cc577a4SJonathan Peyton if (*threads_data_p != NULL) { 25637cc577a4SJonathan Peyton kmp_thread_data_t *old_data = *threads_data_p; 25647cc577a4SJonathan Peyton kmp_thread_data_t *new_data = NULL; 25657cc577a4SJonathan Peyton 25663041982dSJonathan Peyton KE_TRACE( 25673041982dSJonathan Peyton 10, 25683041982dSJonathan Peyton ("__kmp_realloc_task_threads_data: T#%d reallocating " 25697cc577a4SJonathan Peyton "threads data for task_team %p, new_size = %d, old_size = %d\n", 25703041982dSJonathan Peyton __kmp_gtid_from_thread(thread), task_team, nthreads, maxthreads)); 25717cc577a4SJonathan Peyton // Reallocate threads_data to have more elements than current array 25727cc577a4SJonathan Peyton // Cannot use __kmp_thread_realloc() because threads not around for 25737cc577a4SJonathan Peyton // kmp_reap_task_team( ). Note all new array entries are initialized 25747cc577a4SJonathan Peyton // to zero by __kmp_allocate(). 25753041982dSJonathan Peyton new_data = (kmp_thread_data_t *)__kmp_allocate( 25763041982dSJonathan Peyton nthreads * sizeof(kmp_thread_data_t)); 25777cc577a4SJonathan Peyton // copy old data to new data 25787cc577a4SJonathan Peyton KMP_MEMCPY_S((void *)new_data, nthreads * sizeof(kmp_thread_data_t), 25793041982dSJonathan Peyton (void *)old_data, maxthreads * sizeof(kmp_taskdata_t *)); 25807cc577a4SJonathan Peyton 25817cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK 25827cc577a4SJonathan Peyton // GEH: Figure out if this is the right thing to do 25837cc577a4SJonathan Peyton for (i = maxthreads; i < nthreads; i++) { 25847cc577a4SJonathan Peyton kmp_thread_data_t *thread_data = &(*threads_data_p)[i]; 25857cc577a4SJonathan Peyton __kmp_init_task_stack(__kmp_gtid_from_thread(thread), thread_data); 25867cc577a4SJonathan Peyton } 25877cc577a4SJonathan Peyton #endif // BUILD_TIED_TASK_STACK 25887cc577a4SJonathan Peyton // Install the new data and free the old data 25897cc577a4SJonathan Peyton (*threads_data_p) = new_data; 25907cc577a4SJonathan Peyton __kmp_free(old_data); 25913041982dSJonathan Peyton } else { 25927cc577a4SJonathan Peyton KE_TRACE(10, ("__kmp_realloc_task_threads_data: T#%d allocating " 25937cc577a4SJonathan Peyton "threads data for task_team %p, size = %d\n", 25947cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread), task_team, nthreads)); 25957cc577a4SJonathan Peyton // Make the initial allocate for threads_data array, and zero entries 25967cc577a4SJonathan Peyton // Cannot use __kmp_thread_calloc() because threads not around for 25977cc577a4SJonathan Peyton // kmp_reap_task_team( ). 25987cc577a4SJonathan Peyton ANNOTATE_IGNORE_WRITES_BEGIN(); 25993041982dSJonathan Peyton *threads_data_p = (kmp_thread_data_t *)__kmp_allocate( 26003041982dSJonathan Peyton nthreads * sizeof(kmp_thread_data_t)); 26017cc577a4SJonathan Peyton ANNOTATE_IGNORE_WRITES_END(); 26027cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK 26037cc577a4SJonathan Peyton // GEH: Figure out if this is the right thing to do 26047cc577a4SJonathan Peyton for (i = 0; i < nthreads; i++) { 26057cc577a4SJonathan Peyton kmp_thread_data_t *thread_data = &(*threads_data_p)[i]; 26067cc577a4SJonathan Peyton __kmp_init_task_stack(__kmp_gtid_from_thread(thread), thread_data); 26077cc577a4SJonathan Peyton } 26087cc577a4SJonathan Peyton #endif // BUILD_TIED_TASK_STACK 26097cc577a4SJonathan Peyton } 26107cc577a4SJonathan Peyton task_team->tt.tt_max_threads = nthreads; 26113041982dSJonathan Peyton } else { 26127cc577a4SJonathan Peyton // If array has (more than) enough elements, go ahead and use it 26137cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(*threads_data_p != NULL); 26147cc577a4SJonathan Peyton } 26157cc577a4SJonathan Peyton 26167cc577a4SJonathan Peyton // initialize threads_data pointers back to thread_info structures 26177cc577a4SJonathan Peyton for (i = 0; i < nthreads; i++) { 26187cc577a4SJonathan Peyton kmp_thread_data_t *thread_data = &(*threads_data_p)[i]; 26197cc577a4SJonathan Peyton thread_data->td.td_thr = team->t.t_threads[i]; 26207cc577a4SJonathan Peyton 26217cc577a4SJonathan Peyton if (thread_data->td.td_deque_last_stolen >= nthreads) { 26227cc577a4SJonathan Peyton // The last stolen field survives across teams / barrier, and the number 26237cc577a4SJonathan Peyton // of threads may have changed. It's possible (likely?) that a new 26243041982dSJonathan Peyton // parallel region will exhibit the same behavior as previous region. 26257cc577a4SJonathan Peyton thread_data->td.td_deque_last_stolen = -1; 26267cc577a4SJonathan Peyton } 26277cc577a4SJonathan Peyton } 26287cc577a4SJonathan Peyton 26297cc577a4SJonathan Peyton KMP_MB(); 26307cc577a4SJonathan Peyton TCW_SYNC_4(task_team->tt.tt_found_tasks, TRUE); 26317cc577a4SJonathan Peyton } 26327cc577a4SJonathan Peyton 26337cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&task_team->tt.tt_threads_lock); 26347cc577a4SJonathan Peyton return is_init_thread; 26357cc577a4SJonathan Peyton } 26367cc577a4SJonathan Peyton 26377cc577a4SJonathan Peyton // __kmp_free_task_threads_data: 26387cc577a4SJonathan Peyton // Deallocates a threads_data array for a task team, including any attached 26397cc577a4SJonathan Peyton // tasking deques. Only occurs at library shutdown. 26403041982dSJonathan Peyton static void __kmp_free_task_threads_data(kmp_task_team_t *task_team) { 26417cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&task_team->tt.tt_threads_lock); 26427cc577a4SJonathan Peyton if (task_team->tt.tt_threads_data != NULL) { 26437cc577a4SJonathan Peyton int i; 26447cc577a4SJonathan Peyton for (i = 0; i < task_team->tt.tt_max_threads; i++) { 26457cc577a4SJonathan Peyton __kmp_free_task_deque(&task_team->tt.tt_threads_data[i]); 26467cc577a4SJonathan Peyton } 26477cc577a4SJonathan Peyton __kmp_free(task_team->tt.tt_threads_data); 26487cc577a4SJonathan Peyton task_team->tt.tt_threads_data = NULL; 26497cc577a4SJonathan Peyton } 26507cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&task_team->tt.tt_threads_lock); 26517cc577a4SJonathan Peyton } 26527cc577a4SJonathan Peyton 26537cc577a4SJonathan Peyton // __kmp_allocate_task_team: 26547cc577a4SJonathan Peyton // Allocates a task team associated with a specific team, taking it from 26553041982dSJonathan Peyton // the global task team free list if possible. Also initializes data 26563041982dSJonathan Peyton // structures. 26573041982dSJonathan Peyton static kmp_task_team_t *__kmp_allocate_task_team(kmp_info_t *thread, 26583041982dSJonathan Peyton kmp_team_t *team) { 26597cc577a4SJonathan Peyton kmp_task_team_t *task_team = NULL; 26607cc577a4SJonathan Peyton int nthreads; 26617cc577a4SJonathan Peyton 26627cc577a4SJonathan Peyton KA_TRACE(20, ("__kmp_allocate_task_team: T#%d entering; team = %p\n", 26637cc577a4SJonathan Peyton (thread ? __kmp_gtid_from_thread(thread) : -1), team)); 26647cc577a4SJonathan Peyton 26657cc577a4SJonathan Peyton if (TCR_PTR(__kmp_free_task_teams) != NULL) { 26667cc577a4SJonathan Peyton // Take a task team from the task team pool 26677cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&__kmp_task_team_lock); 26687cc577a4SJonathan Peyton if (__kmp_free_task_teams != NULL) { 26697cc577a4SJonathan Peyton task_team = __kmp_free_task_teams; 26707cc577a4SJonathan Peyton TCW_PTR(__kmp_free_task_teams, task_team->tt.tt_next); 26717cc577a4SJonathan Peyton task_team->tt.tt_next = NULL; 26727cc577a4SJonathan Peyton } 26737cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&__kmp_task_team_lock); 26747cc577a4SJonathan Peyton } 26757cc577a4SJonathan Peyton 26767cc577a4SJonathan Peyton if (task_team == NULL) { 26777cc577a4SJonathan Peyton KE_TRACE(10, ("__kmp_allocate_task_team: T#%d allocating " 26787cc577a4SJonathan Peyton "task team for team %p\n", 26797cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread), team)); 26807cc577a4SJonathan Peyton // Allocate a new task team if one is not available. 26817cc577a4SJonathan Peyton // Cannot use __kmp_thread_malloc() because threads not around for 26827cc577a4SJonathan Peyton // kmp_reap_task_team( ). 26837cc577a4SJonathan Peyton task_team = (kmp_task_team_t *)__kmp_allocate(sizeof(kmp_task_team_t)); 26847cc577a4SJonathan Peyton __kmp_init_bootstrap_lock(&task_team->tt.tt_threads_lock); 26853041982dSJonathan Peyton // AC: __kmp_allocate zeroes returned memory 26863041982dSJonathan Peyton // task_team -> tt.tt_threads_data = NULL; 26877cc577a4SJonathan Peyton // task_team -> tt.tt_max_threads = 0; 26887cc577a4SJonathan Peyton // task_team -> tt.tt_next = NULL; 26897cc577a4SJonathan Peyton } 26907cc577a4SJonathan Peyton 26917cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_found_tasks, FALSE); 26927cc577a4SJonathan Peyton #if OMP_45_ENABLED 26937cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_found_proxy_tasks, FALSE); 26947cc577a4SJonathan Peyton #endif 26957cc577a4SJonathan Peyton task_team->tt.tt_nproc = nthreads = team->t.t_nproc; 26967cc577a4SJonathan Peyton 26977cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_unfinished_threads, nthreads); 26987cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_active, TRUE); 26997cc577a4SJonathan Peyton 27003041982dSJonathan Peyton KA_TRACE(20, ("__kmp_allocate_task_team: T#%d exiting; task_team = %p " 27013041982dSJonathan Peyton "unfinished_threads init'd to %d\n", 27023041982dSJonathan Peyton (thread ? __kmp_gtid_from_thread(thread) : -1), task_team, 27033041982dSJonathan Peyton task_team->tt.tt_unfinished_threads)); 27047cc577a4SJonathan Peyton return task_team; 27057cc577a4SJonathan Peyton } 27067cc577a4SJonathan Peyton 27077cc577a4SJonathan Peyton // __kmp_free_task_team: 27087cc577a4SJonathan Peyton // Frees the task team associated with a specific thread, and adds it 27097cc577a4SJonathan Peyton // to the global task team free list. 27103041982dSJonathan Peyton void __kmp_free_task_team(kmp_info_t *thread, kmp_task_team_t *task_team) { 27117cc577a4SJonathan Peyton KA_TRACE(20, ("__kmp_free_task_team: T#%d task_team = %p\n", 27127cc577a4SJonathan Peyton thread ? __kmp_gtid_from_thread(thread) : -1, task_team)); 27137cc577a4SJonathan Peyton 27147cc577a4SJonathan Peyton // Put task team back on free list 27157cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&__kmp_task_team_lock); 27167cc577a4SJonathan Peyton 27177cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_team->tt.tt_next == NULL); 27187cc577a4SJonathan Peyton task_team->tt.tt_next = __kmp_free_task_teams; 27197cc577a4SJonathan Peyton TCW_PTR(__kmp_free_task_teams, task_team); 27207cc577a4SJonathan Peyton 27217cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&__kmp_task_team_lock); 27227cc577a4SJonathan Peyton } 27237cc577a4SJonathan Peyton 27247cc577a4SJonathan Peyton // __kmp_reap_task_teams: 27257cc577a4SJonathan Peyton // Free all the task teams on the task team free list. 27267cc577a4SJonathan Peyton // Should only be done during library shutdown. 27273041982dSJonathan Peyton // Cannot do anything that needs a thread structure or gtid since they are 27283041982dSJonathan Peyton // already gone. 27293041982dSJonathan Peyton void __kmp_reap_task_teams(void) { 27307cc577a4SJonathan Peyton kmp_task_team_t *task_team; 27317cc577a4SJonathan Peyton 27327cc577a4SJonathan Peyton if (TCR_PTR(__kmp_free_task_teams) != NULL) { 27337cc577a4SJonathan Peyton // Free all task_teams on the free list 27347cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&__kmp_task_team_lock); 27357cc577a4SJonathan Peyton while ((task_team = __kmp_free_task_teams) != NULL) { 27367cc577a4SJonathan Peyton __kmp_free_task_teams = task_team->tt.tt_next; 27377cc577a4SJonathan Peyton task_team->tt.tt_next = NULL; 27387cc577a4SJonathan Peyton 27397cc577a4SJonathan Peyton // Free threads_data if necessary 27407cc577a4SJonathan Peyton if (task_team->tt.tt_threads_data != NULL) { 27417cc577a4SJonathan Peyton __kmp_free_task_threads_data(task_team); 27427cc577a4SJonathan Peyton } 27437cc577a4SJonathan Peyton __kmp_free(task_team); 27447cc577a4SJonathan Peyton } 27457cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&__kmp_task_team_lock); 27467cc577a4SJonathan Peyton } 27477cc577a4SJonathan Peyton } 27487cc577a4SJonathan Peyton 27497cc577a4SJonathan Peyton // __kmp_wait_to_unref_task_teams: 27507cc577a4SJonathan Peyton // Some threads could still be in the fork barrier release code, possibly 27517cc577a4SJonathan Peyton // trying to steal tasks. Wait for each thread to unreference its task team. 27523041982dSJonathan Peyton void __kmp_wait_to_unref_task_teams(void) { 27537cc577a4SJonathan Peyton kmp_info_t *thread; 27547cc577a4SJonathan Peyton kmp_uint32 spins; 27557cc577a4SJonathan Peyton int done; 27567cc577a4SJonathan Peyton 27577cc577a4SJonathan Peyton KMP_INIT_YIELD(spins); 27587cc577a4SJonathan Peyton 27597cc577a4SJonathan Peyton for (;;) { 27607cc577a4SJonathan Peyton done = TRUE; 27617cc577a4SJonathan Peyton 27627cc577a4SJonathan Peyton // TODO: GEH - this may be is wrong because some sync would be necessary 27633041982dSJonathan Peyton // in case threads are added to the pool during the traversal. Need to 27643041982dSJonathan Peyton // verify that lock for thread pool is held when calling this routine. 27653041982dSJonathan Peyton for (thread = (kmp_info_t *)__kmp_thread_pool; thread != NULL; 27663041982dSJonathan Peyton thread = thread->th.th_next_pool) { 27677cc577a4SJonathan Peyton #if KMP_OS_WINDOWS 27687cc577a4SJonathan Peyton DWORD exit_val; 27697cc577a4SJonathan Peyton #endif 27707cc577a4SJonathan Peyton if (TCR_PTR(thread->th.th_task_team) == NULL) { 27717cc577a4SJonathan Peyton KA_TRACE(10, ("__kmp_wait_to_unref_task_team: T#%d task_team == NULL\n", 27727cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread))); 27737cc577a4SJonathan Peyton continue; 27747cc577a4SJonathan Peyton } 27757cc577a4SJonathan Peyton #if KMP_OS_WINDOWS 27767cc577a4SJonathan Peyton // TODO: GEH - add this check for Linux* OS / OS X* as well? 27777cc577a4SJonathan Peyton if (!__kmp_is_thread_alive(thread, &exit_val)) { 27787cc577a4SJonathan Peyton thread->th.th_task_team = NULL; 27797cc577a4SJonathan Peyton continue; 27807cc577a4SJonathan Peyton } 27817cc577a4SJonathan Peyton #endif 27827cc577a4SJonathan Peyton 27837cc577a4SJonathan Peyton done = FALSE; // Because th_task_team pointer is not NULL for this thread 27847cc577a4SJonathan Peyton 27853041982dSJonathan Peyton KA_TRACE(10, ("__kmp_wait_to_unref_task_team: Waiting for T#%d to " 27863041982dSJonathan Peyton "unreference task_team\n", 27877cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread))); 27887cc577a4SJonathan Peyton 27897cc577a4SJonathan Peyton if (__kmp_dflt_blocktime != KMP_MAX_BLOCKTIME) { 27907cc577a4SJonathan Peyton volatile void *sleep_loc; 27917cc577a4SJonathan Peyton // If the thread is sleeping, awaken it. 27927cc577a4SJonathan Peyton if ((sleep_loc = TCR_PTR(thread->th.th_sleep_loc)) != NULL) { 27933041982dSJonathan Peyton KA_TRACE( 27943041982dSJonathan Peyton 10, 27953041982dSJonathan Peyton ("__kmp_wait_to_unref_task_team: T#%d waking up thread T#%d\n", 27967cc577a4SJonathan Peyton __kmp_gtid_from_thread(thread), __kmp_gtid_from_thread(thread))); 27977cc577a4SJonathan Peyton __kmp_null_resume_wrapper(__kmp_gtid_from_thread(thread), sleep_loc); 27987cc577a4SJonathan Peyton } 27997cc577a4SJonathan Peyton } 28007cc577a4SJonathan Peyton } 28017cc577a4SJonathan Peyton if (done) { 28027cc577a4SJonathan Peyton break; 28037cc577a4SJonathan Peyton } 28047cc577a4SJonathan Peyton 28053041982dSJonathan Peyton // If we are oversubscribed, or have waited a bit (and library mode is 28063041982dSJonathan Peyton // throughput), yield. Pause is in the following code. 28077cc577a4SJonathan Peyton KMP_YIELD(TCR_4(__kmp_nth) > __kmp_avail_proc); 28087cc577a4SJonathan Peyton KMP_YIELD_SPIN(spins); // Yields only if KMP_LIBRARY=throughput 28097cc577a4SJonathan Peyton } 28107cc577a4SJonathan Peyton } 28117cc577a4SJonathan Peyton 28127cc577a4SJonathan Peyton // __kmp_task_team_setup: Create a task_team for the current team, but use 28137cc577a4SJonathan Peyton // an already created, unused one if it already exists. 28143041982dSJonathan Peyton void __kmp_task_team_setup(kmp_info_t *this_thr, kmp_team_t *team, int always) { 28157cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(__kmp_tasking_mode != tskm_immediate_exec); 28167cc577a4SJonathan Peyton 28173041982dSJonathan Peyton // If this task_team hasn't been created yet, allocate it. It will be used in 28183041982dSJonathan Peyton // the region after the next. 28193041982dSJonathan Peyton // If it exists, it is the current task team and shouldn't be touched yet as 28203041982dSJonathan Peyton // it may still be in use. 28213041982dSJonathan Peyton if (team->t.t_task_team[this_thr->th.th_task_state] == NULL && 28223041982dSJonathan Peyton (always || team->t.t_nproc > 1)) { 28233041982dSJonathan Peyton team->t.t_task_team[this_thr->th.th_task_state] = 28243041982dSJonathan Peyton __kmp_allocate_task_team(this_thr, team); 28253041982dSJonathan Peyton KA_TRACE(20, ("__kmp_task_team_setup: Master T#%d created new task_team %p " 28263041982dSJonathan Peyton "for team %d at parity=%d\n", 28273041982dSJonathan Peyton __kmp_gtid_from_thread(this_thr), 28283041982dSJonathan Peyton team->t.t_task_team[this_thr->th.th_task_state], 28293041982dSJonathan Peyton ((team != NULL) ? team->t.t_id : -1), 28303041982dSJonathan Peyton this_thr->th.th_task_state)); 28317cc577a4SJonathan Peyton } 28327cc577a4SJonathan Peyton 28333041982dSJonathan Peyton // After threads exit the release, they will call sync, and then point to this 28343041982dSJonathan Peyton // other task_team; make sure it is allocated and properly initialized. As 28353041982dSJonathan Peyton // threads spin in the barrier release phase, they will continue to use the 28363041982dSJonathan Peyton // previous task_team struct(above), until they receive the signal to stop 28373041982dSJonathan Peyton // checking for tasks (they can't safely reference the kmp_team_t struct, 28383041982dSJonathan Peyton // which could be reallocated by the master thread). No task teams are formed 28393041982dSJonathan Peyton // for serialized teams. 28407cc577a4SJonathan Peyton if (team->t.t_nproc > 1) { 28417cc577a4SJonathan Peyton int other_team = 1 - this_thr->th.th_task_state; 28427cc577a4SJonathan Peyton if (team->t.t_task_team[other_team] == NULL) { // setup other team as well 28433041982dSJonathan Peyton team->t.t_task_team[other_team] = 28443041982dSJonathan Peyton __kmp_allocate_task_team(this_thr, team); 28453041982dSJonathan Peyton KA_TRACE(20, ("__kmp_task_team_setup: Master T#%d created second new " 28463041982dSJonathan Peyton "task_team %p for team %d at parity=%d\n", 28473041982dSJonathan Peyton __kmp_gtid_from_thread(this_thr), 28483041982dSJonathan Peyton team->t.t_task_team[other_team], 28497cc577a4SJonathan Peyton ((team != NULL) ? team->t.t_id : -1), other_team)); 28503041982dSJonathan Peyton } else { // Leave the old task team struct in place for the upcoming region; 28513041982dSJonathan Peyton // adjust as needed 28527cc577a4SJonathan Peyton kmp_task_team_t *task_team = team->t.t_task_team[other_team]; 28533041982dSJonathan Peyton if (!task_team->tt.tt_active || 28543041982dSJonathan Peyton team->t.t_nproc != task_team->tt.tt_nproc) { 28557cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_nproc, team->t.t_nproc); 28567cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_found_tasks, FALSE); 28577cc577a4SJonathan Peyton #if OMP_45_ENABLED 28587cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_found_proxy_tasks, FALSE); 28597cc577a4SJonathan Peyton #endif 28607cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_unfinished_threads, team->t.t_nproc); 28617cc577a4SJonathan Peyton TCW_4(task_team->tt.tt_active, TRUE); 28627cc577a4SJonathan Peyton } 28633041982dSJonathan Peyton // if team size has changed, the first thread to enable tasking will 28643041982dSJonathan Peyton // realloc threads_data if necessary 28653041982dSJonathan Peyton KA_TRACE(20, ("__kmp_task_team_setup: Master T#%d reset next task_team " 28663041982dSJonathan Peyton "%p for team %d at parity=%d\n", 28673041982dSJonathan Peyton __kmp_gtid_from_thread(this_thr), 28683041982dSJonathan Peyton team->t.t_task_team[other_team], 28697cc577a4SJonathan Peyton ((team != NULL) ? team->t.t_id : -1), other_team)); 28707cc577a4SJonathan Peyton } 28717cc577a4SJonathan Peyton } 28727cc577a4SJonathan Peyton } 28737cc577a4SJonathan Peyton 28747cc577a4SJonathan Peyton // __kmp_task_team_sync: Propagation of task team data from team to threads 28757cc577a4SJonathan Peyton // which happens just after the release phase of a team barrier. This may be 28767cc577a4SJonathan Peyton // called by any thread, but only for teams with # threads > 1. 28773041982dSJonathan Peyton void __kmp_task_team_sync(kmp_info_t *this_thr, kmp_team_t *team) { 28787cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(__kmp_tasking_mode != tskm_immediate_exec); 28797cc577a4SJonathan Peyton 28803041982dSJonathan Peyton // Toggle the th_task_state field, to switch which task_team this thread 28813041982dSJonathan Peyton // refers to 28827cc577a4SJonathan Peyton this_thr->th.th_task_state = 1 - this_thr->th.th_task_state; 28833041982dSJonathan Peyton // It is now safe to propagate the task team pointer from the team struct to 28843041982dSJonathan Peyton // the current thread. 28853041982dSJonathan Peyton TCW_PTR(this_thr->th.th_task_team, 28863041982dSJonathan Peyton team->t.t_task_team[this_thr->th.th_task_state]); 28873041982dSJonathan Peyton KA_TRACE(20, 28883041982dSJonathan Peyton ("__kmp_task_team_sync: Thread T#%d task team switched to task_team " 28893041982dSJonathan Peyton "%p from Team #%d (parity=%d)\n", 28907cc577a4SJonathan Peyton __kmp_gtid_from_thread(this_thr), this_thr->th.th_task_team, 28917cc577a4SJonathan Peyton ((team != NULL) ? team->t.t_id : -1), this_thr->th.th_task_state)); 28927cc577a4SJonathan Peyton } 28937cc577a4SJonathan Peyton 28943041982dSJonathan Peyton // __kmp_task_team_wait: Master thread waits for outstanding tasks after the 28953041982dSJonathan Peyton // barrier gather phase. Only called by master thread if #threads in team > 1 or 28963041982dSJonathan Peyton // if proxy tasks were created. 28973041982dSJonathan Peyton // 28983041982dSJonathan Peyton // wait is a flag that defaults to 1 (see kmp.h), but waiting can be turned off 28993041982dSJonathan Peyton // by passing in 0 optionally as the last argument. When wait is zero, master 29003041982dSJonathan Peyton // thread does not wait for unfinished_threads to reach 0. 29013041982dSJonathan Peyton void __kmp_task_team_wait( 29023041982dSJonathan Peyton kmp_info_t *this_thr, 29033041982dSJonathan Peyton kmp_team_t *team USE_ITT_BUILD_ARG(void *itt_sync_obj), int wait) { 29047cc577a4SJonathan Peyton kmp_task_team_t *task_team = team->t.t_task_team[this_thr->th.th_task_state]; 29057cc577a4SJonathan Peyton 29067cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(__kmp_tasking_mode != tskm_immediate_exec); 29077cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_team == this_thr->th.th_task_team); 29087cc577a4SJonathan Peyton 29097cc577a4SJonathan Peyton if ((task_team != NULL) && KMP_TASKING_ENABLED(task_team)) { 29107cc577a4SJonathan Peyton if (wait) { 29113041982dSJonathan Peyton KA_TRACE(20, ("__kmp_task_team_wait: Master T#%d waiting for all tasks " 29123041982dSJonathan Peyton "(for unfinished_threads to reach 0) on task_team = %p\n", 29137cc577a4SJonathan Peyton __kmp_gtid_from_thread(this_thr), task_team)); 29143041982dSJonathan Peyton // Worker threads may have dropped through to release phase, but could 29153041982dSJonathan Peyton // still be executing tasks. Wait here for tasks to complete. To avoid 29163041982dSJonathan Peyton // memory contention, only master thread checks termination condition. 29177cc577a4SJonathan Peyton kmp_flag_32 flag(&task_team->tt.tt_unfinished_threads, 0U); 29183041982dSJonathan Peyton flag.wait(this_thr, TRUE USE_ITT_BUILD_ARG(itt_sync_obj)); 29197cc577a4SJonathan Peyton } 29203041982dSJonathan Peyton // Deactivate the old task team, so that the worker threads will stop 29213041982dSJonathan Peyton // referencing it while spinning. 29223041982dSJonathan Peyton KA_TRACE( 29233041982dSJonathan Peyton 20, 29243041982dSJonathan Peyton ("__kmp_task_team_wait: Master T#%d deactivating task_team %p: " 29257cc577a4SJonathan Peyton "setting active to false, setting local and team's pointer to NULL\n", 29267cc577a4SJonathan Peyton __kmp_gtid_from_thread(this_thr), task_team)); 29277cc577a4SJonathan Peyton #if OMP_45_ENABLED 29283041982dSJonathan Peyton KMP_DEBUG_ASSERT(task_team->tt.tt_nproc > 1 || 29293041982dSJonathan Peyton task_team->tt.tt_found_proxy_tasks == TRUE); 29307cc577a4SJonathan Peyton TCW_SYNC_4(task_team->tt.tt_found_proxy_tasks, FALSE); 29317cc577a4SJonathan Peyton #else 29327cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_team->tt.tt_nproc > 1); 29337cc577a4SJonathan Peyton #endif 29347cc577a4SJonathan Peyton TCW_SYNC_4(task_team->tt.tt_active, FALSE); 29357cc577a4SJonathan Peyton KMP_MB(); 29367cc577a4SJonathan Peyton 29377cc577a4SJonathan Peyton TCW_PTR(this_thr->th.th_task_team, NULL); 29387cc577a4SJonathan Peyton } 29397cc577a4SJonathan Peyton } 29407cc577a4SJonathan Peyton 29417cc577a4SJonathan Peyton // __kmp_tasking_barrier: 29427cc577a4SJonathan Peyton // This routine may only called when __kmp_tasking_mode == tskm_extra_barrier. 29433041982dSJonathan Peyton // Internal function to execute all tasks prior to a regular barrier or a join 29443041982dSJonathan Peyton // barrier. It is a full barrier itself, which unfortunately turns regular 29453041982dSJonathan Peyton // barriers into double barriers and join barriers into 1 1/2 barriers. 29463041982dSJonathan Peyton void __kmp_tasking_barrier(kmp_team_t *team, kmp_info_t *thread, int gtid) { 29473041982dSJonathan Peyton volatile kmp_uint32 *spin = 29483041982dSJonathan Peyton &team->t.t_task_team[thread->th.th_task_state]->tt.tt_unfinished_threads; 29497cc577a4SJonathan Peyton int flag = FALSE; 29507cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(__kmp_tasking_mode == tskm_extra_barrier); 29517cc577a4SJonathan Peyton 29527cc577a4SJonathan Peyton #if USE_ITT_BUILD 29537cc577a4SJonathan Peyton KMP_FSYNC_SPIN_INIT(spin, (kmp_uint32 *)NULL); 29547cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 29557cc577a4SJonathan Peyton kmp_flag_32 spin_flag(spin, 0U); 29563041982dSJonathan Peyton while (!spin_flag.execute_tasks(thread, gtid, TRUE, 29573041982dSJonathan Peyton &flag USE_ITT_BUILD_ARG(NULL), 0)) { 29587cc577a4SJonathan Peyton #if USE_ITT_BUILD 29597cc577a4SJonathan Peyton // TODO: What about itt_sync_obj?? 29607cc577a4SJonathan Peyton KMP_FSYNC_SPIN_PREPARE(spin); 29617cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 29627cc577a4SJonathan Peyton 29637cc577a4SJonathan Peyton if (TCR_4(__kmp_global.g.g_done)) { 29647cc577a4SJonathan Peyton if (__kmp_global.g.g_abort) 29657cc577a4SJonathan Peyton __kmp_abort_thread(); 29667cc577a4SJonathan Peyton break; 29677cc577a4SJonathan Peyton } 29687cc577a4SJonathan Peyton KMP_YIELD(TRUE); // GH: We always yield here 29697cc577a4SJonathan Peyton } 29707cc577a4SJonathan Peyton #if USE_ITT_BUILD 29717cc577a4SJonathan Peyton KMP_FSYNC_SPIN_ACQUIRED((void *)spin); 29727cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */ 29737cc577a4SJonathan Peyton } 29747cc577a4SJonathan Peyton 29757cc577a4SJonathan Peyton #if OMP_45_ENABLED 29767cc577a4SJonathan Peyton 29773041982dSJonathan Peyton // __kmp_give_task puts a task into a given thread queue if: 29783041982dSJonathan Peyton // - the queue for that thread was created 29793041982dSJonathan Peyton // - there's space in that queue 29803041982dSJonathan Peyton // Because of this, __kmp_push_task needs to check if there's space after 29813041982dSJonathan Peyton // getting the lock 29823041982dSJonathan Peyton static bool __kmp_give_task(kmp_info_t *thread, kmp_int32 tid, kmp_task_t *task, 29833041982dSJonathan Peyton kmp_int32 pass) { 29847cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task); 29857cc577a4SJonathan Peyton kmp_task_team_t *task_team = taskdata->td_task_team; 29867cc577a4SJonathan Peyton 29873041982dSJonathan Peyton KA_TRACE(20, ("__kmp_give_task: trying to give task %p to thread %d.\n", 29883041982dSJonathan Peyton taskdata, tid)); 29897cc577a4SJonathan Peyton 29907cc577a4SJonathan Peyton // If task_team is NULL something went really bad... 29917cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task_team != NULL); 29927cc577a4SJonathan Peyton 29937cc577a4SJonathan Peyton bool result = false; 29947cc577a4SJonathan Peyton kmp_thread_data_t *thread_data = &task_team->tt.tt_threads_data[tid]; 29957cc577a4SJonathan Peyton 29967cc577a4SJonathan Peyton if (thread_data->td.td_deque == NULL) { 29977cc577a4SJonathan Peyton // There's no queue in this thread, go find another one 29987cc577a4SJonathan Peyton // We're guaranteed that at least one thread has a queue 29993041982dSJonathan Peyton KA_TRACE(30, 30003041982dSJonathan Peyton ("__kmp_give_task: thread %d has no queue while giving task %p.\n", 30013041982dSJonathan Peyton tid, taskdata)); 30027cc577a4SJonathan Peyton return result; 30037cc577a4SJonathan Peyton } 30047cc577a4SJonathan Peyton 30053041982dSJonathan Peyton if (TCR_4(thread_data->td.td_deque_ntasks) >= 30063041982dSJonathan Peyton TASK_DEQUE_SIZE(thread_data->td)) { 30073041982dSJonathan Peyton KA_TRACE( 30083041982dSJonathan Peyton 30, 30093041982dSJonathan Peyton ("__kmp_give_task: queue is full while giving task %p to thread %d.\n", 30103041982dSJonathan Peyton taskdata, tid)); 30117cc577a4SJonathan Peyton 30123041982dSJonathan Peyton // if this deque is bigger than the pass ratio give a chance to another 30133041982dSJonathan Peyton // thread 30143041982dSJonathan Peyton if (TASK_DEQUE_SIZE(thread_data->td) / INITIAL_TASK_DEQUE_SIZE >= pass) 30153041982dSJonathan Peyton return result; 30167cc577a4SJonathan Peyton 30177cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&thread_data->td.td_deque_lock); 30187cc577a4SJonathan Peyton __kmp_realloc_task_deque(thread, thread_data); 30197cc577a4SJonathan Peyton 30207cc577a4SJonathan Peyton } else { 30217cc577a4SJonathan Peyton 30227cc577a4SJonathan Peyton __kmp_acquire_bootstrap_lock(&thread_data->td.td_deque_lock); 30237cc577a4SJonathan Peyton 30243041982dSJonathan Peyton if (TCR_4(thread_data->td.td_deque_ntasks) >= 30253041982dSJonathan Peyton TASK_DEQUE_SIZE(thread_data->td)) { 30263041982dSJonathan Peyton KA_TRACE(30, ("__kmp_give_task: queue is full while giving task %p to " 30273041982dSJonathan Peyton "thread %d.\n", 30283041982dSJonathan Peyton taskdata, tid)); 30297cc577a4SJonathan Peyton 30303041982dSJonathan Peyton // if this deque is bigger than the pass ratio give a chance to another 30313041982dSJonathan Peyton // thread 30327cc577a4SJonathan Peyton if (TASK_DEQUE_SIZE(thread_data->td) / INITIAL_TASK_DEQUE_SIZE >= pass) 30337cc577a4SJonathan Peyton goto release_and_exit; 30347cc577a4SJonathan Peyton 30357cc577a4SJonathan Peyton __kmp_realloc_task_deque(thread, thread_data); 30367cc577a4SJonathan Peyton } 30377cc577a4SJonathan Peyton } 30387cc577a4SJonathan Peyton 30397cc577a4SJonathan Peyton // lock is held here, and there is space in the deque 30407cc577a4SJonathan Peyton 30417cc577a4SJonathan Peyton thread_data->td.td_deque[thread_data->td.td_deque_tail] = taskdata; 30427cc577a4SJonathan Peyton // Wrap index. 30433041982dSJonathan Peyton thread_data->td.td_deque_tail = 30443041982dSJonathan Peyton (thread_data->td.td_deque_tail + 1) & TASK_DEQUE_MASK(thread_data->td); 30453041982dSJonathan Peyton TCW_4(thread_data->td.td_deque_ntasks, 30463041982dSJonathan Peyton TCR_4(thread_data->td.td_deque_ntasks) + 1); 30477cc577a4SJonathan Peyton 30487cc577a4SJonathan Peyton result = true; 30493041982dSJonathan Peyton KA_TRACE(30, ("__kmp_give_task: successfully gave task %p to thread %d.\n", 30503041982dSJonathan Peyton taskdata, tid)); 30517cc577a4SJonathan Peyton 30527cc577a4SJonathan Peyton release_and_exit: 30537cc577a4SJonathan Peyton __kmp_release_bootstrap_lock(&thread_data->td.td_deque_lock); 30547cc577a4SJonathan Peyton 30557cc577a4SJonathan Peyton return result; 30567cc577a4SJonathan Peyton } 30577cc577a4SJonathan Peyton 30583041982dSJonathan Peyton /* The finish of the proxy tasks is divided in two pieces: 30597cc577a4SJonathan Peyton - the top half is the one that can be done from a thread outside the team 30607cc577a4SJonathan Peyton - the bottom half must be run from a them within the team 30617cc577a4SJonathan Peyton 30623041982dSJonathan Peyton In order to run the bottom half the task gets queued back into one of the 30633041982dSJonathan Peyton threads of the team. Once the td_incomplete_child_task counter of the parent 30643041982dSJonathan Peyton is decremented the threads can leave the barriers. So, the bottom half needs 30653041982dSJonathan Peyton to be queued before the counter is decremented. The top half is therefore 30663041982dSJonathan Peyton divided in two parts: 30677cc577a4SJonathan Peyton - things that can be run before queuing the bottom half 30687cc577a4SJonathan Peyton - things that must be run after queuing the bottom half 30697cc577a4SJonathan Peyton 30703041982dSJonathan Peyton This creates a second race as the bottom half can free the task before the 30713041982dSJonathan Peyton second top half is executed. To avoid this we use the 30723041982dSJonathan Peyton td_incomplete_child_task of the proxy task to synchronize the top and bottom 30733041982dSJonathan Peyton half. */ 30743041982dSJonathan Peyton static void __kmp_first_top_half_finish_proxy(kmp_taskdata_t *taskdata) { 30757cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.tasktype == TASK_EXPLICIT); 30767cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.proxy == TASK_PROXY); 30777cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.complete == 0); 30787cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.freed == 0); 30797cc577a4SJonathan Peyton 30807cc577a4SJonathan Peyton taskdata->td_flags.complete = 1; // mark the task as completed 30817cc577a4SJonathan Peyton 30827cc577a4SJonathan Peyton if (taskdata->td_taskgroup) 30837cc577a4SJonathan Peyton KMP_TEST_THEN_DEC32((kmp_int32 *)(&taskdata->td_taskgroup->count)); 30847cc577a4SJonathan Peyton 30853041982dSJonathan Peyton // Create an imaginary children for this task so the bottom half cannot 30863041982dSJonathan Peyton // release the task before we have completed the second top half 30877cc577a4SJonathan Peyton TCI_4(taskdata->td_incomplete_child_tasks); 30887cc577a4SJonathan Peyton } 30897cc577a4SJonathan Peyton 30903041982dSJonathan Peyton static void __kmp_second_top_half_finish_proxy(kmp_taskdata_t *taskdata) { 30917cc577a4SJonathan Peyton kmp_int32 children = 0; 30927cc577a4SJonathan Peyton 30937cc577a4SJonathan Peyton // Predecrement simulated by "- 1" calculation 30943041982dSJonathan Peyton children = 30953041982dSJonathan Peyton KMP_TEST_THEN_DEC32( 30963041982dSJonathan Peyton (kmp_int32 *)(&taskdata->td_parent->td_incomplete_child_tasks)) - 30973041982dSJonathan Peyton 1; 30987cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(children >= 0); 30997cc577a4SJonathan Peyton 31007cc577a4SJonathan Peyton // Remove the imaginary children 31017cc577a4SJonathan Peyton TCD_4(taskdata->td_incomplete_child_tasks); 31027cc577a4SJonathan Peyton } 31037cc577a4SJonathan Peyton 31043041982dSJonathan Peyton static void __kmp_bottom_half_finish_proxy(kmp_int32 gtid, kmp_task_t *ptask) { 31057cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(ptask); 31067cc577a4SJonathan Peyton kmp_info_t *thread = __kmp_threads[gtid]; 31077cc577a4SJonathan Peyton 31087cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.proxy == TASK_PROXY); 31093041982dSJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.complete == 31103041982dSJonathan Peyton 1); // top half must run before bottom half 31117cc577a4SJonathan Peyton 31127cc577a4SJonathan Peyton // We need to wait to make sure the top half is finished 31137cc577a4SJonathan Peyton // Spinning here should be ok as this should happen quickly 31143041982dSJonathan Peyton while (TCR_4(taskdata->td_incomplete_child_tasks) > 0) 31153041982dSJonathan Peyton ; 31167cc577a4SJonathan Peyton 31177cc577a4SJonathan Peyton __kmp_release_deps(gtid, taskdata); 31187cc577a4SJonathan Peyton __kmp_free_task_and_ancestors(gtid, taskdata, thread); 31197cc577a4SJonathan Peyton } 31207cc577a4SJonathan Peyton 31217cc577a4SJonathan Peyton /*! 31227cc577a4SJonathan Peyton @ingroup TASKING 31237cc577a4SJonathan Peyton @param gtid Global Thread ID of encountering thread 31247cc577a4SJonathan Peyton @param ptask Task which execution is completed 31257cc577a4SJonathan Peyton 31263041982dSJonathan Peyton Execute the completation of a proxy task from a thread of that is part of the 31273041982dSJonathan Peyton team. Run first and bottom halves directly. 31287cc577a4SJonathan Peyton */ 31293041982dSJonathan Peyton void __kmpc_proxy_task_completed(kmp_int32 gtid, kmp_task_t *ptask) { 31307cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(ptask != NULL); 31317cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(ptask); 31323041982dSJonathan Peyton KA_TRACE( 31333041982dSJonathan Peyton 10, ("__kmp_proxy_task_completed(enter): T#%d proxy task %p completing\n", 31343041982dSJonathan Peyton gtid, taskdata)); 31357cc577a4SJonathan Peyton 31367cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.proxy == TASK_PROXY); 31377cc577a4SJonathan Peyton 31387cc577a4SJonathan Peyton __kmp_first_top_half_finish_proxy(taskdata); 31397cc577a4SJonathan Peyton __kmp_second_top_half_finish_proxy(taskdata); 31407cc577a4SJonathan Peyton __kmp_bottom_half_finish_proxy(gtid, ptask); 31417cc577a4SJonathan Peyton 31423041982dSJonathan Peyton KA_TRACE(10, 31433041982dSJonathan Peyton ("__kmp_proxy_task_completed(exit): T#%d proxy task %p completing\n", 31443041982dSJonathan Peyton gtid, taskdata)); 31457cc577a4SJonathan Peyton } 31467cc577a4SJonathan Peyton 31477cc577a4SJonathan Peyton /*! 31487cc577a4SJonathan Peyton @ingroup TASKING 31497cc577a4SJonathan Peyton @param ptask Task which execution is completed 31507cc577a4SJonathan Peyton 31513041982dSJonathan Peyton Execute the completation of a proxy task from a thread that could not belong to 31523041982dSJonathan Peyton the team. 31537cc577a4SJonathan Peyton */ 31543041982dSJonathan Peyton void __kmpc_proxy_task_completed_ooo(kmp_task_t *ptask) { 31557cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(ptask != NULL); 31567cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(ptask); 31577cc577a4SJonathan Peyton 31583041982dSJonathan Peyton KA_TRACE( 31593041982dSJonathan Peyton 10, 31603041982dSJonathan Peyton ("__kmp_proxy_task_completed_ooo(enter): proxy task completing ooo %p\n", 31613041982dSJonathan Peyton taskdata)); 31627cc577a4SJonathan Peyton 31637cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata->td_flags.proxy == TASK_PROXY); 31647cc577a4SJonathan Peyton 31657cc577a4SJonathan Peyton __kmp_first_top_half_finish_proxy(taskdata); 31667cc577a4SJonathan Peyton 31673041982dSJonathan Peyton // Enqueue task to complete bottom half completion from a thread within the 31683041982dSJonathan Peyton // corresponding team 31697cc577a4SJonathan Peyton kmp_team_t *team = taskdata->td_team; 31707cc577a4SJonathan Peyton kmp_int32 nthreads = team->t.t_nproc; 31717cc577a4SJonathan Peyton kmp_info_t *thread; 31727cc577a4SJonathan Peyton 31733041982dSJonathan Peyton // This should be similar to start_k = __kmp_get_random( thread ) % nthreads 31743041982dSJonathan Peyton // but we cannot use __kmp_get_random here 31757cc577a4SJonathan Peyton kmp_int32 start_k = 0; 31767cc577a4SJonathan Peyton kmp_int32 pass = 1; 31777cc577a4SJonathan Peyton kmp_int32 k = start_k; 31787cc577a4SJonathan Peyton 31797cc577a4SJonathan Peyton do { 31807cc577a4SJonathan Peyton // For now we're just linearly trying to find a thread 31817cc577a4SJonathan Peyton thread = team->t.t_threads[k]; 31827cc577a4SJonathan Peyton k = (k + 1) % nthreads; 31837cc577a4SJonathan Peyton 31847cc577a4SJonathan Peyton // we did a full pass through all the threads 31853041982dSJonathan Peyton if (k == start_k) 31863041982dSJonathan Peyton pass = pass << 1; 31877cc577a4SJonathan Peyton 31887cc577a4SJonathan Peyton } while (!__kmp_give_task(thread, k, ptask, pass)); 31897cc577a4SJonathan Peyton 31907cc577a4SJonathan Peyton __kmp_second_top_half_finish_proxy(taskdata); 31917cc577a4SJonathan Peyton 31923041982dSJonathan Peyton KA_TRACE( 31933041982dSJonathan Peyton 10, 31943041982dSJonathan Peyton ("__kmp_proxy_task_completed_ooo(exit): proxy task completing ooo %p\n", 31953041982dSJonathan Peyton taskdata)); 31967cc577a4SJonathan Peyton } 31977cc577a4SJonathan Peyton 31983041982dSJonathan Peyton // __kmp_task_dup_alloc: Allocate the taskdata and make a copy of source task 31993041982dSJonathan Peyton // for taskloop 32007cc577a4SJonathan Peyton // 32017cc577a4SJonathan Peyton // thread: allocating thread 32027cc577a4SJonathan Peyton // task_src: pointer to source task to be duplicated 32037cc577a4SJonathan Peyton // returns: a pointer to the allocated kmp_task_t structure (task). 32043041982dSJonathan Peyton kmp_task_t *__kmp_task_dup_alloc(kmp_info_t *thread, kmp_task_t *task_src) { 32057cc577a4SJonathan Peyton kmp_task_t *task; 32067cc577a4SJonathan Peyton kmp_taskdata_t *taskdata; 32077cc577a4SJonathan Peyton kmp_taskdata_t *taskdata_src; 32087cc577a4SJonathan Peyton kmp_taskdata_t *parent_task = thread->th.th_current_task; 32097cc577a4SJonathan Peyton size_t shareds_offset; 32107cc577a4SJonathan Peyton size_t task_size; 32117cc577a4SJonathan Peyton 32123041982dSJonathan Peyton KA_TRACE(10, ("__kmp_task_dup_alloc(enter): Th %p, source task %p\n", thread, 32133041982dSJonathan Peyton task_src)); 32147cc577a4SJonathan Peyton taskdata_src = KMP_TASK_TO_TASKDATA(task_src); 32153041982dSJonathan Peyton KMP_DEBUG_ASSERT(taskdata_src->td_flags.proxy == 32163041982dSJonathan Peyton TASK_FULL); // it should not be proxy task 32177cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(taskdata_src->td_flags.tasktype == TASK_EXPLICIT); 32187cc577a4SJonathan Peyton task_size = taskdata_src->td_size_alloc; 32197cc577a4SJonathan Peyton 32207cc577a4SJonathan Peyton // Allocate a kmp_taskdata_t block and a kmp_task_t block. 32213041982dSJonathan Peyton KA_TRACE(30, ("__kmp_task_dup_alloc: Th %p, malloc size %ld\n", thread, 32223041982dSJonathan Peyton task_size)); 32237cc577a4SJonathan Peyton #if USE_FAST_MEMORY 32247cc577a4SJonathan Peyton taskdata = (kmp_taskdata_t *)__kmp_fast_allocate(thread, task_size); 32257cc577a4SJonathan Peyton #else 32267cc577a4SJonathan Peyton taskdata = (kmp_taskdata_t *)__kmp_thread_malloc(thread, task_size); 32277cc577a4SJonathan Peyton #endif /* USE_FAST_MEMORY */ 32287cc577a4SJonathan Peyton KMP_MEMCPY(taskdata, taskdata_src, task_size); 32297cc577a4SJonathan Peyton 32307cc577a4SJonathan Peyton task = KMP_TASKDATA_TO_TASK(taskdata); 32317cc577a4SJonathan Peyton 32327cc577a4SJonathan Peyton // Initialize new task (only specific fields not affected by memcpy) 32337cc577a4SJonathan Peyton taskdata->td_task_id = KMP_GEN_TASK_ID(); 32347cc577a4SJonathan Peyton if (task->shareds != NULL) { // need setup shareds pointer 32357cc577a4SJonathan Peyton shareds_offset = (char *)task_src->shareds - (char *)taskdata_src; 32367cc577a4SJonathan Peyton task->shareds = &((char *)taskdata)[shareds_offset]; 32373041982dSJonathan Peyton KMP_DEBUG_ASSERT((((kmp_uintptr_t)task->shareds) & (sizeof(void *) - 1)) == 32383041982dSJonathan Peyton 0); 32397cc577a4SJonathan Peyton } 32407cc577a4SJonathan Peyton taskdata->td_alloc_thread = thread; 32413041982dSJonathan Peyton taskdata->td_taskgroup = 32423041982dSJonathan Peyton parent_task 32433041982dSJonathan Peyton ->td_taskgroup; // task inherits the taskgroup from the parent task 32447cc577a4SJonathan Peyton 32453041982dSJonathan Peyton // Only need to keep track of child task counts if team parallel and tasking 32463041982dSJonathan Peyton // not serialized 32477cc577a4SJonathan Peyton if (!(taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser)) { 32487cc577a4SJonathan Peyton KMP_TEST_THEN_INC32((kmp_int32 *)(&parent_task->td_incomplete_child_tasks)); 32497cc577a4SJonathan Peyton if (parent_task->td_taskgroup) 32507cc577a4SJonathan Peyton KMP_TEST_THEN_INC32((kmp_int32 *)(&parent_task->td_taskgroup->count)); 32513041982dSJonathan Peyton // Only need to keep track of allocated child tasks for explicit tasks since 32523041982dSJonathan Peyton // implicit not deallocated 32537cc577a4SJonathan Peyton if (taskdata->td_parent->td_flags.tasktype == TASK_EXPLICIT) 32543041982dSJonathan Peyton KMP_TEST_THEN_INC32( 32553041982dSJonathan Peyton (kmp_int32 *)(&taskdata->td_parent->td_allocated_child_tasks)); 32567cc577a4SJonathan Peyton } 32577cc577a4SJonathan Peyton 32583041982dSJonathan Peyton KA_TRACE(20, 32593041982dSJonathan Peyton ("__kmp_task_dup_alloc(exit): Th %p, created task %p, parent=%p\n", 32607cc577a4SJonathan Peyton thread, taskdata, taskdata->td_parent)); 32617cc577a4SJonathan Peyton #if OMPT_SUPPORT 32623041982dSJonathan Peyton __kmp_task_init_ompt(taskdata, thread->th.th_info.ds.ds_gtid, 32633041982dSJonathan Peyton (void *)task->routine); 32647cc577a4SJonathan Peyton #endif 32657cc577a4SJonathan Peyton return task; 32667cc577a4SJonathan Peyton } 32677cc577a4SJonathan Peyton 32687cc577a4SJonathan Peyton // Routine optionally generated by th ecompiler for setting the lastprivate flag 32697cc577a4SJonathan Peyton // and calling needed constructors for private/firstprivate objects 32707cc577a4SJonathan Peyton // (used to form taskloop tasks from pattern task) 32717cc577a4SJonathan Peyton typedef void (*p_task_dup_t)(kmp_task_t *, kmp_task_t *, kmp_int32); 32727cc577a4SJonathan Peyton 32737cc577a4SJonathan Peyton // __kmp_taskloop_linear: Start tasks of the taskloop linearly 32747cc577a4SJonathan Peyton // 32757cc577a4SJonathan Peyton // loc Source location information 32767cc577a4SJonathan Peyton // gtid Global thread ID 32777cc577a4SJonathan Peyton // task Task with whole loop iteration range 32787cc577a4SJonathan Peyton // lb Pointer to loop lower bound 32797cc577a4SJonathan Peyton // ub Pointer to loop upper bound 32807cc577a4SJonathan Peyton // st Loop stride 32817cc577a4SJonathan Peyton // sched Schedule specified 0/1/2 for none/grainsize/num_tasks 32827cc577a4SJonathan Peyton // grainsize Schedule value if specified 32837cc577a4SJonathan Peyton // task_dup Tasks duplication routine 32843041982dSJonathan Peyton void __kmp_taskloop_linear(ident_t *loc, int gtid, kmp_task_t *task, 32857cc577a4SJonathan Peyton kmp_uint64 *lb, kmp_uint64 *ub, kmp_int64 st, 32863041982dSJonathan Peyton int sched, kmp_uint64 grainsize, void *task_dup) { 32877cc577a4SJonathan Peyton KMP_COUNT_BLOCK(OMP_TASKLOOP); 32887cc577a4SJonathan Peyton KMP_TIME_PARTITIONED_BLOCK(OMP_taskloop_scheduling); 32897cc577a4SJonathan Peyton p_task_dup_t ptask_dup = (p_task_dup_t)task_dup; 32907cc577a4SJonathan Peyton kmp_uint64 tc; 32917cc577a4SJonathan Peyton kmp_uint64 lower = *lb; // compiler provides global bounds here 32927cc577a4SJonathan Peyton kmp_uint64 upper = *ub; 32937cc577a4SJonathan Peyton kmp_uint64 i, num_tasks = 0, extras = 0; 32947cc577a4SJonathan Peyton kmp_info_t *thread = __kmp_threads[gtid]; 32957cc577a4SJonathan Peyton kmp_taskdata_t *current_task = thread->th.th_current_task; 32967cc577a4SJonathan Peyton kmp_task_t *next_task; 32977cc577a4SJonathan Peyton kmp_int32 lastpriv = 0; 32983041982dSJonathan Peyton size_t lower_offset = 32993041982dSJonathan Peyton (char *)lb - (char *)task; // remember offset of lb in the task structure 33003041982dSJonathan Peyton size_t upper_offset = 33013041982dSJonathan Peyton (char *)ub - (char *)task; // remember offset of ub in the task structure 33027cc577a4SJonathan Peyton 33037cc577a4SJonathan Peyton // compute trip count 33047cc577a4SJonathan Peyton if (st == 1) { // most common case 33057cc577a4SJonathan Peyton tc = upper - lower + 1; 33067cc577a4SJonathan Peyton } else if (st < 0) { 33077cc577a4SJonathan Peyton tc = (lower - upper) / (-st) + 1; 33087cc577a4SJonathan Peyton } else { // st > 0 33097cc577a4SJonathan Peyton tc = (upper - lower) / st + 1; 33107cc577a4SJonathan Peyton } 33117cc577a4SJonathan Peyton if (tc == 0) { 33127cc577a4SJonathan Peyton KA_TRACE(20, ("__kmpc_taskloop(exit): T#%d zero-trip loop\n", gtid)); 33137cc577a4SJonathan Peyton // free the pattern task and exit 33147cc577a4SJonathan Peyton __kmp_task_start(gtid, task, current_task); 33157cc577a4SJonathan Peyton // do not execute anything for zero-trip loop 33167cc577a4SJonathan Peyton __kmp_task_finish(gtid, task, current_task); 33177cc577a4SJonathan Peyton return; 33187cc577a4SJonathan Peyton } 33197cc577a4SJonathan Peyton 33207cc577a4SJonathan Peyton // compute num_tasks/grainsize based on the input provided 33217cc577a4SJonathan Peyton switch (sched) { 33227cc577a4SJonathan Peyton case 0: // no schedule clause specified, we can choose the default 33237cc577a4SJonathan Peyton // let's try to schedule (team_size*10) tasks 33247cc577a4SJonathan Peyton grainsize = thread->th.th_team_nproc * 10; 33257cc577a4SJonathan Peyton case 2: // num_tasks provided 33267cc577a4SJonathan Peyton if (grainsize > tc) { 33277cc577a4SJonathan Peyton num_tasks = tc; // too big num_tasks requested, adjust values 33287cc577a4SJonathan Peyton grainsize = 1; 33297cc577a4SJonathan Peyton extras = 0; 33307cc577a4SJonathan Peyton } else { 33317cc577a4SJonathan Peyton num_tasks = grainsize; 33327cc577a4SJonathan Peyton grainsize = tc / num_tasks; 33337cc577a4SJonathan Peyton extras = tc % num_tasks; 33347cc577a4SJonathan Peyton } 33357cc577a4SJonathan Peyton break; 33367cc577a4SJonathan Peyton case 1: // grainsize provided 33377cc577a4SJonathan Peyton if (grainsize > tc) { 33387cc577a4SJonathan Peyton num_tasks = 1; // too big grainsize requested, adjust values 33397cc577a4SJonathan Peyton grainsize = tc; 33407cc577a4SJonathan Peyton extras = 0; 33417cc577a4SJonathan Peyton } else { 33427cc577a4SJonathan Peyton num_tasks = tc / grainsize; 33433041982dSJonathan Peyton grainsize = 33443041982dSJonathan Peyton tc / 33453041982dSJonathan Peyton num_tasks; // adjust grainsize for balanced distribution of iterations 33467cc577a4SJonathan Peyton extras = tc % num_tasks; 33477cc577a4SJonathan Peyton } 33487cc577a4SJonathan Peyton break; 33497cc577a4SJonathan Peyton default: 33507cc577a4SJonathan Peyton KMP_ASSERT2(0, "unknown scheduling of taskloop"); 33517cc577a4SJonathan Peyton } 33527cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(tc == num_tasks * grainsize + extras); 33537cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(num_tasks > extras); 33547cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(num_tasks > 0); 33553041982dSJonathan Peyton KA_TRACE(20, ("__kmpc_taskloop: T#%d will launch: num_tasks %lld, grainsize " 33563041982dSJonathan Peyton "%lld, extras %lld\n", 33577cc577a4SJonathan Peyton gtid, num_tasks, grainsize, extras)); 33587cc577a4SJonathan Peyton 33597cc577a4SJonathan Peyton // Main loop, launch num_tasks tasks, assign grainsize iterations each task 33607cc577a4SJonathan Peyton for (i = 0; i < num_tasks; ++i) { 33617cc577a4SJonathan Peyton kmp_uint64 chunk_minus_1; 33627cc577a4SJonathan Peyton if (extras == 0) { 33637cc577a4SJonathan Peyton chunk_minus_1 = grainsize - 1; 33647cc577a4SJonathan Peyton } else { 33657cc577a4SJonathan Peyton chunk_minus_1 = grainsize; 33667cc577a4SJonathan Peyton --extras; // first extras iterations get bigger chunk (grainsize+1) 33677cc577a4SJonathan Peyton } 33687cc577a4SJonathan Peyton upper = lower + st * chunk_minus_1; 33697cc577a4SJonathan Peyton if (i == num_tasks - 1) { 33707cc577a4SJonathan Peyton // schedule the last task, set lastprivate flag 33717cc577a4SJonathan Peyton lastpriv = 1; 33727cc577a4SJonathan Peyton #if KMP_DEBUG 33737cc577a4SJonathan Peyton if (st == 1) 33747cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(upper == *ub); 33757cc577a4SJonathan Peyton else if (st > 0) 33767cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(upper + st > *ub); 33777cc577a4SJonathan Peyton else 33787cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(upper + st < *ub); 33797cc577a4SJonathan Peyton #endif 33807cc577a4SJonathan Peyton } 33817cc577a4SJonathan Peyton next_task = __kmp_task_dup_alloc(thread, task); // allocate new task 33823041982dSJonathan Peyton *(kmp_uint64 *)((char *)next_task + lower_offset) = 33833041982dSJonathan Peyton lower; // adjust task-specific bounds 33847cc577a4SJonathan Peyton *(kmp_uint64 *)((char *)next_task + upper_offset) = upper; 33857cc577a4SJonathan Peyton if (ptask_dup != NULL) 33863041982dSJonathan Peyton ptask_dup(next_task, task, 33873041982dSJonathan Peyton lastpriv); // set lastprivate flag, construct fistprivates, etc. 33883041982dSJonathan Peyton KA_TRACE(20, ("__kmpc_taskloop: T#%d schedule task %p: lower %lld, upper " 33893041982dSJonathan Peyton "%lld (offsets %p %p)\n", 33907cc577a4SJonathan Peyton gtid, next_task, lower, upper, lower_offset, upper_offset)); 33917cc577a4SJonathan Peyton __kmp_omp_task(gtid, next_task, true); // schedule new task 33927cc577a4SJonathan Peyton lower = upper + st; // adjust lower bound for the next iteration 33937cc577a4SJonathan Peyton } 33947cc577a4SJonathan Peyton // free the pattern task and exit 33957cc577a4SJonathan Peyton __kmp_task_start(gtid, task, current_task); 33967cc577a4SJonathan Peyton // do not execute the pattern task, just do bookkeeping 33977cc577a4SJonathan Peyton __kmp_task_finish(gtid, task, current_task); 33987cc577a4SJonathan Peyton } 33997cc577a4SJonathan Peyton 34007cc577a4SJonathan Peyton /*! 34017cc577a4SJonathan Peyton @ingroup TASKING 34027cc577a4SJonathan Peyton @param loc Source location information 34037cc577a4SJonathan Peyton @param gtid Global thread ID 34047cc577a4SJonathan Peyton @param task Task structure 34057cc577a4SJonathan Peyton @param if_val Value of the if clause 34067cc577a4SJonathan Peyton @param lb Pointer to loop lower bound 34077cc577a4SJonathan Peyton @param ub Pointer to loop upper bound 34087cc577a4SJonathan Peyton @param st Loop stride 34097cc577a4SJonathan Peyton @param nogroup Flag, 1 if nogroup clause specified, 0 otherwise 34107cc577a4SJonathan Peyton @param sched Schedule specified 0/1/2 for none/grainsize/num_tasks 34117cc577a4SJonathan Peyton @param grainsize Schedule value if specified 34127cc577a4SJonathan Peyton @param task_dup Tasks duplication routine 34137cc577a4SJonathan Peyton 34147cc577a4SJonathan Peyton Execute the taskloop construct. 34157cc577a4SJonathan Peyton */ 34163041982dSJonathan Peyton void __kmpc_taskloop(ident_t *loc, int gtid, kmp_task_t *task, int if_val, 34173041982dSJonathan Peyton kmp_uint64 *lb, kmp_uint64 *ub, kmp_int64 st, int nogroup, 34183041982dSJonathan Peyton int sched, kmp_uint64 grainsize, void *task_dup) { 34197cc577a4SJonathan Peyton kmp_taskdata_t *taskdata = KMP_TASK_TO_TASKDATA(task); 34207cc577a4SJonathan Peyton KMP_DEBUG_ASSERT(task != NULL); 34217cc577a4SJonathan Peyton 34223041982dSJonathan Peyton KA_TRACE(10, ("__kmpc_taskloop(enter): T#%d, pattern task %p, lb %lld ub " 34233041982dSJonathan Peyton "%lld st %lld, grain %llu(%d)\n", 34247cc577a4SJonathan Peyton gtid, taskdata, *lb, *ub, st, grainsize, sched)); 34257cc577a4SJonathan Peyton 34267cc577a4SJonathan Peyton // check if clause value first 34277cc577a4SJonathan Peyton if (if_val == 0) { // if(0) specified, mark task as serial 34287cc577a4SJonathan Peyton taskdata->td_flags.task_serial = 1; 34297cc577a4SJonathan Peyton taskdata->td_flags.tiedness = TASK_TIED; // AC: serial task cannot be untied 34307cc577a4SJonathan Peyton } 34317cc577a4SJonathan Peyton if (nogroup == 0) { 34327cc577a4SJonathan Peyton __kmpc_taskgroup(loc, gtid); 34337cc577a4SJonathan Peyton } 34347cc577a4SJonathan Peyton 34357cc577a4SJonathan Peyton if (1 /* AC: use some heuristic here to choose task scheduling method */) { 34363041982dSJonathan Peyton __kmp_taskloop_linear(loc, gtid, task, lb, ub, st, sched, grainsize, 34373041982dSJonathan Peyton task_dup); 34387cc577a4SJonathan Peyton } 34397cc577a4SJonathan Peyton 34407cc577a4SJonathan Peyton if (nogroup == 0) { 34417cc577a4SJonathan Peyton __kmpc_end_taskgroup(loc, gtid); 34427cc577a4SJonathan Peyton } 34437cc577a4SJonathan Peyton KA_TRACE(10, ("__kmpc_taskloop(exit): T#%d\n", gtid)); 34447cc577a4SJonathan Peyton } 34457cc577a4SJonathan Peyton 34467cc577a4SJonathan Peyton #endif 3447