diff --git a/python/cuvs_bench/cuvs_bench/config/algos/cuvs_cagra.yaml b/python/cuvs_bench/cuvs_bench/config/algos/cuvs_cagra.yaml index 240d2ac8ee..76b819e9d4 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/cuvs_cagra.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/cuvs_cagra.yaml @@ -5,17 +5,18 @@ constraints: groups: base: build: - graph_degree: [32, 64, 96, 128] - intermediate_graph_degree: [32, 64, 96, 128] - graph_build_algo: ["NN_DESCENT"] + graph_degree: [32, 64, 96] + intermediate_graph_degree: [128] + graph_build_algo: ["IVF_PQ"] search: itopk: [32, 64, 128, 256, 512] - search_width: [1, 2, 4, 8, 16, 32, 64] + max_iterations: [0, 16] + search_width: [1, 2, 4, 8] test: build: graph_degree: [32] intermediate_graph_degree: [32] - graph_build_algo: ["NN_DESCENT"] + graph_build_algo: ["IVF_PQ"] search: itopk: [32] search_width: [1, 2] diff --git a/python/cuvs_bench/cuvs_bench/config/algos/cuvs_mg_cagra.yaml b/python/cuvs_bench/cuvs_bench/config/algos/cuvs_mg_cagra.yaml index 195755b321..85e0191cc4 100644 --- a/python/cuvs_bench/cuvs_bench/config/algos/cuvs_mg_cagra.yaml +++ b/python/cuvs_bench/cuvs_bench/config/algos/cuvs_mg_cagra.yaml @@ -5,17 +5,18 @@ constraints: groups: base: build: - graph_degree: [32, 64, 96, 128] - intermediate_graph_degree: [32, 64, 96, 128] - graph_build_algo: ["NN_DESCENT"] + graph_degree: [32, 64, 96] + intermediate_graph_degree: [128] + graph_build_algo: ["IVF_PQ"] search: itopk: [32, 64, 128, 256, 512] - search_width: [1, 2, 4, 8, 16, 32, 64] + max_iterations: [0, 16] + search_width: [1, 2, 4, 8] test: build: graph_degree: [32] intermediate_graph_degree: [32] - graph_build_algo: ["NN_DESCENT"] + graph_build_algo: ["IVF_PQ"] search: itopk: [32] search_width: [1, 2]