include/oneapi/dpl/pstl/hetero/dpcpp/parallel_backend_sycl.h - analyz…

…e compare_exchange_strong result Signed-off-by: Sergey Kopienko <[email protected]>
uxlfoundation · Jun 26, 2024 · a934c68 · a934c68
1 parent 3538d15
commit a934c68
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/include/oneapi/dpl/pstl/hetero/dpcpp/parallel_backend_sycl.h b/include/oneapi/dpl/pstl/hetero/dpcpp/parallel_backend_sycl.h
@@ -1075,7 +1075,8 @@ struct __early_exit_find_or
                 {
                     for (auto __old = __found_local.load(); __comp(__shifted_idx, __old); __old = __found_local.load())
                     {
-                        __found_local.compare_exchange_strong(__old, __shifted_idx);
+                        if (__found_local.compare_exchange_strong(__old, __shifted_idx))
+                            break;
                     }
                 }
 
@@ -1176,7 +1177,8 @@ __parallel_find_or(oneapi::dpl::__internal::__device_backend_tag, _ExecutionPoli
                             for (auto __old = __found.load(); __comp(__found_local.load(), __old);
                                  __old = __found.load())
                             {
-                                __found.compare_exchange_strong(__old, __found_local.load());
+                                if (__found.compare_exchange_strong(__old, __found_local.load()))
+                                    break;
                             }
                         }
                     }