Update cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh

Co-authored-by: Micka <[email protected]>
rapidsai · Sep 8, 2023 · 5eae9a3 · 5eae9a3
1 parent b536eed
commit 5eae9a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
@@ -35,7 +35,7 @@ namespace raft::neighbors::cagra::detail {
 
 template <class CagraSampleFilterT>
 struct CagraSampleFilterWithQueryIdOffset {
-  const std::size_t offset;
+  const uint32_t offset;
   CagraSampleFilterT filter;
 
   CagraSampleFilterWithQueryIdOffset(const uint32_t offset, const CagraSampleFilterT filter)