commit 6a1be003ffd1e677d107fa259d38b73cf20953cb
author Andrey Churbanov <Andrey.Churbanov@intel.com> Wed May 06 18:34:15 2015 +0000
committer Copybara-Service <copybara-worker@google.com> Mon Aug 31 20:09:09 2020 -0700
tree 370319c3b7da4fca2e61b9b736e215b49a13f74c
parent bb8bfdd4015c5b48ad9bc0914ec352cf16743953

D9302.partial: bug fix for ittnotify: added missed timing on hierarchical barrier.

llvm-svn: 236623
GitOrigin-RevId: e6bfb73fd56af3649577f6e269f1e087218bf46a

runtime/src/kmp_barrier.cpp

diff --git a/runtime/src/kmp_barrier.cpp b/runtime/src/kmp_barrier.cpp
index 72c9fa8..429390b 100644
--- a/runtime/src/kmp_barrier.cpp
+++ b/runtime/src/kmp_barrier.cpp
@@ -743,6 +743,13 @@
                   gtid, team->t.t_id, tid, bt));
     KMP_DEBUG_ASSERT(this_thr == other_threads[this_thr->th.th_info.ds.ds_tid]);
 
+#if USE_ITT_BUILD && USE_ITT_NOTIFY
+    // Barrier imbalance - save arrive time to the thread
+    if(__kmp_forkjoin_frames_mode == 3 || __kmp_forkjoin_frames_mode == 2) {
+        this_thr->th.th_bar_arrive_time = __itt_get_timestamp();
+    }
+#endif
+
     (void)__kmp_init_hierarchical_barrier_thread(bt, thr_bar, nproc, gtid, tid, team);
 
     if (thr_bar->my_level) { // not a leaf (my_level==0 means leaf)