From 79f69e9a38f9890e7a64ca9834e1b028d7f35748 Mon Sep 17 00:00:00 2001
From: Hiroyuki Ootomo <hootomo@nvidia.com>
Date: Fri, 8 Sep 2023 18:13:52 +0900
Subject: [PATCH] Fix data type of query id offset

---
 cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
index 13efa277ed..15ea700412 100644
--- a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
+++ b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
@@ -38,7 +38,7 @@ struct CagraSampleFilterWithQueryIdOffset {
   const std::size_t offset;
   CagraSampleFilterT filter;
 
-  CagraSampleFilterWithQueryIdOffset(const std::size_t offset, const CagraSampleFilterT filter)
+  CagraSampleFilterWithQueryIdOffset(const uint32_t offset, const CagraSampleFilterT filter)
     : offset(offset), filter(filter)
   {
   }