contains.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */
#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