Fix query_set initialization

rapidsai · Nov 2, 2023 · 777ded7 · 777ded7
1 parent c310095
commit 777ded7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cpp/bench/ann/src/common/benchmark.hpp b/cpp/bench/ann/src/common/benchmark.hpp
@@ -257,6 +257,7 @@ void bench_search(::benchmark::State& state,
     // benchmark loop, therefore the synchronization here is necessary.
   }
   const auto algo_property = *current_algo_props;
+  query_set                = dataset->query_set(algo_property.query_memory_type);
 
   /**
    * Each thread will manage its own outputs