Temporal sampling implementation

cpp/CMakeLists.txt

@@ -153,10 +153,35 @@ endif()
 #       which should give us a better parallel schedule.
 
 set(CUGRAPH_SOURCES
+    src/sampling/temporal_sampling_sg_v32_e32.cu
+    src/sampling/temporal_sampling_sg_v64_e64.cu
+    src/sampling/temporal_sampling_mg_v32_e32.cu
+    src/sampling/temporal_sampling_mg_v64_e64.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v32_w32_t32_sg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v32_w32_t32_mg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v32_w32_t64_sg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v32_w32_t64_mg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v32_w64_t32_sg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v32_w64_t32_mg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v32_w64_t64_sg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v32_w64_t64_mg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v64_w32_t32_sg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v64_w32_t32_mg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v64_w32_t64_sg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v64_w32_t64_mg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v64_w64_t32_sg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v64_w64_t32_mg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v64_w64_t64_sg.cu
+    src/sampling/detail/temporal_label_gather_one_hop_edgelist_v64_w64_t64_mg.cu
+    src/sampling/detail/update_temporal_edge_mask_sg_v32_e32.cu
+    src/sampling/detail/update_temporal_edge_mask_sg_v64_e64.cu
+    src/sampling/detail/update_temporal_edge_mask_mg_v32_e32.cu
+    src/sampling/detail/update_temporal_edge_mask_mg_v64_e64.cu
     src/utilities/shuffle_vertices_mg_v32_fp.cu
     src/utilities/shuffle_vertices_mg_v32_integral.cu
     src/utilities/shuffle_vertices_mg_v64_fp.cu
     src/utilities/shuffle_vertices_mg_v64_integral.cu
+    src/utilities/validation_checks.cu
     src/detail/permute_range_v32.cu
     src/detail/permute_range_v64.cu
     src/utilities/shuffle_vertex_pairs_mg_v32_e32.cu
@@ -263,10 +288,64 @@ set(CUGRAPH_SOURCES
     src/sampling/detail/check_edge_bias_values_sg_v32_e32.cu
     src/sampling/detail/check_edge_bias_values_mg_v64_e64.cu
     src/sampling/detail/check_edge_bias_values_mg_v32_e32.cu
+    src/sampling/detail/temporal_partition_vertices_v64.cu
+    src/sampling/detail/temporal_partition_vertices_v32.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v32_w32_t32_sg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v32_w32_t32_mg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v32_w32_t64_sg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v32_w32_t64_mg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v32_w64_t32_sg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v32_w64_t32_mg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v32_w64_t64_sg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v32_w64_t64_mg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v64_w32_t32_sg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v64_w32_t32_mg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v64_w32_t64_sg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v64_w32_t64_mg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v64_w64_t32_sg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v64_w64_t32_mg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v64_w64_t64_sg.cu
+    src/sampling/detail/temporal_simple_gather_one_hop_edgelist_v64_w64_t64_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w32_t32_tag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w32_t32_tag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w32_t64_tag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w32_t64_tag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w64_t32_tag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w64_t32_tag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w64_t64_tag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w64_t64_tag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w32_t32_tag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w32_t32_tag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w32_t64_tag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w32_t64_tag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w64_t32_tag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w64_t32_tag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w64_t64_tag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w64_t64_tag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w32_t32_notag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w32_t32_notag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w32_t64_notag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w32_t64_notag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w64_t32_notag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w64_t32_notag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w64_t64_notag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v32_w64_t64_notag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w32_t32_notag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w32_t32_notag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w32_t64_notag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w32_t64_notag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w64_t32_notag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w64_t32_notag_sg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w64_t64_notag_mg.cu
+    src/sampling/detail/simple_gather_one_hop_edgelist_v64_w64_t64_notag_sg.cu
     src/sampling/detail/sample_edges_sg_v64_e64.cu
     src/sampling/detail/sample_edges_sg_v32_e32.cu
     src/sampling/detail/sample_edges_mg_v64_e64.cu
     src/sampling/detail/sample_edges_mg_v32_e32.cu
+    src/sampling/detail/temporal_sample_edges_sg_v64_e64.cu
+    src/sampling/detail/temporal_sample_edges_sg_v32_e32.cu
+    src/sampling/detail/temporal_sample_edges_mg_v64_e64.cu
+    src/sampling/detail/temporal_sample_edges_mg_v32_e32.cu
     src/sampling/detail/shuffle_and_organize_output.cu
     src/sampling/neighbor_sampling_mg_v32_e32.cu
     src/sampling/neighbor_sampling_mg_v64_e64.cu

cpp/include/cugraph/arithmetic_variant_types.hpp

@@ -45,9 +45,11 @@ using const_arithmetic_device_span_t = std::variant<std::monostate,
                                                     raft::device_span<int64_t const>,
                                                     raft::device_span<size_t const>>;
 
-template <typename edge_t>
+template <typename vertex_t, typename edge_t>
 using edge_arithmetic_property_view_t =
   std::variant<std::monostate,
+               cugraph::edge_dummy_property_view_t,
+               cugraph::edge_multi_index_property_view_t<edge_t, vertex_t>,
                cugraph::edge_property_view_t<edge_t, float const*>,
                cugraph::edge_property_view_t<edge_t, double const*>,
                cugraph::edge_property_view_t<edge_t, int32_t const*>,
@@ -158,31 +160,18 @@ auto variant_type_dispatch(const_arithmetic_device_span_t& property, func_t func
   }
 }
 
-template <typename edge_t, typename func_t>
-auto variant_type_dispatch(edge_arithmetic_property_view_t<edge_t>& property, func_t func)
+template <typename vertex_t, typename edge_t, typename func_t>
+auto variant_type_dispatch(edge_arithmetic_property_view_t<edge_t, vertex_t>& property, func_t func)
 {
-  if (std::holds_alternative<cugraph::edge_property_view_t<edge_t, float const*>>(property)) {
-    auto& prop = std::get<cugraph::edge_property_view_t<edge_t, float const*>>(property);
-    return func(prop);
-  } else if (std::holds_alternative<cugraph::edge_property_view_t<edge_t, double const*>>(
-               property)) {
-    auto& prop = std::get<cugraph::edge_property_view_t<edge_t, double const*>>(property);
-    return func(prop);
-  } else if (std::holds_alternative<cugraph::edge_property_view_t<edge_t, int32_t const*>>(
-               property)) {
-    auto& prop = std::get<cugraph::edge_property_view_t<edge_t, int32_t const*>>(property);
-    return func(prop);
-  } else if (std::holds_alternative<cugraph::edge_property_view_t<edge_t, int64_t const*>>(
-               property)) {
-    auto& prop = std::get<cugraph::edge_property_view_t<edge_t, int64_t const*>>(property);
-    return func(prop);
-  } else {
-    CUGRAPH_EXPECTS(
-      (std::holds_alternative<cugraph::edge_property_view_t<edge_t, size_t const*>>(property)),
-      "unsupported variant type -- shouldn't happen");
-
-    auto& prop = std::get<cugraph::edge_property_view_t<edge_t, size_t const*>>(property);
-    return func(prop);
+  switch (property.index()) {
+    case 1: return func(std::get<1>(property));
+    case 2: return func(std::get<2>(property));
+    case 3: return func(std::get<3>(property));
+    case 4: return func(std::get<4>(property));
+    case 5: return func(std::get<5>(property));
+    case 6: return func(std::get<6>(property));
+    case 7: return func(std::get<7>(property));
+    default: CUGRAPH_FAIL("Variant not initialized");
   }
 }
 

cpp/include/cugraph/edge_property.hpp

@@ -29,9 +29,19 @@
 #include <thrust/iterator/iterator_traits.h>
 
 #include <optional>
+#include <tuple>
 #include <type_traits>
 
 namespace cugraph {
+namespace detail {
+
+template <typename... Ts, std::size_t... Is>
+auto view_concat_impl(std::tuple<Ts...> const& tuple, std::index_sequence<Is...>)
+{
+  return view_concat(std::get<Is>(tuple)...);
+}
+
+}  // namespace detail
 
 template <typename edge_t,
           typename ValueIterator,
@@ -273,4 +283,10 @@ auto view_concat(edge_property_view_t<edge_t, Iters, Types> const&... views)
     edge_partition_concat_value_firsts, first_view.edge_counts());
 }
 
+template <typename... Ts>
+auto view_concat(std::tuple<Ts...> const& tuple)
+{
+  return detail::view_concat_impl(tuple, std::index_sequence_for<Ts...>{});
+}
+
 }  // namespace cugraph