Skip to content

Commit

Permalink
Update cpp/src/neighbors/cagra_c.cpp
Browse files Browse the repository at this point in the history
Co-authored-by: Micka <[email protected]>
  • Loading branch information
ajit283 and lowener authored Nov 29, 2024
1 parent 0d0184a commit e3d33a0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cpp/src/neighbors/cagra_c.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ void _search(cuvsResources_t res,
cuvs::neighbors::cagra::search(
*res_ptr, search_params, *index_ptr, queries_mds, neighbors_mds, distances_mds);
} else if (filter.type == BITSET) {
using filter_mdspan_type = raft::device_vector_view<int64_t, int64_t, raft::row_major>;
using filter_mdspan_type = raft::device_vector_view<std::uint32_t, int64_t, raft::row_major>;
auto removed_indices_tensor = reinterpret_cast<DLManagedTensor*>(filter.addr);
auto removed_indices = cuvs::core::from_dlpack<filter_mdspan_type>(removed_indices_tensor);
cuvs::core::bitset<std::uint32_t, int64_t> removed_indices_bitset(
Expand Down

0 comments on commit e3d33a0

Please sign in to comment.