lists/stream_compaction.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2022, 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 
21 #include <rmm/mr/device/device_memory_resource.hpp>
23 
24 namespace cudf::lists {
25 
60 std::unique_ptr<column> apply_boolean_mask(
61  lists_column_view const& input,
62  lists_column_view const& boolean_mask,
63  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
64 
84 std::unique_ptr<column> distinct(
85  lists_column_view const& input,
86  null_equality nulls_equal = null_equality::EQUAL,
87  nan_equality nans_equal = nan_equality::ALL_EQUAL,
88  rmm::mr::device_memory_resource* mr = rmm::mr::get_current_device_resource());
89  // end of group
91 
92 } // namespace cudf::lists
cudf::lists::distinct
std::unique_ptr< column > distinct(lists_column_view const &input, null_equality nulls_equal=null_equality::EQUAL, nan_equality nans_equal=nan_equality::ALL_EQUAL, rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
Create a new list column without duplicate elements in each list.
per_device_resource.hpp
column.hpp
Class definition for cudf::column.
cudf::nan_equality
nan_equality
Enum to consider different elements (of floating point types) holding NaN value as equal or unequal.
Definition: types.hpp:128
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::null_equality
null_equality
Enum to consider two nulls as equal or unequal.
Definition: types.hpp:136
rmm::mr::device_memory_resource
cudf::lists::apply_boolean_mask
std::unique_ptr< column > apply_boolean_mask(lists_column_view const &input, lists_column_view const &boolean_mask, rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
Filters elements in each row of input LIST column using boolean_mask LIST of booleans as a mask.
lists_column_view.hpp
Class definition for cudf::lists_column_view.
cudf::lists
Lists column APIs.
Definition: lists/combine.hpp:26