search.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2019-2024, NVIDIA CORPORATION.
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>
13 #include <cudf/utilities/memory_resource.hpp>
14 
15 #include <vector>
16 
17 namespace CUDF_EXPORT cudf {
59 std::unique_ptr<column> lower_bound(
60  table_view const& haystack,
61  table_view const& needles,
62  std::vector<order> const& column_order,
63  std::vector<null_order> const& null_precedence,
64  rmm::cuda_stream_view stream = cudf::get_default_stream(),
65  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
66 
101 std::unique_ptr<column> upper_bound(
102  table_view const& haystack,
103  table_view const& needles,
104  std::vector<order> const& column_order,
105  std::vector<null_order> const& null_precedence,
106  rmm::cuda_stream_view stream = cudf::get_default_stream(),
107  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
108 
127 bool contains(column_view const& haystack,
128  scalar const& needle,
129  rmm::cuda_stream_view stream = cudf::get_default_stream());
130 
152 std::unique_ptr<column> contains(
153  column_view const& haystack,
154  column_view const& needles,
155  rmm::cuda_stream_view stream = cudf::get_default_stream(),
156  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
157  // end of group
159 } // namespace CUDF_EXPORT cudf
cudf::column_view
A non-owning, immutable view of device data as a column of elements, some of which may be null as ind...
Definition: column_view.hpp:316
cudf::scalar
An owning class to represent a singular value.
Definition: scalar.hpp:40
cudf::table_view
A set of cudf::column_view's of the same size.
Definition: table_view.hpp:189
rmm::cuda_stream_view
column.hpp
Class definition for cudf::column.
cudf::upper_bound
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.
cudf::lower_bound
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.
cudf::contains
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.
cudf::get_default_stream
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
cudf::get_current_device_resource_ref
rmm::device_async_resource_ref get_current_device_resource_ref()
Get the current device memory resource reference.
Definition: memory_resource.hpp:40
device_async_resource_ref
detail::cccl_async_resource_ref< cuda::mr::resource_ref< cuda::mr::device_accessible > > device_async_resource_ref
memory_resource.hpp
cudf
cuDF interfaces
Definition: host_udf.hpp:26
scalar.hpp
Class definitions for cudf::scalar.
table.hpp
Class definition for cudf::table.
types.hpp
Type declarations for libcudf.
Generated by doxygen 1.9.1