updates

rapidsai · Sep 10, 2024 · fafc4f9 · fafc4f9
1 parent b838ff8
commit fafc4f9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/cpp/src/neighbors/detail/nn_descent.cuh b/cpp/src/neighbors/detail/nn_descent.cuh
@@ -1349,14 +1349,6 @@ void GNND<Data_t, Index_t>::build(Data_t* data, const Index_t nrow, Index_t* out
   raft::resource::sync_stream(res);
   graph_.sort_lists();
 
-  std::cout << "NN Descent distances: " << std::endl;
-  for (size_t i = 0; i < nrow_; i++) {
-    for (size_t j = 0; j < graph_.node_degree; j++) {
-      std::cout << graph_.h_dists.data_handle()[i * graph_.node_degree + j] << " ";
-    }
-    std::cout << std::endl;
-  }
-
   // Reuse graph_.h_dists as the buffer for shrink the lists in graph
   static_assert(sizeof(decltype(*(graph_.h_dists.data_handle()))) >= sizeof(Index_t));
   Index_t* graph_shrink_buffer = (Index_t*)graph_.h_dists.data_handle();

diff --git a/python/cuvs/cuvs/test/test_cagra.py b/python/cuvs/cuvs/test/test_cagra.py
@@ -124,7 +124,7 @@ def run_cagra_build_search_test(
 @pytest.mark.parametrize("build_algo", ["ivf_pq", "nn_descent"])
 @pytest.mark.parametrize("metric", ["euclidean", "inner_product"])
 def test_cagra_dataset_dtype_host_device(
-    dtype, array_type, inplace, build_algo
+    dtype, array_type, inplace, build_algo, metric
 ):
     # Note that inner_product tests use normalized input which we cannot
     # represent in int8, therefore we test only sqeuclidean metric here.
@@ -133,6 +133,7 @@ def test_cagra_dataset_dtype_host_device(
         inplace=inplace,
         array_type=array_type,
         build_algo=build_algo,
+        metric=metric,
     )