diff options
| author | Ron Lieberman <ronlieb.g@gmail.com> | 2019-07-17 17:07:52 +0000 |
|---|---|---|
| committer | Ron Lieberman <ronlieb.g@gmail.com> | 2019-07-17 17:07:52 +0000 |
| commit | 59532488b1d65c52aa5122bd7e191fde9a9df589 (patch) | |
| tree | 9d883bdb1dfa76307431e5b7b92b6146521dbed2 /openmp/libomptarget/src | |
| parent | d912a9ba9b1647984dc65d34ba4422874bc28855 (diff) | |
| download | bcm5719-llvm-59532488b1d65c52aa5122bd7e191fde9a9df589.tar.gz bcm5719-llvm-59532488b1d65c52aa5122bd7e191fde9a9df589.zip | |
[OPENMP] Resolve lost LoopTripCnt for subsequent loops in same thread.
Remove loopTripCnt from threaded device stack after consuming it.
Added a libomptarget DP message to aid in future debugging and to
validate the added testcase, which only runs in Debug build.
Differential Revision: https://reviews.llvm.org/D64808
llvm-svn: 366349
Diffstat (limited to 'openmp/libomptarget/src')
| -rw-r--r-- | openmp/libomptarget/src/omptarget.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/openmp/libomptarget/src/omptarget.cpp b/openmp/libomptarget/src/omptarget.cpp index 39b5cb7da86..c41bf3167cc 100644 --- a/openmp/libomptarget/src/omptarget.cpp +++ b/openmp/libomptarget/src/omptarget.cpp @@ -732,8 +732,11 @@ int target(int64_t device_id, void *host_ptr, int32_t arg_num, uint64_t ltc = 0; TblMapMtx.lock(); auto I = Device.LoopTripCnt.find(__kmpc_global_thread_num(NULL)); - if (I != Device.LoopTripCnt.end()) - std::swap(ltc, I->second); + if (I != Device.LoopTripCnt.end()) { + ltc = I->second; + Device.LoopTripCnt.erase(I); + DP("loop trip count is %lu.\n", ltc); + } TblMapMtx.unlock(); // Launch device execution. |
