diff --git a/cpp/src/neighbors/detail/hnsw.hpp b/cpp/src/neighbors/detail/hnsw.hpp index ce1e03264..b33c2fe01 100644 --- a/cpp/src/neighbors/detail/hnsw.hpp +++ b/cpp/src/neighbors/detail/hnsw.hpp @@ -114,7 +114,7 @@ template void get_search_knn_results(hnswlib::HierarchicalNSW::type> const* idx, const T* query, int k, - uint64_t* indices, + int64_t* indices, float* distances) { auto result = idx->searchKnn(query, k); @@ -132,7 +132,7 @@ void search(raft::resources const& res, const search_params& params, const index& idx, raft::host_matrix_view queries, - raft::host_matrix_view neighbors, + raft::host_matrix_view neighbors, raft::host_matrix_view distances) { RAFT_EXPECTS( diff --git a/cpp/src/neighbors/hnsw_c.cpp b/cpp/src/neighbors/hnsw_c.cpp index a19875641..872a494fe 100644 --- a/cpp/src/neighbors/hnsw_c.cpp +++ b/cpp/src/neighbors/hnsw_c.cpp @@ -47,7 +47,7 @@ void _search(cuvsResources_t res, search_params.num_threads = params.numThreads; using queries_mdspan_type = raft::host_matrix_view; - using neighbors_mdspan_type = raft::host_matrix_view; + using neighbors_mdspan_type = raft::host_matrix_view; using distances_mdspan_type = raft::host_matrix_view; auto queries_mds = cuvs::core::from_dlpack(queries_tensor); auto neighbors_mds = cuvs::core::from_dlpack(neighbors_tensor);