Skip to content

Commit

Permalink
Update cpp/include/raft/neighbors/detail/cagra/search_single_cta_kern…
Browse files Browse the repository at this point in the history
…el-inl.cuh

Co-authored-by: Micka <[email protected]>
  • Loading branch information
enp1s0 and lowener authored Sep 8, 2023
1 parent 79f69e9 commit b536eed
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -531,7 +531,7 @@ __launch_bounds__(BLOCK_SIZE, BLOCK_COUNT) __global__
auto smem_working_ptr = reinterpret_cast<std::uint32_t*>(terminate_flag + 1);

// A flag for filtering.
auto filter_frag = terminate_flag;
auto filter_flag = terminate_flag;

const DATA_T* const query_ptr = queries_ptr + query_id * dataset_dim;
for (unsigned i = threadIdx.x; i < MAX_DATASET_DIM; i += BLOCK_SIZE) {
Expand Down

0 comments on commit b536eed

Please sign in to comment.