Fix data type of query id offset

rapidsai · Sep 8, 2023 · 79f69e9 · 79f69e9
1 parent a083244
commit 79f69e9
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
@@ -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)
   {
   }