Fix duplication check

rapidsai · Jan 17, 2025 · daa9807 · daa9807
1 parent bd7de7f
commit daa9807
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/src/neighbors/detail/cagra/graph_core.cuh b/cpp/src/neighbors/detail/cagra/graph_core.cuh
@@ -1365,7 +1365,7 @@ void optimize(
         for (uint32_t pruned_i = 0;
              (pruned_i < output_graph_degree) && (k + l < output_graph_degree);
              pruned_i++) {
-          const auto v = output_graph_ptr[pruned_i];
+          const auto v = output_graph_ptr[output_graph_degree * i + pruned_i];
 
           // duplication check
           bool dup = false;
@@ -1377,7 +1377,7 @@ void optimize(
           }
 
           if (!dup) {
-            my_out_graph[k + l] = output_graph_ptr[(output_graph_degree * i) + pruned_i];
+            my_out_graph[k + l] = v;
             l++;
           }
         }