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_wait_release.h"
207cc577a4SJonathan Peyton #include "kmp_stats.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 /* ------------------------------------------------------------------------ */
297cc577a4SJonathan Peyton /* ------------------------------------------------------------------------ */
307cc577a4SJonathan Peyton 
317cc577a4SJonathan Peyton 
327cc577a4SJonathan Peyton /* forward declaration */
337cc577a4SJonathan Peyton static void __kmp_enable_tasking( kmp_task_team_t *task_team, kmp_info_t *this_thr );
347cc577a4SJonathan Peyton static void __kmp_alloc_task_deque( kmp_info_t *thread, kmp_thread_data_t *thread_data );
357cc577a4SJonathan Peyton static int  __kmp_realloc_task_threads_data( kmp_info_t *thread, kmp_task_team_t *task_team );
367cc577a4SJonathan Peyton 
377cc577a4SJonathan Peyton #ifdef OMP_45_ENABLED
387cc577a4SJonathan Peyton static void __kmp_bottom_half_finish_proxy( kmp_int32 gtid, kmp_task_t * ptask );
397cc577a4SJonathan Peyton #endif
407cc577a4SJonathan Peyton 
417cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK
427cc577a4SJonathan Peyton 
437cc577a4SJonathan Peyton //---------------------------------------------------------------------------
447cc577a4SJonathan Peyton //  __kmp_trace_task_stack: print the tied tasks from the task stack in order
457cc577a4SJonathan Peyton //     from top do bottom
467cc577a4SJonathan Peyton //
477cc577a4SJonathan Peyton //  gtid: global thread identifier for thread containing stack
487cc577a4SJonathan Peyton //  thread_data: thread data for task team thread containing stack
497cc577a4SJonathan Peyton //  threshold: value above which the trace statement triggers
507cc577a4SJonathan Peyton //  location: string identifying call site of this function (for trace)
517cc577a4SJonathan Peyton 
527cc577a4SJonathan Peyton static void
537cc577a4SJonathan Peyton __kmp_trace_task_stack( kmp_int32 gtid, kmp_thread_data_t *thread_data, int threshold, char *location )
547cc577a4SJonathan Peyton {
557cc577a4SJonathan Peyton     kmp_task_stack_t *task_stack = & thread_data->td.td_susp_tied_tasks;
567cc577a4SJonathan Peyton     kmp_taskdata_t **stack_top = task_stack -> ts_top;
577cc577a4SJonathan Peyton     kmp_int32 entries = task_stack -> ts_entries;
587cc577a4SJonathan Peyton     kmp_taskdata_t *tied_task;
597cc577a4SJonathan Peyton 
607cc577a4SJonathan Peyton     KA_TRACE(threshold, ("__kmp_trace_task_stack(start): location = %s, gtid = %d, entries = %d, "
617cc577a4SJonathan Peyton                          "first_block = %p, stack_top = %p \n",
627cc577a4SJonathan Peyton                          location, gtid, entries, task_stack->ts_first_block, stack_top ) );
637cc577a4SJonathan Peyton 
647cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( stack_top != NULL );
657cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( entries > 0 );
667cc577a4SJonathan Peyton 
677cc577a4SJonathan Peyton     while ( entries != 0 )
687cc577a4SJonathan Peyton     {
697cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( stack_top != & task_stack->ts_first_block.sb_block[0] );
707cc577a4SJonathan Peyton         // fix up ts_top if we need to pop from previous block
717cc577a4SJonathan Peyton         if ( entries & TASK_STACK_INDEX_MASK == 0 )
727cc577a4SJonathan Peyton         {
737cc577a4SJonathan Peyton             kmp_stack_block_t *stack_block = (kmp_stack_block_t *) (stack_top) ;
747cc577a4SJonathan Peyton 
757cc577a4SJonathan Peyton             stack_block = stack_block -> sb_prev;
767cc577a4SJonathan Peyton             stack_top = & stack_block -> sb_block[TASK_STACK_BLOCK_SIZE];
777cc577a4SJonathan Peyton         }
787cc577a4SJonathan Peyton 
797cc577a4SJonathan Peyton         // finish bookkeeping
807cc577a4SJonathan Peyton         stack_top--;
817cc577a4SJonathan Peyton         entries--;
827cc577a4SJonathan Peyton 
837cc577a4SJonathan Peyton         tied_task = * stack_top;
847cc577a4SJonathan Peyton 
857cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( tied_task != NULL );
867cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( tied_task -> td_flags.tasktype == TASK_TIED );
877cc577a4SJonathan Peyton 
887cc577a4SJonathan Peyton         KA_TRACE(threshold, ("__kmp_trace_task_stack(%s):             gtid=%d, entry=%d, "
897cc577a4SJonathan Peyton                              "stack_top=%p, tied_task=%p\n",
907cc577a4SJonathan Peyton                              location, gtid, entries, stack_top, tied_task ) );
917cc577a4SJonathan Peyton     }
927cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( stack_top == & task_stack->ts_first_block.sb_block[0] );
937cc577a4SJonathan Peyton 
947cc577a4SJonathan Peyton     KA_TRACE(threshold, ("__kmp_trace_task_stack(exit): location = %s, gtid = %d\n",
957cc577a4SJonathan Peyton                          location, gtid ) );
967cc577a4SJonathan Peyton }
977cc577a4SJonathan Peyton 
987cc577a4SJonathan Peyton //---------------------------------------------------------------------------
997cc577a4SJonathan Peyton //  __kmp_init_task_stack: initialize the task stack for the first time
1007cc577a4SJonathan Peyton //    after a thread_data structure is created.
1017cc577a4SJonathan Peyton //    It should not be necessary to do this again (assuming the stack works).
1027cc577a4SJonathan Peyton //
1037cc577a4SJonathan Peyton //  gtid: global thread identifier of calling thread
1047cc577a4SJonathan Peyton //  thread_data: thread data for task team thread containing stack
1057cc577a4SJonathan Peyton 
1067cc577a4SJonathan Peyton static void
1077cc577a4SJonathan Peyton __kmp_init_task_stack( kmp_int32 gtid, kmp_thread_data_t *thread_data )
1087cc577a4SJonathan Peyton {
1097cc577a4SJonathan Peyton     kmp_task_stack_t *task_stack = & thread_data->td.td_susp_tied_tasks;
1107cc577a4SJonathan Peyton     kmp_stack_block_t *first_block;
1117cc577a4SJonathan Peyton 
1127cc577a4SJonathan Peyton     // set up the first block of the stack
1137cc577a4SJonathan Peyton     first_block = & task_stack -> ts_first_block;
1147cc577a4SJonathan Peyton     task_stack -> ts_top = (kmp_taskdata_t **) first_block;
1157cc577a4SJonathan Peyton     memset( (void *) first_block, '\0', TASK_STACK_BLOCK_SIZE * sizeof(kmp_taskdata_t *));
1167cc577a4SJonathan Peyton 
1177cc577a4SJonathan Peyton     // initialize the stack to be empty
1187cc577a4SJonathan Peyton     task_stack  -> ts_entries = TASK_STACK_EMPTY;
1197cc577a4SJonathan Peyton     first_block -> sb_next = NULL;
1207cc577a4SJonathan Peyton     first_block -> sb_prev = NULL;
1217cc577a4SJonathan Peyton }
1227cc577a4SJonathan Peyton 
1237cc577a4SJonathan Peyton 
1247cc577a4SJonathan Peyton //---------------------------------------------------------------------------
1257cc577a4SJonathan Peyton //  __kmp_free_task_stack: free the task stack when thread_data is destroyed.
1267cc577a4SJonathan Peyton //
1277cc577a4SJonathan Peyton //  gtid: global thread identifier for calling thread
1287cc577a4SJonathan Peyton //  thread_data: thread info for thread containing stack
1297cc577a4SJonathan Peyton 
1307cc577a4SJonathan Peyton static void
1317cc577a4SJonathan Peyton __kmp_free_task_stack( kmp_int32 gtid, kmp_thread_data_t *thread_data )
1327cc577a4SJonathan Peyton {
1337cc577a4SJonathan Peyton     kmp_task_stack_t *task_stack = & thread_data->td.td_susp_tied_tasks;
1347cc577a4SJonathan Peyton     kmp_stack_block_t *stack_block = & task_stack -> ts_first_block;
1357cc577a4SJonathan Peyton 
1367cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task_stack -> ts_entries == TASK_STACK_EMPTY );
1377cc577a4SJonathan Peyton     // free from the second block of the stack
1387cc577a4SJonathan Peyton     while ( stack_block != NULL ) {
1397cc577a4SJonathan Peyton         kmp_stack_block_t *next_block = (stack_block) ? stack_block -> sb_next : NULL;
1407cc577a4SJonathan Peyton 
1417cc577a4SJonathan Peyton         stack_block -> sb_next = NULL;
1427cc577a4SJonathan Peyton         stack_block -> sb_prev = NULL;
1437cc577a4SJonathan Peyton         if (stack_block != & task_stack -> ts_first_block) {
1447cc577a4SJonathan Peyton             __kmp_thread_free( thread, stack_block );  // free the block, if not the first
1457cc577a4SJonathan Peyton         }
1467cc577a4SJonathan Peyton         stack_block = next_block;
1477cc577a4SJonathan Peyton     }
1487cc577a4SJonathan Peyton     // initialize the stack to be empty
1497cc577a4SJonathan Peyton     task_stack -> ts_entries = 0;
1507cc577a4SJonathan Peyton     task_stack -> ts_top = NULL;
1517cc577a4SJonathan Peyton }
1527cc577a4SJonathan Peyton 
1537cc577a4SJonathan Peyton 
1547cc577a4SJonathan Peyton //---------------------------------------------------------------------------
1557cc577a4SJonathan Peyton //  __kmp_push_task_stack: Push the tied task onto the task stack.
1567cc577a4SJonathan Peyton //     Grow the stack if necessary by allocating another block.
1577cc577a4SJonathan Peyton //
1587cc577a4SJonathan Peyton //  gtid: global thread identifier for calling thread
1597cc577a4SJonathan Peyton //  thread: thread info for thread containing stack
1607cc577a4SJonathan Peyton //  tied_task: the task to push on the stack
1617cc577a4SJonathan Peyton 
1627cc577a4SJonathan Peyton static void
1637cc577a4SJonathan Peyton __kmp_push_task_stack( kmp_int32 gtid, kmp_info_t *thread, kmp_taskdata_t * tied_task )
1647cc577a4SJonathan Peyton {
1657cc577a4SJonathan Peyton     // GEH - need to consider what to do if tt_threads_data not allocated yet
1667cc577a4SJonathan Peyton     kmp_thread_data_t *thread_data = & thread -> th.th_task_team ->
1677cc577a4SJonathan Peyton                                         tt.tt_threads_data[ __kmp_tid_from_gtid( gtid ) ];
1687cc577a4SJonathan Peyton     kmp_task_stack_t *task_stack = & thread_data->td.td_susp_tied_tasks ;
1697cc577a4SJonathan Peyton 
1707cc577a4SJonathan Peyton     if ( tied_task->td_flags.team_serial || tied_task->td_flags.tasking_ser ) {
1717cc577a4SJonathan Peyton         return;  // Don't push anything on stack if team or team tasks are serialized
1727cc577a4SJonathan Peyton     }
1737cc577a4SJonathan Peyton 
1747cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( tied_task -> td_flags.tasktype == TASK_TIED );
1757cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task_stack -> ts_top != NULL );
1767cc577a4SJonathan Peyton 
1777cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_push_task_stack(enter): GTID: %d; THREAD: %p; TASK: %p\n",
1787cc577a4SJonathan Peyton                   gtid, thread, tied_task ) );
1797cc577a4SJonathan Peyton     // Store entry
1807cc577a4SJonathan Peyton     * (task_stack -> ts_top) = tied_task;
1817cc577a4SJonathan Peyton 
1827cc577a4SJonathan Peyton     // Do bookkeeping for next push
1837cc577a4SJonathan Peyton     task_stack -> ts_top++;
1847cc577a4SJonathan Peyton     task_stack -> ts_entries++;
1857cc577a4SJonathan Peyton 
1867cc577a4SJonathan Peyton     if ( task_stack -> ts_entries & TASK_STACK_INDEX_MASK == 0 )
1877cc577a4SJonathan Peyton     {
1887cc577a4SJonathan Peyton         // Find beginning of this task block
1897cc577a4SJonathan Peyton         kmp_stack_block_t *stack_block =
1907cc577a4SJonathan Peyton              (kmp_stack_block_t *) (task_stack -> ts_top - TASK_STACK_BLOCK_SIZE);
1917cc577a4SJonathan Peyton 
1927cc577a4SJonathan Peyton         // Check if we already have a block
1937cc577a4SJonathan Peyton         if ( stack_block -> sb_next != NULL )
1947cc577a4SJonathan Peyton         {    // reset ts_top to beginning of next block
1957cc577a4SJonathan Peyton             task_stack -> ts_top = & stack_block -> sb_next -> sb_block[0];
1967cc577a4SJonathan Peyton         }
1977cc577a4SJonathan Peyton         else
1987cc577a4SJonathan Peyton         {   // Alloc new block and link it up
1997cc577a4SJonathan Peyton             kmp_stack_block_t *new_block = (kmp_stack_block_t *)
2007cc577a4SJonathan Peyton               __kmp_thread_calloc(thread, sizeof(kmp_stack_block_t));
2017cc577a4SJonathan Peyton 
2027cc577a4SJonathan Peyton             task_stack -> ts_top  = & new_block -> sb_block[0];
2037cc577a4SJonathan Peyton             stack_block -> sb_next = new_block;
2047cc577a4SJonathan Peyton             new_block  -> sb_prev = stack_block;
2057cc577a4SJonathan Peyton             new_block  -> sb_next = NULL;
2067cc577a4SJonathan Peyton 
2077cc577a4SJonathan Peyton             KA_TRACE(30, ("__kmp_push_task_stack(): GTID: %d; TASK: %p; Alloc new block: %p\n",
2087cc577a4SJonathan Peyton                           gtid, tied_task, new_block ) );
2097cc577a4SJonathan Peyton         }
2107cc577a4SJonathan Peyton     }
2117cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_push_task_stack(exit): GTID: %d; TASK: %p\n", gtid, tied_task ) );
2127cc577a4SJonathan Peyton }
2137cc577a4SJonathan Peyton 
2147cc577a4SJonathan Peyton //---------------------------------------------------------------------------
2157cc577a4SJonathan Peyton //  __kmp_pop_task_stack: Pop the tied task from the task stack.  Don't return
2167cc577a4SJonathan Peyton //     the task, just check to make sure it matches the ending task passed in.
2177cc577a4SJonathan Peyton //
2187cc577a4SJonathan Peyton //  gtid: global thread identifier for the calling thread
2197cc577a4SJonathan Peyton //  thread: thread info structure containing stack
2207cc577a4SJonathan Peyton //  tied_task: the task popped off the stack
2217cc577a4SJonathan Peyton //  ending_task: the task that is ending (should match popped task)
2227cc577a4SJonathan Peyton 
2237cc577a4SJonathan Peyton static void
2247cc577a4SJonathan Peyton __kmp_pop_task_stack( kmp_int32 gtid, kmp_info_t *thread, kmp_taskdata_t *ending_task )
2257cc577a4SJonathan Peyton {
2267cc577a4SJonathan Peyton     // GEH - need to consider what to do if tt_threads_data not allocated yet
2277cc577a4SJonathan Peyton     kmp_thread_data_t *thread_data = & thread -> th.th_task_team -> tt_threads_data[ __kmp_tid_from_gtid( gtid ) ];
2287cc577a4SJonathan Peyton     kmp_task_stack_t *task_stack = & thread_data->td.td_susp_tied_tasks ;
2297cc577a4SJonathan Peyton     kmp_taskdata_t *tied_task;
2307cc577a4SJonathan Peyton 
2317cc577a4SJonathan Peyton     if ( ending_task->td_flags.team_serial || ending_task->td_flags.tasking_ser ) {
2327cc577a4SJonathan Peyton         return;  // Don't pop anything from stack if team or team tasks are serialized
2337cc577a4SJonathan Peyton     }
2347cc577a4SJonathan Peyton 
2357cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task_stack -> ts_top != NULL );
2367cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task_stack -> ts_entries > 0 );
2377cc577a4SJonathan Peyton 
2387cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_pop_task_stack(enter): GTID: %d; THREAD: %p\n", gtid, thread ) );
2397cc577a4SJonathan Peyton 
2407cc577a4SJonathan Peyton     // fix up ts_top if we need to pop from previous block
2417cc577a4SJonathan Peyton     if ( task_stack -> ts_entries & TASK_STACK_INDEX_MASK == 0 )
2427cc577a4SJonathan Peyton     {
2437cc577a4SJonathan Peyton         kmp_stack_block_t *stack_block =
2447cc577a4SJonathan Peyton            (kmp_stack_block_t *) (task_stack -> ts_top) ;
2457cc577a4SJonathan Peyton 
2467cc577a4SJonathan Peyton         stack_block = stack_block -> sb_prev;
2477cc577a4SJonathan Peyton         task_stack -> ts_top = & stack_block -> sb_block[TASK_STACK_BLOCK_SIZE];
2487cc577a4SJonathan Peyton     }
2497cc577a4SJonathan Peyton 
2507cc577a4SJonathan Peyton     // finish bookkeeping
2517cc577a4SJonathan Peyton     task_stack -> ts_top--;
2527cc577a4SJonathan Peyton     task_stack -> ts_entries--;
2537cc577a4SJonathan Peyton 
2547cc577a4SJonathan Peyton     tied_task = * (task_stack -> ts_top );
2557cc577a4SJonathan Peyton 
2567cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( tied_task != NULL );
2577cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( tied_task -> td_flags.tasktype == TASK_TIED );
2587cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( tied_task == ending_task );  // If we built the stack correctly
2597cc577a4SJonathan Peyton 
2607cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_pop_task_stack(exit): GTID: %d; TASK: %p\n", gtid, tied_task ) );
2617cc577a4SJonathan Peyton     return;
2627cc577a4SJonathan Peyton }
2637cc577a4SJonathan Peyton #endif /* BUILD_TIED_TASK_STACK */
2647cc577a4SJonathan Peyton 
2657cc577a4SJonathan Peyton //---------------------------------------------------
2667cc577a4SJonathan Peyton //  __kmp_push_task: Add a task to the thread's deque
2677cc577a4SJonathan Peyton 
2687cc577a4SJonathan Peyton static kmp_int32
2697cc577a4SJonathan Peyton __kmp_push_task(kmp_int32 gtid, kmp_task_t * task )
2707cc577a4SJonathan Peyton {
2717cc577a4SJonathan Peyton     kmp_info_t *        thread = __kmp_threads[ gtid ];
2727cc577a4SJonathan Peyton     kmp_taskdata_t *    taskdata = KMP_TASK_TO_TASKDATA(task);
2737cc577a4SJonathan Peyton     kmp_task_team_t *   task_team = thread->th.th_task_team;
2747cc577a4SJonathan Peyton     kmp_int32           tid = __kmp_tid_from_gtid( gtid );
2757cc577a4SJonathan Peyton     kmp_thread_data_t * thread_data;
2767cc577a4SJonathan Peyton 
2777cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_push_task: T#%d trying to push task %p.\n", gtid, taskdata ) );
2787cc577a4SJonathan Peyton 
2797cc577a4SJonathan Peyton     if ( taskdata->td_flags.tiedness == TASK_UNTIED ) {
2807cc577a4SJonathan Peyton         // untied task needs to increment counter so that the task structure is not freed prematurely
2817cc577a4SJonathan Peyton         kmp_int32 counter = 1 + KMP_TEST_THEN_INC32(&taskdata->td_untied_count);
2827cc577a4SJonathan Peyton         KA_TRACE(20, ( "__kmp_push_task: T#%d untied_count (%d) incremented for task %p\n",
2837cc577a4SJonathan Peyton                        gtid, counter, taskdata ) );
2847cc577a4SJonathan Peyton     }
2857cc577a4SJonathan Peyton 
2867cc577a4SJonathan Peyton     // The first check avoids building task_team thread data if serialized
2877cc577a4SJonathan Peyton     if ( taskdata->td_flags.task_serial ) {
2887cc577a4SJonathan Peyton         KA_TRACE(20, ( "__kmp_push_task: T#%d team serialized; returning TASK_NOT_PUSHED for task %p\n",
2897cc577a4SJonathan Peyton                        gtid, taskdata ) );
2907cc577a4SJonathan Peyton         return TASK_NOT_PUSHED;
2917cc577a4SJonathan Peyton     }
2927cc577a4SJonathan Peyton 
2937cc577a4SJonathan Peyton     // Now that serialized tasks have returned, we can assume that we are not in immediate exec mode
2947cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( __kmp_tasking_mode != tskm_immediate_exec );
2957cc577a4SJonathan Peyton     if ( ! KMP_TASKING_ENABLED(task_team) ) {
2967cc577a4SJonathan Peyton          __kmp_enable_tasking( task_team, thread );
2977cc577a4SJonathan Peyton     }
2987cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( TCR_4(task_team -> tt.tt_found_tasks) == TRUE );
2997cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( TCR_PTR(task_team -> tt.tt_threads_data) != NULL );
3007cc577a4SJonathan Peyton 
3017cc577a4SJonathan Peyton     // Find tasking deque specific to encountering thread
3027cc577a4SJonathan Peyton     thread_data = & task_team -> tt.tt_threads_data[ tid ];
3037cc577a4SJonathan Peyton 
3047cc577a4SJonathan Peyton     // No lock needed since only owner can allocate
3057cc577a4SJonathan Peyton     if (thread_data -> td.td_deque == NULL ) {
3067cc577a4SJonathan Peyton         __kmp_alloc_task_deque( thread, thread_data );
3077cc577a4SJonathan Peyton     }
3087cc577a4SJonathan Peyton 
3097cc577a4SJonathan Peyton     // Check if deque is full
3107cc577a4SJonathan Peyton     if ( TCR_4(thread_data -> td.td_deque_ntasks) >= TASK_DEQUE_SIZE(thread_data->td) )
3117cc577a4SJonathan Peyton     {
3127cc577a4SJonathan Peyton         KA_TRACE(20, ( "__kmp_push_task: T#%d deque is full; returning TASK_NOT_PUSHED for task %p\n",
3137cc577a4SJonathan Peyton                        gtid, taskdata ) );
3147cc577a4SJonathan Peyton         return TASK_NOT_PUSHED;
3157cc577a4SJonathan Peyton     }
3167cc577a4SJonathan Peyton 
3177cc577a4SJonathan Peyton     // Lock the deque for the task push operation
3187cc577a4SJonathan Peyton     __kmp_acquire_bootstrap_lock( & thread_data -> td.td_deque_lock );
3197cc577a4SJonathan Peyton 
3207cc577a4SJonathan Peyton #if OMP_45_ENABLED
3217cc577a4SJonathan Peyton     // Need to recheck as we can get a proxy task from a thread outside of OpenMP
3227cc577a4SJonathan Peyton     if ( TCR_4(thread_data -> td.td_deque_ntasks) >= TASK_DEQUE_SIZE(thread_data->td) )
3237cc577a4SJonathan Peyton     {
3247cc577a4SJonathan Peyton         __kmp_release_bootstrap_lock( & thread_data -> td.td_deque_lock );
3257cc577a4SJonathan Peyton         KA_TRACE(20, ( "__kmp_push_task: T#%d deque is full on 2nd check; returning TASK_NOT_PUSHED for task %p\n",
3267cc577a4SJonathan Peyton                        gtid, taskdata ) );
3277cc577a4SJonathan Peyton         return TASK_NOT_PUSHED;
3287cc577a4SJonathan Peyton     }
3297cc577a4SJonathan Peyton #else
3307cc577a4SJonathan Peyton     // Must have room since no thread can add tasks but calling thread
3317cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( TCR_4(thread_data -> td.td_deque_ntasks) < TASK_DEQUE_SIZE(thread_data->td) );
3327cc577a4SJonathan Peyton #endif
3337cc577a4SJonathan Peyton 
3347cc577a4SJonathan Peyton     thread_data -> td.td_deque[ thread_data -> td.td_deque_tail ] = taskdata;  // Push taskdata
3357cc577a4SJonathan Peyton     // Wrap index.
3367cc577a4SJonathan Peyton     thread_data -> td.td_deque_tail = ( thread_data -> td.td_deque_tail + 1 ) & TASK_DEQUE_MASK(thread_data->td);
3377cc577a4SJonathan Peyton     TCW_4(thread_data -> td.td_deque_ntasks, TCR_4(thread_data -> td.td_deque_ntasks) + 1);             // Adjust task count
3387cc577a4SJonathan Peyton 
3397cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_push_task: T#%d returning TASK_SUCCESSFULLY_PUSHED: "
3407cc577a4SJonathan Peyton           "task=%p ntasks=%d head=%u tail=%u\n",
3417cc577a4SJonathan Peyton           gtid, taskdata, thread_data->td.td_deque_ntasks,
3427cc577a4SJonathan Peyton           thread_data->td.td_deque_head, thread_data->td.td_deque_tail) );
3437cc577a4SJonathan Peyton 
3447cc577a4SJonathan Peyton     __kmp_release_bootstrap_lock( & thread_data->td.td_deque_lock );
3457cc577a4SJonathan Peyton 
3467cc577a4SJonathan Peyton     return TASK_SUCCESSFULLY_PUSHED;
3477cc577a4SJonathan Peyton }
3487cc577a4SJonathan Peyton 
3497cc577a4SJonathan Peyton 
3507cc577a4SJonathan Peyton //-----------------------------------------------------------------------------------------
3517cc577a4SJonathan Peyton // __kmp_pop_current_task_from_thread: set up current task from called thread when team ends
3527cc577a4SJonathan Peyton // this_thr: thread structure to set current_task in.
3537cc577a4SJonathan Peyton 
3547cc577a4SJonathan Peyton void
3557cc577a4SJonathan Peyton __kmp_pop_current_task_from_thread( kmp_info_t *this_thr )
3567cc577a4SJonathan Peyton {
3577cc577a4SJonathan Peyton     KF_TRACE( 10, ("__kmp_pop_current_task_from_thread(enter): T#%d this_thread=%p, curtask=%p, "
3587cc577a4SJonathan Peyton                    "curtask_parent=%p\n",
3597cc577a4SJonathan Peyton                    0, this_thr, this_thr -> th.th_current_task,
3607cc577a4SJonathan Peyton                    this_thr -> th.th_current_task -> td_parent ) );
3617cc577a4SJonathan Peyton 
3627cc577a4SJonathan Peyton     this_thr -> th.th_current_task = this_thr -> th.th_current_task -> td_parent;
3637cc577a4SJonathan Peyton 
3647cc577a4SJonathan Peyton     KF_TRACE( 10, ("__kmp_pop_current_task_from_thread(exit): T#%d this_thread=%p, curtask=%p, "
3657cc577a4SJonathan Peyton                    "curtask_parent=%p\n",
3667cc577a4SJonathan Peyton                    0, this_thr, this_thr -> th.th_current_task,
3677cc577a4SJonathan Peyton                    this_thr -> th.th_current_task -> td_parent ) );
3687cc577a4SJonathan Peyton }
3697cc577a4SJonathan Peyton 
3707cc577a4SJonathan Peyton 
3717cc577a4SJonathan Peyton //---------------------------------------------------------------------------------------
3727cc577a4SJonathan Peyton // __kmp_push_current_task_to_thread: set up current task in called thread for a new team
3737cc577a4SJonathan Peyton // this_thr: thread structure to set up
3747cc577a4SJonathan Peyton // team: team for implicit task data
3757cc577a4SJonathan Peyton // tid: thread within team to set up
3767cc577a4SJonathan Peyton 
3777cc577a4SJonathan Peyton void
3787cc577a4SJonathan Peyton __kmp_push_current_task_to_thread( kmp_info_t *this_thr, kmp_team_t *team, int tid )
3797cc577a4SJonathan Peyton {
3807cc577a4SJonathan Peyton     // current task of the thread is a parent of the new just created implicit tasks of new team
3817cc577a4SJonathan Peyton     KF_TRACE( 10, ( "__kmp_push_current_task_to_thread(enter): T#%d this_thread=%p curtask=%p "
3827cc577a4SJonathan Peyton                     "parent_task=%p\n",
3837cc577a4SJonathan Peyton                     tid, this_thr, this_thr->th.th_current_task,
3847cc577a4SJonathan Peyton                     team->t.t_implicit_task_taskdata[tid].td_parent ) );
3857cc577a4SJonathan Peyton 
3867cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT (this_thr != NULL);
3877cc577a4SJonathan Peyton 
3887cc577a4SJonathan Peyton     if( tid == 0 ) {
3897cc577a4SJonathan Peyton         if( this_thr->th.th_current_task != & team -> t.t_implicit_task_taskdata[ 0 ] ) {
3907cc577a4SJonathan Peyton             team -> t.t_implicit_task_taskdata[ 0 ].td_parent = this_thr->th.th_current_task;
3917cc577a4SJonathan Peyton             this_thr->th.th_current_task = & team -> t.t_implicit_task_taskdata[ 0 ];
3927cc577a4SJonathan Peyton         }
3937cc577a4SJonathan Peyton     } else {
3947cc577a4SJonathan Peyton         team -> t.t_implicit_task_taskdata[ tid ].td_parent = team -> t.t_implicit_task_taskdata[ 0 ].td_parent;
3957cc577a4SJonathan Peyton         this_thr->th.th_current_task = & team -> t.t_implicit_task_taskdata[ tid ];
3967cc577a4SJonathan Peyton     }
3977cc577a4SJonathan Peyton 
3987cc577a4SJonathan Peyton     KF_TRACE( 10, ( "__kmp_push_current_task_to_thread(exit): T#%d this_thread=%p 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 
4057cc577a4SJonathan Peyton //----------------------------------------------------------------------
4067cc577a4SJonathan Peyton // __kmp_task_start: bookkeeping for a task starting execution
4077cc577a4SJonathan Peyton // GTID: global thread id of calling thread
4087cc577a4SJonathan Peyton // task: task starting execution
4097cc577a4SJonathan Peyton // current_task: task suspending
4107cc577a4SJonathan Peyton 
4117cc577a4SJonathan Peyton static void
4127cc577a4SJonathan Peyton __kmp_task_start( kmp_int32 gtid, kmp_task_t * task, kmp_taskdata_t * current_task )
4137cc577a4SJonathan Peyton {
4147cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(task);
4157cc577a4SJonathan Peyton     kmp_info_t * thread = __kmp_threads[ gtid ];
4167cc577a4SJonathan Peyton 
4177cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_task_start(enter): T#%d starting task %p: current_task=%p\n",
4187cc577a4SJonathan Peyton                   gtid, taskdata, current_task) );
4197cc577a4SJonathan Peyton 
4207cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.tasktype == TASK_EXPLICIT );
4217cc577a4SJonathan Peyton 
4227cc577a4SJonathan Peyton     // mark currently executing task as suspended
4237cc577a4SJonathan Peyton     // TODO: GEH - make sure root team implicit task is initialized properly.
4247cc577a4SJonathan Peyton     // KMP_DEBUG_ASSERT( current_task -> td_flags.executing == 1 );
4257cc577a4SJonathan Peyton     current_task -> td_flags.executing = 0;
4267cc577a4SJonathan Peyton 
4277cc577a4SJonathan Peyton     // Add task to stack if tied
4287cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK
4297cc577a4SJonathan Peyton     if ( taskdata -> td_flags.tiedness == TASK_TIED )
4307cc577a4SJonathan Peyton     {
4317cc577a4SJonathan Peyton         __kmp_push_task_stack( gtid, thread, taskdata );
4327cc577a4SJonathan Peyton     }
4337cc577a4SJonathan Peyton #endif /* BUILD_TIED_TASK_STACK */
4347cc577a4SJonathan Peyton 
4357cc577a4SJonathan Peyton     // mark starting task as executing and as current task
4367cc577a4SJonathan Peyton     thread -> th.th_current_task = taskdata;
4377cc577a4SJonathan Peyton 
4387cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_flags.started == 0 || taskdata->td_flags.tiedness == TASK_UNTIED );
4397cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_flags.executing == 0 || taskdata->td_flags.tiedness == TASK_UNTIED );
4407cc577a4SJonathan Peyton     taskdata -> td_flags.started = 1;
4417cc577a4SJonathan Peyton     taskdata -> td_flags.executing = 1;
4427cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.complete == 0 );
4437cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.freed == 0 );
4447cc577a4SJonathan Peyton 
4457cc577a4SJonathan Peyton     // GEH TODO: shouldn't we pass some sort of location identifier here?
4467cc577a4SJonathan Peyton     // APT: yes, we will pass location here.
4477cc577a4SJonathan Peyton     // need to store current thread state (in a thread or taskdata structure)
4487cc577a4SJonathan Peyton     // before setting work_state, otherwise wrong state is set after end of task
4497cc577a4SJonathan Peyton 
4507cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_task_start(exit): T#%d task=%p\n",
4517cc577a4SJonathan Peyton                   gtid, taskdata ) );
4527cc577a4SJonathan Peyton 
4537cc577a4SJonathan Peyton #if OMPT_SUPPORT
4547cc577a4SJonathan Peyton     if (ompt_enabled &&
4557cc577a4SJonathan Peyton         ompt_callbacks.ompt_callback(ompt_event_task_begin)) {
4567cc577a4SJonathan Peyton         kmp_taskdata_t *parent = taskdata->td_parent;
4577cc577a4SJonathan Peyton         ompt_callbacks.ompt_callback(ompt_event_task_begin)(
4587cc577a4SJonathan Peyton             parent ? parent->ompt_task_info.task_id : ompt_task_id_none,
4597cc577a4SJonathan Peyton             parent ? &(parent->ompt_task_info.frame) : NULL,
4607cc577a4SJonathan Peyton             taskdata->ompt_task_info.task_id,
4617cc577a4SJonathan Peyton             taskdata->ompt_task_info.function);
4627cc577a4SJonathan Peyton     }
4637cc577a4SJonathan Peyton #endif
4647cc577a4SJonathan Peyton #if OMP_40_ENABLED && OMPT_SUPPORT && OMPT_TRACE
4657cc577a4SJonathan Peyton     /* OMPT emit all dependences if requested by the tool */
4667cc577a4SJonathan Peyton     if (ompt_enabled && taskdata->ompt_task_info.ndeps > 0 &&
4677cc577a4SJonathan Peyton         ompt_callbacks.ompt_callback(ompt_event_task_dependences))
4687cc577a4SJonathan Peyton 	{
4697cc577a4SJonathan Peyton         ompt_callbacks.ompt_callback(ompt_event_task_dependences)(
4707cc577a4SJonathan Peyton             taskdata->ompt_task_info.task_id,
4717cc577a4SJonathan Peyton             taskdata->ompt_task_info.deps,
4727cc577a4SJonathan Peyton             taskdata->ompt_task_info.ndeps
4737cc577a4SJonathan Peyton         );
4747cc577a4SJonathan Peyton 		/* We can now free the allocated memory for the dependencies */
4757cc577a4SJonathan Peyton 		KMP_OMPT_DEPS_FREE (thread, taskdata->ompt_task_info.deps);
4767cc577a4SJonathan Peyton         taskdata->ompt_task_info.deps = NULL;
4777cc577a4SJonathan Peyton         taskdata->ompt_task_info.ndeps = 0;
4787cc577a4SJonathan Peyton     }
4797cc577a4SJonathan Peyton #endif /* OMP_40_ENABLED && OMPT_SUPPORT && OMPT_TRACE */
4807cc577a4SJonathan Peyton 
4817cc577a4SJonathan Peyton     return;
4827cc577a4SJonathan Peyton }
4837cc577a4SJonathan Peyton 
4847cc577a4SJonathan Peyton 
4857cc577a4SJonathan Peyton //----------------------------------------------------------------------
4867cc577a4SJonathan Peyton // __kmpc_omp_task_begin_if0: report that a given serialized task has started execution
4877cc577a4SJonathan Peyton // loc_ref: source location information; points to beginning of task block.
4887cc577a4SJonathan Peyton // gtid: global thread number.
4897cc577a4SJonathan Peyton // task: task thunk for the started task.
4907cc577a4SJonathan Peyton 
4917cc577a4SJonathan Peyton void
4927cc577a4SJonathan Peyton __kmpc_omp_task_begin_if0( ident_t *loc_ref, kmp_int32 gtid, kmp_task_t * task )
4937cc577a4SJonathan Peyton {
4947cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(task);
4957cc577a4SJonathan Peyton     kmp_taskdata_t * current_task = __kmp_threads[ gtid ] -> th.th_current_task;
4967cc577a4SJonathan Peyton 
4977cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_begin_if0(enter): T#%d loc=%p task=%p current_task=%p\n",
4987cc577a4SJonathan Peyton                   gtid, loc_ref, taskdata, current_task ) );
4997cc577a4SJonathan Peyton 
5007cc577a4SJonathan Peyton     if ( taskdata->td_flags.tiedness == TASK_UNTIED ) {
5017cc577a4SJonathan Peyton         // untied task needs to increment counter so that the task structure is not freed prematurely
5027cc577a4SJonathan Peyton         kmp_int32 counter = 1 + KMP_TEST_THEN_INC32(&taskdata->td_untied_count);
5037cc577a4SJonathan Peyton         KA_TRACE(20, ( "__kmpc_omp_task_begin_if0: T#%d untied_count (%d) incremented for task %p\n",
5047cc577a4SJonathan Peyton                        gtid, counter, taskdata ) );
5057cc577a4SJonathan Peyton     }
5067cc577a4SJonathan Peyton 
5077cc577a4SJonathan Peyton     taskdata -> td_flags.task_serial = 1;  // Execute this task immediately, not deferred.
5087cc577a4SJonathan Peyton     __kmp_task_start( gtid, task, current_task );
5097cc577a4SJonathan Peyton 
5107cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_begin_if0(exit): T#%d loc=%p task=%p,\n",
5117cc577a4SJonathan Peyton                   gtid, loc_ref, taskdata ) );
5127cc577a4SJonathan Peyton 
5137cc577a4SJonathan Peyton     return;
5147cc577a4SJonathan Peyton }
5157cc577a4SJonathan Peyton 
5167cc577a4SJonathan Peyton #ifdef TASK_UNUSED
5177cc577a4SJonathan Peyton //----------------------------------------------------------------------
5187cc577a4SJonathan Peyton // __kmpc_omp_task_begin: report that a given task has started execution
5197cc577a4SJonathan Peyton // NEVER GENERATED BY COMPILER, DEPRECATED!!!
5207cc577a4SJonathan Peyton 
5217cc577a4SJonathan Peyton void
5227cc577a4SJonathan Peyton __kmpc_omp_task_begin( ident_t *loc_ref, kmp_int32 gtid, kmp_task_t * task )
5237cc577a4SJonathan Peyton {
5247cc577a4SJonathan Peyton     kmp_taskdata_t * current_task = __kmp_threads[ gtid ] -> th.th_current_task;
5257cc577a4SJonathan Peyton 
5267cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_begin(enter): T#%d loc=%p task=%p current_task=%p\n",
5277cc577a4SJonathan Peyton                   gtid, loc_ref, KMP_TASK_TO_TASKDATA(task), current_task ) );
5287cc577a4SJonathan Peyton 
5297cc577a4SJonathan Peyton     __kmp_task_start( gtid, task, current_task );
5307cc577a4SJonathan Peyton 
5317cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_begin(exit): T#%d loc=%p task=%p,\n",
5327cc577a4SJonathan Peyton                   gtid, loc_ref, KMP_TASK_TO_TASKDATA(task) ) );
5337cc577a4SJonathan Peyton 
5347cc577a4SJonathan Peyton     return;
5357cc577a4SJonathan Peyton }
5367cc577a4SJonathan Peyton #endif // TASK_UNUSED
5377cc577a4SJonathan Peyton 
5387cc577a4SJonathan Peyton 
5397cc577a4SJonathan Peyton //-------------------------------------------------------------------------------------
5407cc577a4SJonathan Peyton // __kmp_free_task: free the current task space and the space for shareds
5417cc577a4SJonathan Peyton // gtid: Global thread ID of calling thread
5427cc577a4SJonathan Peyton // taskdata: task to free
5437cc577a4SJonathan Peyton // thread: thread data structure of caller
5447cc577a4SJonathan Peyton 
5457cc577a4SJonathan Peyton static void
5467cc577a4SJonathan Peyton __kmp_free_task( kmp_int32 gtid, kmp_taskdata_t * taskdata, kmp_info_t * thread )
5477cc577a4SJonathan Peyton {
5487cc577a4SJonathan Peyton     KA_TRACE(30, ("__kmp_free_task: T#%d freeing data from task %p\n",
5497cc577a4SJonathan Peyton                   gtid, taskdata) );
5507cc577a4SJonathan Peyton 
5517cc577a4SJonathan Peyton     // Check to make sure all flags and counters have the correct values
5527cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_flags.tasktype == TASK_EXPLICIT );
5537cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_flags.executing == 0 );
5547cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_flags.complete == 1 );
5557cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_flags.freed == 0 );
5567cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( TCR_4(taskdata->td_allocated_child_tasks) == 0  || taskdata->td_flags.task_serial == 1);
5577cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( TCR_4(taskdata->td_incomplete_child_tasks) == 0 );
5587cc577a4SJonathan Peyton 
5597cc577a4SJonathan Peyton     taskdata->td_flags.freed = 1;
5607cc577a4SJonathan Peyton     ANNOTATE_HAPPENS_BEFORE(taskdata);
5617cc577a4SJonathan Peyton     // deallocate the taskdata and shared variable blocks associated with this task
5627cc577a4SJonathan Peyton     #if USE_FAST_MEMORY
5637cc577a4SJonathan Peyton         __kmp_fast_free( thread, taskdata );
5647cc577a4SJonathan Peyton     #else /* ! USE_FAST_MEMORY */
5657cc577a4SJonathan Peyton         __kmp_thread_free( thread, taskdata );
5667cc577a4SJonathan Peyton     #endif
5677cc577a4SJonathan Peyton 
5687cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_free_task: T#%d freed task %p\n",
5697cc577a4SJonathan Peyton                   gtid, taskdata) );
5707cc577a4SJonathan Peyton }
5717cc577a4SJonathan Peyton 
5727cc577a4SJonathan Peyton //-------------------------------------------------------------------------------------
5737cc577a4SJonathan Peyton // __kmp_free_task_and_ancestors: free the current task and ancestors without children
5747cc577a4SJonathan Peyton //
5757cc577a4SJonathan Peyton // gtid: Global thread ID of calling thread
5767cc577a4SJonathan Peyton // taskdata: task to free
5777cc577a4SJonathan Peyton // thread: thread data structure of caller
5787cc577a4SJonathan Peyton 
5797cc577a4SJonathan Peyton static void
5807cc577a4SJonathan Peyton __kmp_free_task_and_ancestors( kmp_int32 gtid, kmp_taskdata_t * taskdata, kmp_info_t * thread )
5817cc577a4SJonathan Peyton {
5827cc577a4SJonathan Peyton #if OMP_45_ENABLED
5837cc577a4SJonathan Peyton     // Proxy tasks must always be allowed to free their parents
5847cc577a4SJonathan Peyton     // because they can be run in background even in serial mode.
5857cc577a4SJonathan Peyton     kmp_int32 team_serial = ( taskdata->td_flags.team_serial ||
5867cc577a4SJonathan Peyton         taskdata->td_flags.tasking_ser ) && !taskdata->td_flags.proxy;
5877cc577a4SJonathan Peyton #else
5887cc577a4SJonathan Peyton     kmp_int32 team_serial = taskdata->td_flags.team_serial ||
5897cc577a4SJonathan Peyton         taskdata->td_flags.tasking_ser;
5907cc577a4SJonathan Peyton #endif
5917cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.tasktype == TASK_EXPLICIT );
5927cc577a4SJonathan Peyton 
5937cc577a4SJonathan Peyton     kmp_int32 children = KMP_TEST_THEN_DEC32( (kmp_int32 *)(& taskdata -> td_allocated_child_tasks) ) - 1;
5947cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( children >= 0 );
5957cc577a4SJonathan Peyton 
5967cc577a4SJonathan Peyton     // Now, go up the ancestor tree to see if any ancestors can now be freed.
5977cc577a4SJonathan Peyton     while ( children == 0 )
5987cc577a4SJonathan Peyton     {
5997cc577a4SJonathan Peyton         kmp_taskdata_t * parent_taskdata = taskdata -> td_parent;
6007cc577a4SJonathan Peyton 
6017cc577a4SJonathan Peyton         KA_TRACE(20, ("__kmp_free_task_and_ancestors(enter): T#%d task %p complete "
6027cc577a4SJonathan Peyton                       "and freeing itself\n", gtid, taskdata) );
6037cc577a4SJonathan Peyton 
6047cc577a4SJonathan Peyton         // --- Deallocate my ancestor task ---
6057cc577a4SJonathan Peyton         __kmp_free_task( gtid, taskdata, thread );
6067cc577a4SJonathan Peyton 
6077cc577a4SJonathan Peyton         taskdata = parent_taskdata;
6087cc577a4SJonathan Peyton 
6097cc577a4SJonathan Peyton         // Stop checking ancestors at implicit task
6107cc577a4SJonathan Peyton         // instead of walking up ancestor tree to avoid premature deallocation of ancestors.
6117cc577a4SJonathan Peyton         if ( team_serial || taskdata -> td_flags.tasktype == TASK_IMPLICIT )
6127cc577a4SJonathan Peyton             return;
6137cc577a4SJonathan Peyton 
6147cc577a4SJonathan Peyton         // Predecrement simulated by "- 1" calculation
6157cc577a4SJonathan Peyton         children = KMP_TEST_THEN_DEC32( (kmp_int32 *)(& taskdata -> td_allocated_child_tasks) ) - 1;
6167cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( children >= 0 );
6177cc577a4SJonathan Peyton     }
6187cc577a4SJonathan Peyton 
6197cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_free_task_and_ancestors(exit): T#%d task %p has %d children; "
6207cc577a4SJonathan Peyton                   "not freeing it yet\n", gtid, taskdata, children) );
6217cc577a4SJonathan Peyton }
6227cc577a4SJonathan Peyton 
6237cc577a4SJonathan Peyton //---------------------------------------------------------------------
6247cc577a4SJonathan Peyton // __kmp_task_finish: bookkeeping to do when a task finishes execution
6257cc577a4SJonathan Peyton // gtid: global thread ID for calling thread
6267cc577a4SJonathan Peyton // task: task to be finished
6277cc577a4SJonathan Peyton // resumed_task: task to be resumed.  (may be NULL if task is serialized)
6287cc577a4SJonathan Peyton 
6297cc577a4SJonathan Peyton static void
6307cc577a4SJonathan Peyton __kmp_task_finish( kmp_int32 gtid, kmp_task_t *task, kmp_taskdata_t *resumed_task )
6317cc577a4SJonathan Peyton {
6327cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(task);
6337cc577a4SJonathan Peyton     kmp_info_t * thread = __kmp_threads[ gtid ];
6347cc577a4SJonathan Peyton     kmp_task_team_t * task_team = thread->th.th_task_team; // might be NULL for serial teams...
6357cc577a4SJonathan Peyton     kmp_int32 children = 0;
6367cc577a4SJonathan Peyton 
6377cc577a4SJonathan Peyton #if OMPT_SUPPORT
6387cc577a4SJonathan Peyton     if (ompt_enabled &&
6397cc577a4SJonathan Peyton         ompt_callbacks.ompt_callback(ompt_event_task_end)) {
6407cc577a4SJonathan Peyton         kmp_taskdata_t *parent = taskdata->td_parent;
6417cc577a4SJonathan Peyton         ompt_callbacks.ompt_callback(ompt_event_task_end)(
6427cc577a4SJonathan Peyton             taskdata->ompt_task_info.task_id);
6437cc577a4SJonathan Peyton     }
6447cc577a4SJonathan Peyton #endif
6457cc577a4SJonathan Peyton 
6467cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_task_finish(enter): T#%d finishing task %p and resuming task %p\n",
6477cc577a4SJonathan Peyton                   gtid, taskdata, resumed_task) );
6487cc577a4SJonathan Peyton 
6497cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.tasktype == TASK_EXPLICIT );
6507cc577a4SJonathan Peyton 
6517cc577a4SJonathan Peyton     // Pop task from stack if tied
6527cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK
6537cc577a4SJonathan Peyton     if ( taskdata -> td_flags.tiedness == TASK_TIED )
6547cc577a4SJonathan Peyton     {
6557cc577a4SJonathan Peyton         __kmp_pop_task_stack( gtid, thread, taskdata );
6567cc577a4SJonathan Peyton     }
6577cc577a4SJonathan Peyton #endif /* BUILD_TIED_TASK_STACK */
6587cc577a4SJonathan Peyton 
6597cc577a4SJonathan Peyton     if ( taskdata->td_flags.tiedness == TASK_UNTIED ) {
6607cc577a4SJonathan Peyton         // untied task needs to check the counter so that the task structure is not freed prematurely
6617cc577a4SJonathan Peyton         kmp_int32 counter = KMP_TEST_THEN_DEC32(&taskdata->td_untied_count) - 1;
6627cc577a4SJonathan Peyton         KA_TRACE(20, ( "__kmp_task_finish: T#%d untied_count (%d) decremented for task %p\n",
6637cc577a4SJonathan Peyton                        gtid, counter, taskdata ) );
6647cc577a4SJonathan Peyton         if ( counter > 0 ) {
6657cc577a4SJonathan Peyton             // untied task is not done, to be continued possibly by other thread, do not free it now
6667cc577a4SJonathan Peyton             if (resumed_task == NULL) {
6677cc577a4SJonathan Peyton                 KMP_DEBUG_ASSERT( taskdata->td_flags.task_serial );
6687cc577a4SJonathan Peyton                 resumed_task = taskdata->td_parent;  // In a serialized task, the resumed task is the parent
6697cc577a4SJonathan Peyton             }
6707cc577a4SJonathan Peyton             thread->th.th_current_task = resumed_task; // restore current_task
6717cc577a4SJonathan Peyton             resumed_task->td_flags.executing = 1;  // resume previous task
6727cc577a4SJonathan Peyton             KA_TRACE(10, ("__kmp_task_finish(exit): T#%d partially done task %p, resuming task %p\n",
6737cc577a4SJonathan Peyton                           gtid, taskdata, resumed_task) );
6747cc577a4SJonathan Peyton             return;
6757cc577a4SJonathan Peyton         }
6767cc577a4SJonathan Peyton     }
6777cc577a4SJonathan Peyton 
6787cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.complete == 0 );
6797cc577a4SJonathan Peyton     taskdata -> td_flags.complete = 1;   // mark the task as completed
6807cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.started == 1 );
6817cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.freed == 0 );
6827cc577a4SJonathan Peyton 
6837cc577a4SJonathan Peyton     // Only need to keep track of count if team parallel and tasking not serialized
6847cc577a4SJonathan Peyton     if ( !( taskdata -> td_flags.team_serial || taskdata -> td_flags.tasking_ser ) ) {
6857cc577a4SJonathan Peyton         // Predecrement simulated by "- 1" calculation
6867cc577a4SJonathan Peyton         children = KMP_TEST_THEN_DEC32( (kmp_int32 *)(& taskdata -> td_parent -> td_incomplete_child_tasks) ) - 1;
6877cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( children >= 0 );
6887cc577a4SJonathan Peyton #if OMP_40_ENABLED
6897cc577a4SJonathan Peyton         if ( taskdata->td_taskgroup )
6907cc577a4SJonathan Peyton             KMP_TEST_THEN_DEC32( (kmp_int32 *)(& taskdata->td_taskgroup->count) );
6917cc577a4SJonathan Peyton #if OMP_45_ENABLED
6927cc577a4SJonathan Peyton     }
6937cc577a4SJonathan Peyton     // if we found proxy tasks there could exist a dependency chain
6947cc577a4SJonathan Peyton     // with the proxy task as origin
6957cc577a4SJonathan Peyton     if ( !( taskdata -> td_flags.team_serial || taskdata -> td_flags.tasking_ser ) || (task_team && task_team->tt.tt_found_proxy_tasks) ) {
6967cc577a4SJonathan Peyton #endif
6977cc577a4SJonathan Peyton         __kmp_release_deps(gtid,taskdata);
6987cc577a4SJonathan Peyton #endif
6997cc577a4SJonathan Peyton     }
7007cc577a4SJonathan Peyton 
7017cc577a4SJonathan Peyton     // td_flags.executing  must be marked as 0 after __kmp_release_deps has been called
7027cc577a4SJonathan Peyton     // Othertwise, if a task is executed immediately from the release_deps code
7037cc577a4SJonathan Peyton     // the flag will be reset to 1 again by this same function
7047cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.executing == 1 );
7057cc577a4SJonathan Peyton     taskdata -> td_flags.executing = 0;  // suspend the finishing task
7067cc577a4SJonathan Peyton 
7077cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_task_finish: T#%d finished task %p, %d incomplete children\n",
7087cc577a4SJonathan Peyton                   gtid, taskdata, children) );
7097cc577a4SJonathan Peyton 
7107cc577a4SJonathan Peyton #if OMP_40_ENABLED
7117cc577a4SJonathan Peyton     /* If the tasks' destructor thunk flag has been set, we need to invoke the
7127cc577a4SJonathan Peyton        destructor thunk that has been generated by the compiler.
7137cc577a4SJonathan Peyton        The code is placed here, since at this point other tasks might have been released
7147cc577a4SJonathan Peyton        hence overlapping the destructor invokations with some other work in the
7157cc577a4SJonathan Peyton        released tasks.  The OpenMP spec is not specific on when the destructors are
7167cc577a4SJonathan Peyton        invoked, so we should be free to choose.
7177cc577a4SJonathan Peyton     */
7187cc577a4SJonathan Peyton     if (taskdata->td_flags.destructors_thunk) {
7197cc577a4SJonathan Peyton         kmp_routine_entry_t destr_thunk = task->data1.destructors;
7207cc577a4SJonathan Peyton         KMP_ASSERT(destr_thunk);
7217cc577a4SJonathan Peyton         destr_thunk(gtid, task);
7227cc577a4SJonathan Peyton     }
7237cc577a4SJonathan Peyton #endif // OMP_40_ENABLED
7247cc577a4SJonathan Peyton 
7257cc577a4SJonathan Peyton     // bookkeeping for resuming task:
7267cc577a4SJonathan Peyton     // GEH - note tasking_ser => task_serial
7277cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( (taskdata->td_flags.tasking_ser || taskdata->td_flags.task_serial) ==
7287cc577a4SJonathan Peyton                        taskdata->td_flags.task_serial);
7297cc577a4SJonathan Peyton     if ( taskdata->td_flags.task_serial )
7307cc577a4SJonathan Peyton     {
7317cc577a4SJonathan Peyton         if (resumed_task == NULL) {
7327cc577a4SJonathan Peyton             resumed_task = taskdata->td_parent;  // In a serialized task, the resumed task is the parent
7337cc577a4SJonathan Peyton         }
7347cc577a4SJonathan Peyton         else
7357cc577a4SJonathan Peyton #if OMP_45_ENABLED
7367cc577a4SJonathan Peyton              if ( !(task_team && task_team->tt.tt_found_proxy_tasks) )
7377cc577a4SJonathan Peyton #endif
7387cc577a4SJonathan Peyton         {
7397cc577a4SJonathan Peyton             // verify resumed task passed in points to parent
7407cc577a4SJonathan Peyton             KMP_DEBUG_ASSERT( resumed_task == taskdata->td_parent );
7417cc577a4SJonathan Peyton         }
7427cc577a4SJonathan Peyton     }
7437cc577a4SJonathan Peyton     else {
7447cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( resumed_task != NULL );        // verify that resumed task is passed as arguemnt
7457cc577a4SJonathan Peyton     }
7467cc577a4SJonathan Peyton 
7477cc577a4SJonathan Peyton     // Free this task and then ancestor tasks if they have no children.
7487cc577a4SJonathan Peyton     // Restore th_current_task first as suggested by John:
7497cc577a4SJonathan Peyton     // johnmc: if an asynchronous inquiry peers into the runtime system
7507cc577a4SJonathan Peyton     // it doesn't see the freed task as the current task.
7517cc577a4SJonathan Peyton     thread->th.th_current_task = resumed_task;
7527cc577a4SJonathan Peyton     __kmp_free_task_and_ancestors(gtid, taskdata, thread);
7537cc577a4SJonathan Peyton 
7547cc577a4SJonathan Peyton     // TODO: GEH - make sure root team implicit task is initialized properly.
7557cc577a4SJonathan Peyton     // KMP_DEBUG_ASSERT( resumed_task->td_flags.executing == 0 );
7567cc577a4SJonathan Peyton     resumed_task->td_flags.executing = 1;  // resume previous task
7577cc577a4SJonathan Peyton 
7587cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_task_finish(exit): T#%d finished task %p, resuming task %p\n",
7597cc577a4SJonathan Peyton                   gtid, taskdata, resumed_task) );
7607cc577a4SJonathan Peyton 
7617cc577a4SJonathan Peyton     return;
7627cc577a4SJonathan Peyton }
7637cc577a4SJonathan Peyton 
7647cc577a4SJonathan Peyton //---------------------------------------------------------------------
7657cc577a4SJonathan Peyton // __kmpc_omp_task_complete_if0: report that a task has completed execution
7667cc577a4SJonathan Peyton // loc_ref: source location information; points to end of task block.
7677cc577a4SJonathan Peyton // gtid: global thread number.
7687cc577a4SJonathan Peyton // task: task thunk for the completed task.
7697cc577a4SJonathan Peyton 
7707cc577a4SJonathan Peyton void
7717cc577a4SJonathan Peyton __kmpc_omp_task_complete_if0( ident_t *loc_ref, kmp_int32 gtid, kmp_task_t *task )
7727cc577a4SJonathan Peyton {
7737cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_complete_if0(enter): T#%d loc=%p task=%p\n",
7747cc577a4SJonathan Peyton                   gtid, loc_ref, KMP_TASK_TO_TASKDATA(task) ) );
7757cc577a4SJonathan Peyton 
7767cc577a4SJonathan Peyton     __kmp_task_finish( gtid, task, NULL );  // this routine will provide task to resume
7777cc577a4SJonathan Peyton 
7787cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_complete_if0(exit): T#%d loc=%p task=%p\n",
7797cc577a4SJonathan Peyton                   gtid, loc_ref, KMP_TASK_TO_TASKDATA(task) ) );
7807cc577a4SJonathan Peyton 
7817cc577a4SJonathan Peyton     return;
7827cc577a4SJonathan Peyton }
7837cc577a4SJonathan Peyton 
7847cc577a4SJonathan Peyton #ifdef TASK_UNUSED
7857cc577a4SJonathan Peyton //---------------------------------------------------------------------
7867cc577a4SJonathan Peyton // __kmpc_omp_task_complete: report that a task has completed execution
7877cc577a4SJonathan Peyton // NEVER GENERATED BY COMPILER, DEPRECATED!!!
7887cc577a4SJonathan Peyton 
7897cc577a4SJonathan Peyton void
7907cc577a4SJonathan Peyton __kmpc_omp_task_complete( ident_t *loc_ref, kmp_int32 gtid, kmp_task_t *task )
7917cc577a4SJonathan Peyton {
7927cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_complete(enter): T#%d loc=%p task=%p\n",
7937cc577a4SJonathan Peyton                   gtid, loc_ref, KMP_TASK_TO_TASKDATA(task) ) );
7947cc577a4SJonathan Peyton 
7957cc577a4SJonathan Peyton     __kmp_task_finish( gtid, task, NULL );  // Not sure how to find task to resume
7967cc577a4SJonathan Peyton 
7977cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_complete(exit): T#%d loc=%p task=%p\n",
7987cc577a4SJonathan Peyton                   gtid, loc_ref, KMP_TASK_TO_TASKDATA(task) ) );
7997cc577a4SJonathan Peyton     return;
8007cc577a4SJonathan Peyton }
8017cc577a4SJonathan Peyton #endif // TASK_UNUSED
8027cc577a4SJonathan Peyton 
8037cc577a4SJonathan Peyton 
8047cc577a4SJonathan Peyton #if OMPT_SUPPORT
8057cc577a4SJonathan Peyton //----------------------------------------------------------------------------------------------------
8067cc577a4SJonathan Peyton // __kmp_task_init_ompt:
8077cc577a4SJonathan Peyton //   Initialize OMPT fields maintained by a task. This will only be called after
8087cc577a4SJonathan Peyton //   ompt_tool, so we already know whether ompt is enabled or not.
8097cc577a4SJonathan Peyton 
8107cc577a4SJonathan Peyton static inline void
8117cc577a4SJonathan Peyton __kmp_task_init_ompt( kmp_taskdata_t * task, int tid, void * function )
8127cc577a4SJonathan Peyton {
8137cc577a4SJonathan Peyton     if (ompt_enabled) {
8147cc577a4SJonathan Peyton         task->ompt_task_info.task_id = __ompt_task_id_new(tid);
8157cc577a4SJonathan Peyton         task->ompt_task_info.function = function;
8167cc577a4SJonathan Peyton         task->ompt_task_info.frame.exit_runtime_frame = NULL;
8177cc577a4SJonathan Peyton         task->ompt_task_info.frame.reenter_runtime_frame = NULL;
8187cc577a4SJonathan Peyton #if OMP_40_ENABLED
8197cc577a4SJonathan Peyton         task->ompt_task_info.ndeps = 0;
8207cc577a4SJonathan Peyton         task->ompt_task_info.deps = NULL;
8217cc577a4SJonathan Peyton #endif /* OMP_40_ENABLED */
8227cc577a4SJonathan Peyton     }
8237cc577a4SJonathan Peyton }
8247cc577a4SJonathan Peyton #endif
8257cc577a4SJonathan Peyton 
8267cc577a4SJonathan Peyton 
8277cc577a4SJonathan Peyton //----------------------------------------------------------------------------------------------------
8287cc577a4SJonathan Peyton // __kmp_init_implicit_task: Initialize the appropriate fields in the implicit task for a given thread
8297cc577a4SJonathan Peyton //
8307cc577a4SJonathan Peyton // loc_ref:  reference to source location of parallel region
8317cc577a4SJonathan Peyton // this_thr:  thread data structure corresponding to implicit task
8327cc577a4SJonathan Peyton // team: team for this_thr
8337cc577a4SJonathan Peyton // tid: thread id of given thread within team
8347cc577a4SJonathan Peyton // set_curr_task: TRUE if need to push current task to thread
8357cc577a4SJonathan Peyton // NOTE: Routine does not set up the implicit task ICVS.  This is assumed to have already been done elsewhere.
8367cc577a4SJonathan Peyton // TODO: Get better loc_ref.  Value passed in may be NULL
8377cc577a4SJonathan Peyton 
8387cc577a4SJonathan Peyton void
8397cc577a4SJonathan Peyton __kmp_init_implicit_task( ident_t *loc_ref, kmp_info_t *this_thr, kmp_team_t *team, int tid, int set_curr_task )
8407cc577a4SJonathan Peyton {
8417cc577a4SJonathan Peyton     kmp_taskdata_t * task   = & team->t.t_implicit_task_taskdata[ tid ];
8427cc577a4SJonathan Peyton 
8437cc577a4SJonathan Peyton     KF_TRACE(10, ("__kmp_init_implicit_task(enter): T#:%d team=%p task=%p, reinit=%s\n",
8447cc577a4SJonathan Peyton                   tid, team, task, set_curr_task ? "TRUE" : "FALSE" ) );
8457cc577a4SJonathan Peyton 
8467cc577a4SJonathan Peyton     task->td_task_id  = KMP_GEN_TASK_ID();
8477cc577a4SJonathan Peyton     task->td_team     = team;
8487cc577a4SJonathan Peyton //    task->td_parent   = NULL;  // fix for CQ230101 (broken parent task info in debugger)
8497cc577a4SJonathan Peyton     task->td_ident    = loc_ref;
8507cc577a4SJonathan Peyton     task->td_taskwait_ident   = NULL;
8517cc577a4SJonathan Peyton     task->td_taskwait_counter = 0;
8527cc577a4SJonathan Peyton     task->td_taskwait_thread  = 0;
8537cc577a4SJonathan Peyton 
8547cc577a4SJonathan Peyton     task->td_flags.tiedness    = TASK_TIED;
8557cc577a4SJonathan Peyton     task->td_flags.tasktype    = TASK_IMPLICIT;
8567cc577a4SJonathan Peyton #if OMP_45_ENABLED
8577cc577a4SJonathan Peyton     task->td_flags.proxy       = TASK_FULL;
8587cc577a4SJonathan Peyton #endif
8597cc577a4SJonathan Peyton 
8607cc577a4SJonathan Peyton     // All implicit tasks are executed immediately, not deferred
8617cc577a4SJonathan Peyton     task->td_flags.task_serial = 1;
8627cc577a4SJonathan Peyton     task->td_flags.tasking_ser = ( __kmp_tasking_mode == tskm_immediate_exec );
8637cc577a4SJonathan Peyton     task->td_flags.team_serial = ( team->t.t_serialized ) ? 1 : 0;
8647cc577a4SJonathan Peyton 
8657cc577a4SJonathan Peyton     task->td_flags.started     = 1;
8667cc577a4SJonathan Peyton     task->td_flags.executing   = 1;
8677cc577a4SJonathan Peyton     task->td_flags.complete    = 0;
8687cc577a4SJonathan Peyton     task->td_flags.freed       = 0;
8697cc577a4SJonathan Peyton 
8707cc577a4SJonathan Peyton #if OMP_40_ENABLED
8717cc577a4SJonathan Peyton     task->td_depnode = NULL;
8727cc577a4SJonathan Peyton #endif
8737cc577a4SJonathan Peyton 
8747cc577a4SJonathan Peyton     if (set_curr_task) {  // only do this initialization the first time a thread is created
8757cc577a4SJonathan Peyton         task->td_incomplete_child_tasks = 0;
8767cc577a4SJonathan Peyton         task->td_allocated_child_tasks  = 0; // Not used because do not need to deallocate implicit task
8777cc577a4SJonathan Peyton #if OMP_40_ENABLED
8787cc577a4SJonathan Peyton         task->td_taskgroup = NULL;           // An implicit task does not have taskgroup
8797cc577a4SJonathan Peyton         task->td_dephash = NULL;
8807cc577a4SJonathan Peyton #endif
8817cc577a4SJonathan Peyton         __kmp_push_current_task_to_thread( this_thr, team, tid );
8827cc577a4SJonathan Peyton     } else {
8837cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT(task->td_incomplete_child_tasks == 0);
8847cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT(task->td_allocated_child_tasks  == 0);
8857cc577a4SJonathan Peyton     }
8867cc577a4SJonathan Peyton 
8877cc577a4SJonathan Peyton #if OMPT_SUPPORT
8887cc577a4SJonathan Peyton     __kmp_task_init_ompt(task, tid, NULL);
8897cc577a4SJonathan Peyton #endif
8907cc577a4SJonathan Peyton 
8917cc577a4SJonathan Peyton     KF_TRACE(10, ("__kmp_init_implicit_task(exit): T#:%d team=%p task=%p\n",
8927cc577a4SJonathan Peyton                   tid, team, task ) );
8937cc577a4SJonathan Peyton }
8947cc577a4SJonathan Peyton 
8957cc577a4SJonathan Peyton 
8967cc577a4SJonathan Peyton //-----------------------------------------------------------------------------
8977cc577a4SJonathan Peyton //// __kmp_finish_implicit_task: Release resources associated to implicit tasks
8987cc577a4SJonathan Peyton //// at the end of parallel regions. Some resources are kept for reuse in the
8997cc577a4SJonathan Peyton //// next parallel region.
9007cc577a4SJonathan Peyton ////
9017cc577a4SJonathan Peyton //// thread:  thread data structure corresponding to implicit task
9027cc577a4SJonathan Peyton //
9037cc577a4SJonathan Peyton void
9047cc577a4SJonathan Peyton __kmp_finish_implicit_task(kmp_info_t *thread)
9057cc577a4SJonathan Peyton {
9067cc577a4SJonathan Peyton     kmp_taskdata_t *task = thread->th.th_current_task;
9077cc577a4SJonathan Peyton     if (task->td_dephash)
9087cc577a4SJonathan Peyton         __kmp_dephash_free_entries(thread, task->td_dephash);
9097cc577a4SJonathan Peyton }
9107cc577a4SJonathan Peyton 
9117cc577a4SJonathan Peyton 
9127cc577a4SJonathan Peyton //-----------------------------------------------------------------------------
9137cc577a4SJonathan Peyton //// __kmp_free_implicit_task: Release resources associated to implicit tasks
9147cc577a4SJonathan Peyton //// when these are destroyed regions
9157cc577a4SJonathan Peyton ////
9167cc577a4SJonathan Peyton //// thread:  thread data structure corresponding to implicit task
9177cc577a4SJonathan Peyton //
9187cc577a4SJonathan Peyton void
9197cc577a4SJonathan Peyton __kmp_free_implicit_task(kmp_info_t *thread)
9207cc577a4SJonathan Peyton {
9217cc577a4SJonathan Peyton     kmp_taskdata_t *task = thread->th.th_current_task;
9227cc577a4SJonathan Peyton     if (task->td_dephash)
9237cc577a4SJonathan Peyton         __kmp_dephash_free(thread, task->td_dephash);
9247cc577a4SJonathan Peyton     task->td_dephash = NULL;
9257cc577a4SJonathan Peyton }
9267cc577a4SJonathan Peyton 
9277cc577a4SJonathan Peyton 
9287cc577a4SJonathan Peyton // Round up a size to a power of two specified by val
9297cc577a4SJonathan Peyton // Used to insert padding between structures co-allocated using a single malloc() call
9307cc577a4SJonathan Peyton static size_t
9317cc577a4SJonathan Peyton __kmp_round_up_to_val( size_t size, size_t val ) {
9327cc577a4SJonathan Peyton     if ( size & ( val - 1 ) ) {
9337cc577a4SJonathan Peyton         size &= ~ ( val - 1 );
9347cc577a4SJonathan Peyton         if ( size <= KMP_SIZE_T_MAX - val ) {
9357cc577a4SJonathan Peyton             size += val;    // Round up if there is no overflow.
9367cc577a4SJonathan Peyton         }; // if
9377cc577a4SJonathan Peyton     }; // if
9387cc577a4SJonathan Peyton     return size;
9397cc577a4SJonathan Peyton } // __kmp_round_up_to_va
9407cc577a4SJonathan Peyton 
9417cc577a4SJonathan Peyton 
9427cc577a4SJonathan Peyton //---------------------------------------------------------------------------------
9437cc577a4SJonathan Peyton // __kmp_task_alloc: Allocate the taskdata and task data structures for a task
9447cc577a4SJonathan Peyton //
9457cc577a4SJonathan Peyton // loc_ref: source location information
9467cc577a4SJonathan Peyton // gtid: global thread number.
9477cc577a4SJonathan Peyton // flags: include tiedness & task type (explicit vs. implicit) of the ''new'' task encountered.
9487cc577a4SJonathan Peyton //        Converted from kmp_int32 to kmp_tasking_flags_t in routine.
9497cc577a4SJonathan Peyton // sizeof_kmp_task_t:  Size in bytes of kmp_task_t data structure including private vars accessed in task.
9507cc577a4SJonathan Peyton // sizeof_shareds:  Size in bytes of array of pointers to shared vars accessed in task.
9517cc577a4SJonathan Peyton // task_entry: Pointer to task code entry point generated by compiler.
9527cc577a4SJonathan Peyton // returns: a pointer to the allocated kmp_task_t structure (task).
9537cc577a4SJonathan Peyton 
9547cc577a4SJonathan Peyton kmp_task_t *
9557cc577a4SJonathan Peyton __kmp_task_alloc( ident_t *loc_ref, kmp_int32 gtid, kmp_tasking_flags_t *flags,
9567cc577a4SJonathan Peyton                   size_t sizeof_kmp_task_t, size_t sizeof_shareds,
9577cc577a4SJonathan Peyton                   kmp_routine_entry_t task_entry )
9587cc577a4SJonathan Peyton {
9597cc577a4SJonathan Peyton     kmp_task_t *task;
9607cc577a4SJonathan Peyton     kmp_taskdata_t *taskdata;
9617cc577a4SJonathan Peyton     kmp_info_t *thread = __kmp_threads[ gtid ];
9627cc577a4SJonathan Peyton     kmp_team_t *team = thread->th.th_team;
9637cc577a4SJonathan Peyton     kmp_taskdata_t *parent_task = thread->th.th_current_task;
9647cc577a4SJonathan Peyton     size_t shareds_offset;
9657cc577a4SJonathan Peyton 
9667cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_task_alloc(enter): T#%d loc=%p, flags=(0x%x) "
9677cc577a4SJonathan Peyton                   "sizeof_task=%ld sizeof_shared=%ld entry=%p\n",
9687cc577a4SJonathan Peyton                   gtid, loc_ref, *((kmp_int32 *)flags), sizeof_kmp_task_t,
9697cc577a4SJonathan Peyton                   sizeof_shareds, task_entry) );
9707cc577a4SJonathan Peyton 
9717cc577a4SJonathan Peyton     if ( parent_task->td_flags.final ) {
9727cc577a4SJonathan Peyton         if (flags->merged_if0) {
9737cc577a4SJonathan Peyton         }
9747cc577a4SJonathan Peyton         flags->final = 1;
9757cc577a4SJonathan Peyton     }
9767cc577a4SJonathan Peyton 
9777cc577a4SJonathan Peyton #if OMP_45_ENABLED
9787cc577a4SJonathan Peyton     if ( flags->proxy == TASK_PROXY ) {
9797cc577a4SJonathan Peyton         flags->tiedness = TASK_UNTIED;
9807cc577a4SJonathan Peyton         flags->merged_if0 = 1;
9817cc577a4SJonathan Peyton 
9827cc577a4SJonathan Peyton         /* are we running in a sequential parallel or tskm_immediate_exec... we need tasking support enabled */
9837cc577a4SJonathan Peyton         if ( (thread->th.th_task_team) == NULL ) {
9847cc577a4SJonathan Peyton             /* This should only happen if the team is serialized
9857cc577a4SJonathan Peyton                 setup a task team and propagate it to the thread
9867cc577a4SJonathan Peyton             */
9877cc577a4SJonathan Peyton             KMP_DEBUG_ASSERT(team->t.t_serialized);
9887cc577a4SJonathan Peyton             KA_TRACE(30,("T#%d creating task team in __kmp_task_alloc for proxy task\n", gtid));
9897cc577a4SJonathan Peyton             __kmp_task_team_setup(thread,team,1); // 1 indicates setup the current team regardless of nthreads
9907cc577a4SJonathan Peyton             thread->th.th_task_team = team->t.t_task_team[thread->th.th_task_state];
9917cc577a4SJonathan Peyton         }
9927cc577a4SJonathan Peyton         kmp_task_team_t * task_team = thread->th.th_task_team;
9937cc577a4SJonathan Peyton 
9947cc577a4SJonathan Peyton         /* tasking must be enabled now as the task might not be pushed */
9957cc577a4SJonathan Peyton         if ( !KMP_TASKING_ENABLED( task_team ) ) {
9967cc577a4SJonathan Peyton             KA_TRACE(30,("T#%d enabling tasking in __kmp_task_alloc for proxy task\n", gtid));
9977cc577a4SJonathan Peyton             __kmp_enable_tasking( task_team, thread );
9987cc577a4SJonathan Peyton             kmp_int32 tid = thread->th.th_info.ds.ds_tid;
9997cc577a4SJonathan Peyton             kmp_thread_data_t * thread_data = & task_team -> tt.tt_threads_data[ tid ];
10007cc577a4SJonathan Peyton             // No lock needed since only owner can allocate
10017cc577a4SJonathan Peyton             if (thread_data -> td.td_deque == NULL ) {
10027cc577a4SJonathan Peyton                 __kmp_alloc_task_deque( thread, thread_data );
10037cc577a4SJonathan Peyton             }
10047cc577a4SJonathan Peyton         }
10057cc577a4SJonathan Peyton 
10067cc577a4SJonathan Peyton         if ( task_team->tt.tt_found_proxy_tasks == FALSE )
10077cc577a4SJonathan Peyton           TCW_4(task_team -> tt.tt_found_proxy_tasks, TRUE);
10087cc577a4SJonathan Peyton     }
10097cc577a4SJonathan Peyton #endif
10107cc577a4SJonathan Peyton 
10117cc577a4SJonathan Peyton     // Calculate shared structure offset including padding after kmp_task_t struct
10127cc577a4SJonathan Peyton     // to align pointers in shared struct
10137cc577a4SJonathan Peyton     shareds_offset = sizeof( kmp_taskdata_t ) + sizeof_kmp_task_t;
10147cc577a4SJonathan Peyton     shareds_offset = __kmp_round_up_to_val( shareds_offset, sizeof( void * ));
10157cc577a4SJonathan Peyton 
10167cc577a4SJonathan Peyton     // Allocate a kmp_taskdata_t block and a kmp_task_t block.
10177cc577a4SJonathan Peyton     KA_TRACE(30, ("__kmp_task_alloc: T#%d First malloc size: %ld\n",
10187cc577a4SJonathan Peyton                   gtid, shareds_offset) );
10197cc577a4SJonathan Peyton     KA_TRACE(30, ("__kmp_task_alloc: T#%d Second malloc size: %ld\n",
10207cc577a4SJonathan Peyton                   gtid, sizeof_shareds) );
10217cc577a4SJonathan Peyton 
10227cc577a4SJonathan Peyton     // Avoid double allocation here by combining shareds with taskdata
10237cc577a4SJonathan Peyton     #if USE_FAST_MEMORY
10247cc577a4SJonathan Peyton     taskdata = (kmp_taskdata_t *) __kmp_fast_allocate( thread, shareds_offset + sizeof_shareds );
10257cc577a4SJonathan Peyton     #else /* ! USE_FAST_MEMORY */
10267cc577a4SJonathan Peyton     taskdata = (kmp_taskdata_t *) __kmp_thread_malloc( thread, shareds_offset + sizeof_shareds );
10277cc577a4SJonathan Peyton     #endif /* USE_FAST_MEMORY */
10287cc577a4SJonathan Peyton     ANNOTATE_HAPPENS_AFTER(taskdata);
10297cc577a4SJonathan Peyton 
10307cc577a4SJonathan Peyton     task                      = KMP_TASKDATA_TO_TASK(taskdata);
10317cc577a4SJonathan Peyton 
10327cc577a4SJonathan Peyton     // Make sure task & taskdata are aligned appropriately
10337cc577a4SJonathan Peyton #if KMP_ARCH_X86 || KMP_ARCH_PPC64 || !KMP_HAVE_QUAD
10347cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( ( ((kmp_uintptr_t)taskdata) & (sizeof(double)-1) ) == 0 );
10357cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( ( ((kmp_uintptr_t)task) & (sizeof(double)-1) ) == 0 );
10367cc577a4SJonathan Peyton #else
10377cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( ( ((kmp_uintptr_t)taskdata) & (sizeof(_Quad)-1) ) == 0 );
10387cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( ( ((kmp_uintptr_t)task) & (sizeof(_Quad)-1) ) == 0 );
10397cc577a4SJonathan Peyton #endif
10407cc577a4SJonathan Peyton     if (sizeof_shareds > 0) {
10417cc577a4SJonathan Peyton         // Avoid double allocation here by combining shareds with taskdata
10427cc577a4SJonathan Peyton         task->shareds         = & ((char *) taskdata)[ shareds_offset ];
10437cc577a4SJonathan Peyton         // Make sure shareds struct is aligned to pointer size
10447cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( ( ((kmp_uintptr_t)task->shareds) & (sizeof(void *)-1) ) == 0 );
10457cc577a4SJonathan Peyton     } else {
10467cc577a4SJonathan Peyton         task->shareds         = NULL;
10477cc577a4SJonathan Peyton     }
10487cc577a4SJonathan Peyton     task->routine             = task_entry;
10497cc577a4SJonathan Peyton     task->part_id             = 0;      // AC: Always start with 0 part id
10507cc577a4SJonathan Peyton 
10517cc577a4SJonathan Peyton     taskdata->td_task_id      = KMP_GEN_TASK_ID();
10527cc577a4SJonathan Peyton     taskdata->td_team         = team;
10537cc577a4SJonathan Peyton     taskdata->td_alloc_thread = thread;
10547cc577a4SJonathan Peyton     taskdata->td_parent       = parent_task;
10557cc577a4SJonathan Peyton     taskdata->td_level        = parent_task->td_level + 1; // increment nesting level
10567cc577a4SJonathan Peyton     taskdata->td_untied_count = 0;
10577cc577a4SJonathan Peyton     taskdata->td_ident        = loc_ref;
10587cc577a4SJonathan Peyton     taskdata->td_taskwait_ident   = NULL;
10597cc577a4SJonathan Peyton     taskdata->td_taskwait_counter = 0;
10607cc577a4SJonathan Peyton     taskdata->td_taskwait_thread  = 0;
10617cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_parent != NULL );
10627cc577a4SJonathan Peyton #if OMP_45_ENABLED
10637cc577a4SJonathan Peyton     // avoid copying icvs for proxy tasks
10647cc577a4SJonathan Peyton     if ( flags->proxy == TASK_FULL )
10657cc577a4SJonathan Peyton #endif
10667cc577a4SJonathan Peyton        copy_icvs( &taskdata->td_icvs, &taskdata->td_parent->td_icvs );
10677cc577a4SJonathan Peyton 
10687cc577a4SJonathan Peyton     taskdata->td_flags.tiedness    = flags->tiedness;
10697cc577a4SJonathan Peyton     taskdata->td_flags.final       = flags->final;
10707cc577a4SJonathan Peyton     taskdata->td_flags.merged_if0  = flags->merged_if0;
10717cc577a4SJonathan Peyton #if OMP_40_ENABLED
10727cc577a4SJonathan Peyton     taskdata->td_flags.destructors_thunk = flags->destructors_thunk;
10737cc577a4SJonathan Peyton #endif // OMP_40_ENABLED
10747cc577a4SJonathan Peyton #if OMP_45_ENABLED
10757cc577a4SJonathan Peyton     taskdata->td_flags.proxy           = flags->proxy;
10767cc577a4SJonathan Peyton     taskdata->td_task_team         = thread->th.th_task_team;
10777cc577a4SJonathan Peyton     taskdata->td_size_alloc        = shareds_offset + sizeof_shareds;
10787cc577a4SJonathan Peyton #endif
10797cc577a4SJonathan Peyton     taskdata->td_flags.tasktype    = TASK_EXPLICIT;
10807cc577a4SJonathan Peyton 
10817cc577a4SJonathan Peyton     // GEH - TODO: fix this to copy parent task's value of tasking_ser flag
10827cc577a4SJonathan Peyton     taskdata->td_flags.tasking_ser = ( __kmp_tasking_mode == tskm_immediate_exec );
10837cc577a4SJonathan Peyton 
10847cc577a4SJonathan Peyton     // GEH - TODO: fix this to copy parent task's value of team_serial flag
10857cc577a4SJonathan Peyton     taskdata->td_flags.team_serial = ( team->t.t_serialized ) ? 1 : 0;
10867cc577a4SJonathan Peyton 
10877cc577a4SJonathan Peyton     // GEH - Note we serialize the task if the team is serialized to make sure implicit parallel region
10887cc577a4SJonathan Peyton     //       tasks are not left until program termination to execute.  Also, it helps locality to execute
10897cc577a4SJonathan Peyton     //       immediately.
10907cc577a4SJonathan Peyton     taskdata->td_flags.task_serial = ( parent_task->td_flags.final
10917cc577a4SJonathan Peyton       || taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser );
10927cc577a4SJonathan Peyton 
10937cc577a4SJonathan Peyton     taskdata->td_flags.started     = 0;
10947cc577a4SJonathan Peyton     taskdata->td_flags.executing   = 0;
10957cc577a4SJonathan Peyton     taskdata->td_flags.complete    = 0;
10967cc577a4SJonathan Peyton     taskdata->td_flags.freed       = 0;
10977cc577a4SJonathan Peyton 
10987cc577a4SJonathan Peyton     taskdata->td_flags.native      = flags->native;
10997cc577a4SJonathan Peyton 
11007cc577a4SJonathan Peyton     taskdata->td_incomplete_child_tasks = 0;
11017cc577a4SJonathan Peyton     taskdata->td_allocated_child_tasks  = 1; // start at one because counts current task and children
11027cc577a4SJonathan Peyton #if OMP_40_ENABLED
11037cc577a4SJonathan Peyton     taskdata->td_taskgroup = parent_task->td_taskgroup; // task inherits the taskgroup from the parent task
11047cc577a4SJonathan Peyton     taskdata->td_dephash = NULL;
11057cc577a4SJonathan Peyton     taskdata->td_depnode = NULL;
11067cc577a4SJonathan Peyton #endif
11077cc577a4SJonathan Peyton 
11087cc577a4SJonathan Peyton     // Only need to keep track of child task counts if team parallel and tasking not serialized or if it is a proxy task
11097cc577a4SJonathan Peyton #if OMP_45_ENABLED
11107cc577a4SJonathan Peyton     if ( flags->proxy == TASK_PROXY || !( taskdata -> td_flags.team_serial || taskdata -> td_flags.tasking_ser ) )
11117cc577a4SJonathan Peyton #else
11127cc577a4SJonathan Peyton     if ( !( taskdata -> td_flags.team_serial || taskdata -> td_flags.tasking_ser ) )
11137cc577a4SJonathan Peyton #endif
11147cc577a4SJonathan Peyton     {
11157cc577a4SJonathan Peyton         KMP_TEST_THEN_INC32( (kmp_int32 *)(& parent_task->td_incomplete_child_tasks) );
11167cc577a4SJonathan Peyton #if OMP_40_ENABLED
11177cc577a4SJonathan Peyton         if ( parent_task->td_taskgroup )
11187cc577a4SJonathan Peyton             KMP_TEST_THEN_INC32( (kmp_int32 *)(& parent_task->td_taskgroup->count) );
11197cc577a4SJonathan Peyton #endif
11207cc577a4SJonathan Peyton         // Only need to keep track of allocated child tasks for explicit tasks since implicit not deallocated
11217cc577a4SJonathan Peyton         if ( taskdata->td_parent->td_flags.tasktype == TASK_EXPLICIT ) {
11227cc577a4SJonathan Peyton             KMP_TEST_THEN_INC32( (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 
11377cc577a4SJonathan Peyton 
11387cc577a4SJonathan Peyton kmp_task_t *
11397cc577a4SJonathan Peyton __kmpc_omp_task_alloc( ident_t *loc_ref, kmp_int32 gtid, kmp_int32 flags,
11407cc577a4SJonathan Peyton                        size_t sizeof_kmp_task_t, size_t sizeof_shareds,
11417cc577a4SJonathan Peyton                        kmp_routine_entry_t task_entry )
11427cc577a4SJonathan Peyton {
11437cc577a4SJonathan Peyton     kmp_task_t *retval;
11447cc577a4SJonathan Peyton     kmp_tasking_flags_t *input_flags = (kmp_tasking_flags_t *) & flags;
11457cc577a4SJonathan Peyton 
11467cc577a4SJonathan Peyton     input_flags->native = FALSE;
11477cc577a4SJonathan Peyton     // __kmp_task_alloc() sets up all other runtime flags
11487cc577a4SJonathan Peyton 
11497cc577a4SJonathan Peyton #if OMP_45_ENABLED
11507cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_alloc(enter): T#%d loc=%p, flags=(%s %s) "
11517cc577a4SJonathan Peyton                   "sizeof_task=%ld sizeof_shared=%ld entry=%p\n",
11527cc577a4SJonathan Peyton                   gtid, loc_ref, input_flags->tiedness ? "tied  " : "untied",
11537cc577a4SJonathan Peyton                   input_flags->proxy ? "proxy" : "",
11547cc577a4SJonathan Peyton                   sizeof_kmp_task_t, sizeof_shareds, task_entry) );
11557cc577a4SJonathan Peyton #else
11567cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_alloc(enter): T#%d loc=%p, flags=(%s) "
11577cc577a4SJonathan Peyton                   "sizeof_task=%ld sizeof_shared=%ld entry=%p\n",
11587cc577a4SJonathan Peyton                   gtid, loc_ref, input_flags->tiedness ? "tied  " : "untied",
11597cc577a4SJonathan Peyton                   sizeof_kmp_task_t, sizeof_shareds, task_entry) );
11607cc577a4SJonathan Peyton #endif
11617cc577a4SJonathan Peyton 
11627cc577a4SJonathan Peyton     retval = __kmp_task_alloc( loc_ref, gtid, input_flags, sizeof_kmp_task_t,
11637cc577a4SJonathan Peyton                                sizeof_shareds, task_entry );
11647cc577a4SJonathan Peyton 
11657cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmpc_omp_task_alloc(exit): T#%d retval %p\n", gtid, retval) );
11667cc577a4SJonathan Peyton 
11677cc577a4SJonathan Peyton     return retval;
11687cc577a4SJonathan Peyton }
11697cc577a4SJonathan Peyton 
11707cc577a4SJonathan Peyton //-----------------------------------------------------------
11717cc577a4SJonathan Peyton //  __kmp_invoke_task: invoke the specified task
11727cc577a4SJonathan Peyton //
11737cc577a4SJonathan Peyton // gtid: global thread ID of caller
11747cc577a4SJonathan Peyton // task: the task to invoke
11757cc577a4SJonathan Peyton // current_task: the task to resume after task invokation
11767cc577a4SJonathan Peyton 
11777cc577a4SJonathan Peyton static void
11787cc577a4SJonathan Peyton __kmp_invoke_task( kmp_int32 gtid, kmp_task_t *task, kmp_taskdata_t * current_task )
11797cc577a4SJonathan Peyton {
11807cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(task);
11817cc577a4SJonathan Peyton     kmp_uint64 cur_time;
11827cc577a4SJonathan Peyton #if OMP_40_ENABLED
11837cc577a4SJonathan Peyton     int discard = 0 /* false */;
11847cc577a4SJonathan Peyton #endif
11857cc577a4SJonathan Peyton     KA_TRACE(30, ("__kmp_invoke_task(enter): T#%d invoking task %p, current_task=%p\n",
11867cc577a4SJonathan Peyton                   gtid, taskdata, current_task) );
11877cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT(task);
11887cc577a4SJonathan Peyton #if OMP_45_ENABLED
11897cc577a4SJonathan Peyton     if ( taskdata->td_flags.proxy == TASK_PROXY &&
11907cc577a4SJonathan Peyton          taskdata->td_flags.complete == 1)
11917cc577a4SJonathan Peyton          {
11927cc577a4SJonathan Peyton             // This is a proxy task that was already completed but it needs to run
11937cc577a4SJonathan Peyton             // its bottom-half finish
11947cc577a4SJonathan Peyton             KA_TRACE(30, ("__kmp_invoke_task: T#%d running bottom finish for proxy task %p\n",
11957cc577a4SJonathan Peyton                   gtid, taskdata) );
11967cc577a4SJonathan Peyton 
11977cc577a4SJonathan Peyton             __kmp_bottom_half_finish_proxy(gtid,task);
11987cc577a4SJonathan Peyton 
11997cc577a4SJonathan Peyton             KA_TRACE(30, ("__kmp_invoke_task(exit): T#%d completed bottom finish for proxy task %p, resuming task %p\n", gtid, taskdata, current_task) );
12007cc577a4SJonathan Peyton 
12017cc577a4SJonathan Peyton             return;
12027cc577a4SJonathan Peyton          }
12037cc577a4SJonathan Peyton #endif
12047cc577a4SJonathan Peyton 
12057cc577a4SJonathan Peyton #if USE_ITT_BUILD && USE_ITT_NOTIFY
12067cc577a4SJonathan Peyton     if(__kmp_forkjoin_frames_mode == 3) {
12077cc577a4SJonathan Peyton         // Get the current time stamp to measure task execution time to correct 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;
12317cc577a4SJonathan Peyton         taskdata->ompt_task_info.frame.exit_runtime_frame = __builtin_frame_address(0);
12327cc577a4SJonathan Peyton     }
12337cc577a4SJonathan Peyton #endif
12347cc577a4SJonathan Peyton 
12357cc577a4SJonathan Peyton #if OMP_40_ENABLED
12367cc577a4SJonathan Peyton     // TODO: cancel tasks if the parallel region has also been cancelled
12377cc577a4SJonathan Peyton     // TODO: check if this sequence can be hoisted above __kmp_task_start
12387cc577a4SJonathan Peyton     // if cancellation has been enabled for this run ...
12397cc577a4SJonathan Peyton     if (__kmp_omp_cancellation) {
12407cc577a4SJonathan Peyton         kmp_info_t *this_thr = __kmp_threads [ gtid ];
12417cc577a4SJonathan Peyton         kmp_team_t * this_team = this_thr->th.th_team;
12427cc577a4SJonathan Peyton         kmp_taskgroup_t * taskgroup = taskdata->td_taskgroup;
12437cc577a4SJonathan Peyton         if ((taskgroup && taskgroup->cancel_request) || (this_team->t.t_cancel_request == cancel_parallel)) {
12447cc577a4SJonathan Peyton             KMP_COUNT_BLOCK(TASK_cancelled);
12457cc577a4SJonathan Peyton             // this task belongs to a task group and we need to cancel it
12467cc577a4SJonathan Peyton             discard = 1 /* true */;
12477cc577a4SJonathan Peyton         }
12487cc577a4SJonathan Peyton     }
12497cc577a4SJonathan Peyton 
12507cc577a4SJonathan Peyton     //
12517cc577a4SJonathan Peyton     // Invoke the task routine and pass in relevant data.
12527cc577a4SJonathan Peyton     // Thunks generated by gcc take a different argument list.
12537cc577a4SJonathan Peyton     //
12547cc577a4SJonathan Peyton     if (!discard) {
12557cc577a4SJonathan Peyton #if KMP_STATS_ENABLED
12567cc577a4SJonathan Peyton         KMP_COUNT_BLOCK(TASK_executed);
12577cc577a4SJonathan Peyton         switch(KMP_GET_THREAD_STATE()) {
12587cc577a4SJonathan Peyton          case FORK_JOIN_BARRIER: KMP_PUSH_PARTITIONED_TIMER(OMP_task_join_bar); break;
12597cc577a4SJonathan Peyton          case PLAIN_BARRIER: KMP_PUSH_PARTITIONED_TIMER(OMP_task_plain_bar); break;
12607cc577a4SJonathan Peyton          case TASKYIELD: KMP_PUSH_PARTITIONED_TIMER(OMP_task_taskyield); break;
12617cc577a4SJonathan Peyton          case TASKWAIT: KMP_PUSH_PARTITIONED_TIMER(OMP_task_taskwait); break;
12627cc577a4SJonathan Peyton          case TASKGROUP: KMP_PUSH_PARTITIONED_TIMER(OMP_task_taskgroup); break;
12637cc577a4SJonathan Peyton          default: KMP_PUSH_PARTITIONED_TIMER(OMP_task_immediate); break;
12647cc577a4SJonathan Peyton         }
12657cc577a4SJonathan Peyton #endif // KMP_STATS_ENABLED
12667cc577a4SJonathan Peyton #endif // OMP_40_ENABLED
12677cc577a4SJonathan Peyton 
12687cc577a4SJonathan Peyton #if OMPT_SUPPORT && OMPT_TRACE
12697cc577a4SJonathan Peyton         /* let OMPT know that we're about to run this task */
12707cc577a4SJonathan Peyton         if (ompt_enabled &&
12717cc577a4SJonathan Peyton              ompt_callbacks.ompt_callback(ompt_event_task_switch))
12727cc577a4SJonathan Peyton         {
12737cc577a4SJonathan Peyton           ompt_callbacks.ompt_callback(ompt_event_task_switch)(
12747cc577a4SJonathan Peyton             current_task->ompt_task_info.task_id,
12757cc577a4SJonathan Peyton             taskdata->ompt_task_info.task_id);
12767cc577a4SJonathan Peyton         }
12777cc577a4SJonathan Peyton #endif
12787cc577a4SJonathan Peyton 
12797cc577a4SJonathan Peyton #ifdef KMP_GOMP_COMPAT
12807cc577a4SJonathan Peyton         if (taskdata->td_flags.native) {
12817cc577a4SJonathan Peyton             ((void (*)(void *))(*(task->routine)))(task->shareds);
12827cc577a4SJonathan Peyton         }
12837cc577a4SJonathan Peyton         else
12847cc577a4SJonathan Peyton #endif /* KMP_GOMP_COMPAT */
12857cc577a4SJonathan Peyton         {
12867cc577a4SJonathan Peyton             (*(task->routine))(gtid, task);
12877cc577a4SJonathan Peyton         }
12887cc577a4SJonathan Peyton         KMP_POP_PARTITIONED_TIMER();
12897cc577a4SJonathan Peyton 
12907cc577a4SJonathan Peyton #if OMPT_SUPPORT && OMPT_TRACE
12917cc577a4SJonathan Peyton         /* let OMPT know that we're returning to the callee task */
12927cc577a4SJonathan Peyton         if (ompt_enabled &&
12937cc577a4SJonathan Peyton              ompt_callbacks.ompt_callback(ompt_event_task_switch))
12947cc577a4SJonathan Peyton         {
12957cc577a4SJonathan Peyton           ompt_callbacks.ompt_callback(ompt_event_task_switch)(
12967cc577a4SJonathan Peyton             taskdata->ompt_task_info.task_id,
12977cc577a4SJonathan Peyton             current_task->ompt_task_info.task_id);
12987cc577a4SJonathan Peyton         }
12997cc577a4SJonathan Peyton #endif
13007cc577a4SJonathan Peyton 
13017cc577a4SJonathan Peyton #if OMP_40_ENABLED
13027cc577a4SJonathan Peyton     }
13037cc577a4SJonathan Peyton #endif // OMP_40_ENABLED
13047cc577a4SJonathan Peyton 
13057cc577a4SJonathan Peyton 
13067cc577a4SJonathan Peyton #if OMPT_SUPPORT
13077cc577a4SJonathan Peyton     if (ompt_enabled) {
13087cc577a4SJonathan Peyton         thread->th.ompt_thread_info = oldInfo;
13097cc577a4SJonathan Peyton         taskdata->ompt_task_info.frame.exit_runtime_frame = NULL;
13107cc577a4SJonathan Peyton     }
13117cc577a4SJonathan Peyton #endif
13127cc577a4SJonathan Peyton 
13137cc577a4SJonathan Peyton #if OMP_45_ENABLED
13147cc577a4SJonathan Peyton     // Proxy tasks are not handled by the runtime
13157cc577a4SJonathan Peyton     if ( taskdata->td_flags.proxy != TASK_PROXY ) {
13167cc577a4SJonathan Peyton #endif
13177cc577a4SJonathan Peyton       ANNOTATE_HAPPENS_BEFORE(taskdata->td_parent);
13187cc577a4SJonathan Peyton       __kmp_task_finish( gtid, task, current_task );
13197cc577a4SJonathan Peyton #if OMP_45_ENABLED
13207cc577a4SJonathan Peyton     }
13217cc577a4SJonathan Peyton #endif
13227cc577a4SJonathan Peyton 
13237cc577a4SJonathan Peyton #if USE_ITT_BUILD && USE_ITT_NOTIFY
13247cc577a4SJonathan Peyton     // Barrier imbalance - correct arrive time after the task finished
13257cc577a4SJonathan Peyton     if(__kmp_forkjoin_frames_mode == 3) {
13267cc577a4SJonathan Peyton         kmp_info_t *this_thr = __kmp_threads [ gtid ];
13277cc577a4SJonathan Peyton         if(this_thr->th.th_bar_arrive_time) {
13287cc577a4SJonathan Peyton             this_thr->th.th_bar_arrive_time += (__itt_get_timestamp() - cur_time);
13297cc577a4SJonathan Peyton         }
13307cc577a4SJonathan Peyton     }
13317cc577a4SJonathan Peyton #endif
13327cc577a4SJonathan Peyton     KA_TRACE(30, ("__kmp_invoke_task(exit): T#%d completed task %p, resuming task %p\n",
13337cc577a4SJonathan Peyton                   gtid, taskdata, current_task) );
13347cc577a4SJonathan Peyton     return;
13357cc577a4SJonathan Peyton }
13367cc577a4SJonathan Peyton 
13377cc577a4SJonathan Peyton //-----------------------------------------------------------------------
13387cc577a4SJonathan Peyton // __kmpc_omp_task_parts: Schedule a thread-switchable task for execution
13397cc577a4SJonathan Peyton //
13407cc577a4SJonathan Peyton // loc_ref: location of original task pragma (ignored)
13417cc577a4SJonathan Peyton // gtid: Global Thread ID of encountering thread
13427cc577a4SJonathan Peyton // new_task: task thunk allocated by __kmp_omp_task_alloc() for the ''new task''
13437cc577a4SJonathan Peyton // Returns:
13447cc577a4SJonathan Peyton //    TASK_CURRENT_NOT_QUEUED (0) if did not suspend and queue current task to be resumed later.
13457cc577a4SJonathan Peyton //    TASK_CURRENT_QUEUED (1) if suspended and queued the current task to be resumed later.
13467cc577a4SJonathan Peyton 
13477cc577a4SJonathan Peyton kmp_int32
13487cc577a4SJonathan Peyton __kmpc_omp_task_parts( ident_t *loc_ref, kmp_int32 gtid, kmp_task_t * new_task)
13497cc577a4SJonathan Peyton {
13507cc577a4SJonathan Peyton     kmp_taskdata_t * new_taskdata = KMP_TASK_TO_TASKDATA(new_task);
13517cc577a4SJonathan Peyton 
13527cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_parts(enter): T#%d loc=%p task=%p\n",
13537cc577a4SJonathan Peyton                   gtid, loc_ref, new_taskdata ) );
13547cc577a4SJonathan Peyton 
13557cc577a4SJonathan Peyton     /* Should we execute the new task or queue it?   For now, let's just always try to
13567cc577a4SJonathan Peyton        queue it.  If the queue fills up, then we'll execute it.  */
13577cc577a4SJonathan Peyton 
13587cc577a4SJonathan Peyton     if ( __kmp_push_task( gtid, new_task ) == TASK_NOT_PUSHED ) // if cannot defer
13597cc577a4SJonathan Peyton     {                                                           // Execute this task immediately
13607cc577a4SJonathan Peyton         kmp_taskdata_t * current_task = __kmp_threads[ gtid ] -> th.th_current_task;
13617cc577a4SJonathan Peyton         new_taskdata->td_flags.task_serial = 1;
13627cc577a4SJonathan Peyton         __kmp_invoke_task( gtid, new_task, current_task );
13637cc577a4SJonathan Peyton     }
13647cc577a4SJonathan Peyton 
13657cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task_parts(exit): T#%d returning TASK_CURRENT_NOT_QUEUED: "
13667cc577a4SJonathan Peyton                   "loc=%p task=%p, return: TASK_CURRENT_NOT_QUEUED\n", gtid, loc_ref,
13677cc577a4SJonathan Peyton                   new_taskdata ) );
13687cc577a4SJonathan Peyton 
13697cc577a4SJonathan Peyton     ANNOTATE_HAPPENS_BEFORE(new_task);
13707cc577a4SJonathan Peyton     return TASK_CURRENT_NOT_QUEUED;
13717cc577a4SJonathan Peyton }
13727cc577a4SJonathan Peyton 
13737cc577a4SJonathan Peyton //---------------------------------------------------------------------
13747cc577a4SJonathan Peyton // __kmp_omp_task: Schedule a non-thread-switchable task for execution
13757cc577a4SJonathan Peyton // gtid: Global Thread ID of encountering thread
13767cc577a4SJonathan Peyton // new_task: non-thread-switchable task thunk allocated by __kmp_omp_task_alloc()
13777cc577a4SJonathan Peyton // serialize_immediate: if TRUE then if the task is executed immediately its execution will be serialized
13787cc577a4SJonathan Peyton // returns:
13797cc577a4SJonathan Peyton //
13807cc577a4SJonathan Peyton //    TASK_CURRENT_NOT_QUEUED (0) if did not suspend and queue current task to be resumed later.
13817cc577a4SJonathan Peyton //    TASK_CURRENT_QUEUED (1) if suspended and queued the current task to be resumed later.
13827cc577a4SJonathan Peyton kmp_int32
13837cc577a4SJonathan Peyton __kmp_omp_task( kmp_int32 gtid, kmp_task_t * new_task, bool serialize_immediate )
13847cc577a4SJonathan Peyton {
13857cc577a4SJonathan Peyton     kmp_taskdata_t * new_taskdata = KMP_TASK_TO_TASKDATA(new_task);
13867cc577a4SJonathan Peyton 
13877cc577a4SJonathan Peyton #if OMPT_SUPPORT
13887cc577a4SJonathan Peyton     if (ompt_enabled) {
13897cc577a4SJonathan Peyton         new_taskdata->ompt_task_info.frame.reenter_runtime_frame =
13907cc577a4SJonathan Peyton             __builtin_frame_address(1);
13917cc577a4SJonathan Peyton     }
13927cc577a4SJonathan Peyton #endif
13937cc577a4SJonathan Peyton 
13947cc577a4SJonathan Peyton     /* Should we execute the new task or queue it?   For now, let's just always try to
13957cc577a4SJonathan Peyton        queue it.  If the queue fills up, then we'll execute it.  */
13967cc577a4SJonathan Peyton #if OMP_45_ENABLED
13977cc577a4SJonathan Peyton     if ( new_taskdata->td_flags.proxy == TASK_PROXY || __kmp_push_task( gtid, new_task ) == TASK_NOT_PUSHED ) // if cannot defer
13987cc577a4SJonathan Peyton #else
13997cc577a4SJonathan Peyton     if ( __kmp_push_task( gtid, new_task ) == TASK_NOT_PUSHED ) // if cannot defer
14007cc577a4SJonathan Peyton #endif
14017cc577a4SJonathan Peyton     {                                                           // Execute this task immediately
14027cc577a4SJonathan Peyton         kmp_taskdata_t * current_task = __kmp_threads[ gtid ] -> th.th_current_task;
14037cc577a4SJonathan Peyton         if ( serialize_immediate )
14047cc577a4SJonathan Peyton           new_taskdata -> td_flags.task_serial = 1;
14057cc577a4SJonathan Peyton         __kmp_invoke_task( gtid, new_task, current_task );
14067cc577a4SJonathan Peyton     }
14077cc577a4SJonathan Peyton 
14087cc577a4SJonathan Peyton #if OMPT_SUPPORT
14097cc577a4SJonathan Peyton     if (ompt_enabled) {
14107cc577a4SJonathan Peyton         new_taskdata->ompt_task_info.frame.reenter_runtime_frame = NULL;
14117cc577a4SJonathan Peyton     }
14127cc577a4SJonathan Peyton #endif
14137cc577a4SJonathan Peyton 
14147cc577a4SJonathan Peyton     ANNOTATE_HAPPENS_BEFORE(new_task);
14157cc577a4SJonathan Peyton     return TASK_CURRENT_NOT_QUEUED;
14167cc577a4SJonathan Peyton }
14177cc577a4SJonathan Peyton 
14187cc577a4SJonathan Peyton //---------------------------------------------------------------------
14197cc577a4SJonathan Peyton // __kmpc_omp_task: Wrapper around __kmp_omp_task to schedule a non-thread-switchable task from
14207cc577a4SJonathan Peyton // the parent thread only!
14217cc577a4SJonathan Peyton // loc_ref: location of original task pragma (ignored)
14227cc577a4SJonathan Peyton // gtid: Global Thread ID of encountering thread
14237cc577a4SJonathan Peyton // new_task: non-thread-switchable task thunk allocated by __kmp_omp_task_alloc()
14247cc577a4SJonathan Peyton // returns:
14257cc577a4SJonathan Peyton //
14267cc577a4SJonathan Peyton //    TASK_CURRENT_NOT_QUEUED (0) if did not suspend and queue current task to be resumed later.
14277cc577a4SJonathan Peyton //    TASK_CURRENT_QUEUED (1) if suspended and queued the current task to be resumed later.
14287cc577a4SJonathan Peyton 
14297cc577a4SJonathan Peyton kmp_int32
14307cc577a4SJonathan Peyton __kmpc_omp_task( ident_t *loc_ref, kmp_int32 gtid, kmp_task_t * new_task)
14317cc577a4SJonathan Peyton {
14327cc577a4SJonathan Peyton     kmp_int32 res;
14337cc577a4SJonathan Peyton     KMP_SET_THREAD_STATE_BLOCK(EXPLICIT_TASK);
14347cc577a4SJonathan Peyton 
14357cc577a4SJonathan Peyton #if KMP_DEBUG
14367cc577a4SJonathan Peyton     kmp_taskdata_t * new_taskdata = KMP_TASK_TO_TASKDATA(new_task);
14377cc577a4SJonathan Peyton #endif
14387cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task(enter): T#%d loc=%p task=%p\n",
14397cc577a4SJonathan Peyton                   gtid, loc_ref, new_taskdata ) );
14407cc577a4SJonathan Peyton 
14417cc577a4SJonathan Peyton     res =  __kmp_omp_task(gtid,new_task,true);
14427cc577a4SJonathan Peyton 
14437cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_task(exit): T#%d returning TASK_CURRENT_NOT_QUEUED: loc=%p task=%p\n",
14447cc577a4SJonathan Peyton                   gtid, loc_ref, new_taskdata ) );
14457cc577a4SJonathan Peyton     return res;
14467cc577a4SJonathan Peyton }
14477cc577a4SJonathan Peyton 
14487cc577a4SJonathan Peyton //-------------------------------------------------------------------------------------
14497cc577a4SJonathan Peyton // __kmpc_omp_taskwait: Wait until all tasks generated by the current task are complete
14507cc577a4SJonathan Peyton 
14517cc577a4SJonathan Peyton kmp_int32
14527cc577a4SJonathan Peyton __kmpc_omp_taskwait( ident_t *loc_ref, kmp_int32 gtid )
14537cc577a4SJonathan Peyton {
14547cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata;
14557cc577a4SJonathan Peyton     kmp_info_t * thread;
14567cc577a4SJonathan Peyton     int thread_finished = FALSE;
14577cc577a4SJonathan Peyton     KMP_SET_THREAD_STATE_BLOCK(TASKWAIT);
14587cc577a4SJonathan Peyton 
14597cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_taskwait(enter): T#%d loc=%p\n", gtid, loc_ref) );
14607cc577a4SJonathan Peyton 
14617cc577a4SJonathan Peyton     if ( __kmp_tasking_mode != tskm_immediate_exec ) {
14627cc577a4SJonathan Peyton         // GEH TODO: shouldn't we have some sort of OMPRAP API calls here to mark begin wait?
14637cc577a4SJonathan Peyton 
14647cc577a4SJonathan Peyton         thread = __kmp_threads[ gtid ];
14657cc577a4SJonathan Peyton         taskdata = thread -> th.th_current_task;
14667cc577a4SJonathan Peyton 
14677cc577a4SJonathan Peyton #if OMPT_SUPPORT && OMPT_TRACE
14687cc577a4SJonathan Peyton         ompt_task_id_t my_task_id;
14697cc577a4SJonathan Peyton         ompt_parallel_id_t my_parallel_id;
14707cc577a4SJonathan Peyton 
14717cc577a4SJonathan Peyton         if (ompt_enabled) {
14727cc577a4SJonathan Peyton             kmp_team_t *team = thread->th.th_team;
14737cc577a4SJonathan Peyton             my_task_id = taskdata->ompt_task_info.task_id;
14747cc577a4SJonathan Peyton             my_parallel_id = team->t.ompt_team_info.parallel_id;
14757cc577a4SJonathan Peyton 
14767cc577a4SJonathan Peyton             taskdata->ompt_task_info.frame.reenter_runtime_frame = __builtin_frame_address(1);
14777cc577a4SJonathan Peyton             if (ompt_callbacks.ompt_callback(ompt_event_taskwait_begin)) {
14787cc577a4SJonathan Peyton                 ompt_callbacks.ompt_callback(ompt_event_taskwait_begin)(
14797cc577a4SJonathan Peyton                                 my_parallel_id, my_task_id);
14807cc577a4SJonathan Peyton             }
14817cc577a4SJonathan Peyton         }
14827cc577a4SJonathan Peyton #endif
14837cc577a4SJonathan Peyton 
14847cc577a4SJonathan Peyton         // Debugger: The taskwait is active. Store location and thread encountered the taskwait.
14857cc577a4SJonathan Peyton #if USE_ITT_BUILD
14867cc577a4SJonathan Peyton         // Note: These values are used by ITT events as well.
14877cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
14887cc577a4SJonathan Peyton         taskdata->td_taskwait_counter += 1;
14897cc577a4SJonathan Peyton         taskdata->td_taskwait_ident    = loc_ref;
14907cc577a4SJonathan Peyton         taskdata->td_taskwait_thread   = gtid + 1;
14917cc577a4SJonathan Peyton 
14927cc577a4SJonathan Peyton #if USE_ITT_BUILD
14937cc577a4SJonathan Peyton         void * itt_sync_obj = __kmp_itt_taskwait_object( gtid );
14947cc577a4SJonathan Peyton         if ( itt_sync_obj != NULL )
14957cc577a4SJonathan Peyton             __kmp_itt_taskwait_starting( gtid, itt_sync_obj );
14967cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
14977cc577a4SJonathan Peyton 
14987cc577a4SJonathan Peyton         bool must_wait = ! taskdata->td_flags.team_serial && ! taskdata->td_flags.final;
14997cc577a4SJonathan Peyton 
15007cc577a4SJonathan Peyton #if OMP_45_ENABLED
15017cc577a4SJonathan Peyton         must_wait = must_wait || (thread->th.th_task_team != NULL && thread->th.th_task_team->tt.tt_found_proxy_tasks);
15027cc577a4SJonathan Peyton #endif
15037cc577a4SJonathan Peyton         if (must_wait)
15047cc577a4SJonathan Peyton         {
15057cc577a4SJonathan Peyton             kmp_flag_32 flag(&(taskdata->td_incomplete_child_tasks), 0U);
15067cc577a4SJonathan Peyton             while ( TCR_4(taskdata -> td_incomplete_child_tasks) != 0 ) {
15077cc577a4SJonathan Peyton                 flag.execute_tasks(thread, gtid, FALSE, &thread_finished
15087cc577a4SJonathan Peyton                                    USE_ITT_BUILD_ARG(itt_sync_obj), __kmp_task_stealing_constraint );
15097cc577a4SJonathan Peyton             }
15107cc577a4SJonathan Peyton         }
15117cc577a4SJonathan Peyton #if USE_ITT_BUILD
15127cc577a4SJonathan Peyton         if ( itt_sync_obj != NULL )
15137cc577a4SJonathan Peyton             __kmp_itt_taskwait_finished( gtid, itt_sync_obj );
15147cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
15157cc577a4SJonathan Peyton 
15167cc577a4SJonathan Peyton         // GEH TODO: shouldn't we have some sort of OMPRAP API calls here to mark end of wait?
15177cc577a4SJonathan Peyton         // Debugger:  The taskwait is completed. Location remains, but thread is negated.
15187cc577a4SJonathan Peyton         taskdata->td_taskwait_thread = - taskdata->td_taskwait_thread;
15197cc577a4SJonathan Peyton 
15207cc577a4SJonathan Peyton #if OMPT_SUPPORT && OMPT_TRACE
15217cc577a4SJonathan Peyton         if (ompt_enabled) {
15227cc577a4SJonathan Peyton             if (ompt_callbacks.ompt_callback(ompt_event_taskwait_end)) {
15237cc577a4SJonathan Peyton                 ompt_callbacks.ompt_callback(ompt_event_taskwait_end)(
15247cc577a4SJonathan Peyton                                 my_parallel_id, my_task_id);
15257cc577a4SJonathan Peyton             }
15267cc577a4SJonathan Peyton             taskdata->ompt_task_info.frame.reenter_runtime_frame = NULL;
15277cc577a4SJonathan Peyton         }
15287cc577a4SJonathan Peyton #endif
15297cc577a4SJonathan Peyton         ANNOTATE_HAPPENS_AFTER(taskdata);
15307cc577a4SJonathan Peyton     }
15317cc577a4SJonathan Peyton 
15327cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_taskwait(exit): T#%d task %p finished waiting, "
15337cc577a4SJonathan Peyton                   "returning TASK_CURRENT_NOT_QUEUED\n", gtid, taskdata) );
15347cc577a4SJonathan Peyton 
15357cc577a4SJonathan Peyton     return TASK_CURRENT_NOT_QUEUED;
15367cc577a4SJonathan Peyton }
15377cc577a4SJonathan Peyton 
15387cc577a4SJonathan Peyton 
15397cc577a4SJonathan Peyton //-------------------------------------------------
15407cc577a4SJonathan Peyton // __kmpc_omp_taskyield: switch to a different task
15417cc577a4SJonathan Peyton 
15427cc577a4SJonathan Peyton kmp_int32
15437cc577a4SJonathan Peyton __kmpc_omp_taskyield( ident_t *loc_ref, kmp_int32 gtid, int end_part )
15447cc577a4SJonathan Peyton {
15457cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata;
15467cc577a4SJonathan Peyton     kmp_info_t * thread;
15477cc577a4SJonathan Peyton     int thread_finished = FALSE;
15487cc577a4SJonathan Peyton 
15497cc577a4SJonathan Peyton     KMP_COUNT_BLOCK(OMP_TASKYIELD);
15507cc577a4SJonathan Peyton     KMP_SET_THREAD_STATE_BLOCK(TASKYIELD);
15517cc577a4SJonathan Peyton 
15527cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_taskyield(enter): T#%d loc=%p end_part = %d\n",
15537cc577a4SJonathan Peyton                   gtid, loc_ref, end_part) );
15547cc577a4SJonathan Peyton 
15557cc577a4SJonathan Peyton     if ( __kmp_tasking_mode != tskm_immediate_exec && __kmp_init_parallel ) {
15567cc577a4SJonathan Peyton         // GEH TODO: shouldn't we have some sort of OMPRAP API calls here to mark begin wait?
15577cc577a4SJonathan Peyton 
15587cc577a4SJonathan Peyton         thread = __kmp_threads[ gtid ];
15597cc577a4SJonathan Peyton         taskdata = thread -> th.th_current_task;
15607cc577a4SJonathan Peyton         // Should we model this as a task wait or not?
15617cc577a4SJonathan Peyton         // Debugger: The taskwait is active. Store location and thread encountered the taskwait.
15627cc577a4SJonathan Peyton #if USE_ITT_BUILD
15637cc577a4SJonathan Peyton         // Note: These values are used by ITT events as well.
15647cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
15657cc577a4SJonathan Peyton         taskdata->td_taskwait_counter += 1;
15667cc577a4SJonathan Peyton         taskdata->td_taskwait_ident    = loc_ref;
15677cc577a4SJonathan Peyton         taskdata->td_taskwait_thread   = gtid + 1;
15687cc577a4SJonathan Peyton 
15697cc577a4SJonathan Peyton #if USE_ITT_BUILD
15707cc577a4SJonathan Peyton         void * itt_sync_obj = __kmp_itt_taskwait_object( gtid );
15717cc577a4SJonathan Peyton         if ( itt_sync_obj != NULL )
15727cc577a4SJonathan Peyton             __kmp_itt_taskwait_starting( gtid, itt_sync_obj );
15737cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
15747cc577a4SJonathan Peyton         if ( ! taskdata->td_flags.team_serial ) {
15757cc577a4SJonathan Peyton             kmp_task_team_t * task_team = thread->th.th_task_team;
15767cc577a4SJonathan Peyton             if (task_team != NULL) {
15777cc577a4SJonathan Peyton                 if (KMP_TASKING_ENABLED(task_team)) {
15787cc577a4SJonathan Peyton                     __kmp_execute_tasks_32( thread, gtid, NULL, FALSE, &thread_finished
15797cc577a4SJonathan Peyton                                             USE_ITT_BUILD_ARG(itt_sync_obj), __kmp_task_stealing_constraint );
15807cc577a4SJonathan Peyton                 }
15817cc577a4SJonathan Peyton             }
15827cc577a4SJonathan Peyton         }
15837cc577a4SJonathan Peyton #if USE_ITT_BUILD
15847cc577a4SJonathan Peyton         if ( itt_sync_obj != NULL )
15857cc577a4SJonathan Peyton             __kmp_itt_taskwait_finished( gtid, itt_sync_obj );
15867cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
15877cc577a4SJonathan Peyton 
15887cc577a4SJonathan Peyton         // GEH TODO: shouldn't we have some sort of OMPRAP API calls here to mark end of wait?
15897cc577a4SJonathan Peyton         // Debugger:  The taskwait is completed. Location remains, but thread is negated.
15907cc577a4SJonathan Peyton         taskdata->td_taskwait_thread = - taskdata->td_taskwait_thread;
15917cc577a4SJonathan Peyton     }
15927cc577a4SJonathan Peyton 
15937cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_omp_taskyield(exit): T#%d task %p resuming, "
15947cc577a4SJonathan Peyton                   "returning TASK_CURRENT_NOT_QUEUED\n", gtid, taskdata) );
15957cc577a4SJonathan Peyton 
15967cc577a4SJonathan Peyton     return TASK_CURRENT_NOT_QUEUED;
15977cc577a4SJonathan Peyton }
15987cc577a4SJonathan Peyton 
1599*72ba2109SAndrey Churbanov // TODO: change to OMP_50_ENABLED, need to change build tools for this to work
1600*72ba2109SAndrey Churbanov #if OMP_45_ENABLED
1601*72ba2109SAndrey Churbanov //
1602*72ba2109SAndrey Churbanov // Task Reduction implementation
1603*72ba2109SAndrey Churbanov //
1604*72ba2109SAndrey Churbanov 
1605*72ba2109SAndrey Churbanov typedef struct kmp_task_red_flags {
1606*72ba2109SAndrey Churbanov     unsigned  lazy_priv : 1;  // hint: (1) use lazy allocation (big objects)
1607*72ba2109SAndrey Churbanov     unsigned  reserved31 : 31;
1608*72ba2109SAndrey Churbanov } kmp_task_red_flags_t;
1609*72ba2109SAndrey Churbanov 
1610*72ba2109SAndrey Churbanov // internal structure for reduction data item related info
1611*72ba2109SAndrey Churbanov typedef struct kmp_task_red_data {
1612*72ba2109SAndrey Churbanov     void       *reduce_shar; // shared reduction item
1613*72ba2109SAndrey Churbanov     size_t      reduce_size; // size of data item
1614*72ba2109SAndrey Churbanov     void       *reduce_priv; // thread specific data
1615*72ba2109SAndrey Churbanov     void       *reduce_pend; // end of private data for comparison op
1616*72ba2109SAndrey Churbanov     void       *reduce_init; // data initialization routine
1617*72ba2109SAndrey Churbanov     void       *reduce_fini; // data finalization routine
1618*72ba2109SAndrey Churbanov     void       *reduce_comb; // data combiner routine
1619*72ba2109SAndrey Churbanov     kmp_task_red_flags_t flags; // flags for additional info from compiler
1620*72ba2109SAndrey Churbanov } kmp_task_red_data_t;
1621*72ba2109SAndrey Churbanov 
1622*72ba2109SAndrey Churbanov // structure sent us by compiler - one per reduction item
1623*72ba2109SAndrey Churbanov typedef struct kmp_task_red_input {
1624*72ba2109SAndrey Churbanov     void       *reduce_shar; // shared reduction item
1625*72ba2109SAndrey Churbanov     size_t      reduce_size; // size of data item
1626*72ba2109SAndrey Churbanov     void       *reduce_init; // data initialization routine
1627*72ba2109SAndrey Churbanov     void       *reduce_fini; // data finalization routine
1628*72ba2109SAndrey Churbanov     void       *reduce_comb; // data combiner routine
1629*72ba2109SAndrey Churbanov     kmp_task_red_flags_t flags; // flags for additional info from compiler
1630*72ba2109SAndrey Churbanov } kmp_task_red_input_t;
1631*72ba2109SAndrey Churbanov 
1632*72ba2109SAndrey Churbanov /*!
1633*72ba2109SAndrey Churbanov @ingroup TASKING
1634*72ba2109SAndrey Churbanov @param gtid      Global thread ID
1635*72ba2109SAndrey Churbanov @param num       Number of data items to reduce
1636*72ba2109SAndrey Churbanov @param data      Array of data for reduction
1637*72ba2109SAndrey Churbanov @return The taskgroup identifier
1638*72ba2109SAndrey Churbanov 
1639*72ba2109SAndrey Churbanov Initialize task reduction for the taskgroup.
1640*72ba2109SAndrey Churbanov */
1641*72ba2109SAndrey Churbanov void*
1642*72ba2109SAndrey Churbanov __kmpc_task_reduction_init(int gtid, int num, void *data)
1643*72ba2109SAndrey Churbanov {
1644*72ba2109SAndrey Churbanov     kmp_info_t * thread = __kmp_threads[gtid];
1645*72ba2109SAndrey Churbanov     kmp_taskgroup_t * tg = thread->th.th_current_task->td_taskgroup;
1646*72ba2109SAndrey Churbanov     kmp_int32 nth = thread->th.th_team_nproc;
1647*72ba2109SAndrey Churbanov     kmp_task_red_input_t *input = (kmp_task_red_input_t*)data;
1648*72ba2109SAndrey Churbanov     kmp_task_red_data_t *arr;
1649*72ba2109SAndrey Churbanov 
1650*72ba2109SAndrey Churbanov     // check input data just in case
1651*72ba2109SAndrey Churbanov     KMP_ASSERT(tg != NULL);
1652*72ba2109SAndrey Churbanov     KMP_ASSERT(data != NULL);
1653*72ba2109SAndrey Churbanov     KMP_ASSERT(num > 0);
1654*72ba2109SAndrey Churbanov     if (nth == 1) {
1655*72ba2109SAndrey Churbanov         KA_TRACE(10, ("__kmpc_task_reduction_init: T#%d, tg %p, exiting nth=1\n",
1656*72ba2109SAndrey Churbanov                 gtid, tg));
1657*72ba2109SAndrey Churbanov         return (void*)tg;
1658*72ba2109SAndrey Churbanov     }
1659*72ba2109SAndrey Churbanov     KA_TRACE(10,("__kmpc_task_reduction_init: T#%d, taskgroup %p, #items %d\n",
1660*72ba2109SAndrey Churbanov                  gtid, tg, num));
1661*72ba2109SAndrey Churbanov     arr = (kmp_task_red_data_t*)__kmp_thread_malloc(thread, num * sizeof(kmp_task_red_data_t));
1662*72ba2109SAndrey Churbanov     for (int i = 0; i < num; ++i) {
1663*72ba2109SAndrey Churbanov         void(*f_init)(void*) = (void(*)(void*))(input[i].reduce_init);
1664*72ba2109SAndrey Churbanov         size_t size = input[i].reduce_size - 1;
1665*72ba2109SAndrey Churbanov         // round the size up to cache line per thread-specific item
1666*72ba2109SAndrey Churbanov         size += CACHE_LINE - size % CACHE_LINE;
1667*72ba2109SAndrey Churbanov         KMP_ASSERT(input[i].reduce_comb != NULL); // combiner is mandatory
1668*72ba2109SAndrey Churbanov         arr[i].reduce_shar = input[i].reduce_shar;
1669*72ba2109SAndrey Churbanov         arr[i].reduce_size = size;
1670*72ba2109SAndrey Churbanov         arr[i].reduce_init = input[i].reduce_init;
1671*72ba2109SAndrey Churbanov         arr[i].reduce_fini = input[i].reduce_fini;
1672*72ba2109SAndrey Churbanov         arr[i].reduce_comb = input[i].reduce_comb;
1673*72ba2109SAndrey Churbanov         arr[i].flags       = input[i].flags;
1674*72ba2109SAndrey Churbanov         if (!input[i].flags.lazy_priv) {
1675*72ba2109SAndrey Churbanov             // allocate cache-line aligned block and fill it with zeros
1676*72ba2109SAndrey Churbanov             arr[i].reduce_priv = __kmp_allocate(nth * size);
1677*72ba2109SAndrey Churbanov             arr[i].reduce_pend = (char*)(arr[i].reduce_priv) + nth * size;
1678*72ba2109SAndrey Churbanov             if (f_init != NULL) {
1679*72ba2109SAndrey Churbanov                 // initialize thread-specific items
1680*72ba2109SAndrey Churbanov                 for (int j = 0; j < nth; ++j) {
1681*72ba2109SAndrey Churbanov                     f_init((char*)(arr[i].reduce_priv) + j * size);
1682*72ba2109SAndrey Churbanov                 }
1683*72ba2109SAndrey Churbanov             }
1684*72ba2109SAndrey Churbanov         } else {
1685*72ba2109SAndrey Churbanov             // only allocate space for pointers now,
1686*72ba2109SAndrey Churbanov             // objects will be lazily allocated/initialized once requested
1687*72ba2109SAndrey Churbanov             arr[i].reduce_priv = __kmp_allocate(nth * sizeof(void*));
1688*72ba2109SAndrey Churbanov         }
1689*72ba2109SAndrey Churbanov     }
1690*72ba2109SAndrey Churbanov     tg->reduce_data = (void*)arr;
1691*72ba2109SAndrey Churbanov     tg->reduce_num_data = num;
1692*72ba2109SAndrey Churbanov     return (void*)tg;
1693*72ba2109SAndrey Churbanov }
1694*72ba2109SAndrey Churbanov 
1695*72ba2109SAndrey Churbanov /*!
1696*72ba2109SAndrey Churbanov @ingroup TASKING
1697*72ba2109SAndrey Churbanov @param gtid    Global thread ID
1698*72ba2109SAndrey Churbanov @param tskgrp  The taskgroup ID (optional)
1699*72ba2109SAndrey Churbanov @param data    Shared location of the item
1700*72ba2109SAndrey Churbanov @return The pointer to per-thread data
1701*72ba2109SAndrey Churbanov 
1702*72ba2109SAndrey Churbanov Get thread-specific location of data item
1703*72ba2109SAndrey Churbanov */
1704*72ba2109SAndrey Churbanov void*
1705*72ba2109SAndrey Churbanov __kmpc_task_reduction_get_th_data(int gtid, void *tskgrp, void *data)
1706*72ba2109SAndrey Churbanov {
1707*72ba2109SAndrey Churbanov     kmp_info_t * thread = __kmp_threads[gtid];
1708*72ba2109SAndrey Churbanov     kmp_int32 nth = thread->th.th_team_nproc;
1709*72ba2109SAndrey Churbanov     if (nth == 1)
1710*72ba2109SAndrey Churbanov         return data; // nothing to do
1711*72ba2109SAndrey Churbanov 
1712*72ba2109SAndrey Churbanov     kmp_taskgroup_t *tg = (kmp_taskgroup_t*)tskgrp;
1713*72ba2109SAndrey Churbanov     if (tg == NULL)
1714*72ba2109SAndrey Churbanov         tg = thread->th.th_current_task->td_taskgroup;
1715*72ba2109SAndrey Churbanov     KMP_ASSERT(tg != NULL);
1716*72ba2109SAndrey Churbanov     kmp_task_red_data_t *arr = (kmp_task_red_data_t*)(tg->reduce_data);
1717*72ba2109SAndrey Churbanov     kmp_int32 num = tg->reduce_num_data;
1718*72ba2109SAndrey Churbanov     kmp_int32 tid = thread->th.th_info.ds.ds_tid;
1719*72ba2109SAndrey Churbanov 
1720*72ba2109SAndrey Churbanov     KMP_ASSERT(data != NULL);
1721*72ba2109SAndrey Churbanov     while (tg != NULL) {
1722*72ba2109SAndrey Churbanov       for (int i = 0; i < num; ++i) {
1723*72ba2109SAndrey Churbanov         if (!arr[i].flags.lazy_priv) {
1724*72ba2109SAndrey Churbanov           if (data == arr[i].reduce_shar ||
1725*72ba2109SAndrey Churbanov              (data >= arr[i].reduce_priv && data < arr[i].reduce_pend))
1726*72ba2109SAndrey Churbanov             return (char*)(arr[i].reduce_priv) + tid * arr[i].reduce_size;
1727*72ba2109SAndrey Churbanov         } else {
1728*72ba2109SAndrey Churbanov           // check shared location first
1729*72ba2109SAndrey Churbanov           void **p_priv = (void**)(arr[i].reduce_priv);
1730*72ba2109SAndrey Churbanov           if (data == arr[i].reduce_shar)
1731*72ba2109SAndrey Churbanov             goto found;
1732*72ba2109SAndrey Churbanov           // check if we get some thread specific location as parameter
1733*72ba2109SAndrey Churbanov           for (int j = 0; j < nth; ++j)
1734*72ba2109SAndrey Churbanov             if (data == p_priv[j])
1735*72ba2109SAndrey Churbanov               goto found;
1736*72ba2109SAndrey Churbanov           continue; // not found, continue search
1737*72ba2109SAndrey Churbanov         found:
1738*72ba2109SAndrey Churbanov           if (p_priv[tid] == NULL) {
1739*72ba2109SAndrey Churbanov             // allocate thread specific object lazily
1740*72ba2109SAndrey Churbanov             void(*f_init)(void*) = (void(*)(void*))(arr[i].reduce_init);
1741*72ba2109SAndrey Churbanov             p_priv[tid] = __kmp_allocate(arr[i].reduce_size);
1742*72ba2109SAndrey Churbanov             if (f_init != NULL) {
1743*72ba2109SAndrey Churbanov               f_init(p_priv[tid]);
1744*72ba2109SAndrey Churbanov             }
1745*72ba2109SAndrey Churbanov           }
1746*72ba2109SAndrey Churbanov           return p_priv[tid];
1747*72ba2109SAndrey Churbanov         }
1748*72ba2109SAndrey Churbanov       }
1749*72ba2109SAndrey Churbanov       tg = tg->parent;
1750*72ba2109SAndrey Churbanov       arr = (kmp_task_red_data_t*)(tg->reduce_data);
1751*72ba2109SAndrey Churbanov       num = tg->reduce_num_data;
1752*72ba2109SAndrey Churbanov     }
1753*72ba2109SAndrey Churbanov     KMP_ASSERT2(0, "Unknown task reduction item");
1754*72ba2109SAndrey Churbanov     return NULL; // ERROR, this line never executed
1755*72ba2109SAndrey Churbanov }
1756*72ba2109SAndrey Churbanov 
1757*72ba2109SAndrey Churbanov // Finalize task reduction.
1758*72ba2109SAndrey Churbanov // Called from __kmpc_end_taskgroup()
1759*72ba2109SAndrey Churbanov static void
1760*72ba2109SAndrey Churbanov __kmp_task_reduction_fini(kmp_info_t *th, kmp_taskgroup_t *tg)
1761*72ba2109SAndrey Churbanov {
1762*72ba2109SAndrey Churbanov     kmp_int32 nth = th->th.th_team_nproc;
1763*72ba2109SAndrey Churbanov     KMP_DEBUG_ASSERT(nth > 1); // should not be called if nth == 1
1764*72ba2109SAndrey Churbanov     kmp_task_red_data_t *arr = (kmp_task_red_data_t*)tg->reduce_data;
1765*72ba2109SAndrey Churbanov     kmp_int32 num = tg->reduce_num_data;
1766*72ba2109SAndrey Churbanov     for (int i = 0; i < num; ++i) {
1767*72ba2109SAndrey Churbanov         void *sh_data = arr[i].reduce_shar;
1768*72ba2109SAndrey Churbanov         void(*f_fini)(void*) = (void(*)(void*))(arr[i].reduce_fini);
1769*72ba2109SAndrey Churbanov         void(*f_comb)(void*,void*) = (void(*)(void*,void*))(arr[i].reduce_comb);
1770*72ba2109SAndrey Churbanov         if (!arr[i].flags.lazy_priv) {
1771*72ba2109SAndrey Churbanov             void *pr_data = arr[i].reduce_priv;
1772*72ba2109SAndrey Churbanov             size_t size = arr[i].reduce_size;
1773*72ba2109SAndrey Churbanov             for (int j = 0; j < nth; ++j) {
1774*72ba2109SAndrey Churbanov                 void * priv_data = (char*)pr_data + j * size;
1775*72ba2109SAndrey Churbanov                 f_comb(sh_data, priv_data); // combine results
1776*72ba2109SAndrey Churbanov                 if (f_fini)
1777*72ba2109SAndrey Churbanov                     f_fini(priv_data); // finalize if needed
1778*72ba2109SAndrey Churbanov             }
1779*72ba2109SAndrey Churbanov         } else {
1780*72ba2109SAndrey Churbanov             void **pr_data = (void**)(arr[i].reduce_priv);
1781*72ba2109SAndrey Churbanov             for (int j = 0; j < nth; ++j) {
1782*72ba2109SAndrey Churbanov                 if (pr_data[j] != NULL) {
1783*72ba2109SAndrey Churbanov                     f_comb(sh_data, pr_data[j]); // combine results
1784*72ba2109SAndrey Churbanov                     if (f_fini)
1785*72ba2109SAndrey Churbanov                         f_fini(pr_data[j]); // finalize if needed
1786*72ba2109SAndrey Churbanov                     __kmp_free(pr_data[j]);
1787*72ba2109SAndrey Churbanov                 }
1788*72ba2109SAndrey Churbanov             }
1789*72ba2109SAndrey Churbanov         }
1790*72ba2109SAndrey Churbanov         __kmp_free(arr[i].reduce_priv);
1791*72ba2109SAndrey Churbanov     }
1792*72ba2109SAndrey Churbanov     __kmp_thread_free(th, arr);
1793*72ba2109SAndrey Churbanov     tg->reduce_data = NULL;
1794*72ba2109SAndrey Churbanov     tg->reduce_num_data = 0;
1795*72ba2109SAndrey Churbanov }
1796*72ba2109SAndrey Churbanov #endif
17977cc577a4SJonathan Peyton 
17987cc577a4SJonathan Peyton #if OMP_40_ENABLED
17997cc577a4SJonathan Peyton //-------------------------------------------------------------------------------------
18007cc577a4SJonathan Peyton // __kmpc_taskgroup: Start a new taskgroup
18017cc577a4SJonathan Peyton 
18027cc577a4SJonathan Peyton void
18037cc577a4SJonathan Peyton __kmpc_taskgroup( ident_t* loc, int gtid )
18047cc577a4SJonathan Peyton {
18057cc577a4SJonathan Peyton     kmp_info_t      * thread = __kmp_threads[ gtid ];
18067cc577a4SJonathan Peyton     kmp_taskdata_t  * taskdata = thread->th.th_current_task;
18077cc577a4SJonathan Peyton     kmp_taskgroup_t * tg_new =
18087cc577a4SJonathan Peyton         (kmp_taskgroup_t *)__kmp_thread_malloc( thread, sizeof( kmp_taskgroup_t ) );
18097cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_taskgroup: T#%d loc=%p group=%p\n", gtid, loc, tg_new) );
18107cc577a4SJonathan Peyton     tg_new->count = 0;
18117cc577a4SJonathan Peyton     tg_new->cancel_request = cancel_noreq;
18127cc577a4SJonathan Peyton     tg_new->parent = taskdata->td_taskgroup;
1813*72ba2109SAndrey Churbanov // TODO: change to OMP_50_ENABLED, need to change build tools for this to work
1814*72ba2109SAndrey Churbanov #if OMP_45_ENABLED
1815*72ba2109SAndrey Churbanov     tg_new->reduce_data = NULL;
1816*72ba2109SAndrey Churbanov     tg_new->reduce_num_data = 0;
1817*72ba2109SAndrey Churbanov #endif
18187cc577a4SJonathan Peyton     taskdata->td_taskgroup = tg_new;
18197cc577a4SJonathan Peyton }
18207cc577a4SJonathan Peyton 
18217cc577a4SJonathan Peyton 
18227cc577a4SJonathan Peyton //-------------------------------------------------------------------------------------
18237cc577a4SJonathan Peyton // __kmpc_end_taskgroup: Wait until all tasks generated by the current task
18247cc577a4SJonathan Peyton //                       and its descendants are complete
18257cc577a4SJonathan Peyton 
18267cc577a4SJonathan Peyton void
18277cc577a4SJonathan Peyton __kmpc_end_taskgroup( ident_t* loc, int gtid )
18287cc577a4SJonathan Peyton {
18297cc577a4SJonathan Peyton     kmp_info_t      * thread = __kmp_threads[ gtid ];
18307cc577a4SJonathan Peyton     kmp_taskdata_t  * taskdata = thread->th.th_current_task;
18317cc577a4SJonathan Peyton     kmp_taskgroup_t * taskgroup = taskdata->td_taskgroup;
18327cc577a4SJonathan Peyton     int thread_finished = FALSE;
18337cc577a4SJonathan Peyton 
18347cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_end_taskgroup(enter): T#%d loc=%p\n", gtid, loc) );
18357cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskgroup != NULL );
18367cc577a4SJonathan Peyton     KMP_SET_THREAD_STATE_BLOCK(TASKGROUP);
18377cc577a4SJonathan Peyton 
18387cc577a4SJonathan Peyton     if ( __kmp_tasking_mode != tskm_immediate_exec ) {
18397cc577a4SJonathan Peyton #if USE_ITT_BUILD
18407cc577a4SJonathan Peyton         // For ITT the taskgroup wait is similar to taskwait until we need to distinguish them
18417cc577a4SJonathan Peyton         void * itt_sync_obj = __kmp_itt_taskwait_object( gtid );
18427cc577a4SJonathan Peyton         if ( itt_sync_obj != NULL )
18437cc577a4SJonathan Peyton             __kmp_itt_taskwait_starting( gtid, itt_sync_obj );
18447cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
18457cc577a4SJonathan Peyton 
18467cc577a4SJonathan Peyton #if OMP_45_ENABLED
18477cc577a4SJonathan Peyton         if ( ! taskdata->td_flags.team_serial || (thread->th.th_task_team != NULL && thread->th.th_task_team->tt.tt_found_proxy_tasks) )
18487cc577a4SJonathan Peyton #else
18497cc577a4SJonathan Peyton         if ( ! taskdata->td_flags.team_serial )
18507cc577a4SJonathan Peyton #endif
18517cc577a4SJonathan Peyton         {
18527cc577a4SJonathan Peyton             kmp_flag_32 flag(&(taskgroup->count), 0U);
18537cc577a4SJonathan Peyton             while ( TCR_4(taskgroup->count) != 0 ) {
18547cc577a4SJonathan Peyton                 flag.execute_tasks(thread, gtid, FALSE, &thread_finished
18557cc577a4SJonathan Peyton                                    USE_ITT_BUILD_ARG(itt_sync_obj), __kmp_task_stealing_constraint );
18567cc577a4SJonathan Peyton             }
18577cc577a4SJonathan Peyton         }
18587cc577a4SJonathan Peyton 
18597cc577a4SJonathan Peyton #if USE_ITT_BUILD
18607cc577a4SJonathan Peyton         if ( itt_sync_obj != NULL )
18617cc577a4SJonathan Peyton             __kmp_itt_taskwait_finished( gtid, itt_sync_obj );
18627cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
18637cc577a4SJonathan Peyton     }
18647cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskgroup->count == 0 );
18657cc577a4SJonathan Peyton 
1866*72ba2109SAndrey Churbanov // TODO: change to OMP_50_ENABLED, need to change build tools for this to work
1867*72ba2109SAndrey Churbanov #if OMP_45_ENABLED
1868*72ba2109SAndrey Churbanov     if( taskgroup->reduce_data != NULL ) // need to reduce?
1869*72ba2109SAndrey Churbanov         __kmp_task_reduction_fini(thread, taskgroup);
1870*72ba2109SAndrey Churbanov #endif
18717cc577a4SJonathan Peyton     // Restore parent taskgroup for the current task
18727cc577a4SJonathan Peyton     taskdata->td_taskgroup = taskgroup->parent;
18737cc577a4SJonathan Peyton     __kmp_thread_free( thread, taskgroup );
18747cc577a4SJonathan Peyton 
18757cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_end_taskgroup(exit): T#%d task %p finished waiting\n", gtid, taskdata) );
18767cc577a4SJonathan Peyton     ANNOTATE_HAPPENS_AFTER(taskdata);
18777cc577a4SJonathan Peyton }
18787cc577a4SJonathan Peyton #endif
18797cc577a4SJonathan Peyton 
18807cc577a4SJonathan Peyton 
18817cc577a4SJonathan Peyton //------------------------------------------------------
18827cc577a4SJonathan Peyton // __kmp_remove_my_task: remove a task from my own deque
18837cc577a4SJonathan Peyton 
18847cc577a4SJonathan Peyton static kmp_task_t *
18857cc577a4SJonathan Peyton __kmp_remove_my_task( kmp_info_t * thread, kmp_int32 gtid, kmp_task_team_t *task_team,
18867cc577a4SJonathan Peyton                       kmp_int32 is_constrained )
18877cc577a4SJonathan Peyton {
18887cc577a4SJonathan Peyton     kmp_task_t * task;
18897cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata;
18907cc577a4SJonathan Peyton     kmp_thread_data_t *thread_data;
18917cc577a4SJonathan Peyton     kmp_uint32 tail;
18927cc577a4SJonathan Peyton 
18937cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( __kmp_tasking_mode != tskm_immediate_exec );
18947cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task_team -> tt.tt_threads_data != NULL ); // Caller should check this condition
18957cc577a4SJonathan Peyton 
18967cc577a4SJonathan Peyton         thread_data = & task_team -> tt.tt_threads_data[ __kmp_tid_from_gtid( gtid ) ];
18977cc577a4SJonathan Peyton 
18987cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_remove_my_task(enter): T#%d ntasks=%d head=%u tail=%u\n",
18997cc577a4SJonathan Peyton                   gtid, thread_data->td.td_deque_ntasks, thread_data->td.td_deque_head,
19007cc577a4SJonathan Peyton                   thread_data->td.td_deque_tail) );
19017cc577a4SJonathan Peyton 
19027cc577a4SJonathan Peyton     if (TCR_4(thread_data -> td.td_deque_ntasks) == 0) {
19037cc577a4SJonathan Peyton         KA_TRACE(10, ("__kmp_remove_my_task(exit #1): T#%d No tasks to remove: ntasks=%d head=%u tail=%u\n",
19047cc577a4SJonathan Peyton                       gtid, thread_data->td.td_deque_ntasks, thread_data->td.td_deque_head,
19057cc577a4SJonathan Peyton                       thread_data->td.td_deque_tail) );
19067cc577a4SJonathan Peyton         return NULL;
19077cc577a4SJonathan Peyton     }
19087cc577a4SJonathan Peyton 
19097cc577a4SJonathan Peyton     __kmp_acquire_bootstrap_lock( & thread_data -> td.td_deque_lock );
19107cc577a4SJonathan Peyton 
19117cc577a4SJonathan Peyton     if (TCR_4(thread_data -> td.td_deque_ntasks) == 0) {
19127cc577a4SJonathan Peyton         __kmp_release_bootstrap_lock( & thread_data -> td.td_deque_lock );
19137cc577a4SJonathan Peyton         KA_TRACE(10, ("__kmp_remove_my_task(exit #2): T#%d No tasks to remove: ntasks=%d head=%u tail=%u\n",
19147cc577a4SJonathan Peyton                       gtid, thread_data->td.td_deque_ntasks, thread_data->td.td_deque_head,
19157cc577a4SJonathan Peyton                       thread_data->td.td_deque_tail) );
19167cc577a4SJonathan Peyton         return NULL;
19177cc577a4SJonathan Peyton     }
19187cc577a4SJonathan Peyton 
19197cc577a4SJonathan Peyton     tail = ( thread_data -> td.td_deque_tail - 1 ) & 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 ) {
19297cc577a4SJonathan Peyton             parent = parent->td_parent;  // check generation up to the level of the current task
19307cc577a4SJonathan Peyton             KMP_DEBUG_ASSERT(parent != NULL);
19317cc577a4SJonathan Peyton         }
19327cc577a4SJonathan Peyton         if ( parent != current ) {
19337cc577a4SJonathan Peyton             // If the tail task is not a child, then no other child can appear in the deque.
19347cc577a4SJonathan Peyton             __kmp_release_bootstrap_lock( & thread_data -> td.td_deque_lock );
19357cc577a4SJonathan Peyton             KA_TRACE(10, ("__kmp_remove_my_task(exit #2): T#%d No tasks to remove: ntasks=%d head=%u tail=%u\n",
19367cc577a4SJonathan Peyton                           gtid, thread_data->td.td_deque_ntasks, thread_data->td.td_deque_head,
19377cc577a4SJonathan Peyton                           thread_data->td.td_deque_tail) );
19387cc577a4SJonathan Peyton             return NULL;
19397cc577a4SJonathan Peyton         }
19407cc577a4SJonathan Peyton     }
19417cc577a4SJonathan Peyton 
19427cc577a4SJonathan Peyton     thread_data -> td.td_deque_tail = tail;
19437cc577a4SJonathan Peyton     TCW_4(thread_data -> td.td_deque_ntasks, thread_data -> td.td_deque_ntasks - 1);
19447cc577a4SJonathan Peyton 
19457cc577a4SJonathan Peyton     __kmp_release_bootstrap_lock( & thread_data->td.td_deque_lock );
19467cc577a4SJonathan Peyton 
19477cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_remove_my_task(exit #2): T#%d task %p removed: ntasks=%d head=%u tail=%u\n",
19487cc577a4SJonathan Peyton                   gtid, taskdata, thread_data->td.td_deque_ntasks, thread_data->td.td_deque_head,
19497cc577a4SJonathan Peyton                   thread_data->td.td_deque_tail) );
19507cc577a4SJonathan Peyton 
19517cc577a4SJonathan Peyton     task = KMP_TASKDATA_TO_TASK( taskdata );
19527cc577a4SJonathan Peyton     return task;
19537cc577a4SJonathan Peyton }
19547cc577a4SJonathan Peyton 
19557cc577a4SJonathan Peyton 
19567cc577a4SJonathan Peyton //-----------------------------------------------------------
19577cc577a4SJonathan Peyton // __kmp_steal_task: remove a task from another thread's deque
19587cc577a4SJonathan Peyton // Assume that calling thread has already checked existence of
19597cc577a4SJonathan Peyton // task_team thread_data before calling this routine.
19607cc577a4SJonathan Peyton 
19617cc577a4SJonathan Peyton static kmp_task_t *
19627cc577a4SJonathan Peyton __kmp_steal_task( kmp_info_t *victim, kmp_int32 gtid, kmp_task_team_t *task_team,
19637cc577a4SJonathan Peyton                   volatile kmp_uint32 *unfinished_threads, int *thread_finished,
19647cc577a4SJonathan Peyton                   kmp_int32 is_constrained )
19657cc577a4SJonathan Peyton {
19667cc577a4SJonathan Peyton     kmp_task_t * task;
19677cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata;
19687cc577a4SJonathan Peyton     kmp_thread_data_t *victim_td, *threads_data;
19697cc577a4SJonathan Peyton     kmp_int32 victim_tid;
19707cc577a4SJonathan Peyton 
19717cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( __kmp_tasking_mode != tskm_immediate_exec );
19727cc577a4SJonathan Peyton 
19737cc577a4SJonathan Peyton     threads_data = task_team -> tt.tt_threads_data;
19747cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( threads_data != NULL );  // Caller should check this condition
19757cc577a4SJonathan Peyton 
19767cc577a4SJonathan Peyton     victim_tid = victim->th.th_info.ds.ds_tid;
19777cc577a4SJonathan Peyton     victim_td = & threads_data[ victim_tid ];
19787cc577a4SJonathan Peyton 
19797cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_steal_task(enter): T#%d try to steal from T#%d: task_team=%p ntasks=%d "
19807cc577a4SJonathan Peyton                   "head=%u tail=%u\n",
19817cc577a4SJonathan Peyton                   gtid, __kmp_gtid_from_thread( victim ), task_team, victim_td->td.td_deque_ntasks,
19827cc577a4SJonathan Peyton                   victim_td->td.td_deque_head, victim_td->td.td_deque_tail) );
19837cc577a4SJonathan Peyton 
19847cc577a4SJonathan Peyton     if ( (TCR_4(victim_td -> td.td_deque_ntasks) == 0) || // Caller should not check this condition
19857cc577a4SJonathan Peyton          (TCR_PTR(victim->th.th_task_team) != task_team)) // GEH: why would this happen?
19867cc577a4SJonathan Peyton     {
19877cc577a4SJonathan Peyton         KA_TRACE(10, ("__kmp_steal_task(exit #1): T#%d could not steal from T#%d: task_team=%p "
19887cc577a4SJonathan Peyton                       "ntasks=%d head=%u tail=%u\n",
19897cc577a4SJonathan Peyton                       gtid, __kmp_gtid_from_thread( victim ), task_team, victim_td->td.td_deque_ntasks,
19907cc577a4SJonathan Peyton                       victim_td->td.td_deque_head, victim_td->td.td_deque_tail) );
19917cc577a4SJonathan Peyton         return NULL;
19927cc577a4SJonathan Peyton     }
19937cc577a4SJonathan Peyton 
19947cc577a4SJonathan Peyton     __kmp_acquire_bootstrap_lock( & victim_td -> td.td_deque_lock );
19957cc577a4SJonathan Peyton 
19967cc577a4SJonathan Peyton     // Check again after we acquire the lock
19977cc577a4SJonathan Peyton     if ( (TCR_4(victim_td -> td.td_deque_ntasks) == 0) ||
19987cc577a4SJonathan Peyton          (TCR_PTR(victim->th.th_task_team) != task_team)) // GEH: why would this happen?
19997cc577a4SJonathan Peyton     {
20007cc577a4SJonathan Peyton         __kmp_release_bootstrap_lock( & victim_td -> td.td_deque_lock );
20017cc577a4SJonathan Peyton         KA_TRACE(10, ("__kmp_steal_task(exit #2): T#%d could not steal from T#%d: task_team=%p "
20027cc577a4SJonathan Peyton                       "ntasks=%d head=%u tail=%u\n",
20037cc577a4SJonathan Peyton                       gtid, __kmp_gtid_from_thread( victim ), task_team, victim_td->td.td_deque_ntasks,
20047cc577a4SJonathan Peyton                       victim_td->td.td_deque_head, victim_td->td.td_deque_tail) );
20057cc577a4SJonathan Peyton         return NULL;
20067cc577a4SJonathan Peyton     }
20077cc577a4SJonathan Peyton 
20087cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( victim_td -> td.td_deque != NULL );
20097cc577a4SJonathan Peyton 
20107cc577a4SJonathan Peyton     taskdata = victim_td->td.td_deque[victim_td->td.td_deque_head];
20117cc577a4SJonathan Peyton     if ( is_constrained ) {
20127cc577a4SJonathan Peyton         // we need to check if the candidate obeys task scheduling constraint:
20137cc577a4SJonathan Peyton         // only descendant of current task can be scheduled
20147cc577a4SJonathan Peyton         kmp_taskdata_t * current = __kmp_threads[ gtid ]->th.th_current_task;
20157cc577a4SJonathan Peyton         kmp_int32        level = current->td_level;
20167cc577a4SJonathan Peyton         kmp_taskdata_t * parent = taskdata->td_parent;
20177cc577a4SJonathan Peyton         while ( parent != current && parent->td_level > level ) {
20187cc577a4SJonathan Peyton             parent = parent->td_parent;  // check generation up to the level of the current task
20197cc577a4SJonathan Peyton             KMP_DEBUG_ASSERT(parent != NULL);
20207cc577a4SJonathan Peyton         }
20217cc577a4SJonathan Peyton         if ( parent != current ) {
20227cc577a4SJonathan Peyton             // If the head task is not a descendant of the current task then do not
20237cc577a4SJonathan Peyton             // steal it. No other task in victim's deque can be a descendant of the
20247cc577a4SJonathan Peyton             // current task.
20257cc577a4SJonathan Peyton             __kmp_release_bootstrap_lock( & victim_td -> td.td_deque_lock );
20267cc577a4SJonathan Peyton             KA_TRACE(10, ("__kmp_steal_task(exit #2): T#%d could not steal from T#%d: task_team=%p "
20277cc577a4SJonathan Peyton                           "ntasks=%d head=%u tail=%u\n",
20287cc577a4SJonathan Peyton                           gtid, __kmp_gtid_from_thread( threads_data[victim_tid].td.td_thr ),
20297cc577a4SJonathan Peyton                           task_team, victim_td->td.td_deque_ntasks,
20307cc577a4SJonathan Peyton                           victim_td->td.td_deque_head, victim_td->td.td_deque_tail) );
20317cc577a4SJonathan Peyton             return NULL;
20327cc577a4SJonathan Peyton         }
20337cc577a4SJonathan Peyton     }
20347cc577a4SJonathan Peyton     // Bump head pointer and Wrap.
20357cc577a4SJonathan Peyton     victim_td->td.td_deque_head = (victim_td->td.td_deque_head + 1) & TASK_DEQUE_MASK(victim_td->td);
20367cc577a4SJonathan Peyton     if (*thread_finished) {
20377cc577a4SJonathan Peyton         // We need to un-mark this victim as a finished victim.  This must be done before
20387cc577a4SJonathan Peyton         // releasing the lock, or else other threads (starting with the master victim)
20397cc577a4SJonathan Peyton         // might be prematurely released from the barrier!!!
20407cc577a4SJonathan Peyton         kmp_uint32 count;
20417cc577a4SJonathan Peyton 
20427cc577a4SJonathan Peyton         count = KMP_TEST_THEN_INC32( (kmp_int32 *)unfinished_threads );
20437cc577a4SJonathan Peyton 
20447cc577a4SJonathan Peyton         KA_TRACE(20, ("__kmp_steal_task: T#%d inc unfinished_threads to %d: task_team=%p\n",
20457cc577a4SJonathan Peyton                       gtid, count + 1, task_team) );
20467cc577a4SJonathan Peyton 
20477cc577a4SJonathan Peyton         *thread_finished = FALSE;
20487cc577a4SJonathan Peyton     }
20497cc577a4SJonathan Peyton     TCW_4(victim_td -> td.td_deque_ntasks, TCR_4(victim_td -> td.td_deque_ntasks) - 1);
20507cc577a4SJonathan Peyton 
20517cc577a4SJonathan Peyton     __kmp_release_bootstrap_lock( & victim_td -> td.td_deque_lock );
20527cc577a4SJonathan Peyton 
20537cc577a4SJonathan Peyton     KMP_COUNT_BLOCK(TASK_stolen);
20547cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_steal_task(exit #3): T#%d stole task %p from T#%d: task_team=%p "
20557cc577a4SJonathan Peyton                   "ntasks=%d head=%u tail=%u\n",
20567cc577a4SJonathan Peyton                   gtid, taskdata, __kmp_gtid_from_thread( victim ), task_team,
20577cc577a4SJonathan Peyton                   victim_td->td.td_deque_ntasks, victim_td->td.td_deque_head,
20587cc577a4SJonathan Peyton                   victim_td->td.td_deque_tail) );
20597cc577a4SJonathan Peyton 
20607cc577a4SJonathan Peyton     task = KMP_TASKDATA_TO_TASK( taskdata );
20617cc577a4SJonathan Peyton     return task;
20627cc577a4SJonathan Peyton }
20637cc577a4SJonathan Peyton 
20647cc577a4SJonathan Peyton 
20657cc577a4SJonathan Peyton //-----------------------------------------------------------------------------
20667cc577a4SJonathan Peyton // __kmp_execute_tasks_template: Choose and execute tasks until either the condition
20677cc577a4SJonathan Peyton // is statisfied (return true) or there are none left (return false).
20687cc577a4SJonathan Peyton // final_spin is TRUE if this is the spin at the release barrier.
20697cc577a4SJonathan Peyton // thread_finished indicates whether the thread is finished executing all
20707cc577a4SJonathan Peyton // the tasks it has on its deque, and is at the release barrier.
20717cc577a4SJonathan Peyton // spinner is the location on which to spin.
20727cc577a4SJonathan Peyton // spinner == NULL means only execute a single task and return.
20737cc577a4SJonathan Peyton // checker is the value to check to terminate the spin.
20747cc577a4SJonathan Peyton template <class C>
20757cc577a4SJonathan Peyton static inline int __kmp_execute_tasks_template(kmp_info_t *thread, kmp_int32 gtid, C *flag, int final_spin,
20767cc577a4SJonathan Peyton                                                int *thread_finished
20777cc577a4SJonathan Peyton                                                USE_ITT_BUILD_ARG(void * itt_sync_obj), kmp_int32 is_constrained)
20787cc577a4SJonathan Peyton {
20797cc577a4SJonathan Peyton     kmp_task_team_t *     task_team = thread->th.th_task_team;
20807cc577a4SJonathan Peyton     kmp_thread_data_t *   threads_data;
20817cc577a4SJonathan Peyton     kmp_task_t *          task;
20827cc577a4SJonathan Peyton     kmp_info_t *          other_thread;
20837cc577a4SJonathan Peyton     kmp_taskdata_t *      current_task = thread -> th.th_current_task;
20847cc577a4SJonathan Peyton     volatile kmp_uint32 * unfinished_threads;
20857cc577a4SJonathan Peyton     kmp_int32             nthreads, victim=-2, use_own_tasks=1, new_victim=0, tid=thread->th.th_info.ds.ds_tid;
20867cc577a4SJonathan Peyton 
20877cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( __kmp_tasking_mode != tskm_immediate_exec );
20887cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( thread == __kmp_threads[ gtid ] );
20897cc577a4SJonathan Peyton 
20907cc577a4SJonathan Peyton     if (task_team == NULL) return FALSE;
20917cc577a4SJonathan Peyton 
20927cc577a4SJonathan Peyton     KA_TRACE(15, ("__kmp_execute_tasks_template(enter): T#%d final_spin=%d *thread_finished=%d\n",
20937cc577a4SJonathan Peyton                   gtid, final_spin, *thread_finished) );
20947cc577a4SJonathan Peyton 
2095581490e7SAndrey Churbanov     thread->th.th_reap_state = KMP_NOT_SAFE_TO_REAP;
20967cc577a4SJonathan Peyton     threads_data = (kmp_thread_data_t *)TCR_PTR(task_team -> tt.tt_threads_data);
20977cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( threads_data != NULL );
20987cc577a4SJonathan Peyton 
20997cc577a4SJonathan Peyton     nthreads = task_team -> tt.tt_nproc;
21007cc577a4SJonathan Peyton     unfinished_threads = &(task_team -> tt.tt_unfinished_threads);
21017cc577a4SJonathan Peyton #if OMP_45_ENABLED
21027cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( nthreads > 1 || task_team->tt.tt_found_proxy_tasks);
21037cc577a4SJonathan Peyton #else
21047cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( nthreads > 1 );
21057cc577a4SJonathan Peyton #endif
21067cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( (int)(TCR_4(*unfinished_threads)) >= 0 );
21077cc577a4SJonathan Peyton 
21087cc577a4SJonathan Peyton     while (1) { // Outer loop keeps trying to find tasks in case of single thread getting tasks from target constructs
21097cc577a4SJonathan Peyton         while (1) { // Inner loop to find a task and execute it
21107cc577a4SJonathan Peyton             task = NULL;
21117cc577a4SJonathan Peyton             if (use_own_tasks) { // check on own queue first
21127cc577a4SJonathan Peyton                 task = __kmp_remove_my_task( thread, gtid, task_team, is_constrained );
21137cc577a4SJonathan Peyton             }
21147cc577a4SJonathan Peyton             if ((task == NULL) && (nthreads > 1)) { // Steal a task
21157cc577a4SJonathan Peyton                 int asleep = 1;
21167cc577a4SJonathan Peyton                 use_own_tasks = 0;
21177cc577a4SJonathan Peyton                 // Try to steal from the last place I stole from successfully.
21187cc577a4SJonathan Peyton                 if (victim == -2) { // haven't stolen anything yet
21197cc577a4SJonathan Peyton                     victim = threads_data[tid].td.td_deque_last_stolen;
21207cc577a4SJonathan Peyton                     if (victim != -1) // if we have a last stolen from victim, get the thread
21217cc577a4SJonathan Peyton                         other_thread = threads_data[victim].td.td_thr;
21227cc577a4SJonathan Peyton                 }
21237cc577a4SJonathan Peyton                 if (victim != -1) { // found last victim
21247cc577a4SJonathan Peyton                     asleep = 0;
21257cc577a4SJonathan Peyton                 }
21267cc577a4SJonathan Peyton                 else if (!new_victim) { // no recent steals and we haven't already used a new victim; select a random thread
21277cc577a4SJonathan Peyton                     do { // Find a different thread to steal work from.
21287cc577a4SJonathan Peyton                         // Pick a random thread. Initial plan was to cycle through all the threads, and only return if
21297cc577a4SJonathan Peyton                         // we tried to steal from every thread, and failed.  Arch says that's not such a great idea.
21307cc577a4SJonathan Peyton                         victim = __kmp_get_random(thread) % (nthreads - 1);
21317cc577a4SJonathan Peyton                         if (victim >= tid) {
21327cc577a4SJonathan Peyton                             ++victim;  // Adjusts random distribution to exclude self
21337cc577a4SJonathan Peyton                         }
21347cc577a4SJonathan Peyton                         // Found a potential victim
21357cc577a4SJonathan Peyton                         other_thread = threads_data[victim].td.td_thr;
21367cc577a4SJonathan Peyton                         // There is a slight chance that __kmp_enable_tasking() did not wake up all threads
21377cc577a4SJonathan Peyton                         // waiting at the barrier.  If victim is sleeping, then wake it up.  Since we were going to
21387cc577a4SJonathan Peyton                         // pay the cache miss penalty for referencing another thread's kmp_info_t struct anyway,
21397cc577a4SJonathan Peyton                         // the check shouldn't cost too much performance at this point. In extra barrier mode, tasks
21407cc577a4SJonathan Peyton                         // do not sleep at the separate tasking barrier, so this isn't a problem.
21417cc577a4SJonathan Peyton                         asleep = 0;
21427cc577a4SJonathan Peyton                         if ( ( __kmp_tasking_mode == tskm_task_teams ) &&
21437cc577a4SJonathan Peyton                              (__kmp_dflt_blocktime != KMP_MAX_BLOCKTIME) &&
21447cc577a4SJonathan Peyton                              (TCR_PTR(other_thread->th.th_sleep_loc) != NULL)) {
21457cc577a4SJonathan Peyton                             asleep = 1;
21467cc577a4SJonathan Peyton                             __kmp_null_resume_wrapper(__kmp_gtid_from_thread(other_thread), other_thread->th.th_sleep_loc);
21477cc577a4SJonathan Peyton                             // A sleeping thread should not have any tasks on it's queue. There is a slight
21487cc577a4SJonathan Peyton                             // possibility that it resumes, steals a task from another thread, which spawns more
21497cc577a4SJonathan Peyton                             // tasks, all in the time that it takes this thread to check => don't write an assertion
21507cc577a4SJonathan Peyton                             // that the victim's queue is empty.  Try stealing from a different thread.
21517cc577a4SJonathan Peyton                         }
21527cc577a4SJonathan Peyton                     } while (asleep);
21537cc577a4SJonathan Peyton                 }
21547cc577a4SJonathan Peyton 
21557cc577a4SJonathan Peyton                 if (!asleep) {
21567cc577a4SJonathan Peyton                     // We have a victim to try to steal from
21577cc577a4SJonathan Peyton                     task = __kmp_steal_task(other_thread, gtid, task_team, unfinished_threads, thread_finished, is_constrained);
21587cc577a4SJonathan Peyton                 }
21597cc577a4SJonathan Peyton                 if (task != NULL) { // set last stolen to victim
21607cc577a4SJonathan Peyton                     if (threads_data[tid].td.td_deque_last_stolen != victim) {
21617cc577a4SJonathan Peyton                         threads_data[tid].td.td_deque_last_stolen = victim;
21627cc577a4SJonathan Peyton                         // The pre-refactored code did not try more than 1 successful new vicitm,
21637cc577a4SJonathan Peyton                         // unless the last one generated more local tasks; new_victim keeps track of this
21647cc577a4SJonathan Peyton                         new_victim = 1;
21657cc577a4SJonathan Peyton                     }
21667cc577a4SJonathan Peyton                 }
21677cc577a4SJonathan Peyton                 else { // No tasks found; unset last_stolen
21687cc577a4SJonathan Peyton                     KMP_CHECK_UPDATE(threads_data[tid].td.td_deque_last_stolen, -1);
21697cc577a4SJonathan Peyton                     victim = -2; // no successful victim found
21707cc577a4SJonathan Peyton                 }
21717cc577a4SJonathan Peyton             }
21727cc577a4SJonathan Peyton 
21737cc577a4SJonathan Peyton             if (task == NULL) // break out of tasking loop
21747cc577a4SJonathan Peyton                 break;
21757cc577a4SJonathan Peyton 
21767cc577a4SJonathan Peyton             // Found a task; execute it
21777cc577a4SJonathan Peyton #if USE_ITT_BUILD && USE_ITT_NOTIFY
21787cc577a4SJonathan Peyton             if ( __itt_sync_create_ptr || KMP_ITT_DEBUG ) {
21797cc577a4SJonathan Peyton                 if ( itt_sync_obj == NULL ) { // we are at fork barrier where we could not get the object reliably
21807cc577a4SJonathan Peyton                     itt_sync_obj = __kmp_itt_barrier_object( gtid, bs_forkjoin_barrier );
21817cc577a4SJonathan Peyton                 }
21827cc577a4SJonathan Peyton                 __kmp_itt_task_starting( itt_sync_obj );
21837cc577a4SJonathan Peyton             }
21847cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD && USE_ITT_NOTIFY */
21857cc577a4SJonathan Peyton             __kmp_invoke_task( gtid, task, current_task );
21867cc577a4SJonathan Peyton #if USE_ITT_BUILD
21877cc577a4SJonathan Peyton             if ( itt_sync_obj != NULL ) __kmp_itt_task_finished( itt_sync_obj );
21887cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
21897cc577a4SJonathan Peyton             // If this thread is only partway through the barrier and the condition is met, then return now,
21907cc577a4SJonathan Peyton             // so that the barrier gather/release pattern can proceed. If this thread is in the last spin loop
21917cc577a4SJonathan Peyton             // in the barrier, waiting to be released, we know that the termination condition will not be
21927cc577a4SJonathan Peyton             // satisified, so don't waste any cycles checking it.
21937cc577a4SJonathan Peyton             if (flag == NULL || (!final_spin && flag->done_check())) {
21947cc577a4SJonathan Peyton                 KA_TRACE(15, ("__kmp_execute_tasks_template: T#%d spin condition satisfied\n", gtid) );
21957cc577a4SJonathan Peyton                 return TRUE;
21967cc577a4SJonathan Peyton             }
21977cc577a4SJonathan Peyton             if (thread->th.th_task_team == NULL) {
21987cc577a4SJonathan Peyton                 break;
21997cc577a4SJonathan Peyton             }
22007cc577a4SJonathan Peyton             KMP_YIELD( __kmp_library == library_throughput );   // Yield before executing next task
22017cc577a4SJonathan Peyton             // If execution of a stolen task results in more tasks being placed on our run queue, reset use_own_tasks
22027cc577a4SJonathan Peyton             if (!use_own_tasks && TCR_4(threads_data[tid].td.td_deque_ntasks) != 0) {
22037cc577a4SJonathan Peyton                 KA_TRACE(20, ("__kmp_execute_tasks_template: T#%d stolen task spawned other tasks, restart\n", gtid));
22047cc577a4SJonathan Peyton                 use_own_tasks = 1;
22057cc577a4SJonathan Peyton                 new_victim = 0;
22067cc577a4SJonathan Peyton             }
22077cc577a4SJonathan Peyton         }
22087cc577a4SJonathan Peyton 
22097cc577a4SJonathan Peyton         // The task source has been exhausted. If in final spin loop of barrier, check if termination condition is satisfied.
22107cc577a4SJonathan Peyton #if OMP_45_ENABLED
22117cc577a4SJonathan Peyton         // The work queue may be empty but there might be proxy tasks still executing
22127cc577a4SJonathan Peyton         if (final_spin && TCR_4(current_task->td_incomplete_child_tasks) == 0)
22137cc577a4SJonathan Peyton #else
22147cc577a4SJonathan Peyton         if (final_spin)
22157cc577a4SJonathan Peyton #endif
22167cc577a4SJonathan Peyton         {
22177cc577a4SJonathan Peyton             // First, decrement the #unfinished threads, if that has not already been done.  This decrement
22187cc577a4SJonathan Peyton             // might be to the spin location, and result in the termination condition being satisfied.
22197cc577a4SJonathan Peyton             if (! *thread_finished) {
22207cc577a4SJonathan Peyton                 kmp_uint32 count;
22217cc577a4SJonathan Peyton 
22227cc577a4SJonathan Peyton                 count = KMP_TEST_THEN_DEC32( (kmp_int32 *)unfinished_threads ) - 1;
22237cc577a4SJonathan Peyton                 KA_TRACE(20, ("__kmp_execute_tasks_template: T#%d dec unfinished_threads to %d task_team=%p\n",
22247cc577a4SJonathan Peyton                               gtid, count, task_team) );
22257cc577a4SJonathan Peyton                 *thread_finished = TRUE;
22267cc577a4SJonathan Peyton             }
22277cc577a4SJonathan Peyton 
22287cc577a4SJonathan Peyton             // It is now unsafe to reference thread->th.th_team !!!
22297cc577a4SJonathan Peyton             // Decrementing task_team->tt.tt_unfinished_threads can allow the master thread to pass through
22307cc577a4SJonathan Peyton             // the barrier, where it might reset each thread's th.th_team field for the next parallel region.
22317cc577a4SJonathan Peyton             // If we can steal more work, we know that this has not happened yet.
22327cc577a4SJonathan Peyton             if (flag != NULL && flag->done_check()) {
22337cc577a4SJonathan Peyton                 KA_TRACE(15, ("__kmp_execute_tasks_template: T#%d spin condition satisfied\n", gtid) );
22347cc577a4SJonathan Peyton                 return TRUE;
22357cc577a4SJonathan Peyton             }
22367cc577a4SJonathan Peyton         }
22377cc577a4SJonathan Peyton 
22387cc577a4SJonathan Peyton         // If this thread's task team is NULL, master has recognized that there are no more tasks; bail out
22397cc577a4SJonathan Peyton         if (thread->th.th_task_team == NULL) {
22407cc577a4SJonathan Peyton             KA_TRACE(15, ("__kmp_execute_tasks_template: T#%d no more tasks\n", gtid) );
22417cc577a4SJonathan Peyton             return FALSE;
22427cc577a4SJonathan Peyton         }
22437cc577a4SJonathan Peyton 
22447cc577a4SJonathan Peyton #if OMP_45_ENABLED
22457cc577a4SJonathan Peyton         // We could be getting tasks from target constructs; if this is the only thread, keep trying to execute
22467cc577a4SJonathan Peyton         // tasks from own queue
22477cc577a4SJonathan Peyton         if (nthreads == 1)
22487cc577a4SJonathan Peyton             use_own_tasks = 1;
22497cc577a4SJonathan Peyton         else
22507cc577a4SJonathan Peyton #endif
22517cc577a4SJonathan Peyton         {
22527cc577a4SJonathan Peyton             KA_TRACE(15, ("__kmp_execute_tasks_template: T#%d can't find work\n", gtid) );
22537cc577a4SJonathan Peyton             return FALSE;
22547cc577a4SJonathan Peyton         }
22557cc577a4SJonathan Peyton     }
22567cc577a4SJonathan Peyton }
22577cc577a4SJonathan Peyton 
22587cc577a4SJonathan Peyton int __kmp_execute_tasks_32(kmp_info_t *thread, kmp_int32 gtid, kmp_flag_32 *flag, int final_spin,
22597cc577a4SJonathan Peyton                            int *thread_finished
22607cc577a4SJonathan Peyton                            USE_ITT_BUILD_ARG(void * itt_sync_obj), kmp_int32 is_constrained)
22617cc577a4SJonathan Peyton {
22627cc577a4SJonathan Peyton     return __kmp_execute_tasks_template(thread, gtid, flag, final_spin, thread_finished
22637cc577a4SJonathan Peyton                                         USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained);
22647cc577a4SJonathan Peyton }
22657cc577a4SJonathan Peyton 
22667cc577a4SJonathan Peyton int __kmp_execute_tasks_64(kmp_info_t *thread, kmp_int32 gtid, kmp_flag_64 *flag, int final_spin,
22677cc577a4SJonathan Peyton                            int *thread_finished
22687cc577a4SJonathan Peyton                            USE_ITT_BUILD_ARG(void * itt_sync_obj), kmp_int32 is_constrained)
22697cc577a4SJonathan Peyton {
22707cc577a4SJonathan Peyton     return __kmp_execute_tasks_template(thread, gtid, flag, final_spin, thread_finished
22717cc577a4SJonathan Peyton                                         USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained);
22727cc577a4SJonathan Peyton }
22737cc577a4SJonathan Peyton 
22747cc577a4SJonathan Peyton int __kmp_execute_tasks_oncore(kmp_info_t *thread, kmp_int32 gtid, kmp_flag_oncore *flag, int final_spin,
22757cc577a4SJonathan Peyton                                int *thread_finished
22767cc577a4SJonathan Peyton                                USE_ITT_BUILD_ARG(void * itt_sync_obj), kmp_int32 is_constrained)
22777cc577a4SJonathan Peyton {
22787cc577a4SJonathan Peyton     return __kmp_execute_tasks_template(thread, gtid, flag, final_spin, thread_finished
22797cc577a4SJonathan Peyton                                         USE_ITT_BUILD_ARG(itt_sync_obj), is_constrained);
22807cc577a4SJonathan Peyton }
22817cc577a4SJonathan Peyton 
22827cc577a4SJonathan Peyton 
22837cc577a4SJonathan Peyton 
22847cc577a4SJonathan Peyton //-----------------------------------------------------------------------------
22857cc577a4SJonathan Peyton // __kmp_enable_tasking: Allocate task team and resume threads sleeping at the
22867cc577a4SJonathan Peyton // next barrier so they can assist in executing enqueued tasks.
22877cc577a4SJonathan Peyton // First thread in allocates the task team atomically.
22887cc577a4SJonathan Peyton 
22897cc577a4SJonathan Peyton static void
22907cc577a4SJonathan Peyton __kmp_enable_tasking( kmp_task_team_t *task_team, kmp_info_t *this_thr )
22917cc577a4SJonathan Peyton {
22927cc577a4SJonathan Peyton     kmp_thread_data_t *threads_data;
22937cc577a4SJonathan Peyton     int nthreads, i, is_init_thread;
22947cc577a4SJonathan Peyton 
22957cc577a4SJonathan Peyton     KA_TRACE( 10, ( "__kmp_enable_tasking(enter): T#%d\n",
22967cc577a4SJonathan Peyton                     __kmp_gtid_from_thread( this_thr ) ) );
22977cc577a4SJonathan Peyton 
22987cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT(task_team != NULL);
22997cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT(this_thr->th.th_team != NULL);
23007cc577a4SJonathan Peyton 
23017cc577a4SJonathan Peyton     nthreads = task_team->tt.tt_nproc;
23027cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT(nthreads > 0);
23037cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT(nthreads == this_thr->th.th_team->t.t_nproc);
23047cc577a4SJonathan Peyton 
23057cc577a4SJonathan Peyton     // Allocate or increase the size of threads_data if necessary
23067cc577a4SJonathan Peyton     is_init_thread = __kmp_realloc_task_threads_data( this_thr, task_team );
23077cc577a4SJonathan Peyton 
23087cc577a4SJonathan Peyton     if (!is_init_thread) {
23097cc577a4SJonathan Peyton         // Some other thread already set up the array.
23107cc577a4SJonathan Peyton         KA_TRACE( 20, ( "__kmp_enable_tasking(exit): T#%d: threads array already set up.\n",
23117cc577a4SJonathan Peyton                         __kmp_gtid_from_thread( this_thr ) ) );
23127cc577a4SJonathan Peyton         return;
23137cc577a4SJonathan Peyton     }
23147cc577a4SJonathan Peyton     threads_data = (kmp_thread_data_t *)TCR_PTR(task_team -> tt.tt_threads_data);
23157cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( threads_data != NULL );
23167cc577a4SJonathan Peyton 
23177cc577a4SJonathan Peyton     if ( ( __kmp_tasking_mode == tskm_task_teams ) &&
23187cc577a4SJonathan Peyton          ( __kmp_dflt_blocktime != KMP_MAX_BLOCKTIME ) )
23197cc577a4SJonathan Peyton     {
23207cc577a4SJonathan Peyton         // Release any threads sleeping at the barrier, so that they can steal
23217cc577a4SJonathan Peyton         // tasks and execute them.  In extra barrier mode, tasks do not sleep
23227cc577a4SJonathan Peyton         // at the separate tasking barrier, so this isn't a problem.
23237cc577a4SJonathan Peyton         for (i = 0; i < nthreads; i++) {
23247cc577a4SJonathan Peyton             volatile void *sleep_loc;
23257cc577a4SJonathan Peyton             kmp_info_t *thread = threads_data[i].td.td_thr;
23267cc577a4SJonathan Peyton 
23277cc577a4SJonathan Peyton             if (i == this_thr->th.th_info.ds.ds_tid) {
23287cc577a4SJonathan Peyton                 continue;
23297cc577a4SJonathan Peyton             }
23307cc577a4SJonathan Peyton             // Since we haven't locked the thread's suspend mutex lock at this
23317cc577a4SJonathan Peyton             // point, there is a small window where a thread might be putting
23327cc577a4SJonathan Peyton             // itself to sleep, but hasn't set the th_sleep_loc field yet.
23337cc577a4SJonathan Peyton             // To work around this, __kmp_execute_tasks_template() periodically checks
23347cc577a4SJonathan Peyton             // see if other threads are sleeping (using the same random
23357cc577a4SJonathan Peyton             // mechanism that is used for task stealing) and awakens them if
23367cc577a4SJonathan Peyton             // they are.
23377cc577a4SJonathan Peyton             if ( ( sleep_loc = TCR_PTR( thread -> th.th_sleep_loc) ) != NULL )
23387cc577a4SJonathan Peyton             {
23397cc577a4SJonathan Peyton                 KF_TRACE( 50, ( "__kmp_enable_tasking: T#%d waking up thread T#%d\n",
23407cc577a4SJonathan Peyton                                  __kmp_gtid_from_thread( this_thr ),
23417cc577a4SJonathan Peyton                                  __kmp_gtid_from_thread( thread ) ) );
23427cc577a4SJonathan Peyton                 __kmp_null_resume_wrapper(__kmp_gtid_from_thread(thread), sleep_loc);
23437cc577a4SJonathan Peyton             }
23447cc577a4SJonathan Peyton             else {
23457cc577a4SJonathan Peyton                 KF_TRACE( 50, ( "__kmp_enable_tasking: T#%d don't wake up thread T#%d\n",
23467cc577a4SJonathan Peyton                                  __kmp_gtid_from_thread( this_thr ),
23477cc577a4SJonathan Peyton                                  __kmp_gtid_from_thread( thread ) ) );
23487cc577a4SJonathan Peyton             }
23497cc577a4SJonathan Peyton         }
23507cc577a4SJonathan Peyton     }
23517cc577a4SJonathan Peyton 
23527cc577a4SJonathan Peyton     KA_TRACE( 10, ( "__kmp_enable_tasking(exit): T#%d\n",
23537cc577a4SJonathan Peyton                     __kmp_gtid_from_thread( this_thr ) ) );
23547cc577a4SJonathan Peyton }
23557cc577a4SJonathan Peyton 
23567cc577a4SJonathan Peyton 
23577cc577a4SJonathan Peyton /* ------------------------------------------------------------------------ */
23587cc577a4SJonathan Peyton /* // TODO: Check the comment consistency
23597cc577a4SJonathan Peyton  * Utility routines for "task teams".  A task team (kmp_task_t) is kind of
23607cc577a4SJonathan Peyton  * like a shadow of the kmp_team_t data struct, with a different lifetime.
23617cc577a4SJonathan Peyton  * After a child * thread checks into a barrier and calls __kmp_release() from
23627cc577a4SJonathan Peyton  * the particular variant of __kmp_<barrier_kind>_barrier_gather(), it can no
23637cc577a4SJonathan Peyton  * longer assume that the kmp_team_t structure is intact (at any moment, the
23647cc577a4SJonathan Peyton  * master thread may exit the barrier code and free the team data structure,
23657cc577a4SJonathan Peyton  * and return the threads to the thread pool).
23667cc577a4SJonathan Peyton  *
23677cc577a4SJonathan Peyton  * This does not work with the the tasking code, as the thread is still
23687cc577a4SJonathan Peyton  * expected to participate in the execution of any tasks that may have been
23697cc577a4SJonathan Peyton  * spawned my a member of the team, and the thread still needs access to all
23707cc577a4SJonathan Peyton  * to each thread in the team, so that it can steal work from it.
23717cc577a4SJonathan Peyton  *
23727cc577a4SJonathan Peyton  * Enter the existence of the kmp_task_team_t struct.  It employs a reference
23737cc577a4SJonathan Peyton  * counting mechanims, and is allocated by the master thread before calling
23747cc577a4SJonathan Peyton  * __kmp_<barrier_kind>_release, and then is release by the last thread to
23757cc577a4SJonathan Peyton  * exit __kmp_<barrier_kind>_release at the next barrier.  I.e. the lifetimes
23767cc577a4SJonathan Peyton  * of the kmp_task_team_t structs for consecutive barriers can overlap
23777cc577a4SJonathan Peyton  * (and will, unless the master thread is the last thread to exit the barrier
23787cc577a4SJonathan Peyton  * release phase, which is not typical).
23797cc577a4SJonathan Peyton  *
23807cc577a4SJonathan Peyton  * The existence of such a struct is useful outside the context of tasking,
23817cc577a4SJonathan Peyton  * but for now, I'm trying to keep it specific to the OMP_30_ENABLED macro,
23827cc577a4SJonathan Peyton  * so that any performance differences show up when comparing the 2.5 vs. 3.0
23837cc577a4SJonathan Peyton  * libraries.
23847cc577a4SJonathan Peyton  *
23857cc577a4SJonathan Peyton  * We currently use the existence of the threads array as an indicator that
23867cc577a4SJonathan Peyton  * tasks were spawned since the last barrier.  If the structure is to be
23877cc577a4SJonathan Peyton  * useful outside the context of tasking, then this will have to change, but
23887cc577a4SJonathan Peyton  * not settting the field minimizes the performance impact of tasking on
23897cc577a4SJonathan Peyton  * barriers, when no explicit tasks were spawned (pushed, actually).
23907cc577a4SJonathan Peyton  */
23917cc577a4SJonathan Peyton 
23927cc577a4SJonathan Peyton 
23937cc577a4SJonathan Peyton static kmp_task_team_t *__kmp_free_task_teams = NULL;           // Free list for task_team data structures
23947cc577a4SJonathan Peyton // Lock for task team data structures
23957cc577a4SJonathan Peyton static kmp_bootstrap_lock_t __kmp_task_team_lock = KMP_BOOTSTRAP_LOCK_INITIALIZER( __kmp_task_team_lock );
23967cc577a4SJonathan Peyton 
23977cc577a4SJonathan Peyton 
23987cc577a4SJonathan Peyton //------------------------------------------------------------------------------
23997cc577a4SJonathan Peyton // __kmp_alloc_task_deque:
24007cc577a4SJonathan Peyton // Allocates a task deque for a particular thread, and initialize the necessary
24017cc577a4SJonathan Peyton // data structures relating to the deque.  This only happens once per thread
24027cc577a4SJonathan Peyton // per task team since task teams are recycled.
24037cc577a4SJonathan Peyton // No lock is needed during allocation since each thread allocates its own
24047cc577a4SJonathan Peyton // deque.
24057cc577a4SJonathan Peyton 
24067cc577a4SJonathan Peyton static void
24077cc577a4SJonathan Peyton __kmp_alloc_task_deque( kmp_info_t *thread, kmp_thread_data_t *thread_data )
24087cc577a4SJonathan Peyton {
24097cc577a4SJonathan Peyton     __kmp_init_bootstrap_lock( & thread_data -> td.td_deque_lock );
24107cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( thread_data -> td.td_deque == NULL );
24117cc577a4SJonathan Peyton 
24127cc577a4SJonathan Peyton     // Initialize last stolen task field to "none"
24137cc577a4SJonathan Peyton     thread_data -> td.td_deque_last_stolen = -1;
24147cc577a4SJonathan Peyton 
24157cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( TCR_4(thread_data -> td.td_deque_ntasks) == 0 );
24167cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( thread_data -> td.td_deque_head == 0 );
24177cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( thread_data -> td.td_deque_tail == 0 );
24187cc577a4SJonathan Peyton 
24197cc577a4SJonathan Peyton     KE_TRACE( 10, ( "__kmp_alloc_task_deque: T#%d allocating deque[%d] for thread_data %p\n",
24207cc577a4SJonathan Peyton                    __kmp_gtid_from_thread( thread ), INITIAL_TASK_DEQUE_SIZE, thread_data ) );
24217cc577a4SJonathan Peyton     // Allocate space for task deque, and zero the deque
24227cc577a4SJonathan Peyton     // Cannot use __kmp_thread_calloc() because threads not around for
24237cc577a4SJonathan Peyton     // kmp_reap_task_team( ).
24247cc577a4SJonathan Peyton     thread_data -> td.td_deque = (kmp_taskdata_t **)
24257cc577a4SJonathan Peyton             __kmp_allocate( INITIAL_TASK_DEQUE_SIZE * sizeof(kmp_taskdata_t *));
24267cc577a4SJonathan Peyton 	thread_data -> td.td_deque_size = INITIAL_TASK_DEQUE_SIZE;
24277cc577a4SJonathan Peyton }
24287cc577a4SJonathan Peyton 
24297cc577a4SJonathan Peyton //------------------------------------------------------------------------------
24307cc577a4SJonathan Peyton // __kmp_realloc_task_deque:
24317cc577a4SJonathan Peyton // Re-allocates a task deque for a particular thread, copies the content from the old deque
24327cc577a4SJonathan Peyton // and adjusts the necessary data structures relating to the deque.
24337cc577a4SJonathan Peyton // This operation must be done with a the deque_lock being held
24347cc577a4SJonathan Peyton 
24357cc577a4SJonathan Peyton static void __kmp_realloc_task_deque ( kmp_info_t *thread, kmp_thread_data_t *thread_data )
24367cc577a4SJonathan Peyton {
24377cc577a4SJonathan Peyton     kmp_int32 size = TASK_DEQUE_SIZE(thread_data->td);
24387cc577a4SJonathan Peyton     kmp_int32 new_size = 2 * size;
24397cc577a4SJonathan Peyton 
24407cc577a4SJonathan Peyton     KE_TRACE( 10, ( "__kmp_realloc_task_deque: T#%d reallocating deque[from %d to %d] for thread_data %p\n",
24417cc577a4SJonathan Peyton                   __kmp_gtid_from_thread( thread ), size, new_size, thread_data ) );
24427cc577a4SJonathan Peyton 
24437cc577a4SJonathan Peyton     kmp_taskdata_t ** new_deque = (kmp_taskdata_t **) __kmp_allocate( new_size * sizeof(kmp_taskdata_t *));
24447cc577a4SJonathan Peyton 
24457cc577a4SJonathan Peyton     int i,j;
24467cc577a4SJonathan Peyton     for ( i = thread_data->td.td_deque_head, j = 0; j < size; i = (i+1) & TASK_DEQUE_MASK(thread_data->td), j++ )
24477cc577a4SJonathan Peyton        new_deque[j] = thread_data->td.td_deque[i];
24487cc577a4SJonathan Peyton 
24497cc577a4SJonathan Peyton     __kmp_free(thread_data->td.td_deque);
24507cc577a4SJonathan Peyton 
24517cc577a4SJonathan Peyton     thread_data -> td.td_deque_head = 0;
24527cc577a4SJonathan Peyton     thread_data -> td.td_deque_tail = size;
24537cc577a4SJonathan Peyton     thread_data -> td.td_deque = new_deque;
24547cc577a4SJonathan Peyton     thread_data -> td.td_deque_size = new_size;
24557cc577a4SJonathan Peyton }
24567cc577a4SJonathan Peyton 
24577cc577a4SJonathan Peyton //------------------------------------------------------------------------------
24587cc577a4SJonathan Peyton // __kmp_free_task_deque:
24597cc577a4SJonathan Peyton // Deallocates a task deque for a particular thread.
24607cc577a4SJonathan Peyton // Happens at library deallocation so don't need to reset all thread data fields.
24617cc577a4SJonathan Peyton 
24627cc577a4SJonathan Peyton static void
24637cc577a4SJonathan Peyton __kmp_free_task_deque( kmp_thread_data_t *thread_data )
24647cc577a4SJonathan Peyton {
24657cc577a4SJonathan Peyton     __kmp_acquire_bootstrap_lock( & thread_data -> td.td_deque_lock );
24667cc577a4SJonathan Peyton 
24677cc577a4SJonathan Peyton     if ( thread_data -> td.td_deque != NULL ) {
24687cc577a4SJonathan Peyton         TCW_4(thread_data -> td.td_deque_ntasks, 0);
24697cc577a4SJonathan Peyton          __kmp_free( thread_data -> td.td_deque );
24707cc577a4SJonathan Peyton         thread_data -> td.td_deque = NULL;
24717cc577a4SJonathan Peyton     }
24727cc577a4SJonathan Peyton     __kmp_release_bootstrap_lock( & thread_data -> td.td_deque_lock );
24737cc577a4SJonathan Peyton 
24747cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK
24757cc577a4SJonathan Peyton     // GEH: Figure out what to do here for td_susp_tied_tasks
24767cc577a4SJonathan Peyton     if ( thread_data -> td.td_susp_tied_tasks.ts_entries != TASK_STACK_EMPTY ) {
24777cc577a4SJonathan Peyton         __kmp_free_task_stack( __kmp_thread_from_gtid( gtid ), thread_data );
24787cc577a4SJonathan Peyton     }
24797cc577a4SJonathan Peyton #endif // BUILD_TIED_TASK_STACK
24807cc577a4SJonathan Peyton }
24817cc577a4SJonathan Peyton 
24827cc577a4SJonathan Peyton 
24837cc577a4SJonathan Peyton //------------------------------------------------------------------------------
24847cc577a4SJonathan Peyton // __kmp_realloc_task_threads_data:
24857cc577a4SJonathan Peyton // Allocates a threads_data array for a task team, either by allocating an initial
24867cc577a4SJonathan Peyton // array or enlarging an existing array.  Only the first thread to get the lock
24877cc577a4SJonathan Peyton // allocs or enlarges the array and re-initializes the array eleemnts.
24887cc577a4SJonathan Peyton // That thread returns "TRUE", the rest return "FALSE".
24897cc577a4SJonathan Peyton // Assumes that the new array size is given by task_team -> tt.tt_nproc.
24907cc577a4SJonathan Peyton // The current size is given by task_team -> tt.tt_max_threads.
24917cc577a4SJonathan Peyton 
24927cc577a4SJonathan Peyton static int
24937cc577a4SJonathan Peyton __kmp_realloc_task_threads_data( kmp_info_t *thread, kmp_task_team_t *task_team )
24947cc577a4SJonathan Peyton {
24957cc577a4SJonathan Peyton     kmp_thread_data_t ** threads_data_p;
24967cc577a4SJonathan Peyton     kmp_int32            nthreads, maxthreads;
24977cc577a4SJonathan Peyton     int                  is_init_thread = FALSE;
24987cc577a4SJonathan Peyton 
24997cc577a4SJonathan Peyton     if ( TCR_4(task_team -> tt.tt_found_tasks) ) {
25007cc577a4SJonathan Peyton         // Already reallocated and initialized.
25017cc577a4SJonathan Peyton         return FALSE;
25027cc577a4SJonathan Peyton     }
25037cc577a4SJonathan Peyton 
25047cc577a4SJonathan Peyton     threads_data_p = & task_team -> tt.tt_threads_data;
25057cc577a4SJonathan Peyton     nthreads   = task_team -> tt.tt_nproc;
25067cc577a4SJonathan Peyton     maxthreads = task_team -> tt.tt_max_threads;
25077cc577a4SJonathan Peyton 
25087cc577a4SJonathan Peyton     // All threads must lock when they encounter the first task of the implicit task
25097cc577a4SJonathan Peyton     // region to make sure threads_data fields are (re)initialized before used.
25107cc577a4SJonathan Peyton     __kmp_acquire_bootstrap_lock( & task_team -> tt.tt_threads_lock );
25117cc577a4SJonathan Peyton 
25127cc577a4SJonathan Peyton     if ( ! TCR_4(task_team -> tt.tt_found_tasks) ) {
25137cc577a4SJonathan Peyton         // first thread to enable tasking
25147cc577a4SJonathan Peyton         kmp_team_t *team = thread -> th.th_team;
25157cc577a4SJonathan Peyton         int i;
25167cc577a4SJonathan Peyton 
25177cc577a4SJonathan Peyton         is_init_thread = TRUE;
25187cc577a4SJonathan Peyton         if ( maxthreads < nthreads ) {
25197cc577a4SJonathan Peyton 
25207cc577a4SJonathan Peyton             if ( *threads_data_p != NULL ) {
25217cc577a4SJonathan Peyton                 kmp_thread_data_t *old_data = *threads_data_p;
25227cc577a4SJonathan Peyton                 kmp_thread_data_t *new_data = NULL;
25237cc577a4SJonathan Peyton 
25247cc577a4SJonathan Peyton                 KE_TRACE( 10, ( "__kmp_realloc_task_threads_data: T#%d reallocating "
25257cc577a4SJonathan Peyton                                "threads data for task_team %p, new_size = %d, old_size = %d\n",
25267cc577a4SJonathan Peyton                                __kmp_gtid_from_thread( thread ), task_team,
25277cc577a4SJonathan Peyton                                nthreads, maxthreads ) );
25287cc577a4SJonathan Peyton                 // Reallocate threads_data to have more elements than current array
25297cc577a4SJonathan Peyton                 // Cannot use __kmp_thread_realloc() because threads not around for
25307cc577a4SJonathan Peyton                 // kmp_reap_task_team( ).  Note all new array entries are initialized
25317cc577a4SJonathan Peyton                 // to zero by __kmp_allocate().
25327cc577a4SJonathan Peyton                 new_data = (kmp_thread_data_t *)
25337cc577a4SJonathan Peyton                             __kmp_allocate( nthreads * sizeof(kmp_thread_data_t) );
25347cc577a4SJonathan Peyton                 // copy old data to new data
25357cc577a4SJonathan Peyton                 KMP_MEMCPY_S( (void *) new_data, nthreads * sizeof(kmp_thread_data_t),
25367cc577a4SJonathan Peyton                               (void *) old_data,
25377cc577a4SJonathan Peyton                               maxthreads * sizeof(kmp_taskdata_t *) );
25387cc577a4SJonathan Peyton 
25397cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK
25407cc577a4SJonathan Peyton                 // GEH: Figure out if this is the right thing to do
25417cc577a4SJonathan Peyton                 for (i = maxthreads; i < nthreads; i++) {
25427cc577a4SJonathan Peyton                     kmp_thread_data_t *thread_data = & (*threads_data_p)[i];
25437cc577a4SJonathan Peyton                     __kmp_init_task_stack( __kmp_gtid_from_thread( thread ), thread_data );
25447cc577a4SJonathan Peyton                 }
25457cc577a4SJonathan Peyton #endif // BUILD_TIED_TASK_STACK
25467cc577a4SJonathan Peyton                 // Install the new data and free the old data
25477cc577a4SJonathan Peyton                 (*threads_data_p) = new_data;
25487cc577a4SJonathan Peyton                 __kmp_free( old_data );
25497cc577a4SJonathan Peyton             }
25507cc577a4SJonathan Peyton             else {
25517cc577a4SJonathan Peyton                 KE_TRACE( 10, ( "__kmp_realloc_task_threads_data: T#%d allocating "
25527cc577a4SJonathan Peyton                                "threads data for task_team %p, size = %d\n",
25537cc577a4SJonathan Peyton                                __kmp_gtid_from_thread( thread ), task_team, nthreads ) );
25547cc577a4SJonathan Peyton                 // Make the initial allocate for threads_data array, and zero entries
25557cc577a4SJonathan Peyton                 // Cannot use __kmp_thread_calloc() because threads not around for
25567cc577a4SJonathan Peyton                 // kmp_reap_task_team( ).
25577cc577a4SJonathan Peyton                 ANNOTATE_IGNORE_WRITES_BEGIN();
25587cc577a4SJonathan Peyton                 *threads_data_p = (kmp_thread_data_t *)
25597cc577a4SJonathan Peyton                                   __kmp_allocate( nthreads * sizeof(kmp_thread_data_t) );
25607cc577a4SJonathan Peyton                 ANNOTATE_IGNORE_WRITES_END();
25617cc577a4SJonathan Peyton #ifdef BUILD_TIED_TASK_STACK
25627cc577a4SJonathan Peyton                 // GEH: Figure out if this is the right thing to do
25637cc577a4SJonathan Peyton                 for (i = 0; i < nthreads; i++) {
25647cc577a4SJonathan Peyton                     kmp_thread_data_t *thread_data = & (*threads_data_p)[i];
25657cc577a4SJonathan Peyton                     __kmp_init_task_stack( __kmp_gtid_from_thread( thread ), thread_data );
25667cc577a4SJonathan Peyton                 }
25677cc577a4SJonathan Peyton #endif // BUILD_TIED_TASK_STACK
25687cc577a4SJonathan Peyton             }
25697cc577a4SJonathan Peyton             task_team -> tt.tt_max_threads = nthreads;
25707cc577a4SJonathan Peyton         }
25717cc577a4SJonathan Peyton         else {
25727cc577a4SJonathan Peyton             // If array has (more than) enough elements, go ahead and use it
25737cc577a4SJonathan Peyton             KMP_DEBUG_ASSERT( *threads_data_p != NULL );
25747cc577a4SJonathan Peyton         }
25757cc577a4SJonathan Peyton 
25767cc577a4SJonathan Peyton         // initialize threads_data pointers back to thread_info structures
25777cc577a4SJonathan Peyton         for (i = 0; i < nthreads; i++) {
25787cc577a4SJonathan Peyton             kmp_thread_data_t *thread_data = & (*threads_data_p)[i];
25797cc577a4SJonathan Peyton             thread_data -> td.td_thr = team -> t.t_threads[i];
25807cc577a4SJonathan Peyton 
25817cc577a4SJonathan Peyton             if ( thread_data -> td.td_deque_last_stolen >= nthreads) {
25827cc577a4SJonathan Peyton                 // The last stolen field survives across teams / barrier, and the number
25837cc577a4SJonathan Peyton                 // of threads may have changed.  It's possible (likely?) that a new
25847cc577a4SJonathan Peyton                 // parallel region will exhibit the same behavior as the previous region.
25857cc577a4SJonathan Peyton                 thread_data -> td.td_deque_last_stolen = -1;
25867cc577a4SJonathan Peyton             }
25877cc577a4SJonathan Peyton         }
25887cc577a4SJonathan Peyton 
25897cc577a4SJonathan Peyton         KMP_MB();
25907cc577a4SJonathan Peyton         TCW_SYNC_4(task_team -> tt.tt_found_tasks, TRUE);
25917cc577a4SJonathan Peyton     }
25927cc577a4SJonathan Peyton 
25937cc577a4SJonathan Peyton     __kmp_release_bootstrap_lock( & task_team -> tt.tt_threads_lock );
25947cc577a4SJonathan Peyton     return is_init_thread;
25957cc577a4SJonathan Peyton }
25967cc577a4SJonathan Peyton 
25977cc577a4SJonathan Peyton 
25987cc577a4SJonathan Peyton //------------------------------------------------------------------------------
25997cc577a4SJonathan Peyton // __kmp_free_task_threads_data:
26007cc577a4SJonathan Peyton // Deallocates a threads_data array for a task team, including any attached
26017cc577a4SJonathan Peyton // tasking deques.  Only occurs at library shutdown.
26027cc577a4SJonathan Peyton 
26037cc577a4SJonathan Peyton static void
26047cc577a4SJonathan Peyton __kmp_free_task_threads_data( kmp_task_team_t *task_team )
26057cc577a4SJonathan Peyton {
26067cc577a4SJonathan Peyton     __kmp_acquire_bootstrap_lock( & task_team -> tt.tt_threads_lock );
26077cc577a4SJonathan Peyton     if ( task_team -> tt.tt_threads_data != NULL ) {
26087cc577a4SJonathan Peyton         int i;
26097cc577a4SJonathan Peyton         for (i = 0; i < task_team->tt.tt_max_threads; i++ ) {
26107cc577a4SJonathan Peyton             __kmp_free_task_deque( & task_team -> tt.tt_threads_data[i] );
26117cc577a4SJonathan Peyton         }
26127cc577a4SJonathan Peyton         __kmp_free( task_team -> tt.tt_threads_data );
26137cc577a4SJonathan Peyton         task_team -> tt.tt_threads_data = NULL;
26147cc577a4SJonathan Peyton     }
26157cc577a4SJonathan Peyton     __kmp_release_bootstrap_lock( & task_team -> tt.tt_threads_lock );
26167cc577a4SJonathan Peyton }
26177cc577a4SJonathan Peyton 
26187cc577a4SJonathan Peyton 
26197cc577a4SJonathan Peyton //------------------------------------------------------------------------------
26207cc577a4SJonathan Peyton // __kmp_allocate_task_team:
26217cc577a4SJonathan Peyton // Allocates a task team associated with a specific team, taking it from
26227cc577a4SJonathan Peyton // the global task team free list if possible.  Also initializes data structures.
26237cc577a4SJonathan Peyton 
26247cc577a4SJonathan Peyton static kmp_task_team_t *
26257cc577a4SJonathan Peyton __kmp_allocate_task_team( kmp_info_t *thread, kmp_team_t *team )
26267cc577a4SJonathan Peyton {
26277cc577a4SJonathan Peyton     kmp_task_team_t *task_team = NULL;
26287cc577a4SJonathan Peyton     int nthreads;
26297cc577a4SJonathan Peyton 
26307cc577a4SJonathan Peyton     KA_TRACE( 20, ( "__kmp_allocate_task_team: T#%d entering; team = %p\n",
26317cc577a4SJonathan Peyton                     (thread ? __kmp_gtid_from_thread( thread ) : -1), team ) );
26327cc577a4SJonathan Peyton 
26337cc577a4SJonathan Peyton     if (TCR_PTR(__kmp_free_task_teams) != NULL) {
26347cc577a4SJonathan Peyton         // Take a task team from the task team pool
26357cc577a4SJonathan Peyton         __kmp_acquire_bootstrap_lock( &__kmp_task_team_lock );
26367cc577a4SJonathan Peyton         if (__kmp_free_task_teams != NULL) {
26377cc577a4SJonathan Peyton             task_team = __kmp_free_task_teams;
26387cc577a4SJonathan Peyton             TCW_PTR(__kmp_free_task_teams, task_team -> tt.tt_next);
26397cc577a4SJonathan Peyton             task_team -> tt.tt_next = NULL;
26407cc577a4SJonathan Peyton         }
26417cc577a4SJonathan Peyton         __kmp_release_bootstrap_lock( &__kmp_task_team_lock );
26427cc577a4SJonathan Peyton     }
26437cc577a4SJonathan Peyton 
26447cc577a4SJonathan Peyton     if (task_team == NULL) {
26457cc577a4SJonathan Peyton         KE_TRACE( 10, ( "__kmp_allocate_task_team: T#%d allocating "
26467cc577a4SJonathan Peyton                        "task team for team %p\n",
26477cc577a4SJonathan Peyton                        __kmp_gtid_from_thread( thread ), team ) );
26487cc577a4SJonathan Peyton         // Allocate a new task team if one is not available.
26497cc577a4SJonathan Peyton         // Cannot use __kmp_thread_malloc() because threads not around for
26507cc577a4SJonathan Peyton         // kmp_reap_task_team( ).
26517cc577a4SJonathan Peyton         task_team = (kmp_task_team_t *) __kmp_allocate( sizeof(kmp_task_team_t) );
26527cc577a4SJonathan Peyton         __kmp_init_bootstrap_lock( & task_team -> tt.tt_threads_lock );
26537cc577a4SJonathan Peyton         //task_team -> tt.tt_threads_data = NULL;   // AC: __kmp_allocate zeroes returned memory
26547cc577a4SJonathan Peyton         //task_team -> tt.tt_max_threads = 0;
26557cc577a4SJonathan Peyton         //task_team -> tt.tt_next = NULL;
26567cc577a4SJonathan Peyton     }
26577cc577a4SJonathan Peyton 
26587cc577a4SJonathan Peyton     TCW_4(task_team -> tt.tt_found_tasks, FALSE);
26597cc577a4SJonathan Peyton #if OMP_45_ENABLED
26607cc577a4SJonathan Peyton     TCW_4(task_team -> tt.tt_found_proxy_tasks, FALSE);
26617cc577a4SJonathan Peyton #endif
26627cc577a4SJonathan Peyton     task_team -> tt.tt_nproc = nthreads = team->t.t_nproc;
26637cc577a4SJonathan Peyton 
26647cc577a4SJonathan Peyton     TCW_4( task_team -> tt.tt_unfinished_threads, nthreads );
26657cc577a4SJonathan Peyton     TCW_4( task_team -> tt.tt_active, TRUE );
26667cc577a4SJonathan Peyton 
26677cc577a4SJonathan Peyton     KA_TRACE( 20, ( "__kmp_allocate_task_team: T#%d exiting; task_team = %p unfinished_threads init'd to %d\n",
26687cc577a4SJonathan Peyton                     (thread ? __kmp_gtid_from_thread( thread ) : -1), task_team, task_team -> tt.tt_unfinished_threads) );
26697cc577a4SJonathan Peyton     return task_team;
26707cc577a4SJonathan Peyton }
26717cc577a4SJonathan Peyton 
26727cc577a4SJonathan Peyton 
26737cc577a4SJonathan Peyton //------------------------------------------------------------------------------
26747cc577a4SJonathan Peyton // __kmp_free_task_team:
26757cc577a4SJonathan Peyton // Frees the task team associated with a specific thread, and adds it
26767cc577a4SJonathan Peyton // to the global task team free list.
26777cc577a4SJonathan Peyton 
26787cc577a4SJonathan Peyton void
26797cc577a4SJonathan Peyton __kmp_free_task_team( kmp_info_t *thread, kmp_task_team_t *task_team )
26807cc577a4SJonathan Peyton {
26817cc577a4SJonathan Peyton     KA_TRACE( 20, ( "__kmp_free_task_team: T#%d task_team = %p\n",
26827cc577a4SJonathan Peyton                     thread ? __kmp_gtid_from_thread( thread ) : -1, task_team ) );
26837cc577a4SJonathan Peyton 
26847cc577a4SJonathan Peyton     // Put task team back on free list
26857cc577a4SJonathan Peyton     __kmp_acquire_bootstrap_lock( & __kmp_task_team_lock );
26867cc577a4SJonathan Peyton 
26877cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task_team -> tt.tt_next == NULL );
26887cc577a4SJonathan Peyton     task_team -> tt.tt_next = __kmp_free_task_teams;
26897cc577a4SJonathan Peyton     TCW_PTR(__kmp_free_task_teams, task_team);
26907cc577a4SJonathan Peyton 
26917cc577a4SJonathan Peyton     __kmp_release_bootstrap_lock( & __kmp_task_team_lock );
26927cc577a4SJonathan Peyton }
26937cc577a4SJonathan Peyton 
26947cc577a4SJonathan Peyton 
26957cc577a4SJonathan Peyton //------------------------------------------------------------------------------
26967cc577a4SJonathan Peyton // __kmp_reap_task_teams:
26977cc577a4SJonathan Peyton // Free all the task teams on the task team free list.
26987cc577a4SJonathan Peyton // Should only be done during library shutdown.
26997cc577a4SJonathan Peyton // Cannot do anything that needs a thread structure or gtid since they are already gone.
27007cc577a4SJonathan Peyton 
27017cc577a4SJonathan Peyton void
27027cc577a4SJonathan Peyton __kmp_reap_task_teams( void )
27037cc577a4SJonathan Peyton {
27047cc577a4SJonathan Peyton     kmp_task_team_t   *task_team;
27057cc577a4SJonathan Peyton 
27067cc577a4SJonathan Peyton     if ( TCR_PTR(__kmp_free_task_teams) != NULL ) {
27077cc577a4SJonathan Peyton         // Free all task_teams on the free list
27087cc577a4SJonathan Peyton         __kmp_acquire_bootstrap_lock( &__kmp_task_team_lock );
27097cc577a4SJonathan Peyton         while ( ( task_team = __kmp_free_task_teams ) != NULL ) {
27107cc577a4SJonathan Peyton             __kmp_free_task_teams = task_team -> tt.tt_next;
27117cc577a4SJonathan Peyton             task_team -> tt.tt_next = NULL;
27127cc577a4SJonathan Peyton 
27137cc577a4SJonathan Peyton             // Free threads_data if necessary
27147cc577a4SJonathan Peyton             if ( task_team -> tt.tt_threads_data != NULL ) {
27157cc577a4SJonathan Peyton                 __kmp_free_task_threads_data( task_team );
27167cc577a4SJonathan Peyton             }
27177cc577a4SJonathan Peyton             __kmp_free( task_team );
27187cc577a4SJonathan Peyton         }
27197cc577a4SJonathan Peyton         __kmp_release_bootstrap_lock( &__kmp_task_team_lock );
27207cc577a4SJonathan Peyton     }
27217cc577a4SJonathan Peyton }
27227cc577a4SJonathan Peyton 
27237cc577a4SJonathan Peyton //------------------------------------------------------------------------------
27247cc577a4SJonathan Peyton // __kmp_wait_to_unref_task_teams:
27257cc577a4SJonathan Peyton // Some threads could still be in the fork barrier release code, possibly
27267cc577a4SJonathan Peyton // trying to steal tasks.  Wait for each thread to unreference its task team.
27277cc577a4SJonathan Peyton //
27287cc577a4SJonathan Peyton void
27297cc577a4SJonathan Peyton __kmp_wait_to_unref_task_teams(void)
27307cc577a4SJonathan Peyton {
27317cc577a4SJonathan Peyton     kmp_info_t *thread;
27327cc577a4SJonathan Peyton     kmp_uint32 spins;
27337cc577a4SJonathan Peyton     int done;
27347cc577a4SJonathan Peyton 
27357cc577a4SJonathan Peyton     KMP_INIT_YIELD( spins );
27367cc577a4SJonathan Peyton 
27377cc577a4SJonathan Peyton     for (;;) {
27387cc577a4SJonathan Peyton         done = TRUE;
27397cc577a4SJonathan Peyton 
27407cc577a4SJonathan Peyton         // TODO: GEH - this may be is wrong because some sync would be necessary
27417cc577a4SJonathan Peyton         //             in case threads are added to the pool during the traversal.
27427cc577a4SJonathan Peyton         //             Need to verify that lock for thread pool is held when calling
27437cc577a4SJonathan Peyton         //             this routine.
27447cc577a4SJonathan Peyton         for (thread = (kmp_info_t *)__kmp_thread_pool;
27457cc577a4SJonathan Peyton              thread != NULL;
27467cc577a4SJonathan Peyton              thread = thread->th.th_next_pool)
27477cc577a4SJonathan Peyton         {
27487cc577a4SJonathan Peyton #if KMP_OS_WINDOWS
27497cc577a4SJonathan Peyton             DWORD exit_val;
27507cc577a4SJonathan Peyton #endif
27517cc577a4SJonathan Peyton             if ( TCR_PTR(thread->th.th_task_team) == NULL ) {
27527cc577a4SJonathan Peyton                 KA_TRACE( 10, ("__kmp_wait_to_unref_task_team: T#%d task_team == NULL\n",
27537cc577a4SJonathan Peyton                                __kmp_gtid_from_thread( thread ) ) );
27547cc577a4SJonathan Peyton                 continue;
27557cc577a4SJonathan Peyton             }
27567cc577a4SJonathan Peyton #if KMP_OS_WINDOWS
27577cc577a4SJonathan Peyton             // TODO: GEH - add this check for Linux* OS / OS X* as well?
27587cc577a4SJonathan Peyton             if (!__kmp_is_thread_alive(thread, &exit_val)) {
27597cc577a4SJonathan Peyton                 thread->th.th_task_team = NULL;
27607cc577a4SJonathan Peyton                 continue;
27617cc577a4SJonathan Peyton             }
27627cc577a4SJonathan Peyton #endif
27637cc577a4SJonathan Peyton 
27647cc577a4SJonathan Peyton             done = FALSE;  // Because th_task_team pointer is not NULL for this thread
27657cc577a4SJonathan Peyton 
27667cc577a4SJonathan Peyton             KA_TRACE( 10, ("__kmp_wait_to_unref_task_team: Waiting for T#%d to unreference task_team\n",
27677cc577a4SJonathan Peyton                            __kmp_gtid_from_thread( thread ) ) );
27687cc577a4SJonathan Peyton 
27697cc577a4SJonathan Peyton             if ( __kmp_dflt_blocktime != KMP_MAX_BLOCKTIME ) {
27707cc577a4SJonathan Peyton                 volatile void *sleep_loc;
27717cc577a4SJonathan Peyton                 // If the thread is sleeping, awaken it.
27727cc577a4SJonathan Peyton                 if ( ( sleep_loc = TCR_PTR( thread->th.th_sleep_loc) ) != NULL ) {
27737cc577a4SJonathan Peyton                     KA_TRACE( 10, ( "__kmp_wait_to_unref_task_team: T#%d waking up thread T#%d\n",
27747cc577a4SJonathan Peyton                                     __kmp_gtid_from_thread( thread ), __kmp_gtid_from_thread( thread ) ) );
27757cc577a4SJonathan Peyton                     __kmp_null_resume_wrapper(__kmp_gtid_from_thread(thread), sleep_loc);
27767cc577a4SJonathan Peyton                 }
27777cc577a4SJonathan Peyton             }
27787cc577a4SJonathan Peyton         }
27797cc577a4SJonathan Peyton         if (done) {
27807cc577a4SJonathan Peyton             break;
27817cc577a4SJonathan Peyton         }
27827cc577a4SJonathan Peyton 
27837cc577a4SJonathan Peyton         // If we are oversubscribed,
27847cc577a4SJonathan Peyton         // or have waited a bit (and library mode is throughput), yield.
27857cc577a4SJonathan Peyton         // Pause is in the following code.
27867cc577a4SJonathan Peyton         KMP_YIELD( TCR_4(__kmp_nth) > __kmp_avail_proc );
27877cc577a4SJonathan Peyton         KMP_YIELD_SPIN( spins );        // Yields only if KMP_LIBRARY=throughput
27887cc577a4SJonathan Peyton     }
27897cc577a4SJonathan Peyton }
27907cc577a4SJonathan Peyton 
27917cc577a4SJonathan Peyton 
27927cc577a4SJonathan Peyton //------------------------------------------------------------------------------
27937cc577a4SJonathan Peyton // __kmp_task_team_setup:  Create a task_team for the current team, but use
27947cc577a4SJonathan Peyton // an already created, unused one if it already exists.
27957cc577a4SJonathan Peyton void
27967cc577a4SJonathan Peyton __kmp_task_team_setup( kmp_info_t *this_thr, kmp_team_t *team, int always )
27977cc577a4SJonathan Peyton {
27987cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( __kmp_tasking_mode != tskm_immediate_exec );
27997cc577a4SJonathan Peyton 
28007cc577a4SJonathan Peyton     // If this task_team hasn't been created yet, allocate it. It will be used in the region after the next.
28017cc577a4SJonathan Peyton     // If it exists, it is the current task team and shouldn't be touched yet as it may still be in use.
28027cc577a4SJonathan Peyton     if (team->t.t_task_team[this_thr->th.th_task_state] == NULL && (always || team->t.t_nproc > 1) ) {
28037cc577a4SJonathan Peyton         team->t.t_task_team[this_thr->th.th_task_state] = __kmp_allocate_task_team( this_thr, team );
28047cc577a4SJonathan Peyton         KA_TRACE(20, ("__kmp_task_team_setup: Master T#%d created new task_team %p for team %d at parity=%d\n",
28057cc577a4SJonathan Peyton                       __kmp_gtid_from_thread(this_thr), team->t.t_task_team[this_thr->th.th_task_state],
28067cc577a4SJonathan Peyton                       ((team != NULL) ? team->t.t_id : -1), this_thr->th.th_task_state));
28077cc577a4SJonathan Peyton     }
28087cc577a4SJonathan Peyton 
28097cc577a4SJonathan Peyton     // After threads exit the release, they will call sync, and then point to this other task_team; make sure it is
28107cc577a4SJonathan Peyton     // allocated and properly initialized. As threads spin in the barrier release phase, they will continue to use the
28117cc577a4SJonathan Peyton     // previous task_team struct(above), until they receive the signal to stop checking for tasks (they can't safely
28127cc577a4SJonathan Peyton     // reference the kmp_team_t struct, which could be reallocated by the master thread). No task teams are formed for
28137cc577a4SJonathan Peyton     // serialized teams.
28147cc577a4SJonathan Peyton     if (team->t.t_nproc > 1) {
28157cc577a4SJonathan Peyton         int other_team = 1 - this_thr->th.th_task_state;
28167cc577a4SJonathan Peyton         if (team->t.t_task_team[other_team] == NULL) { // setup other team as well
28177cc577a4SJonathan Peyton                 team->t.t_task_team[other_team] = __kmp_allocate_task_team( this_thr, team );
28187cc577a4SJonathan Peyton                 KA_TRACE(20, ("__kmp_task_team_setup: Master T#%d created second new task_team %p for team %d at parity=%d\n",
28197cc577a4SJonathan Peyton                                 __kmp_gtid_from_thread( this_thr ), team->t.t_task_team[other_team],
28207cc577a4SJonathan Peyton                               ((team != NULL) ? team->t.t_id : -1), other_team ));
28217cc577a4SJonathan Peyton         }
28227cc577a4SJonathan Peyton         else { // Leave the old task team struct in place for the upcoming region; adjust as needed
28237cc577a4SJonathan Peyton             kmp_task_team_t *task_team = team->t.t_task_team[other_team];
28247cc577a4SJonathan Peyton             if (!task_team->tt.tt_active || team->t.t_nproc != task_team->tt.tt_nproc) {
28257cc577a4SJonathan Peyton                 TCW_4(task_team->tt.tt_nproc, team->t.t_nproc);
28267cc577a4SJonathan Peyton                 TCW_4(task_team->tt.tt_found_tasks, FALSE);
28277cc577a4SJonathan Peyton #if OMP_45_ENABLED
28287cc577a4SJonathan Peyton                 TCW_4(task_team->tt.tt_found_proxy_tasks, FALSE);
28297cc577a4SJonathan Peyton #endif
28307cc577a4SJonathan Peyton                 TCW_4(task_team->tt.tt_unfinished_threads, team->t.t_nproc );
28317cc577a4SJonathan Peyton                 TCW_4(task_team->tt.tt_active, TRUE );
28327cc577a4SJonathan Peyton             }
28337cc577a4SJonathan Peyton             // if team size has changed, the first thread to enable tasking will realloc threads_data if necessary
28347cc577a4SJonathan Peyton             KA_TRACE(20, ("__kmp_task_team_setup: Master T#%d reset next task_team %p for team %d at parity=%d\n",
28357cc577a4SJonathan Peyton                           __kmp_gtid_from_thread( this_thr ), team->t.t_task_team[other_team],
28367cc577a4SJonathan Peyton                           ((team != NULL) ? team->t.t_id : -1), other_team ));
28377cc577a4SJonathan Peyton         }
28387cc577a4SJonathan Peyton     }
28397cc577a4SJonathan Peyton }
28407cc577a4SJonathan Peyton 
28417cc577a4SJonathan Peyton 
28427cc577a4SJonathan Peyton //------------------------------------------------------------------------------
28437cc577a4SJonathan Peyton // __kmp_task_team_sync: Propagation of task team data from team to threads
28447cc577a4SJonathan Peyton // which happens just after the release phase of a team barrier.  This may be
28457cc577a4SJonathan Peyton // called by any thread, but only for teams with # threads > 1.
28467cc577a4SJonathan Peyton 
28477cc577a4SJonathan Peyton void
28487cc577a4SJonathan Peyton __kmp_task_team_sync( kmp_info_t *this_thr, kmp_team_t *team )
28497cc577a4SJonathan Peyton {
28507cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( __kmp_tasking_mode != tskm_immediate_exec );
28517cc577a4SJonathan Peyton 
28527cc577a4SJonathan Peyton     // Toggle the th_task_state field, to switch which task_team this thread refers to
28537cc577a4SJonathan Peyton     this_thr->th.th_task_state = 1 - this_thr->th.th_task_state;
28547cc577a4SJonathan Peyton     // It is now safe to propagate the task team pointer from the team struct to the current thread.
28557cc577a4SJonathan Peyton     TCW_PTR(this_thr->th.th_task_team, team->t.t_task_team[this_thr->th.th_task_state]);
28567cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_task_team_sync: Thread T#%d task team switched to task_team %p from Team #%d (parity=%d)\n",
28577cc577a4SJonathan Peyton                   __kmp_gtid_from_thread( this_thr ), this_thr->th.th_task_team,
28587cc577a4SJonathan Peyton                   ((team != NULL) ? team->t.t_id : -1), this_thr->th.th_task_state));
28597cc577a4SJonathan Peyton }
28607cc577a4SJonathan Peyton 
28617cc577a4SJonathan Peyton 
28627cc577a4SJonathan Peyton //--------------------------------------------------------------------------------------------
28637cc577a4SJonathan Peyton // __kmp_task_team_wait: Master thread waits for outstanding tasks after the barrier gather
28647cc577a4SJonathan Peyton // phase.  Only called by master thread if #threads in team > 1 or if proxy tasks were created.
28657cc577a4SJonathan Peyton // wait is a flag that defaults to 1 (see kmp.h), but waiting can be turned off by passing in 0
28667cc577a4SJonathan Peyton // optionally as the last argument. When wait is zero, master thread does not wait for
28677cc577a4SJonathan Peyton // unfinished_threads to reach 0.
28687cc577a4SJonathan Peyton void
28697cc577a4SJonathan Peyton __kmp_task_team_wait( kmp_info_t *this_thr, kmp_team_t *team
28707cc577a4SJonathan Peyton                       USE_ITT_BUILD_ARG(void * itt_sync_obj)
28717cc577a4SJonathan Peyton                       , int wait)
28727cc577a4SJonathan Peyton {
28737cc577a4SJonathan Peyton     kmp_task_team_t *task_team = team->t.t_task_team[this_thr->th.th_task_state];
28747cc577a4SJonathan Peyton 
28757cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( __kmp_tasking_mode != tskm_immediate_exec );
28767cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task_team == this_thr->th.th_task_team );
28777cc577a4SJonathan Peyton 
28787cc577a4SJonathan Peyton     if ( ( task_team != NULL ) && KMP_TASKING_ENABLED(task_team) ) {
28797cc577a4SJonathan Peyton         if (wait) {
28807cc577a4SJonathan Peyton             KA_TRACE(20, ("__kmp_task_team_wait: Master T#%d waiting for all tasks (for unfinished_threads to reach 0) on task_team = %p\n",
28817cc577a4SJonathan Peyton                           __kmp_gtid_from_thread(this_thr), task_team));
28827cc577a4SJonathan Peyton             // Worker threads may have dropped through to release phase, but could still be executing tasks. Wait
28837cc577a4SJonathan Peyton             // here for tasks to complete. To avoid memory contention, only master thread checks termination condition.
28847cc577a4SJonathan Peyton             kmp_flag_32 flag(&task_team->tt.tt_unfinished_threads, 0U);
28857cc577a4SJonathan Peyton             flag.wait(this_thr, TRUE
28867cc577a4SJonathan Peyton                       USE_ITT_BUILD_ARG(itt_sync_obj));
28877cc577a4SJonathan Peyton         }
28887cc577a4SJonathan Peyton         // Deactivate the old task team, so that the worker threads will stop referencing it while spinning.
28897cc577a4SJonathan Peyton         KA_TRACE(20, ("__kmp_task_team_wait: Master T#%d deactivating task_team %p: "
28907cc577a4SJonathan Peyton                       "setting active to false, setting local and team's pointer to NULL\n",
28917cc577a4SJonathan Peyton                       __kmp_gtid_from_thread(this_thr), task_team));
28927cc577a4SJonathan Peyton #if OMP_45_ENABLED
28937cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( task_team->tt.tt_nproc > 1 || task_team->tt.tt_found_proxy_tasks == TRUE );
28947cc577a4SJonathan Peyton         TCW_SYNC_4( task_team->tt.tt_found_proxy_tasks, FALSE );
28957cc577a4SJonathan Peyton #else
28967cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( task_team->tt.tt_nproc > 1 );
28977cc577a4SJonathan Peyton #endif
28987cc577a4SJonathan Peyton         TCW_SYNC_4( task_team->tt.tt_active, FALSE );
28997cc577a4SJonathan Peyton         KMP_MB();
29007cc577a4SJonathan Peyton 
29017cc577a4SJonathan Peyton         TCW_PTR(this_thr->th.th_task_team, NULL);
29027cc577a4SJonathan Peyton     }
29037cc577a4SJonathan Peyton }
29047cc577a4SJonathan Peyton 
29057cc577a4SJonathan Peyton 
29067cc577a4SJonathan Peyton //------------------------------------------------------------------------------
29077cc577a4SJonathan Peyton // __kmp_tasking_barrier:
29087cc577a4SJonathan Peyton // This routine may only called when __kmp_tasking_mode == tskm_extra_barrier.
29097cc577a4SJonathan Peyton // Internal function to execute all tasks prior to a regular barrier or a
29107cc577a4SJonathan Peyton // join barrier.  It is a full barrier itself, which unfortunately turns
29117cc577a4SJonathan Peyton // regular barriers into double barriers and join barriers into 1 1/2
29127cc577a4SJonathan Peyton // barriers.
29137cc577a4SJonathan Peyton void
29147cc577a4SJonathan Peyton __kmp_tasking_barrier( kmp_team_t *team, kmp_info_t *thread, int gtid )
29157cc577a4SJonathan Peyton {
29167cc577a4SJonathan Peyton     volatile kmp_uint32 *spin = &team->t.t_task_team[thread->th.th_task_state]->tt.tt_unfinished_threads;
29177cc577a4SJonathan Peyton     int flag = FALSE;
29187cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( __kmp_tasking_mode == tskm_extra_barrier );
29197cc577a4SJonathan Peyton 
29207cc577a4SJonathan Peyton #if USE_ITT_BUILD
29217cc577a4SJonathan Peyton     KMP_FSYNC_SPIN_INIT( spin, (kmp_uint32*) NULL );
29227cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
29237cc577a4SJonathan Peyton     kmp_flag_32 spin_flag(spin, 0U);
29247cc577a4SJonathan Peyton     while (! spin_flag.execute_tasks(thread, gtid, TRUE, &flag
29257cc577a4SJonathan Peyton                                      USE_ITT_BUILD_ARG(NULL), 0 ) ) {
29267cc577a4SJonathan Peyton #if USE_ITT_BUILD
29277cc577a4SJonathan Peyton         // TODO: What about itt_sync_obj??
29287cc577a4SJonathan Peyton         KMP_FSYNC_SPIN_PREPARE( spin );
29297cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
29307cc577a4SJonathan Peyton 
29317cc577a4SJonathan Peyton         if( TCR_4(__kmp_global.g.g_done) ) {
29327cc577a4SJonathan Peyton             if( __kmp_global.g.g_abort )
29337cc577a4SJonathan Peyton                 __kmp_abort_thread( );
29347cc577a4SJonathan Peyton             break;
29357cc577a4SJonathan Peyton         }
29367cc577a4SJonathan Peyton         KMP_YIELD( TRUE );       // GH: We always yield here
29377cc577a4SJonathan Peyton     }
29387cc577a4SJonathan Peyton #if USE_ITT_BUILD
29397cc577a4SJonathan Peyton     KMP_FSYNC_SPIN_ACQUIRED( (void*) spin );
29407cc577a4SJonathan Peyton #endif /* USE_ITT_BUILD */
29417cc577a4SJonathan Peyton }
29427cc577a4SJonathan Peyton 
29437cc577a4SJonathan Peyton 
29447cc577a4SJonathan Peyton #if OMP_45_ENABLED
29457cc577a4SJonathan Peyton 
29467cc577a4SJonathan Peyton /* __kmp_give_task puts a task into a given thread queue if:
29477cc577a4SJonathan Peyton     - the queue for that thread was created
29487cc577a4SJonathan Peyton     - there's space in that queue
29497cc577a4SJonathan Peyton 
29507cc577a4SJonathan Peyton     Because of this, __kmp_push_task needs to check if there's space after getting the lock
29517cc577a4SJonathan Peyton  */
29527cc577a4SJonathan Peyton static bool __kmp_give_task ( kmp_info_t *thread, kmp_int32 tid, kmp_task_t * task, kmp_int32 pass )
29537cc577a4SJonathan Peyton {
29547cc577a4SJonathan Peyton     kmp_taskdata_t *    taskdata = KMP_TASK_TO_TASKDATA(task);
29557cc577a4SJonathan Peyton     kmp_task_team_t *	task_team = taskdata->td_task_team;
29567cc577a4SJonathan Peyton 
29577cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_give_task: trying to give task %p to thread %d.\n", taskdata, tid ) );
29587cc577a4SJonathan Peyton 
29597cc577a4SJonathan Peyton     // If task_team is NULL something went really bad...
29607cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task_team != NULL );
29617cc577a4SJonathan Peyton 
29627cc577a4SJonathan Peyton     bool result = false;
29637cc577a4SJonathan Peyton     kmp_thread_data_t * thread_data = & task_team -> tt.tt_threads_data[ tid ];
29647cc577a4SJonathan Peyton 
29657cc577a4SJonathan Peyton     if (thread_data -> td.td_deque == NULL ) {
29667cc577a4SJonathan Peyton         // There's no queue in this thread, go find another one
29677cc577a4SJonathan Peyton         // We're guaranteed that at least one thread has a queue
29687cc577a4SJonathan Peyton         KA_TRACE(30, ("__kmp_give_task: thread %d has no queue while giving task %p.\n", tid, taskdata ) );
29697cc577a4SJonathan Peyton         return result;
29707cc577a4SJonathan Peyton     }
29717cc577a4SJonathan Peyton 
29727cc577a4SJonathan Peyton     if ( TCR_4(thread_data -> td.td_deque_ntasks) >= TASK_DEQUE_SIZE(thread_data->td) )
29737cc577a4SJonathan Peyton     {
29747cc577a4SJonathan Peyton         KA_TRACE(30, ("__kmp_give_task: queue is full while giving task %p to thread %d.\n", taskdata, tid ) );
29757cc577a4SJonathan Peyton 
29767cc577a4SJonathan Peyton         // if this deque is bigger than the pass ratio give a chance to another thread
29777cc577a4SJonathan Peyton         if ( TASK_DEQUE_SIZE(thread_data->td)/INITIAL_TASK_DEQUE_SIZE >= pass ) return result;
29787cc577a4SJonathan Peyton 
29797cc577a4SJonathan Peyton         __kmp_acquire_bootstrap_lock( & thread_data-> td.td_deque_lock );
29807cc577a4SJonathan Peyton         __kmp_realloc_task_deque(thread,thread_data);
29817cc577a4SJonathan Peyton 
29827cc577a4SJonathan Peyton     } else {
29837cc577a4SJonathan Peyton 
29847cc577a4SJonathan Peyton        __kmp_acquire_bootstrap_lock( & thread_data-> td.td_deque_lock );
29857cc577a4SJonathan Peyton 
29867cc577a4SJonathan Peyton        if ( TCR_4(thread_data -> td.td_deque_ntasks) >= TASK_DEQUE_SIZE(thread_data->td) )
29877cc577a4SJonathan Peyton        {
29887cc577a4SJonathan Peyton            KA_TRACE(30, ("__kmp_give_task: queue is full while giving task %p to thread %d.\n", taskdata, tid ) );
29897cc577a4SJonathan Peyton 
29907cc577a4SJonathan Peyton            // if this deque is bigger than the pass ratio give a chance to another thread
29917cc577a4SJonathan Peyton            if ( TASK_DEQUE_SIZE(thread_data->td)/INITIAL_TASK_DEQUE_SIZE >= pass )
29927cc577a4SJonathan Peyton               goto release_and_exit;
29937cc577a4SJonathan Peyton 
29947cc577a4SJonathan Peyton            __kmp_realloc_task_deque(thread,thread_data);
29957cc577a4SJonathan Peyton        }
29967cc577a4SJonathan Peyton     }
29977cc577a4SJonathan Peyton 
29987cc577a4SJonathan Peyton     // lock is held here, and there is space in the deque
29997cc577a4SJonathan Peyton 
30007cc577a4SJonathan Peyton     thread_data -> td.td_deque[ thread_data -> td.td_deque_tail ] = taskdata;
30017cc577a4SJonathan Peyton     // Wrap index.
30027cc577a4SJonathan Peyton     thread_data -> td.td_deque_tail = ( thread_data -> td.td_deque_tail + 1 ) & TASK_DEQUE_MASK(thread_data->td);
30037cc577a4SJonathan Peyton     TCW_4(thread_data -> td.td_deque_ntasks, TCR_4(thread_data -> td.td_deque_ntasks) + 1);
30047cc577a4SJonathan Peyton 
30057cc577a4SJonathan Peyton     result = true;
30067cc577a4SJonathan Peyton     KA_TRACE(30, ("__kmp_give_task: successfully gave task %p to thread %d.\n", taskdata, tid ) );
30077cc577a4SJonathan Peyton 
30087cc577a4SJonathan Peyton release_and_exit:
30097cc577a4SJonathan Peyton     __kmp_release_bootstrap_lock( & thread_data-> td.td_deque_lock );
30107cc577a4SJonathan Peyton 
30117cc577a4SJonathan Peyton      return result;
30127cc577a4SJonathan Peyton }
30137cc577a4SJonathan Peyton 
30147cc577a4SJonathan Peyton 
30157cc577a4SJonathan Peyton /* The finish of the a proxy tasks is divided in two pieces:
30167cc577a4SJonathan Peyton     - the top half is the one that can be done from a thread outside the team
30177cc577a4SJonathan Peyton     - the bottom half must be run from a them within the team
30187cc577a4SJonathan Peyton 
30197cc577a4SJonathan Peyton     In order to run the bottom half the task gets queued back into one of the threads of the team.
30207cc577a4SJonathan Peyton     Once the td_incomplete_child_task counter of the parent is decremented the threads can leave the barriers.
30217cc577a4SJonathan Peyton     So, the bottom half needs to be queued before the counter is decremented. The top half is therefore divided in two parts:
30227cc577a4SJonathan Peyton     - things that can be run before queuing the bottom half
30237cc577a4SJonathan Peyton     - things that must be run after queuing the bottom half
30247cc577a4SJonathan Peyton 
30257cc577a4SJonathan Peyton     This creates a second race as the bottom half can free the task before the second top half is executed. To avoid this
30267cc577a4SJonathan Peyton     we use the td_incomplete_child_task of the proxy task to synchronize the top and bottom half.
30277cc577a4SJonathan Peyton */
30287cc577a4SJonathan Peyton 
30297cc577a4SJonathan Peyton static void __kmp_first_top_half_finish_proxy( kmp_taskdata_t * taskdata )
30307cc577a4SJonathan Peyton {
30317cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.tasktype == TASK_EXPLICIT );
30327cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.proxy == TASK_PROXY );
30337cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.complete == 0 );
30347cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.freed == 0 );
30357cc577a4SJonathan Peyton 
30367cc577a4SJonathan Peyton     taskdata -> td_flags.complete = 1;   // mark the task as completed
30377cc577a4SJonathan Peyton 
30387cc577a4SJonathan Peyton     if ( taskdata->td_taskgroup )
30397cc577a4SJonathan Peyton        KMP_TEST_THEN_DEC32( (kmp_int32 *)(& taskdata->td_taskgroup->count) );
30407cc577a4SJonathan Peyton 
30417cc577a4SJonathan Peyton     // Create an imaginary children for this task so the bottom half cannot release the task before we have completed the second top half
30427cc577a4SJonathan Peyton     TCI_4(taskdata->td_incomplete_child_tasks);
30437cc577a4SJonathan Peyton }
30447cc577a4SJonathan Peyton 
30457cc577a4SJonathan Peyton static void __kmp_second_top_half_finish_proxy( kmp_taskdata_t * taskdata )
30467cc577a4SJonathan Peyton {
30477cc577a4SJonathan Peyton     kmp_int32 children = 0;
30487cc577a4SJonathan Peyton 
30497cc577a4SJonathan Peyton     // Predecrement simulated by "- 1" calculation
30507cc577a4SJonathan Peyton     children = KMP_TEST_THEN_DEC32( (kmp_int32 *)(& taskdata -> td_parent -> td_incomplete_child_tasks) ) - 1;
30517cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( children >= 0 );
30527cc577a4SJonathan Peyton 
30537cc577a4SJonathan Peyton     // Remove the imaginary children
30547cc577a4SJonathan Peyton     TCD_4(taskdata->td_incomplete_child_tasks);
30557cc577a4SJonathan Peyton }
30567cc577a4SJonathan Peyton 
30577cc577a4SJonathan Peyton static void __kmp_bottom_half_finish_proxy( kmp_int32 gtid, kmp_task_t * ptask )
30587cc577a4SJonathan Peyton {
30597cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(ptask);
30607cc577a4SJonathan Peyton     kmp_info_t * thread = __kmp_threads[ gtid ];
30617cc577a4SJonathan Peyton 
30627cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.proxy == TASK_PROXY );
30637cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata -> td_flags.complete == 1 ); // top half must run before bottom half
30647cc577a4SJonathan Peyton 
30657cc577a4SJonathan Peyton     // We need to wait to make sure the top half is finished
30667cc577a4SJonathan Peyton     // Spinning here should be ok as this should happen quickly
30677cc577a4SJonathan Peyton     while ( TCR_4(taskdata->td_incomplete_child_tasks) > 0 ) ;
30687cc577a4SJonathan Peyton 
30697cc577a4SJonathan Peyton     __kmp_release_deps(gtid,taskdata);
30707cc577a4SJonathan Peyton     __kmp_free_task_and_ancestors(gtid, taskdata, thread);
30717cc577a4SJonathan Peyton }
30727cc577a4SJonathan Peyton 
30737cc577a4SJonathan Peyton /*!
30747cc577a4SJonathan Peyton @ingroup TASKING
30757cc577a4SJonathan Peyton @param gtid Global Thread ID of encountering thread
30767cc577a4SJonathan Peyton @param ptask Task which execution is completed
30777cc577a4SJonathan Peyton 
30787cc577a4SJonathan Peyton Execute the completation of a proxy task from a thread of that is part of the team. Run first and bottom halves directly.
30797cc577a4SJonathan Peyton */
30807cc577a4SJonathan Peyton void __kmpc_proxy_task_completed( kmp_int32 gtid, kmp_task_t *ptask )
30817cc577a4SJonathan Peyton {
30827cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( ptask != NULL );
30837cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(ptask);
30847cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_proxy_task_completed(enter): T#%d proxy task %p completing\n", gtid, taskdata ) );
30857cc577a4SJonathan Peyton 
30867cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_flags.proxy == TASK_PROXY );
30877cc577a4SJonathan Peyton 
30887cc577a4SJonathan Peyton     __kmp_first_top_half_finish_proxy(taskdata);
30897cc577a4SJonathan Peyton     __kmp_second_top_half_finish_proxy(taskdata);
30907cc577a4SJonathan Peyton     __kmp_bottom_half_finish_proxy(gtid,ptask);
30917cc577a4SJonathan Peyton 
30927cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_proxy_task_completed(exit): T#%d proxy task %p completing\n", gtid, taskdata ) );
30937cc577a4SJonathan Peyton }
30947cc577a4SJonathan Peyton 
30957cc577a4SJonathan Peyton /*!
30967cc577a4SJonathan Peyton @ingroup TASKING
30977cc577a4SJonathan Peyton @param ptask Task which execution is completed
30987cc577a4SJonathan Peyton 
30997cc577a4SJonathan Peyton Execute the completation of a proxy task from a thread that could not belong to the team.
31007cc577a4SJonathan Peyton */
31017cc577a4SJonathan Peyton void __kmpc_proxy_task_completed_ooo ( kmp_task_t *ptask )
31027cc577a4SJonathan Peyton {
31037cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( ptask != NULL );
31047cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(ptask);
31057cc577a4SJonathan Peyton 
31067cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_proxy_task_completed_ooo(enter): proxy task completing ooo %p\n", taskdata ) );
31077cc577a4SJonathan Peyton 
31087cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata->td_flags.proxy == TASK_PROXY );
31097cc577a4SJonathan Peyton 
31107cc577a4SJonathan Peyton     __kmp_first_top_half_finish_proxy(taskdata);
31117cc577a4SJonathan Peyton 
31127cc577a4SJonathan Peyton     // Enqueue task to complete bottom half completion from a thread within the corresponding team
31137cc577a4SJonathan Peyton     kmp_team_t * team = taskdata->td_team;
31147cc577a4SJonathan Peyton     kmp_int32 nthreads = team->t.t_nproc;
31157cc577a4SJonathan Peyton     kmp_info_t *thread;
31167cc577a4SJonathan Peyton 
31177cc577a4SJonathan Peyton     //This should be similar to start_k = __kmp_get_random( thread ) % nthreads but we cannot use __kmp_get_random here
31187cc577a4SJonathan Peyton     kmp_int32 start_k = 0;
31197cc577a4SJonathan Peyton     kmp_int32 pass = 1;
31207cc577a4SJonathan Peyton     kmp_int32 k = start_k;
31217cc577a4SJonathan Peyton 
31227cc577a4SJonathan Peyton     do {
31237cc577a4SJonathan Peyton         //For now we're just linearly trying to find a thread
31247cc577a4SJonathan Peyton         thread = team->t.t_threads[k];
31257cc577a4SJonathan Peyton         k = (k+1) % nthreads;
31267cc577a4SJonathan Peyton 
31277cc577a4SJonathan Peyton         // we did a full pass through all the threads
31287cc577a4SJonathan Peyton         if ( k == start_k ) pass = pass << 1;
31297cc577a4SJonathan Peyton 
31307cc577a4SJonathan Peyton     } while ( !__kmp_give_task( thread, k,  ptask, pass ) );
31317cc577a4SJonathan Peyton 
31327cc577a4SJonathan Peyton     __kmp_second_top_half_finish_proxy(taskdata);
31337cc577a4SJonathan Peyton 
31347cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_proxy_task_completed_ooo(exit): proxy task completing ooo %p\n", taskdata ) );
31357cc577a4SJonathan Peyton }
31367cc577a4SJonathan Peyton 
31377cc577a4SJonathan Peyton //---------------------------------------------------------------------------------
31387cc577a4SJonathan Peyton // __kmp_task_dup_alloc: Allocate the taskdata and make a copy of source task for taskloop
31397cc577a4SJonathan Peyton //
31407cc577a4SJonathan Peyton // thread:   allocating thread
31417cc577a4SJonathan Peyton // task_src: pointer to source task to be duplicated
31427cc577a4SJonathan Peyton // returns:  a pointer to the allocated kmp_task_t structure (task).
31437cc577a4SJonathan Peyton kmp_task_t *
31447cc577a4SJonathan Peyton __kmp_task_dup_alloc( kmp_info_t *thread, kmp_task_t *task_src )
31457cc577a4SJonathan Peyton {
31467cc577a4SJonathan Peyton     kmp_task_t     *task;
31477cc577a4SJonathan Peyton     kmp_taskdata_t *taskdata;
31487cc577a4SJonathan Peyton     kmp_taskdata_t *taskdata_src;
31497cc577a4SJonathan Peyton     kmp_taskdata_t *parent_task = thread->th.th_current_task;
31507cc577a4SJonathan Peyton     size_t shareds_offset;
31517cc577a4SJonathan Peyton     size_t task_size;
31527cc577a4SJonathan Peyton 
31537cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmp_task_dup_alloc(enter): Th %p, source task %p\n", thread, task_src) );
31547cc577a4SJonathan Peyton     taskdata_src = KMP_TASK_TO_TASKDATA( task_src );
31557cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata_src->td_flags.proxy == TASK_FULL ); // it should not be proxy task
31567cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( taskdata_src->td_flags.tasktype == TASK_EXPLICIT );
31577cc577a4SJonathan Peyton     task_size = taskdata_src->td_size_alloc;
31587cc577a4SJonathan Peyton 
31597cc577a4SJonathan Peyton     // Allocate a kmp_taskdata_t block and a kmp_task_t block.
31607cc577a4SJonathan Peyton     KA_TRACE(30, ("__kmp_task_dup_alloc: Th %p, malloc size %ld\n", thread, task_size) );
31617cc577a4SJonathan Peyton     #if USE_FAST_MEMORY
31627cc577a4SJonathan Peyton     taskdata = (kmp_taskdata_t *)__kmp_fast_allocate( thread, task_size );
31637cc577a4SJonathan Peyton     #else
31647cc577a4SJonathan Peyton     taskdata = (kmp_taskdata_t *)__kmp_thread_malloc( thread, task_size );
31657cc577a4SJonathan Peyton     #endif /* USE_FAST_MEMORY */
31667cc577a4SJonathan Peyton     KMP_MEMCPY(taskdata, taskdata_src, task_size);
31677cc577a4SJonathan Peyton 
31687cc577a4SJonathan Peyton     task = KMP_TASKDATA_TO_TASK(taskdata);
31697cc577a4SJonathan Peyton 
31707cc577a4SJonathan Peyton     // Initialize new task (only specific fields not affected by memcpy)
31717cc577a4SJonathan Peyton     taskdata->td_task_id = KMP_GEN_TASK_ID();
31727cc577a4SJonathan Peyton     if( task->shareds != NULL ) { // need setup shareds pointer
31737cc577a4SJonathan Peyton         shareds_offset = (char*)task_src->shareds - (char*)taskdata_src;
31747cc577a4SJonathan Peyton         task->shareds = &((char*)taskdata)[shareds_offset];
31757cc577a4SJonathan Peyton         KMP_DEBUG_ASSERT( (((kmp_uintptr_t)task->shareds) & (sizeof(void*)-1)) == 0 );
31767cc577a4SJonathan Peyton     }
31777cc577a4SJonathan Peyton     taskdata->td_alloc_thread = thread;
31787cc577a4SJonathan Peyton     taskdata->td_taskgroup = parent_task->td_taskgroup; // task inherits the taskgroup from the parent task
31797cc577a4SJonathan Peyton 
31807cc577a4SJonathan Peyton     // Only need to keep track of child task counts if team parallel and tasking not serialized
31817cc577a4SJonathan Peyton     if ( !( taskdata->td_flags.team_serial || taskdata->td_flags.tasking_ser ) ) {
31827cc577a4SJonathan Peyton         KMP_TEST_THEN_INC32( (kmp_int32 *)(& parent_task->td_incomplete_child_tasks) );
31837cc577a4SJonathan Peyton         if ( parent_task->td_taskgroup )
31847cc577a4SJonathan Peyton             KMP_TEST_THEN_INC32( (kmp_int32 *)(& parent_task->td_taskgroup->count) );
31857cc577a4SJonathan Peyton         // Only need to keep track of allocated child tasks for explicit tasks since implicit not deallocated
31867cc577a4SJonathan Peyton         if ( taskdata->td_parent->td_flags.tasktype == TASK_EXPLICIT )
31877cc577a4SJonathan Peyton             KMP_TEST_THEN_INC32( (kmp_int32 *)(& taskdata->td_parent->td_allocated_child_tasks) );
31887cc577a4SJonathan Peyton     }
31897cc577a4SJonathan Peyton 
31907cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmp_task_dup_alloc(exit): Th %p, created task %p, parent=%p\n",
31917cc577a4SJonathan Peyton                   thread, taskdata, taskdata->td_parent) );
31927cc577a4SJonathan Peyton #if OMPT_SUPPORT
31937cc577a4SJonathan Peyton     __kmp_task_init_ompt(taskdata, thread->th.th_info.ds.ds_gtid, (void*)task->routine);
31947cc577a4SJonathan Peyton #endif
31957cc577a4SJonathan Peyton     return task;
31967cc577a4SJonathan Peyton }
31977cc577a4SJonathan Peyton 
31987cc577a4SJonathan Peyton // Routine optionally generated by th ecompiler for setting the lastprivate flag
31997cc577a4SJonathan Peyton // and calling needed constructors for private/firstprivate objects
32007cc577a4SJonathan Peyton // (used to form taskloop tasks from pattern task)
32017cc577a4SJonathan Peyton typedef void(*p_task_dup_t)(kmp_task_t *, kmp_task_t *, kmp_int32);
32027cc577a4SJonathan Peyton 
32037cc577a4SJonathan Peyton //---------------------------------------------------------------------------------
32047cc577a4SJonathan Peyton // __kmp_taskloop_linear: Start tasks of the taskloop linearly
32057cc577a4SJonathan Peyton //
32067cc577a4SJonathan Peyton // loc       Source location information
32077cc577a4SJonathan Peyton // gtid      Global thread ID
32087cc577a4SJonathan Peyton // task      Task with whole loop iteration range
32097cc577a4SJonathan Peyton // lb        Pointer to loop lower bound
32107cc577a4SJonathan Peyton // ub        Pointer to loop upper bound
32117cc577a4SJonathan Peyton // st        Loop stride
32127cc577a4SJonathan Peyton // sched     Schedule specified 0/1/2 for none/grainsize/num_tasks
32137cc577a4SJonathan Peyton // grainsize Schedule value if specified
32147cc577a4SJonathan Peyton // task_dup  Tasks duplication routine
32157cc577a4SJonathan Peyton void
32167cc577a4SJonathan Peyton __kmp_taskloop_linear(ident_t *loc, int gtid, kmp_task_t *task,
32177cc577a4SJonathan Peyton                 kmp_uint64 *lb, kmp_uint64 *ub, kmp_int64 st,
32187cc577a4SJonathan Peyton                 int sched, kmp_uint64 grainsize, void *task_dup )
32197cc577a4SJonathan Peyton {
32207cc577a4SJonathan Peyton     KMP_COUNT_BLOCK(OMP_TASKLOOP);
32217cc577a4SJonathan Peyton     KMP_TIME_PARTITIONED_BLOCK(OMP_taskloop_scheduling);
32227cc577a4SJonathan Peyton     p_task_dup_t ptask_dup = (p_task_dup_t)task_dup;
32237cc577a4SJonathan Peyton     kmp_uint64 tc;
32247cc577a4SJonathan Peyton     kmp_uint64 lower = *lb; // compiler provides global bounds here
32257cc577a4SJonathan Peyton     kmp_uint64 upper = *ub;
32267cc577a4SJonathan Peyton     kmp_uint64 i, num_tasks = 0, extras = 0;
32277cc577a4SJonathan Peyton     kmp_info_t *thread = __kmp_threads[gtid];
32287cc577a4SJonathan Peyton     kmp_taskdata_t *current_task = thread->th.th_current_task;
32297cc577a4SJonathan Peyton     kmp_task_t *next_task;
32307cc577a4SJonathan Peyton     kmp_int32 lastpriv = 0;
32317cc577a4SJonathan Peyton     size_t lower_offset = (char*)lb - (char*)task; // remember offset of lb in the task structure
32327cc577a4SJonathan Peyton     size_t upper_offset = (char*)ub - (char*)task; // remember offset of ub in the task structure
32337cc577a4SJonathan Peyton 
32347cc577a4SJonathan Peyton     // compute trip count
32357cc577a4SJonathan Peyton     if ( st == 1 ) {   // most common case
32367cc577a4SJonathan Peyton         tc = upper - lower + 1;
32377cc577a4SJonathan Peyton     } else if ( st < 0 ) {
32387cc577a4SJonathan Peyton         tc = (lower - upper) / (-st) + 1;
32397cc577a4SJonathan Peyton     } else {       // st > 0
32407cc577a4SJonathan Peyton         tc = (upper - lower) / st + 1;
32417cc577a4SJonathan Peyton     }
32427cc577a4SJonathan Peyton     if(tc == 0) {
32437cc577a4SJonathan Peyton         KA_TRACE(20, ("__kmpc_taskloop(exit): T#%d zero-trip loop\n", gtid));
32447cc577a4SJonathan Peyton         // free the pattern task and exit
32457cc577a4SJonathan Peyton         __kmp_task_start( gtid, task, current_task );
32467cc577a4SJonathan Peyton         // do not execute anything for zero-trip loop
32477cc577a4SJonathan Peyton         __kmp_task_finish( gtid, task, current_task );
32487cc577a4SJonathan Peyton         return;
32497cc577a4SJonathan Peyton     }
32507cc577a4SJonathan Peyton 
32517cc577a4SJonathan Peyton     // compute num_tasks/grainsize based on the input provided
32527cc577a4SJonathan Peyton     switch( sched ) {
32537cc577a4SJonathan Peyton     case 0: // no schedule clause specified, we can choose the default
32547cc577a4SJonathan Peyton             // let's try to schedule (team_size*10) tasks
32557cc577a4SJonathan Peyton         grainsize = thread->th.th_team_nproc * 10;
32567cc577a4SJonathan Peyton     case 2: // num_tasks provided
32577cc577a4SJonathan Peyton         if( grainsize > tc ) {
32587cc577a4SJonathan Peyton             num_tasks = tc;   // too big num_tasks requested, adjust values
32597cc577a4SJonathan Peyton             grainsize = 1;
32607cc577a4SJonathan Peyton             extras = 0;
32617cc577a4SJonathan Peyton         } else {
32627cc577a4SJonathan Peyton             num_tasks = grainsize;
32637cc577a4SJonathan Peyton             grainsize = tc / num_tasks;
32647cc577a4SJonathan Peyton             extras = tc % num_tasks;
32657cc577a4SJonathan Peyton         }
32667cc577a4SJonathan Peyton         break;
32677cc577a4SJonathan Peyton     case 1: // grainsize provided
32687cc577a4SJonathan Peyton         if( grainsize > tc ) {
32697cc577a4SJonathan Peyton             num_tasks = 1;    // too big grainsize requested, adjust values
32707cc577a4SJonathan Peyton             grainsize = tc;
32717cc577a4SJonathan Peyton             extras = 0;
32727cc577a4SJonathan Peyton         } else {
32737cc577a4SJonathan Peyton             num_tasks = tc / grainsize;
32747cc577a4SJonathan Peyton             grainsize = tc / num_tasks; // adjust grainsize for balanced distribution of iterations
32757cc577a4SJonathan Peyton             extras = tc % num_tasks;
32767cc577a4SJonathan Peyton         }
32777cc577a4SJonathan Peyton         break;
32787cc577a4SJonathan Peyton     default:
32797cc577a4SJonathan Peyton         KMP_ASSERT2(0, "unknown scheduling of taskloop");
32807cc577a4SJonathan Peyton     }
32817cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT(tc == num_tasks * grainsize + extras);
32827cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT(num_tasks > extras);
32837cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT(num_tasks > 0);
32847cc577a4SJonathan Peyton     KA_TRACE(20, ("__kmpc_taskloop: T#%d will launch: num_tasks %lld, grainsize %lld, extras %lld\n",
32857cc577a4SJonathan Peyton                   gtid, num_tasks, grainsize, extras));
32867cc577a4SJonathan Peyton 
32877cc577a4SJonathan Peyton     // Main loop, launch num_tasks tasks, assign grainsize iterations each task
32887cc577a4SJonathan Peyton     for( i = 0; i < num_tasks; ++i ) {
32897cc577a4SJonathan Peyton         kmp_uint64 chunk_minus_1;
32907cc577a4SJonathan Peyton         if( extras == 0 ) {
32917cc577a4SJonathan Peyton             chunk_minus_1 = grainsize - 1;
32927cc577a4SJonathan Peyton         } else {
32937cc577a4SJonathan Peyton             chunk_minus_1 = grainsize;
32947cc577a4SJonathan Peyton             --extras; // first extras iterations get bigger chunk (grainsize+1)
32957cc577a4SJonathan Peyton         }
32967cc577a4SJonathan Peyton         upper = lower + st * chunk_minus_1;
32977cc577a4SJonathan Peyton         if( i == num_tasks - 1 ) {
32987cc577a4SJonathan Peyton             // schedule the last task, set lastprivate flag
32997cc577a4SJonathan Peyton             lastpriv = 1;
33007cc577a4SJonathan Peyton #if KMP_DEBUG
33017cc577a4SJonathan Peyton             if( st == 1 )
33027cc577a4SJonathan Peyton                 KMP_DEBUG_ASSERT(upper == *ub);
33037cc577a4SJonathan Peyton             else if( st > 0 )
33047cc577a4SJonathan Peyton                 KMP_DEBUG_ASSERT(upper+st > *ub);
33057cc577a4SJonathan Peyton             else
33067cc577a4SJonathan Peyton                 KMP_DEBUG_ASSERT(upper+st < *ub);
33077cc577a4SJonathan Peyton #endif
33087cc577a4SJonathan Peyton         }
33097cc577a4SJonathan Peyton         next_task = __kmp_task_dup_alloc(thread, task); // allocate new task
33107cc577a4SJonathan Peyton         *(kmp_uint64*)((char*)next_task + lower_offset) = lower; // adjust task-specific bounds
33117cc577a4SJonathan Peyton         *(kmp_uint64*)((char*)next_task + upper_offset) = upper;
33127cc577a4SJonathan Peyton         if( ptask_dup != NULL )
33137cc577a4SJonathan Peyton             ptask_dup(next_task, task, lastpriv); // set lastprivate flag, construct fistprivates, etc.
33147cc577a4SJonathan Peyton         KA_TRACE(20, ("__kmpc_taskloop: T#%d schedule task %p: lower %lld, upper %lld (offsets %p %p)\n",
33157cc577a4SJonathan Peyton                       gtid, next_task, lower, upper, lower_offset, upper_offset));
33167cc577a4SJonathan Peyton         __kmp_omp_task(gtid, next_task, true); // schedule new task
33177cc577a4SJonathan Peyton         lower = upper + st; // adjust lower bound for the next iteration
33187cc577a4SJonathan Peyton     }
33197cc577a4SJonathan Peyton     // free the pattern task and exit
33207cc577a4SJonathan Peyton     __kmp_task_start( gtid, task, current_task );
33217cc577a4SJonathan Peyton     // do not execute the pattern task, just do bookkeeping
33227cc577a4SJonathan Peyton     __kmp_task_finish( gtid, task, current_task );
33237cc577a4SJonathan Peyton }
33247cc577a4SJonathan Peyton 
33257cc577a4SJonathan Peyton /*!
33267cc577a4SJonathan Peyton @ingroup TASKING
33277cc577a4SJonathan Peyton @param loc       Source location information
33287cc577a4SJonathan Peyton @param gtid      Global thread ID
33297cc577a4SJonathan Peyton @param task      Task structure
33307cc577a4SJonathan Peyton @param if_val    Value of the if clause
33317cc577a4SJonathan Peyton @param lb        Pointer to loop lower bound
33327cc577a4SJonathan Peyton @param ub        Pointer to loop upper bound
33337cc577a4SJonathan Peyton @param st        Loop stride
33347cc577a4SJonathan Peyton @param nogroup   Flag, 1 if nogroup clause specified, 0 otherwise
33357cc577a4SJonathan Peyton @param sched     Schedule specified 0/1/2 for none/grainsize/num_tasks
33367cc577a4SJonathan Peyton @param grainsize Schedule value if specified
33377cc577a4SJonathan Peyton @param task_dup  Tasks duplication routine
33387cc577a4SJonathan Peyton 
33397cc577a4SJonathan Peyton Execute the taskloop construct.
33407cc577a4SJonathan Peyton */
33417cc577a4SJonathan Peyton void
33427cc577a4SJonathan Peyton __kmpc_taskloop(ident_t *loc, int gtid, kmp_task_t *task, int if_val,
33437cc577a4SJonathan Peyton                 kmp_uint64 *lb, kmp_uint64 *ub, kmp_int64 st,
33447cc577a4SJonathan Peyton                 int nogroup, int sched, kmp_uint64 grainsize, void *task_dup )
33457cc577a4SJonathan Peyton {
33467cc577a4SJonathan Peyton     kmp_taskdata_t * taskdata = KMP_TASK_TO_TASKDATA(task);
33477cc577a4SJonathan Peyton     KMP_DEBUG_ASSERT( task != NULL );
33487cc577a4SJonathan Peyton 
33497cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_taskloop(enter): T#%d, pattern task %p, lb %lld ub %lld st %lld, grain %llu(%d)\n",
33507cc577a4SJonathan Peyton         gtid, taskdata, *lb, *ub, st, grainsize, sched));
33517cc577a4SJonathan Peyton 
33527cc577a4SJonathan Peyton     // check if clause value first
33537cc577a4SJonathan Peyton     if( if_val == 0 ) { // if(0) specified, mark task as serial
33547cc577a4SJonathan Peyton         taskdata->td_flags.task_serial = 1;
33557cc577a4SJonathan Peyton         taskdata->td_flags.tiedness = TASK_TIED; // AC: serial task cannot be untied
33567cc577a4SJonathan Peyton     }
33577cc577a4SJonathan Peyton     if( nogroup == 0 ) {
33587cc577a4SJonathan Peyton         __kmpc_taskgroup( loc, gtid );
33597cc577a4SJonathan Peyton     }
33607cc577a4SJonathan Peyton 
33617cc577a4SJonathan Peyton     if( 1 /* AC: use some heuristic here to choose task scheduling method */ ) {
33627cc577a4SJonathan Peyton         __kmp_taskloop_linear( loc, gtid, task, lb, ub, st, sched, grainsize, task_dup );
33637cc577a4SJonathan Peyton     }
33647cc577a4SJonathan Peyton 
33657cc577a4SJonathan Peyton     if( nogroup == 0 ) {
33667cc577a4SJonathan Peyton         __kmpc_end_taskgroup( loc, gtid );
33677cc577a4SJonathan Peyton     }
33687cc577a4SJonathan Peyton     KA_TRACE(10, ("__kmpc_taskloop(exit): T#%d\n", gtid));
33697cc577a4SJonathan Peyton }
33707cc577a4SJonathan Peyton 
33717cc577a4SJonathan Peyton #endif
3372