scalar_factories.hpp

转到此文件的文档。

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

#include <cudf/scalar/scalar.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <cudf/utilities/memory_resource.hpp>

#include <rmm/cuda_stream_view.hpp>

namespace CUDF_EXPORT cudf {
std::unique_ptr<scalar> make_numeric_scalar(
 data_type type,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<scalar> make_timestamp_scalar(
 data_type type,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<scalar> make_duration_scalar(
 data_type type,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<scalar> make_fixed_width_scalar(
 data_type type,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<scalar> make_string_scalar(
 std::string const& string,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<scalar> make_default_constructed_scalar(
 data_type type,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<scalar> make_empty_scalar_like(
 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());

template <typename T>
std::unique_ptr<scalar> make_fixed_width_scalar(
 T value,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref())
{
 return std::make_unique<scalar_type_t<T>>(value, true, stream, mr);
}

template <typename T>
std::unique_ptr<scalar> make_fixed_point_scalar(
 typename T::rep value,
 numeric::scale_type scale,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref())
{
 return std::make_unique<scalar_type_t<T>>(value, scale, true, stream, mr);
}

std::unique_ptr<scalar> make_list_scalar(
 column_view elements,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<scalar> make_struct_scalar(
 table_view const& data,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::unique_ptr<scalar> make_struct_scalar(
 host_span<column_view const> data,
 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