Merge branch 'branch-23.12' into bug-2312-timer_start

rapidsai · Nov 14, 2023 · 88255e8 · 88255e8
2 parents 7d4cbf8 + 9f84648
commit 88255e8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cpp/include/raft/neighbors/detail/cagra/graph_core.cuh b/cpp/include/raft/neighbors/detail/cagra/graph_core.cuh
@@ -528,8 +528,9 @@ void optimize(raft::resources const& res,
     constexpr int _omp_chunk = 1024;
 #pragma omp parallel for schedule(dynamic, _omp_chunk)
     for (uint64_t j = 0; j < graph_size; j++) {
-      for (uint64_t _k = 0; _k < rev_graph_count.data_handle()[j]; _k++) {
-        uint64_t k = rev_graph_count.data_handle()[j] - 1 - _k;
+      uint64_t k = std::min(rev_graph_count.data_handle()[j], output_graph_degree);
+      while (k) {
+        k--;
         uint64_t i = rev_graph.data_handle()[k + (output_graph_degree * j)];
 
         uint64_t pos =