transform_gather_e.cuh#

namespace cugraph

Functions

template<typename GraphViewType, typename EdgeBucketType, typename EdgeSrcValueInputWrapper, typename EdgeDstValueInputWrapper, typename EdgeValueInputWrapper, typename EdgeOp, typename GatheredValueOutputIterator>
void transform_gather_e(
raft::handle_t const &handle,
GraphViewType const &graph_view,
EdgeBucketType const &edge_list,
EdgeSrcValueInputWrapper edge_src_value_input,
EdgeDstValueInputWrapper edge_dst_value_input,
EdgeValueInputWrapper edge_value_input,
EdgeOp e_op,
GatheredValueOutputIterator gathered_value_output_first,
bool do_expensive_check = false
)

Iterate over the edges in the input edge list and gather the edge operator outputs.

This function is inspired by thrust::transform() and thrust::gather().

Template Parameters:

  • GraphViewType – Type of the passed non-owning graph object.

  • EdgeBucketType – Type of the edge bucket class which stores the edge list.

  • EdgeSrcValueInputWrapper – Type of the wrapper for input edge source property values.

  • EdgeDstValueInputWrapper – Type of the wrapper for input edge destination property values.

  • EdgeValueInputWrapper – Type of the wrapper for input edge property values.

  • EdgeOp – Type of the quinary edge operator.

  • GatheredValueOutputIterator – Type of the iterator for gathered output values.

Parameters:

  • handle – RAFT handle object to encapsulate resources (e.g. CUDA stream, communicator, and handles to various CUDA libraries) to run graph algorithms.

  • graph_view – Non-owning graph object.

  • edge_list – EdgeBucketType class object storing the edge list to update edge property values.

  • edge_src_value_input – Wrapper used to access source input property values (for the edge sources assigned to this process in multi-GPU). Use either cugraph::edge_src_property_t::view() (if e_op needs to access source property values) or cugraph::edge_src_dummy_property_t::view() (if e_op does not access source property values). Use update_edge_src_property to fill the wrapper.

  • edge_dst_value_input – Wrapper used to access destination input property values (for the edge destinations assigned to this process in multi-GPU). Use either cugraph::edge_dst_property_t::view() (if e_op needs to access destination property values) or cugraph::edge_dst_dummy_property_t::view() (if e_op does not access destination property values). Use update_edge_dst_property to fill the wrapper.

  • edge_value_input – Wrapper used to access edge input property values (for the edges assigned to this process in multi-GPU). Use either cugraph::edge_property_t::view() (if e_op needs to access edge property values) or cugraph::edge_dummy_property_t::view() (if e_op does not access edge property values).

  • e_op – Quinary operator takes edge source, edge destination, property values for the source, destination, and edge and returns a value to be reduced.

  • gathered_value_output_first – Iterator pointing the beginning of the gathered output buffer. gathered_value_output_last (exclusive) is deduced as gathered_value_output_first + edge_list.size().

  • do_expensive_check – A flag to run expensive checks for input arguments (if set to true).

namespace detail
template<typename GraphViewType, typename EdgePartitionSrcValueInputWrapper, typename EdgePartitionDstValueInputWrapper, typename EdgePartitionEdgeValueInputWrapper, typename EdgeOp>
struct return_e_value_t

Public Types

using e_op_result_t = typename detail::edge_op_result_type<GraphViewType, typename GraphViewType::vertex_type, EdgePartitionSrcValueInputWrapper, EdgePartitionDstValueInputWrapper, EdgePartitionEdgeValueInputWrapper, EdgeOp>::type#

Public Functions

inline __device__ e_op_result_t operator() (cuda::std::tuple< typename GraphViewType::vertex_type, typename GraphViewType::vertex_type, typename GraphViewType::edge_type > edge) const

Public Members

edge_partition_device_view_t<typename GraphViewType::vertex_type, typename GraphViewType::edge_type, GraphViewType::is_multi_gpu> edge_partition = {}#
EdgePartitionSrcValueInputWrapper edge_partition_src_value_input = {}#
EdgePartitionDstValueInputWrapper edge_partition_dst_value_input = {}#
EdgePartitionEdgeValueInputWrapper edge_partition_e_value_input = {}#
EdgeOp e_op = {}#