lists/contains.hpp

前往此文件的文档。

/*
 * 版权所有 (c) 2021-2024, NVIDIA CORPORATION。
 *
 * 根据 Apache 许可，版本 2.0（“许可”）获得许可；
 * 除非遵守许可，否则您不得使用此文件。
 * 您可以在以下位置获取许可的副本
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，根据许可分发的软件
 * 按“原样”分发，不附带任何明示或暗示的担保或条件。
 * 请参阅许可了解特定的语言权限和限制。
 *
 * limitationS under the License.
 */
#pragma once

#include <cudf/column/column.hpp>
#include <cudf/lists/lists_column_view.hpp>
#include <cudf/utilities/export.hpp>
#include <cudf/utilities/memory_resource.hpp>

namespace CUDF_EXPORT cudf {
namespace lists {
std::unique_ptr<column> contains(
 cudf::lists_column_view const& lists,
 cudf::scalar const& search_key,
 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> contains(
 cudf::lists_column_view const& lists,
 cudf::column_view const& search_keys,
 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> contains_nulls(
 cudf::lists_column_view const& lists,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

enum class duplicate_find_option : int32_t {
 FIND_FIRST = 0,
 FIND_LAST
};

std::unique_ptr<column> index_of(
 cudf::lists_column_view const& lists,
 cudf::scalar const& search_key,
 duplicate_find_option find_option = duplicate_find_option::FIND_FIRST,
 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> index_of(
 cudf::lists_column_view const& lists,
 cudf::column_view const& search_keys,
 duplicate_find_option find_option = duplicate_find_option::FIND_FIRST,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
 // end of group
} // namespace lists
} // namespace CUDF_EXPORT cudf