fix logic error

rapidsai · Feb 21, 2024 · 28275b3 · 28275b3
1 parent 0355693
commit 28275b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/community/k_truss_impl.cuh b/cpp/src/community/k_truss_impl.cuh
@@ -807,7 +807,7 @@ void k_truss(raft::handle_t const& handle,
         invalid_edges_buffer, num_edges, handle.get_stream());
 
       // Need to run prefix_sum again to get new ranges because some incoming edges were removed
-      prefix_sum.resize(num_edges + 1, handle.get_stream());
+      prefix_sum.resize(num_invalid_edges + 1, handle.get_stream());
       indices.resize(num_invalid_edges, handle.get_stream());
 
       // FIXME: need to sort 'incoming_vertex_pairs'. No need because a stable partition was