cudf/replace.hpp

前往此文件的文档。

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

#pragma once

#include <cudf/types.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <cudf/utilities/export.hpp>
#include <cudf/utilities/memory_resource.hpp>

#include <memory>

namespace CUDF_EXPORT cudf {
enum class replace_policy : bool { PRECEDING, FOLLOWING };

std::unique_ptr<column> replace_nulls(
 column_view const& input,
 column_view const& replacement,
 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> replace_nulls(
 column_view const& input,
 scalar const& replacement,
 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> replace_nulls(
 column_view const& input,
 replace_policy const& replace_policy,
 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> replace_nans(
 column_view const& input,
 column_view const& replacement,
 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> replace_nans(
 column_view const& input,
 scalar const& replacement,
 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> find_and_replace_all(
 column_view const& input_col,
 column_view const& values_to_replace,
 column_view const& replacement_values,
 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> clamp(
 column_view const& input,
 scalar const& lo,
 scalar const& lo_replace,
 scalar const& hi,
 scalar const& hi_replace,
 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> clamp(
 column_view const& input,
 scalar const& lo,
 scalar const& hi,
 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> normalize_nans_and_zeros(
 column_view const& input,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

void normalize_nans_and_zeros(mutable_column_view& in_out,
 rmm::cuda_stream_view stream = cudf::get_default_stream());
 // end of group
} // namespace CUDF_EXPORT cudf