@@ -502,30 +502,6 @@ inline auto make_pq_params_from_vpq(const cuvs::neighbors::vpq_params& in_params
502502 max_train_points_per_vq_cluster};
503503}
504504
505- inline auto make_pq_params_from_vpq (const cuvs::neighbors::vpq_params& in_params,
506- const uint64_t n_rows)
507- -> cuvs::preprocessing::quantize::pq::params
508- {
509- const uint32_t pq_n_centers = 1 << in_params.pq_bits ;
510- uint32_t max_train_points_per_vq_cluster = in_params.max_train_points_per_vq_cluster ;
511- if (in_params.vq_n_centers > 0 ) {
512- max_train_points_per_vq_cluster =
513- std::min<uint32_t >(max_train_points_per_vq_cluster,
514- n_rows * in_params.vq_kmeans_trainset_fraction / in_params.vq_n_centers );
515- }
516- return cuvs::preprocessing::quantize::pq::params{
517- in_params.pq_bits ,
518- in_params.pq_dim ,
519- true ,
520- true ,
521- in_params.vq_n_centers ,
522- in_params.kmeans_n_iters ,
523- in_params.pq_kmeans_type ,
524- std::min<uint32_t >(in_params.max_train_points_per_pq_code ,
525- n_rows * in_params.pq_kmeans_trainset_fraction / pq_n_centers),
526- max_train_points_per_vq_cluster};
527- }
528-
529505template <typename DatasetT, typename MathT, typename IdxT>
530506auto vpq_build (const raft::resources& res,
531507 const cuvs::neighbors::vpq_params& params,
0 commit comments