Set NN Descent Metric From CAGRA Params (#4540)

Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/4540

Reviewed By: mnorris11

Differential Revision: D80734549

Pulled By: junjieqi

fbshipit-source-id: 0b19e3e6ed05731e7076fb230f45503b36a3faac
This commit is contained in:
Tarang Jain
2025-08-21 15:56:25 -07:00
committed by Facebook GitHub Bot
parent 5c61ed8c94
commit fbbb290a57
+1
View File
@@ -193,6 +193,7 @@ void CuvsCagra<data_t>::train(idx_t n, const data_t* x) {
cuvs::neighbors::cagra::graph_build_params::nn_descent_params
graph_build_params(index_params_.intermediate_graph_degree);
graph_build_params.max_iterations = nn_descent_niter_;
graph_build_params.metric = metricFaissToCuvs(metric_, false);
index_params_.graph_build_params = graph_build_params;
}