diff --git a/cpp/include/cuvs/neighbors/vamana.hpp b/cpp/include/cuvs/neighbors/vamana.hpp index bec17937f..4b4a8a8d3 100644 --- a/cpp/include/cuvs/neighbors/vamana.hpp +++ b/cpp/include/cuvs/neighbors/vamana.hpp @@ -31,7 +31,7 @@ #include #include -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { /** * @defgroup vamana_cpp_index_params Vamana index build parameters * @{ @@ -219,34 +219,34 @@ struct index : cuvs::neighbors::index { * */ auto build(raft::resources const& handle, - const cuvs::neighbors::experimental::vamana::index_params& params, + const cuvs::neighbors::vamana::index_params& params, raft::device_matrix_view dataset) - -> cuvs::neighbors::experimental::vamana::index; + -> cuvs::neighbors::vamana::index; auto build(raft::resources const& handle, - const cuvs::neighbors::experimental::vamana::index_params& params, + const cuvs::neighbors::vamana::index_params& params, raft::host_matrix_view dataset) - -> cuvs::neighbors::experimental::vamana::index; + -> cuvs::neighbors::vamana::index; auto build(raft::resources const& handle, - const cuvs::neighbors::experimental::vamana::index_params& params, + const cuvs::neighbors::vamana::index_params& params, raft::device_matrix_view dataset) - -> cuvs::neighbors::experimental::vamana::index; + -> cuvs::neighbors::vamana::index; auto build(raft::resources const& handle, - const cuvs::neighbors::experimental::vamana::index_params& params, + const cuvs::neighbors::vamana::index_params& params, raft::host_matrix_view dataset) - -> cuvs::neighbors::experimental::vamana::index; + -> cuvs::neighbors::vamana::index; auto build(raft::resources const& handle, - const cuvs::neighbors::experimental::vamana::index_params& params, + const cuvs::neighbors::vamana::index_params& params, raft::device_matrix_view dataset) - -> cuvs::neighbors::experimental::vamana::index; + -> cuvs::neighbors::vamana::index; auto build(raft::resources const& handle, - const cuvs::neighbors::experimental::vamana::index_params& params, + const cuvs::neighbors::vamana::index_params& params, raft::host_matrix_view dataset) - -> cuvs::neighbors::experimental::vamana::index; + -> cuvs::neighbors::vamana::index; /** * @defgroup vamana_cpp_serialize Vamana serialize functions @@ -258,18 +258,18 @@ auto build(raft::resources const& handle, void serialize(raft::resources const& handle, const std::string& file_prefix, - const cuvs::neighbors::experimental::vamana::index& index); + const cuvs::neighbors::vamana::index& index); void serialize(raft::resources const& handle, const std::string& file_prefix, - const cuvs::neighbors::experimental::vamana::index& index); + const cuvs::neighbors::vamana::index& index); void serialize(raft::resources const& handle, const std::string& file_prefix, - const cuvs::neighbors::experimental::vamana::index& index); + const cuvs::neighbors::vamana::index& index); /** * @} */ -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/src/neighbors/detail/vamana/greedy_search.cuh b/cpp/src/neighbors/detail/vamana/greedy_search.cuh index f7dff7a25..6a7938321 100644 --- a/cpp/src/neighbors/detail/vamana/greedy_search.cuh +++ b/cpp/src/neighbors/detail/vamana/greedy_search.cuh @@ -30,7 +30,7 @@ #include #include -namespace cuvs::neighbors::experimental::vamana::detail { +namespace cuvs::neighbors::vamana::detail { /* @defgroup greedy_search_detail greedy search * @{ @@ -286,4 +286,4 @@ __global__ void GreedySearchKernel( * @} */ -} // namespace cuvs::neighbors::experimental::vamana::detail +} // namespace cuvs::neighbors::vamana::detail diff --git a/cpp/src/neighbors/detail/vamana/macros.cuh b/cpp/src/neighbors/detail/vamana/macros.cuh index 5692650a0..c290413a2 100644 --- a/cpp/src/neighbors/detail/vamana/macros.cuh +++ b/cpp/src/neighbors/detail/vamana/macros.cuh @@ -16,7 +16,7 @@ #pragma once -namespace cuvs::neighbors::experimental::vamana::detail { +namespace cuvs::neighbors::vamana::detail { /* Macros to compute the shared memory requirements for CUB primitives used by search and prune */ #define COMPUTE_SMEM_SIZES(degree, visited_size, DEG, CANDS) \ @@ -79,4 +79,4 @@ namespace cuvs::neighbors::experimental::vamana::detail { SEARCH_CALL_SORT(topk, 512); \ SEARCH_CALL_SORT(topk, 1024); -} // namespace cuvs::neighbors::experimental::vamana::detail +} // namespace cuvs::neighbors::vamana::detail diff --git a/cpp/src/neighbors/detail/vamana/priority_queue.cuh b/cpp/src/neighbors/detail/vamana/priority_queue.cuh index 4b3bd8466..6dc1dc94a 100644 --- a/cpp/src/neighbors/detail/vamana/priority_queue.cuh +++ b/cpp/src/neighbors/detail/vamana/priority_queue.cuh @@ -20,7 +20,7 @@ #include #include -namespace cuvs::neighbors::experimental::vamana::detail { +namespace cuvs::neighbors::vamana::detail { /*************************************************************************************** ***************************************************************************************/ @@ -326,4 +326,4 @@ __forceinline__ __device__ void enqueue_all_neighbors(int num_neighbors, } } -} // namespace cuvs::neighbors::experimental::vamana::detail +} // namespace cuvs::neighbors::vamana::detail diff --git a/cpp/src/neighbors/detail/vamana/robust_prune.cuh b/cpp/src/neighbors/detail/vamana/robust_prune.cuh index fe96dc036..ee51776cf 100644 --- a/cpp/src/neighbors/detail/vamana/robust_prune.cuh +++ b/cpp/src/neighbors/detail/vamana/robust_prune.cuh @@ -22,7 +22,7 @@ #include "macros.cuh" #include "vamana_structs.cuh" -namespace cuvs::neighbors::experimental::vamana::detail { +namespace cuvs::neighbors::vamana::detail { // Load candidates (from query) and previous edges (from nbh_list) into registers (tmp) spanning // warp @@ -247,4 +247,4 @@ __global__ void RobustPruneKernel( } // namespace -} // namespace cuvs::neighbors::experimental::vamana::detail +} // namespace cuvs::neighbors::vamana::detail diff --git a/cpp/src/neighbors/detail/vamana/vamana_build.cuh b/cpp/src/neighbors/detail/vamana/vamana_build.cuh index 7d961b6eb..65f683a08 100644 --- a/cpp/src/neighbors/detail/vamana/vamana_build.cuh +++ b/cpp/src/neighbors/detail/vamana/vamana_build.cuh @@ -46,7 +46,7 @@ #include #include -namespace cuvs::neighbors::experimental::vamana::detail { +namespace cuvs::neighbors::vamana::detail { /* @defgroup vamana_build_detail vamana build * @{ @@ -406,4 +406,4 @@ index build( * @} */ -} // namespace cuvs::neighbors::experimental::vamana::detail +} // namespace cuvs::neighbors::vamana::detail diff --git a/cpp/src/neighbors/detail/vamana/vamana_serialize.cuh b/cpp/src/neighbors/detail/vamana/vamana_serialize.cuh index c360ae19a..27a17205e 100644 --- a/cpp/src/neighbors/detail/vamana/vamana_serialize.cuh +++ b/cpp/src/neighbors/detail/vamana/vamana_serialize.cuh @@ -34,7 +34,7 @@ #include #include -namespace cuvs::neighbors::experimental::vamana::detail { +namespace cuvs::neighbors::vamana::detail { /** * Save the index to file. @@ -117,4 +117,4 @@ void serialize(raft::resources const& res, if (!index_of) { RAFT_FAIL("Error writing output %s", file_name.c_str()); } } -} // namespace cuvs::neighbors::experimental::vamana::detail +} // namespace cuvs::neighbors::vamana::detail diff --git a/cpp/src/neighbors/detail/vamana/vamana_structs.cuh b/cpp/src/neighbors/detail/vamana/vamana_structs.cuh index 5485d5e94..fade674dc 100644 --- a/cpp/src/neighbors/detail/vamana/vamana_structs.cuh +++ b/cpp/src/neighbors/detail/vamana/vamana_structs.cuh @@ -34,7 +34,7 @@ #include -namespace cuvs::neighbors::experimental::vamana::detail { +namespace cuvs::neighbors::vamana::detail { /* @defgroup vamana_structures vamana structures * @{ @@ -476,4 +476,4 @@ __global__ void recompute_reverse_dists( * @} */ -} // namespace cuvs::neighbors::experimental::vamana::detail +} // namespace cuvs::neighbors::vamana::detail diff --git a/cpp/src/neighbors/vamana.cuh b/cpp/src/neighbors/vamana.cuh index 9b9e8d271..81b63c52f 100644 --- a/cpp/src/neighbors/vamana.cuh +++ b/cpp/src/neighbors/vamana.cuh @@ -31,7 +31,7 @@ #include -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { /** * @defgroup VAMANA ANN Graph-based nearest neighbor search @@ -85,7 +85,7 @@ index build( const index_params& params, raft::mdspan, raft::row_major, Accessor> dataset) { - return cuvs::neighbors::experimental::vamana::detail::build( + return cuvs::neighbors::vamana::detail::build( res, params, dataset); } @@ -94,9 +94,9 @@ void serialize(raft::resources const& res, const std::string& file_prefix, const index& index_) { - cuvs::neighbors::experimental::vamana::detail::build(res, file_prefix, index_); + cuvs::neighbors::vamana::detail::build(res, file_prefix, index_); } /** @} */ // end group vamana -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/src/neighbors/vamana_build_float.cu b/cpp/src/neighbors/vamana_build_float.cu index b83af6122..33a5381ff 100644 --- a/cpp/src/neighbors/vamana_build_float.cu +++ b/cpp/src/neighbors/vamana_build_float.cu @@ -17,27 +17,27 @@ #include "vamana.cuh" #include -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { #define RAFT_INST_VAMANA_BUILD(T, IdxT) \ auto build(raft::resources const& handle, \ - const cuvs::neighbors::experimental::vamana::index_params& params, \ + const cuvs::neighbors::vamana::index_params& params, \ raft::device_matrix_view dataset) \ - ->cuvs::neighbors::experimental::vamana::index \ + ->cuvs::neighbors::vamana::index \ { \ - return cuvs::neighbors::experimental::vamana::build(handle, params, dataset); \ + return cuvs::neighbors::vamana::build(handle, params, dataset); \ } \ \ auto build(raft::resources const& handle, \ - const cuvs::neighbors::experimental::vamana::index_params& params, \ + const cuvs::neighbors::vamana::index_params& params, \ raft::host_matrix_view dataset) \ - ->cuvs::neighbors::experimental::vamana::index \ + ->cuvs::neighbors::vamana::index \ { \ - return cuvs::neighbors::experimental::vamana::build(handle, params, dataset); \ + return cuvs::neighbors::vamana::build(handle, params, dataset); \ } RAFT_INST_VAMANA_BUILD(float, uint32_t); #undef RAFT_INST_VAMANA_BUILD -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/src/neighbors/vamana_build_int8.cu b/cpp/src/neighbors/vamana_build_int8.cu index 91d2cf028..a942bfbe2 100644 --- a/cpp/src/neighbors/vamana_build_int8.cu +++ b/cpp/src/neighbors/vamana_build_int8.cu @@ -17,27 +17,27 @@ #include "vamana.cuh" #include -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { #define RAFT_INST_VAMANA_BUILD(T, IdxT) \ auto build(raft::resources const& handle, \ - const cuvs::neighbors::experimental::vamana::index_params& params, \ + const cuvs::neighbors::vamana::index_params& params, \ raft::device_matrix_view dataset) \ - ->cuvs::neighbors::experimental::vamana::index \ + ->cuvs::neighbors::vamana::index \ { \ - return cuvs::neighbors::experimental::vamana::build(handle, params, dataset); \ + return cuvs::neighbors::vamana::build(handle, params, dataset); \ } \ \ auto build(raft::resources const& handle, \ - const cuvs::neighbors::experimental::vamana::index_params& params, \ + const cuvs::neighbors::vamana::index_params& params, \ raft::host_matrix_view dataset) \ - ->cuvs::neighbors::experimental::vamana::index \ + ->cuvs::neighbors::vamana::index \ { \ - return cuvs::neighbors::experimental::vamana::build(handle, params, dataset); \ + return cuvs::neighbors::vamana::build(handle, params, dataset); \ } RAFT_INST_VAMANA_BUILD(int8_t, uint32_t); #undef RAFT_INST_VAMANA_BUILD -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/src/neighbors/vamana_build_uint8.cu b/cpp/src/neighbors/vamana_build_uint8.cu index bba93e7f4..e1f8f1d21 100644 --- a/cpp/src/neighbors/vamana_build_uint8.cu +++ b/cpp/src/neighbors/vamana_build_uint8.cu @@ -17,27 +17,27 @@ #include "vamana.cuh" #include -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { #define RAFT_INST_VAMANA_BUILD(T, IdxT) \ auto build(raft::resources const& handle, \ - const cuvs::neighbors::experimental::vamana::index_params& params, \ + const cuvs::neighbors::vamana::index_params& params, \ raft::device_matrix_view dataset) \ - ->cuvs::neighbors::experimental::vamana::index \ + ->cuvs::neighbors::vamana::index \ { \ - return cuvs::neighbors::experimental::vamana::build(handle, params, dataset); \ + return cuvs::neighbors::vamana::build(handle, params, dataset); \ } \ \ auto build(raft::resources const& handle, \ - const cuvs::neighbors::experimental::vamana::index_params& params, \ + const cuvs::neighbors::vamana::index_params& params, \ raft::host_matrix_view dataset) \ - ->cuvs::neighbors::experimental::vamana::index \ + ->cuvs::neighbors::vamana::index \ { \ - return cuvs::neighbors::experimental::vamana::build(handle, params, dataset); \ + return cuvs::neighbors::vamana::build(handle, params, dataset); \ } RAFT_INST_VAMANA_BUILD(uint8_t, uint32_t); #undef RAFT_INST_VAMANA_BUILD -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/src/neighbors/vamana_serialize.cuh b/cpp/src/neighbors/vamana_serialize.cuh index a49d267b3..a800d497a 100644 --- a/cpp/src/neighbors/vamana_serialize.cuh +++ b/cpp/src/neighbors/vamana_serialize.cuh @@ -18,7 +18,7 @@ #include "detail/vamana/vamana_serialize.cuh" -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { /** * @defgroup VAMANA graph serialize/derserialize @@ -28,12 +28,12 @@ namespace cuvs::neighbors::experimental::vamana { #define CUVS_INST_VAMANA_SERIALIZE(DTYPE) \ void serialize(raft::resources const& handle, \ const std::string& file_prefix, \ - const cuvs::neighbors::experimental::vamana::index& index_) \ + const cuvs::neighbors::vamana::index& index_) \ { \ - cuvs::neighbors::experimental::vamana::detail::serialize( \ + cuvs::neighbors::vamana::detail::serialize( \ handle, file_prefix, index_); \ }; /** @} */ // end group vamana -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/src/neighbors/vamana_serialize_float.cu b/cpp/src/neighbors/vamana_serialize_float.cu index f25369368..8bf7ceb1e 100644 --- a/cpp/src/neighbors/vamana_serialize_float.cu +++ b/cpp/src/neighbors/vamana_serialize_float.cu @@ -16,8 +16,8 @@ #include "vamana_serialize.cuh" -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { CUVS_INST_VAMANA_SERIALIZE(float); -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/src/neighbors/vamana_serialize_int8.cu b/cpp/src/neighbors/vamana_serialize_int8.cu index 1cd54b198..0f87f67ce 100644 --- a/cpp/src/neighbors/vamana_serialize_int8.cu +++ b/cpp/src/neighbors/vamana_serialize_int8.cu @@ -16,8 +16,8 @@ #include "vamana_serialize.cuh" -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { CUVS_INST_VAMANA_SERIALIZE(int8_t); -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/src/neighbors/vamana_serialize_uint8.cu b/cpp/src/neighbors/vamana_serialize_uint8.cu index 3e6d945b8..871c30506 100644 --- a/cpp/src/neighbors/vamana_serialize_uint8.cu +++ b/cpp/src/neighbors/vamana_serialize_uint8.cu @@ -16,8 +16,8 @@ #include "vamana_serialize.cuh" -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { CUVS_INST_VAMANA_SERIALIZE(uint8_t); -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/test/neighbors/ann_vamana.cuh b/cpp/test/neighbors/ann_vamana.cuh index 857257124..3d4f88639 100644 --- a/cpp/test/neighbors/ann_vamana.cuh +++ b/cpp/test/neighbors/ann_vamana.cuh @@ -46,7 +46,7 @@ #include #include -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { struct edge_op { template @@ -335,4 +335,4 @@ inline std::vector generate_inputs() const std::vector inputs = generate_inputs(); -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/test/neighbors/ann_vamana/test_float_uint32_t.cu b/cpp/test/neighbors/ann_vamana/test_float_uint32_t.cu index 9aa9da1b8..7b89b6544 100644 --- a/cpp/test/neighbors/ann_vamana/test_float_uint32_t.cu +++ b/cpp/test/neighbors/ann_vamana/test_float_uint32_t.cu @@ -18,11 +18,11 @@ #include "../ann_vamana.cuh" -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { typedef AnnVamanaTest AnnVamanaTestF_U32; TEST_P(AnnVamanaTestF_U32, AnnVamana) { this->testVamana(); } INSTANTIATE_TEST_CASE_P(AnnVamanaTest, AnnVamanaTestF_U32, ::testing::ValuesIn(inputs)); -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/test/neighbors/ann_vamana/test_int8_t_uint32_t.cu b/cpp/test/neighbors/ann_vamana/test_int8_t_uint32_t.cu index 0a6b563b2..843d2274a 100644 --- a/cpp/test/neighbors/ann_vamana/test_int8_t_uint32_t.cu +++ b/cpp/test/neighbors/ann_vamana/test_int8_t_uint32_t.cu @@ -18,11 +18,11 @@ #include "../ann_vamana.cuh" -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { typedef AnnVamanaTest AnnVamanaTestI8_U32; TEST_P(AnnVamanaTestI8_U32, AnnVamana) { this->testVamana(); } INSTANTIATE_TEST_CASE_P(AnnVamanaTest, AnnVamanaTestI8_U32, ::testing::ValuesIn(inputs)); -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/cpp/test/neighbors/ann_vamana/test_uint8_t_uint32_t.cu b/cpp/test/neighbors/ann_vamana/test_uint8_t_uint32_t.cu index c0680dc18..f08db0c49 100644 --- a/cpp/test/neighbors/ann_vamana/test_uint8_t_uint32_t.cu +++ b/cpp/test/neighbors/ann_vamana/test_uint8_t_uint32_t.cu @@ -18,11 +18,11 @@ #include "../ann_vamana.cuh" -namespace cuvs::neighbors::experimental::vamana { +namespace cuvs::neighbors::vamana { typedef AnnVamanaTest AnnVamanaTestU8_U32; TEST_P(AnnVamanaTestU8_U32, AnnVamana) { this->testVamana(); } INSTANTIATE_TEST_CASE_P(AnnVamanaTest, AnnVamanaTestU8_U32, ::testing::ValuesIn(inputs)); -} // namespace cuvs::neighbors::experimental::vamana +} // namespace cuvs::neighbors::vamana diff --git a/docs/source/cpp_api/neighbors.rst b/docs/source/cpp_api/neighbors.rst index ab810ab53..95359558f 100644 --- a/docs/source/cpp_api/neighbors.rst +++ b/docs/source/cpp_api/neighbors.rst @@ -18,3 +18,4 @@ Nearest Neighbors neighbors_nn_descent.rst neighbors_refine.rst neighbors_mg.rst + neighbors_vamana.rst