From cbea2099a9ff8f581180dc9347bdf544e11d8b5d Mon Sep 17 00:00:00 2001
From: Mickael Ide <mide@nvidia.com>
Date: Mon, 2 Oct 2023 11:30:53 +0200
Subject: [PATCH] Use k=255 for cagra bench

---
 cpp/bench/prims/neighbors/cagra_bench.cuh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cpp/bench/prims/neighbors/cagra_bench.cuh b/cpp/bench/prims/neighbors/cagra_bench.cuh
index 6cd8ef0e0b..63f6c14686 100644
--- a/cpp/bench/prims/neighbors/cagra_bench.cuh
+++ b/cpp/bench/prims/neighbors/cagra_bench.cuh
@@ -183,9 +183,9 @@ inline const std::vector<params> generate_inputs()
     {2000000ull, 10000000ull},                 // n_samples
     {128},                                     // dataset dim
     {1, 10, 10000},                            // n_queries
-    {32},                                      // k
+    {255},                                     // k
     {64},                                      // knn graph degree
-    {64},                                      // itopk_size
+    {300},                                     // itopk_size
     {256},                                     // block_size
     {2},                                       // search_width
     {0},                                       // max_iterations