diff --git a/include/pq_flash_index.h b/include/pq_flash_index.h index 61769c1f9..49f192dbe 100644 --- a/include/pq_flash_index.h +++ b/include/pq_flash_index.h @@ -97,6 +97,7 @@ template class PQFlashIndex DISKANN_DLLEXPORT std::vector get_vector(std::uint64_t VID); DISKANN_DLLEXPORT std::vector get_pq_vector(std::uint64_t VID); DISKANN_DLLEXPORT std::vector get_neighbors(std::uint64_t VID); + DISKANN_DLLEXPORT uint64_t get_num_points(); protected: DISKANN_DLLEXPORT void use_medoids_data_as_centroids(); diff --git a/src/pq_flash_index.cpp b/src/pq_flash_index.cpp index e5b25f048..998d12aff 100644 --- a/src/pq_flash_index.cpp +++ b/src/pq_flash_index.cpp @@ -1615,6 +1615,12 @@ std::vector PQFlashIndex::get_neighbors(std::uint64_t return out; } +template +std::uint64_t PQFlashIndex::get_num_points() +{ + return num_points; +} + #ifdef EXEC_ENV_OLS template char *PQFlashIndex::getHeaderBytes() {