Merge branch 'fix_ann_bench_latency' of github.com:tfeher/raft into f…

…ix_ann_bench_latency
rapidsai · Nov 3, 2023 · ab049c3 · ab049c3
2 parents b19fb44 + d903245
commit ab049c3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -38,7 +38,7 @@ fi
 
 if [[ ${package_name} == "raft-dask" ]]; then
     sed -r -i "s/pylibraft==(.*)\"/pylibraft${PACKAGE_CUDA_SUFFIX}==\1${alpha_spec}\"/g" ${pyproject_file}
-    sed -i "s/ucx-py/ucx-py${PACKAGE_CUDA_SUFFIX}/g" python/raft-dask/pyproject.toml
+    sed -r -i "s/ucx-py==(.*)\"/ucx-py${PACKAGE_CUDA_SUFFIX}==\1${alpha_spec}\"/g" ${pyproject_file}
 else
     sed -r -i "s/rmm(.*)\"/rmm${PACKAGE_CUDA_SUFFIX}\1${alpha_spec}\"/g" ${pyproject_file}
 fi

diff --git a/cpp/bench/ann/src/common/benchmark.hpp b/cpp/bench/ann/src/common/benchmark.hpp
@@ -295,7 +295,7 @@ void bench_search(::benchmark::State& state,
       queries_processed += n_queries;
     }
   }
-  2 auto end    = std::chrono::high_resolution_clock::now();
+  auto end      = std::chrono::high_resolution_clock::now();
   auto duration = std::chrono::duration_cast<std::chrono::duration<double>>(end - start).count();
   if (state.thread_index() == 0) { state.counters.insert({{"end_to_end", duration}}); }
   state.counters.insert(