commit 4200e2aa1250692e48654355e373897d9fb2e91c
author Hansang Bae <hansang.bae@intel.com> Fri Jan 15 17:37:13 2021 -0600
committer Copybara-Service <copybara-worker@google.com> Tue Jan 19 17:23:27 2021 -0800
tree 61cdd8169929a6a71f64ed19b50a7ea74a8d7ec0
parent 2176448297e2d9ca2b2e3e570904b1f75368dfe3

[OpenMP] Fix atomic entries for captured logical operation

Added missing code for the captured atomic operation.

Differential Revision: https://reviews.llvm.org/D94848

GitOrigin-RevId: 2d911f7c72f9174a34f74abe2909f992b03caaf1

runtime/src/kmp_atomic.cpp

diff --git a/runtime/src/kmp_atomic.cpp b/runtime/src/kmp_atomic.cpp
index 4d60b55..a9d5257 100644
--- a/runtime/src/kmp_atomic.cpp
+++ b/runtime/src/kmp_atomic.cpp
@@ -2536,8 +2536,11 @@
                                                                                \
   if (flag) {                                                                  \
     new_value OP rhs;                                                          \
-  } else                                                                       \
+    (*lhs) = new_value;                                                        \
+  } else {                                                                     \
     new_value = (*lhs);                                                        \
+    (*lhs) OP rhs;                                                             \
+  }                                                                            \
                                                                                \
   __kmp_release_atomic_lock(&ATOMIC_LOCK##LCK_ID, gtid);