Merge branch 'branch-24.10' into cagra-bf

rapidsai · Sep 26, 2024 · 22f868d · 22f868d
2 parents a8ef6d8 + 0f8b097
commit 22f868d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/cpp/test/CMakeLists.txt b/cpp/test/CMakeLists.txt
@@ -440,7 +440,9 @@ if(BUILD_TESTS)
  neighbors/ann_nn_descent/test_float_uint32_t.cu
  neighbors/ann_nn_descent/test_int8_t_uint32_t.cu
  neighbors/ann_nn_descent/test_uint8_t_uint32_t.cu
- neighbors/ann_nn_descent/test_batch_float_uint32_t.cu
+ # TODO: Investigate why this test is failing Reference issue
+ # https:/rapidsai/raft/issues/2450
+ # neighbors/ann_nn_descent/test_batch_float_uint32_t.cu
  LIB
  EXPLICIT_INSTANTIATE_ONLY
  GPUS

diff --git a/cpp/test/neighbors/ann_nn_descent.cuh b/cpp/test/neighbors/ann_nn_descent.cuh
@@ -318,13 +318,15 @@ const std::vector<AnnNNDescentInputs> inputs = raft::util::itertools::product<An
  {false, true},
  {0.90});
 
-const std::vector<AnnNNDescentBatchInputs> inputsBatch =
- raft::util::itertools::product<AnnNNDescentBatchInputs>(
- {std::make_pair(0.9, 3lu), std::make_pair(0.9, 2lu)}, // min_recall, n_clusters
- {4000, 5000}, // n_rows
- {192, 512}, // dim
- {32, 64}, // graph_degree
- {raft::distance::DistanceType::L2Expanded},
- {false, true});
+// TODO: Investigate why this test is failing
+// Reference issue https:/rapidsai/raft/issues/2450
+// const std::vector<AnnNNDescentBatchInputs> inputsBatch =
+// raft::util::itertools::product<AnnNNDescentBatchInputs>(
+// {std::make_pair(0.9, 3lu), std::make_pair(0.9, 2lu)}, // min_recall, n_clusters
+// {4000, 5000}, // n_rows
+// {192, 512}, // dim
+// {32, 64}, // graph_degree
+// {raft::distance::DistanceType::L2Expanded},
+// {false, true});
 
 } // namespace raft::neighbors::experimental::nn_descent