diff --git a/include/pq.h b/include/pq.h index 94602e4ae..a5ad5361f 100644 --- a/include/pq.h +++ b/include/pq.h @@ -67,7 +67,7 @@ DISKANN_DLLEXPORT int generate_opq_pivots(const float *train_data, size_t num_tr unsigned num_pq_chunks, std::string opq_pivots_path, bool make_zero_mean = false); -DISKANN_DLLEXPORT int generate_pq_pivots_simplified(const float *train_data, size_t num_train, +DISKANN_DLLEXPORT int generate_pq_pivots_simplified(const float *train_data, size_t num_train, size_t dim, size_t num_pq_chunks, std::vector &pivot_data_vector); @@ -76,8 +76,8 @@ int generate_pq_data_from_pivots(const std::string &data_file, unsigned num_cent const std::string &pq_pivots_path, const std::string &pq_compressed_vectors_path, bool use_opq = false); -DISKANN_DLLEXPORT int generate_pq_data_from_pivots_simplified(const float *data, const size_t num, - const float *pivot_data, const size_t pivots_num, +DISKANN_DLLEXPORT int generate_pq_data_from_pivots_simplified(const float *data, const size_t num, + const float *pivot_data, const size_t pivots_num, const size_t dim, const size_t num_pq_chunks, std::vector &pq);