gather.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2021-2024, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 #pragma once
6 
7 #include <cudf/column/column.hpp>
9 #include <cudf/copying.hpp>
11 #include <cudf/utilities/export.hpp>
13 
14 namespace CUDF_EXPORT cudf {
15 namespace lists {
62 std::unique_ptr<column> segmented_gather(
63  lists_column_view const& source_column,
64  lists_column_view const& gather_map_list,
65  out_of_bounds_policy bounds_policy = out_of_bounds_policy::DONT_CHECK,
68  // end of group
70 } // namespace lists
71 } // namespace CUDF_EXPORT cudf
Given a column-view of lists type, an instance of this class provides a wrapper on this compound colu...
Class definition for cudf::column.
column view class definitions
Column APIs for gather, scatter, split, slice, etc.
out_of_bounds_policy
Policy to account for possible out-of-bounds indices.
Definition: copying.hpp:37
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
std::unique_ptr< column > segmented_gather(lists_column_view const &source_column, lists_column_view const &gather_map_list, out_of_bounds_policy bounds_policy=out_of_bounds_policy::DONT_CHECK, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Segmented gather of the elements within a list element in each row of a list column.
rmm::device_async_resource_ref get_current_device_resource_ref()
Get the current device memory resource reference.
detail::cccl_async_resource_ref< cuda::mr::resource_ref< cuda::mr::device_accessible > > device_async_resource_ref
Class definition for cudf::lists_column_view.
cuDF interfaces
Definition: host_udf.hpp:26