Skip to content

Commit

Permalink
fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
jnke2016 committed Nov 8, 2024
1 parent b5d0505 commit a759293
Show file tree
Hide file tree
Showing 6 changed files with 6 additions and 14 deletions.
8 changes: 4 additions & 4 deletions cpp/include/cugraph/sampling_functions.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,7 @@ struct sampling_flags_t {
* edge_type and hop is the number of sampled edges while the size of the offsets vector is the
* number of labels + 1. src is the source vertex, dst is the destination vertex, weight
* (optional) is the edge weight, edge_id (optional) identifies the edge id, edge_type (optional)
* identifies the edge type, hop identifies which hop the edge was encountered in.
* identifies the edge type, hop identifies which hop the edge was encountered in.
* The offsets array (optional) identifies the offset for each label.
*
* If @p label_to_output_comm_rank is specified then the data will be shuffled so that all entries
Expand Down Expand Up @@ -356,7 +356,7 @@ homogeneous_uniform_neighbor_sample(
* edge_type and hop is the number of sampled edges while the size of the offsets vector is the
* number of labels + 1. src is the source vertex, dst is the destination vertex, weight
* (optional) is the edge weight, edge_id (optional) identifies the edge id, edge_type (optional)
* identifies the edge type, hop identifies which hop the edge was encountered in.
* identifies the edge type, hop identifies which hop the edge was encountered in.
* The offsets array (optional) identifies the offset for each label.
*
* If @p label_to_output_comm_rank is specified then the data will be shuffled so that all entries
Expand Down Expand Up @@ -438,7 +438,7 @@ homogeneous_biased_neighbor_sample(
* edge_type and hop is the number of sampled edges while the size of the offsets vector is the
* number of labels + 1. src is the source vertex, dst is the destination vertex, weight
* (optional) is the edge weight, edge_id (optional) identifies the edge id, edge_type (optional)
* identifies the edge type, hop identifies which hop the edge was encountered in.
* identifies the edge type, hop identifies which hop the edge was encountered in.
* The offsets array (optional) identifies the offset for each label.
*
* If @p label_to_output_comm_rank is specified then the data will be shuffled so that all entries
Expand Down Expand Up @@ -516,7 +516,7 @@ heterogeneous_uniform_neighbor_sample(
* edge_type and hop is the number of sampled edges while the size of the offsets vector is the
* number of labels + 1. src is the source vertex, dst is the destination vertex, weight
* (optional) is the edge weight, edge_id (optional) identifies the edge id, edge_type (optional)
* identifies the edge type, hop identifies which hop the edge was encountered in.
* identifies the edge type, hop identifies which hop the edge was encountered in.
* The offsets array (optional) identifies the offset for each label.
*
* If @p label_to_output_comm_rank is specified then the data will be shuffled so that all entries
Expand Down
8 changes: 2 additions & 6 deletions cpp/src/sampling/neighbor_sampling_impl.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -151,9 +151,7 @@ neighbor_sample_impl(raft::handle_t const& handle,
level_result_dst_vectors.reserve(fan_out.size());
if (level_result_weight_vectors) { (*level_result_weight_vectors).reserve(fan_out.size()); }
if (level_result_edge_id_vectors) { (*level_result_edge_id_vectors).reserve(fan_out.size()); }
if (level_result_edge_type_vectors) {
(*level_result_edge_type_vectors).reserve(fan_out.size());
}
if (level_result_edge_type_vectors) { (*level_result_edge_type_vectors).reserve(fan_out.size()); }
if (level_result_label_vectors) { (*level_result_label_vectors).reserve(fan_out.size()); }

rmm::device_uvector<vertex_t> frontier_vertices(0, handle.get_stream());
Expand Down Expand Up @@ -228,9 +226,7 @@ neighbor_sample_impl(raft::handle_t const& handle,
if (edge_types) { (*level_result_edge_type_vectors).push_back(std::move(*edge_types)); }
if (labels) { (*level_result_label_vectors).push_back(std::move(*labels)); }

if (num_edge_types > 1) {
modified_graph_view.clear_edge_mask();
}
if (num_edge_types > 1) { modified_graph_view.clear_edge_mask(); }
}

// FIXME: We should modify vertex_partition_range_lasts to return a raft::host_span
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,6 @@ class Tests_MGHeterogeneous_Biased_Neighbor_Sampling
mg_graph_view,
std::optional<raft::device_span<vertex_t const>>{std::nullopt},
rng_state,
// 20,

std::max(static_cast<size_t>(mg_graph_view.number_of_vertices() * select_probability),
std::min(static_cast<size_t>(mg_graph_view.number_of_vertices()), size_t{1})),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,6 @@ class Tests_MGHeterogeneous_Uniform_Neighbor_Sampling
mg_graph_view,
std::optional<raft::device_span<vertex_t const>>{std::nullopt},
rng_state,
// 20,

std::max(static_cast<size_t>(mg_graph_view.number_of_vertices() * select_probability),
std::min(static_cast<size_t>(mg_graph_view.number_of_vertices()), size_t{1})),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,6 @@ class Tests_MGHomogeneous_Biased_Neighbor_Sampling
mg_graph_view,
std::optional<raft::device_span<vertex_t const>>{std::nullopt},
rng_state,
// 20,

std::max(static_cast<size_t>(mg_graph_view.number_of_vertices() * select_probability),
std::min(static_cast<size_t>(mg_graph_view.number_of_vertices()), size_t{1})),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,6 @@ class Tests_MGHomogeneous_Uniform_Neighbor_Sampling
mg_graph_view,
std::optional<raft::device_span<vertex_t const>>{std::nullopt},
rng_state,
// 20,

std::max(static_cast<size_t>(mg_graph_view.number_of_vertices() * select_probability),
std::min(static_cast<size_t>(mg_graph_view.number_of_vertices()), size_t{1})),
Expand Down

0 comments on commit a759293

Please sign in to comment.