Add test

rapidsai · Nov 9, 2023 · a68d7a7 · a68d7a7
1 parent d67fe8d
commit a68d7a7
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/cpp/test/neighbors/ann_ivf_pq.cuh b/cpp/test/neighbors/ann_ivf_pq.cuh
@@ -1053,6 +1053,19 @@ inline auto special_cases() -> test_cases_t
     x.search_params.n_probes     = 69;
   });
 
+  ADD_CASE({
+    x.num_db_vecs                = 4335;
+    x.dim                        = 4;
+    x.num_queries                = 100000;
+    x.k                          = 12;
+    x.index_params.metric        = distance::DistanceType::L2Expanded;
+    x.index_params.codebook_kind = ivf_pq::codebook_gen::PER_CLUSTER;
+    x.index_params.pq_dim        = 2;
+    x.index_params.pq_bits       = 8;
+    x.index_params.n_lists       = 69;
+    x.search_params.n_probes     = 69;
+  });
+
   return xs;
 }