update_keys.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
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 #include <span>
13 
19 namespace CUDF_EXPORT cudf {
20 namespace dictionary {
52 std::unique_ptr<column> add_keys(
53  dictionary_column_view const& dictionary_column,
54  column_view const& new_keys,
57 
84 std::unique_ptr<column> remove_keys(
85  dictionary_column_view const& dictionary_column,
86  column_view const& keys_to_remove,
89 
107 std::unique_ptr<column> remove_unused_keys(
108  dictionary_column_view const& dictionary_column,
111 
140 std::unique_ptr<column> set_keys(
141  dictionary_column_view const& dictionary_column,
142  column_view const& keys,
145 
157 std::vector<std::unique_ptr<column>> match_dictionaries(
158  std::span<dictionary_column_view const> input,
161  // end of group
163 } // namespace dictionary
164 } // 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::vector< std::unique_ptr< column > > match_dictionaries(std::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.
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.
rmm::device_async_resource_ref get_current_device_resource_ref()
Get the current device memory resource reference.
cuda::mr::resource_ref< cuda::mr::device_accessible > device_async_resource_ref
APIs for getting and setting the current device memory resource.
cuDF interfaces
Definition: host_udf.hpp:26
APIs for spans.