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

Co-authored-by: Tamas Bela Feher <[email protected]>
rapidsai · Sep 21, 2023 · 608f5d1 · 608f5d1
1 parent 8148ef2
commit 608f5d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/include/raft/neighbors/detail/cagra/search_multi_kernel.cuh b/cpp/include/raft/neighbors/detail/cagra/search_multi_kernel.cuh
@@ -501,7 +501,7 @@ void apply_filter(INDEX_T* const result_indices_ptr,
                   cudaStream_t cuda_stream)
 {
   const std::size_t block_size = 256;
-  const std::size_t grid_size  = (num_queries * result_buffer_size + block_size - 1) / block_size;
+  const std::size_t grid_size  = ceildiv(num_queries * result_buffer_size, block_size ) ;
 
   apply_filter_kernel<<<grid_size, block_size, 0, cuda_stream>>>(result_indices_ptr,
                                                                  result_distances_ptr,