lists/stream_compaction.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2022-2025, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 #pragma once
6 
7 #include <cudf/column/column.hpp>
10 #include <cudf/utilities/export.hpp>
12 
14 
15 namespace CUDF_EXPORT cudf {
16 namespace lists {
17 
53 std::unique_ptr<column> apply_boolean_mask(
54  lists_column_view const& input,
55  lists_column_view const& boolean_mask,
58 
80 std::unique_ptr<column> distinct(
81  lists_column_view const& input,
82  null_equality nulls_equal = null_equality::EQUAL,
83  nan_equality nans_equal = nan_equality::ALL_EQUAL,
84  duplicate_keep_option keep_option = duplicate_keep_option::KEEP_ANY,
87  // end of group
89 
90 } // namespace lists
91 } // 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.
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
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, duplicate_keep_option keep_option=duplicate_keep_option::KEEP_ANY, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create a new list column without duplicate elements in each list.
std::unique_ptr< column > apply_boolean_mask(lists_column_view const &input, lists_column_view const &boolean_mask, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Filters elements in each row of input LIST column using boolean_mask LIST of booleans as a mask.
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
duplicate_keep_option
Choices for drop_duplicates API for retainment of duplicate rows.
null_equality
Enum to consider two nulls as equal or unequal.
Definition: types.hpp:140
nan_equality
Enum to consider different elements (of floating point types) holding NaN value as equal or unequal.
Definition: types.hpp:132
Class definition for cudf::lists_column_view.
cuDF interfaces
Definition: host_udf.hpp:26
Column APIs for filtering rows.