From fbbb290a5762e265a501324b3bfeb04c1cb5b006 Mon Sep 17 00:00:00 2001
From: Tarang Jain <jaintarang2015@gmail.com>
Date: Thu, 21 Aug 2025 15:56:25 -0700
Subject: [PATCH] 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
---
 faiss/gpu/impl/CuvsCagra.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/faiss/gpu/impl/CuvsCagra.cu b/faiss/gpu/impl/CuvsCagra.cu
index 482e4d672..9ac4e1c5a 100644
--- a/faiss/gpu/impl/CuvsCagra.cu
+++ b/faiss/gpu/impl/CuvsCagra.cu
@@ -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;
     }