gather.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2021-2023, NVIDIA CORPORATION.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 #pragma once
17 
18 #include <cudf/column/column.hpp>
20 #include <cudf/copying.hpp>
22 
24 
25 namespace cudf {
26 namespace lists {
72 std::unique_ptr<column> segmented_gather(
73  lists_column_view const& source_column,
74  lists_column_view const& gather_map_list,
76  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
77  // end of group
79 } // namespace lists
80 } // namespace cudf
per_device_resource.hpp
column.hpp
Class definition for cudf::column.
cudf::lists::segmented_gather
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::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
Segmented gather of the elements within a list element in each row of a list column.
copying.hpp
Column APIs for gather, scatter, split, slice, etc.
cudf::out_of_bounds_policy::NULLIFY
@ NULLIFY
Output values corresponding to out-of-bounds indices are null.
cudf
cuDF interfaces
Definition: aggregation.hpp:34
cudf::lists_column_view
Given a column-view of lists type, an instance of this class provides a wrapper on this compound colu...
Definition: lists_column_view.hpp:39
cudf::out_of_bounds_policy
out_of_bounds_policy
Policy to account for possible out-of-bounds indices.
Definition: copying.hpp:48
rmm::mr::device_memory_resource
column_view.hpp
column view class definitions
lists_column_view.hpp
Class definition for cudf::lists_column_view.