Minor C++17 cleanup of groupby.cu: structured bindings, more concise lambda, etc

cpp/src/groupby/hash/groupby.cu

@@ -537,27 +537,25 @@ void compute_single_pass_aggs(table_view const& keys,
  * `map`.
  */
 template <typename Map>
-std::pair<rmm::device_uvector<size_type>, size_type> extract_populated_keys(
-  Map map, size_type num_keys, rmm::cuda_stream_view stream)
+rmm::device_uvector<size_type> extract_populated_keys(Map map,
+                                                      size_type num_keys,
+                                                      rmm::cuda_stream_view stream)
 {
   rmm::device_uvector<size_type> populated_keys(num_keys, stream);
 
-  auto get_key = [] __device__(auto const& element) {
-    size_type key, value;
-    thrust::tie(key, value) = element;
-    return key;
-  };
+  auto get_key    = [] __device__(auto const& element) { return element.first; };  // first = key
+  auto get_key_it = thrust::make_transform_iterator(map.data(), get_key);
+  auto key_used   = [unused = map.get_unused_key()] __device__(auto key) { return key != unused; };
 
-  auto end_it = thrust::copy_if(
-    rmm::exec_policy(stream),
-    thrust::make_transform_iterator(map.data(), get_key),
-    thrust::make_transform_iterator(map.data() + map.capacity(), get_key),
-    populated_keys.begin(),
-    [unused_key = map.get_unused_key()] __device__(size_type key) { return key != unused_key; });
+  auto end_it = thrust::copy_if(rmm::exec_policy(stream),
+                                get_key_it,
+                                get_key_it + map.capacity(),
+                                populated_keys.begin(),
+                                key_used);
 
-  size_type map_size = end_it - populated_keys.begin();
+  populated_keys.resize(std::distance(populated_keys.begin(), end_it), stream);
 
-  return std::make_pair(std::move(populated_keys), map_size);
+  return populated_keys;
 }
 
 /**
@@ -594,8 +592,8 @@ std::unique_ptr<table> groupby_null_templated(table_view const& keys,
                                               rmm::cuda_stream_view stream,
                                               rmm::mr::device_memory_resource* mr)
 {
-  auto d_keys = table_device_view::create(keys, stream);
-  auto map    = create_hash_map<keys_have_nulls>(*d_keys, include_null_keys, stream);
+  auto d_keys_ptr = table_device_view::create(keys, stream);
+  auto map        = create_hash_map<keys_have_nulls>(*d_keys_ptr, include_null_keys, stream);
 
   // Cache of sparse results where the location of aggregate value in each
   // column is indexed by the hash map
@@ -607,17 +605,15 @@ std::unique_ptr<table> groupby_null_templated(table_view const& keys,
 
   // Extract the populated indices from the hash map and create a gather map.
   // Gathering using this map from sparse results will give dense results.
-  auto map_and_size = extract_populated_keys(*map, keys.num_rows(), stream);
-  rmm::device_uvector<size_type> gather_map{std::move(map_and_size.first)};
-  size_type const map_size = map_and_size.second;
+  auto gather_map = extract_populated_keys(*map, keys.num_rows(), stream);
 
   // Compact all results from sparse_results and insert into cache
   sparse_to_dense_results(keys,
                           requests,
                           &sparse_results,
                           cache,
                           gather_map,
-                          map_size,
+                          gather_map.size(),
                           *map,
                           keys_have_nulls,
                           include_null_keys,
@@ -626,7 +622,7 @@ std::unique_ptr<table> groupby_null_templated(table_view const& keys,
 
   return cudf::detail::gather(keys,
                               gather_map.begin(),
-                              gather_map.begin() + map_size,
+                              gather_map.begin() + gather_map.size(),
                               out_of_bounds_policy::DONT_CHECK,
                               stream,
                               mr);