From 24753a31c26753f31ba8ec20fda0a3aeb578447b Mon Sep 17 00:00:00 2001
From: Yunsong Wang <yunsongw@nvidia.com>
Date: Tue, 23 Jul 2024 11:40:55 -0700
Subject: [PATCH] Fix rehash error

---
 include/cuco/detail/open_addressing/kernels.cuh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/include/cuco/detail/open_addressing/kernels.cuh b/include/cuco/detail/open_addressing/kernels.cuh
index e4aa5bc3f..266335a50 100644
--- a/include/cuco/detail/open_addressing/kernels.cuh
+++ b/include/cuco/detail/open_addressing/kernels.cuh
@@ -577,7 +577,7 @@ CUCO_KERNEL __launch_bounds__(BlockSize) void rehash(
   auto idx                       = cuco::detail::global_thread_id();
   auto const n                   = storage_ref.num_windows();
 
-  while ((idx - thread_rank / cg_size) < n) {
+  while (idx - thread_rank < n) {
     if (thread_rank == 0) { buffer_size = 0; }
     block.sync();