dictionary/search.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2020-2024, 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 
19 #include <cudf/scalar/scalar.hpp>
21 
22 namespace CUDF_EXPORT cudf {
23 namespace dictionary {
43 std::unique_ptr<scalar> get_index(
44  dictionary_column_view const& dictionary,
45  scalar const& key,
48  // end of group
50 } // namespace dictionary
51 } // namespace CUDF_EXPORT cudf
A wrapper class for operations on a dictionary column.
An owning class to represent a singular value.
Definition: scalar.hpp:49
Class definition for cudf::dictionary_column_view.
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
std::unique_ptr< scalar > get_index(dictionary_column_view const &dictionary, scalar const &key, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Return the index value for a given key.
rmm::device_async_resource_ref get_current_device_resource_ref()
Get the current device memory resource reference.
cuda::mr::async_resource_ref< cuda::mr::device_accessible > device_async_resource_ref
cuDF interfaces
Definition: host_udf.hpp:39
Class definitions for cudf::scalar.