From dbcd05da76d1e0f685b958d085ee9d4bb9efcf43 Mon Sep 17 00:00:00 2001 From: Divye Gala Date: Thu, 19 Oct 2023 20:38:13 -0400 Subject: [PATCH] Increase iterations in NN Descent tests to avoid flakiness (#1915) Authors: - Divye Gala (https://github.com/divyegala) Approvers: - Corey J. Nolet (https://github.com/cjnolet) URL: https://github.com/rapidsai/raft/pull/1915 --- cpp/test/neighbors/ann_cagra.cuh | 2 ++ cpp/test/neighbors/ann_nn_descent.cuh | 1 + 2 files changed, 3 insertions(+) diff --git a/cpp/test/neighbors/ann_cagra.cuh b/cpp/test/neighbors/ann_cagra.cuh index 1f4531071f..f8e82acf83 100644 --- a/cpp/test/neighbors/ann_cagra.cuh +++ b/cpp/test/neighbors/ann_cagra.cuh @@ -457,6 +457,7 @@ class AnnCagraFilterTest : public ::testing::TestWithParam { cagra::index_params index_params; index_params.metric = ps.metric; // Note: currently ony the cagra::index_params metric is // not used for knn_graph building. + index_params.nn_descent_niter = 50; cagra::search_params search_params; search_params.algo = ps.algo; search_params.max_queries = ps.max_queries; @@ -571,6 +572,7 @@ class AnnCagraFilterTest : public ::testing::TestWithParam { cagra::index_params index_params; index_params.metric = ps.metric; // Note: currently ony the cagra::index_params metric is // not used for knn_graph building. + index_params.nn_descent_niter = 50; cagra::search_params search_params; search_params.algo = ps.algo; search_params.max_queries = ps.max_queries; diff --git a/cpp/test/neighbors/ann_nn_descent.cuh b/cpp/test/neighbors/ann_nn_descent.cuh index d62b863437..0590fe52e8 100644 --- a/cpp/test/neighbors/ann_nn_descent.cuh +++ b/cpp/test/neighbors/ann_nn_descent.cuh @@ -90,6 +90,7 @@ class AnnNNDescentTest : public ::testing::TestWithParam { index_params.metric = ps.metric; index_params.graph_degree = ps.graph_degree; index_params.intermediate_graph_degree = 2 * ps.graph_degree; + index_params.max_iterations = 50; auto database_view = raft::make_device_matrix_view( (const DataT*)database.data(), ps.n_rows, ps.dim);