Fix a bug in ref initialize

NVIDIA · Dec 13, 2023 · 4080ea5 · 4080ea5
1 parent db9a93f
commit 4080ea5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/include/cuco/detail/open_addressing/open_addressing_ref_impl.cuh b/include/cuco/detail/open_addressing/open_addressing_ref_impl.cuh
@@ -324,10 +324,12 @@ class open_addressing_ref_impl {
   template <typename CG>
   __device__ constexpr void initialize(CG const& tile) noexcept
   {
-    auto tid = tile.thread_rank();
+    auto tid                = tile.thread_rank();
+    auto* const windows_ptr = this->storage_ref().data();
     while (tid < static_cast<size_type>(this->window_extent())) {
+      auto& window = *(windows_ptr + tid);
 #pragma unroll
-      for (auto& slot : this->storage_ref_[tid]) {
+      for (auto& slot : window) {
         slot = this->empty_slot_sentinel();
       }
       tid += tile.size();