search.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 
6 #pragma once
7 
8 #include <cudf/column/column.hpp>
9 #include <cudf/scalar/scalar.hpp>
10 #include <cudf/table/table.hpp>
11 #include <cudf/types.hpp>
12 #include <cudf/utilities/export.hpp>
14 
15 #include <vector>
16 
22 namespace CUDF_EXPORT cudf {
62 std::unique_ptr<column> lower_bound(
63  table_view const& haystack,
64  table_view const& needles,
65  std::vector<order> const& column_order,
66  std::vector<null_order> const& null_precedence,
69 
104 std::unique_ptr<column> upper_bound(
105  table_view const& haystack,
106  table_view const& needles,
107  std::vector<order> const& column_order,
108  std::vector<null_order> const& null_precedence,
111 
130 bool contains(column_view const& haystack,
131  scalar const& needle,
133 
155 std::unique_ptr<column> contains(
156  column_view const& haystack,
157  column_view const& needles,
160  // end of group
162 } // 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...
An owning class to represent a singular value.
Definition: scalar.hpp:41
A set of cudf::column_view's of the same size.
Definition: table_view.hpp:206
Class definition for cudf::column.
std::unique_ptr< column > upper_bound(table_view const &haystack, table_view const &needles, std::vector< order > const &column_order, std::vector< null_order > const &null_precedence, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Find largest indices in a sorted table where values should be inserted to maintain order.
std::unique_ptr< column > lower_bound(table_view const &haystack, table_view const &needles, std::vector< order > const &column_order, std::vector< null_order > const &null_precedence, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Find smallest indices in a sorted table where values should be inserted to maintain order.
std::unique_ptr< column > contains(column_view const &haystack, column_view const &needles, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::device_async_resource_ref mr=cudf::get_current_device_resource_ref())
Check if the given needles values exists in the haystack column.
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
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
Class definitions for cudf::scalar.
Class definition for cudf::table.
Type declarations for libcudf.