fix: restore missing thread sync

This got removed during a rebase.
ROCm · Aug 29, 2024 · e15bcfd · e15bcfd
1 parent 7e4f295
commit e15bcfd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rocprim/include/rocprim/device/detail/device_merge_sort_mergepath.hpp b/rocprim/include/rocprim/device/detail/device_merge_sort_mergepath.hpp
@@ -201,6 +201,7 @@ namespace detail
         unsigned int indices[ItemsPerThread];
 
         serial_merge<false>(keys_shared, keys, indices, range_local, compare_function);
+        rocprim::syncthreads();
 
         if ROCPRIM_IF_CONSTEXPR(with_values){
             reg_to_shared<BlockSize, ItemsPerThread>(values_shared, values);
@@ -336,6 +337,7 @@ namespace detail
         unsigned int indices[ItemsPerThread];
 
         serial_merge<false>(keys_shared, keys, indices, range_local, compare_function);
+        rocprim::syncthreads();
 
         if ROCPRIM_IF_CONSTEXPR(with_values)
         {