search.hpp

转到此文件的文档。

/*
 * Copyright (c) 2019-2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证版本 2.0（“许可证”）授权；
 * 除非遵守许可证，否则不得使用此文件。
 * 您可以在以下位置获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可证分发，基于“原样”基础，
 * 不附带任何明示或默示的保证或条件。
 * 有关管理权限和
 * 限制的特定语言，请参阅许可证。
 */

#pragma once

#include <cudf/column/column.hpp>
#include <cudf/scalar/scalar.hpp>
#include <cudf/table/table.hpp>
#include <cudf/types.hpp>
#include <cudf/utilities/export.hpp>
#include <cudf/utilities/memory_resource.hpp>

#include <vector>

namespace CUDF_EXPORT cudf {
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());

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());

bool contains(column_view const& haystack,
 scalar const& needle,
 rmm::cuda_stream_view stream = cudf::get_default_stream());

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());
 // 结束分组
} // namespace CUDF_EXPORT cudf