Biased RW implementation and test

cpp/include/cugraph/algorithms.hpp

@@ -1583,7 +1583,7 @@ uniform_random_walks(raft::handle_t const& handle,
                      std::optional<edge_property_view_t<edge_t, weight_t const*>> edge_weight_view,
                      raft::device_span<vertex_t const> start_vertices,
                      size_t max_length,
-                     uint64_t seed = std::numeric_limits<uint64_t>::max());
+                     raft::random::RngState& rng_state);
 
 /**
  * @brief returns biased random walks from starting sources, where each path is of given
@@ -1627,7 +1627,7 @@ biased_random_walks(raft::handle_t const& handle,
                     edge_property_view_t<edge_t, weight_t const*> edge_weight_view,
                     raft::device_span<vertex_t const> start_vertices,
                     size_t max_length,
-                    uint64_t seed = std::numeric_limits<uint64_t>::max());
+                    raft::random::RngState& rng_state);
 
 /**
  * @brief returns biased random walks with node2vec biases from starting sources,
@@ -1676,7 +1676,7 @@ node2vec_random_walks(raft::handle_t const& handle,
                       size_t max_length,
                       weight_t p,
                       weight_t q,
-                      uint64_t seed = std::numeric_limits<uint64_t>::max());
+                      raft::random::RngState& rng_state);
 
 #ifndef NO_CUGRAPH_OPS
 /**

cpp/src/c_api/random_walks.cpp

@@ -17,6 +17,7 @@
 #include "c_api/abstract_functor.hpp"
 #include "c_api/graph.hpp"
 #include "c_api/resource_handle.hpp"
+#include "c_api/random.hpp"
 #include "c_api/utils.hpp"
 
 #include <cugraph_c/algorithms.h>
@@ -156,7 +157,7 @@ struct uniform_random_walks_functor : public cugraph::c_api::abstract_functor {
   cugraph::c_api::cugraph_graph_t* graph_{nullptr};
   cugraph::c_api::cugraph_type_erased_device_array_view_t const* start_vertices_{nullptr};
   size_t max_length_{0};
-  size_t seed_{0};
+  cugraph::c_api::cugraph_rng_state_t* rng_state_{nullptr};
   cugraph::c_api::cugraph_random_walk_result_t* result_{nullptr};
 
   uniform_random_walks_functor(cugraph_resource_handle_t const* handle,
@@ -228,7 +229,7 @@ struct uniform_random_walks_functor : public cugraph::c_api::abstract_functor {
         (edge_weights != nullptr) ? std::make_optional(edge_weights->view()) : std::nullopt,
         raft::device_span<vertex_t const>{start_vertices.data(), start_vertices.size()},
         max_length_,
-        seed_);
+        rng_state_->rng_state_);
 
       //
       // Need to unrenumber the vertices in the resulting paths
@@ -259,7 +260,7 @@ struct biased_random_walks_functor : public cugraph::c_api::abstract_functor {
   cugraph::c_api::cugraph_type_erased_device_array_view_t const* start_vertices_{nullptr};
   size_t max_length_{0};
   cugraph::c_api::cugraph_random_walk_result_t* result_{nullptr};
-  uint64_t seed_{0};
+  cugraph::c_api::cugraph_rng_state_t* rng_state_{nullptr};
 
   biased_random_walks_functor(cugraph_resource_handle_t const* handle,
                               cugraph_graph_t* graph,
@@ -332,7 +333,7 @@ struct biased_random_walks_functor : public cugraph::c_api::abstract_functor {
         edge_weights->view(),
         raft::device_span<vertex_t const>{start_vertices.data(), start_vertices.size()},
         max_length_,
-        seed_);
+        rng_state_->rng_state_);
 
       //
       // Need to unrenumber the vertices in the resulting paths
@@ -359,7 +360,7 @@ struct node2vec_random_walks_functor : public cugraph::c_api::abstract_functor {
   size_t max_length_{0};
   double p_{0};
   double q_{0};
-  uint64_t seed_{0};
+  cugraph::c_api::cugraph_rng_state_t* rng_state_{nullptr};
   cugraph::c_api::cugraph_random_walk_result_t* result_{nullptr};
 
   node2vec_random_walks_functor(cugraph_resource_handle_t const* handle,
@@ -439,7 +440,7 @@ struct node2vec_random_walks_functor : public cugraph::c_api::abstract_functor {
         max_length_,
         static_cast<weight_t>(p_),
         static_cast<weight_t>(q_),
-        seed_);
+        rng_state_->rng_state_);
 
       // FIXME:  Need to fix invalid_vtx issue here.  We can't unrenumber max_vertex_id+1
       // properly...

cpp/src/prims/per_v_random_select_transform_outgoing_e.cuh

@@ -353,7 +353,7 @@ per_v_random_select_transform_e(raft::handle_t const& handle,
       uniform_sample_and_compute_local_nbr_indices(
         handle,
         graph_view,
-        (minor_comm_size > 1) ? get_dataframe_buffer_begin(*aggregate_local_frontier)
+        (minor_comm_size > 1) ? get_dataframe_buffer_cbegin(*aggregate_local_frontier)
                               : frontier.begin(),
         local_frontier_displacements,
         local_frontier_sizes,
@@ -365,7 +365,7 @@ per_v_random_select_transform_e(raft::handle_t const& handle,
       biased_sample_and_compute_local_nbr_indices(
         handle,
         graph_view,
-        (minor_comm_size > 1) ? get_dataframe_buffer_begin(*aggregate_local_frontier)
+        (minor_comm_size > 1) ? get_dataframe_buffer_cbegin(*aggregate_local_frontier)
                               : frontier.begin(),
         edge_bias_src_value_input,
         edge_bias_dst_value_input,
@@ -394,7 +394,7 @@ per_v_random_select_transform_e(raft::handle_t const& handle,
         graph_view.local_edge_partition_view(i));
 
     auto edge_partition_frontier_key_first =
-      ((minor_comm_size > 1) ? get_dataframe_buffer_begin(*aggregate_local_frontier)
+      ((minor_comm_size > 1) ? get_dataframe_buffer_cbegin(*aggregate_local_frontier)
                              : frontier.begin()) +
       local_frontier_displacements[i];
     auto edge_partition_sample_local_nbr_index_first =