update_keys.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2020-2024, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 #pragma once
6 
7 #include <cudf/column/column.hpp>
10 #include <cudf/utilities/span.hpp>
11 
12 namespace CUDF_EXPORT cudf {
13 namespace dictionary {
47 std::unique_ptr<column> add_keys(
48  dictionary_column_view const& dictionary_column,
49  column_view const& new_keys,
52 
79 std::unique_ptr<column> remove_keys(
80  dictionary_column_view const& dictionary_column,
81  column_view const& keys_to_remove,
84 
102 std::unique_ptr<column> remove_unused_keys(
103  dictionary_column_view const& dictionary_column,
106 
135 std::unique_ptr<column> set_keys(
136  dictionary_column_view const& dictionary_column,
137  column_view const& keys,
140 
152 std::vector<std::unique_ptr<column>> match_dictionaries(
156  // end of group
158 } // namespace dictionary
159 } // namespace CUDF_EXPORT cudf
A non-owning, immutable view of device data as a column of elements, some of which may be null as ind...
A wrapper class for operations on a dictionary column.
Class definition for cudf::column.
Class definition for cudf::dictionary_column_view.
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
std::unique_ptr< column > set_keys(dictionary_column_view const &dictionary_column, column_view const &keys, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create a new dictionary column by applying only the specified keys to the existing dictionary_column.
std::unique_ptr< column > remove_unused_keys(dictionary_column_view const &dictionary_column, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create a new dictionary column by removing any keys that are not referenced by any of the indices.
std::unique_ptr< column > remove_keys(dictionary_column_view const &dictionary_column, column_view const &keys_to_remove, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create a new dictionary column by removing the specified keys from the existing dictionary_column.
std::unique_ptr< column > add_keys(dictionary_column_view const &dictionary_column, column_view const &new_keys, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create a new dictionary column by adding the new keys elements to the existing dictionary_column.
std::vector< std::unique_ptr< column > > match_dictionaries(cudf::host_span< dictionary_column_view const > input, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Create new dictionaries that have keys merged from the input dictionaries.
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
cuDF interfaces
Definition: host_udf.hpp:26
APIs for spans.
C++20 std::span with reduced feature set.
Definition: span.hpp:182