From a68d7a730f282bb8df74b1821248085bd173e2ab Mon Sep 17 00:00:00 2001 From: Tarang Jain Date: Thu, 9 Nov 2023 13:08:46 -0800 Subject: [PATCH] Add test --- cpp/test/neighbors/ann_ivf_pq.cuh | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/cpp/test/neighbors/ann_ivf_pq.cuh b/cpp/test/neighbors/ann_ivf_pq.cuh index b4db5c972d..241ea5d7b6 100644 --- 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; }