lists/stream_compaction.hpp
前往此文件的文档。
1 /*
2  * 版权所有 (c) 2022-2024, NVIDIA CORPORATION.
3  *
4  * 根据 Apache 许可，版本 2.0（“许可”）获得许可；
5  * 除非符合许可，否则您不得使用此文件。
6  * 您可以在以下位置获得许可的副本：
7  *
8  * https://apache.ac.cn/licenses/LICENSE-2.0
9  *
10  * 除非适用法律要求或书面同意，否则
11  * 根据许可分发的软件按“原样”分发，
12  * 不附带任何明示或默示的保证或条件。
13  * 请参阅许可，了解管理权限和
14  * 限制的特定语言。
15  */
16 #pragma once
17 
18 #include <cudf/column/column.hpp>
19 #include <cudf/lists/lists_column_view.hpp>
20 #include <cudf/utilities/export.hpp>
21 #include <cudf/utilities/memory_resource.hpp>
22 
23 #include <rmm/mr/device/device_memory_resource.hpp>
24 
25 namespace CUDF_EXPORT cudf {
26 namespace lists {
27 
63 /// @defgroup lists_filtering Lists filtering operations
64 /// @ingroup lists Lists
65 /// @{
66 /// @brief 使用布尔值列表作为掩码过滤输入 LIST 列的每一行中的元素。
67 ///
68 /// @param input 输入的 LIST 类型列。
69 /// @param boolean_mask 输入的 LIST 类型列，元素类型为布尔值，形状与输入相同。boolean_mask 的每个元素都必须不可为空。
70 /// @param stream 用于设备内存操作的 CUDA 流。
71 /// @param mr 用于分配设备内存的设备内存资源。
72 /// @return 带有过滤后元素的新列表列。
73 std::unique_ptr<column> apply_boolean_mask(
74  lists_column_view const& input,
75  lists_column_view const& boolean_mask,
76  rmm::cuda_stream_view stream = cudf::get_default_stream(),
77  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
78 
89 /// @brief 创建一个新的列表列，其中每个列表中没有重复元素。
90 ///
91 /// 对于输入列中的每个列表，返回一个只包含每个元素首次出现项的列表。
92 /// @param input 要查找不同元素的列表列。
93 /// @param nulls_equal 是否将 null 视作相等或不相等。
94 /// @param nans_equal 是否将 NaNs 视作相等或不相等。
95 /// @param stream 用于设备内存操作的 CUDA 流。
96 /// @param mr 用于分配设备内存的设备内存资源。
97 /// @return 一个新的列表列，每行包含不同元素。
98 std::unique_ptr<column> distinct(
99  lists_column_view const& input,
100  null_equality nulls_equal = null_equality::EQUAL,
101  nan_equality nans_equal = nan_equality::ALL_EQUAL,
102  rmm::cuda_stream_view stream = cudf::get_default_stream(),
103  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
104  /// @} // end of group
105 
106 } // namespace lists
107 } // namespace CUDF_EXPORT cudf
cudf::lists_column_view
给定列表类型的列视图，此类的实例在此复合列上提供一个包装...
定义： lists_column_view.hpp:40
rmm::cuda_stream_view
column.hpp
cudf::column 的类定义。
device_memory_resource.hpp
cudf::get_default_stream
rmm::cuda_stream_view const get_default_stream()
获取当前的默认流。
cudf::lists::apply_boolean_mask
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())
使用布尔值列表 boolean_mask 作为掩码过滤输入 LIST 列的每一行中的元素。
cudf::lists::distinct
cudf
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::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
创建一个新的列表列，其中每个列表中没有重复元素。
cudf::get_current_device_resource_ref