reduction.hpp

前往此文件的文档。

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

#pragma once

#include <cudf/aggregation.hpp>
#include <cudf/scalar/scalar.hpp>
#include <cudf/utilities/export.hpp>
#include <cudf/utilities/memory_resource.hpp>

#include <optional>

namespace CUDF_EXPORT cudf {
enum class scan_type : bool { INCLUSIVE, EXCLUSIVE };

std::unique_ptr<scalar> reduce(
 column_view const& col,
 reduce_aggregation const& agg,
 data_type output_dtype,
 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> reduce(
 column_view const& col,
 reduce_aggregation const& agg,
 data_type output_dtype,
 std::optional<std::reference_wrapper<scalar const>> init,
 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> segmented_reduce(
 column_view const& segmented_values,
 device_span<size_type const> offsets,
 segmented_reduce_aggregation const& agg,
 data_type output_dtype,
 null_policy null_handling,
 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> segmented_reduce(
 column_view const& segmented_values,
 device_span<size_type const> offsets,
 segmented_reduce_aggregation const& agg,
 data_type output_dtype,
 null_policy null_handling,
 std::optional<std::reference_wrapper<scalar const>> init,
 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> scan(
 column_view const& input,
 scan_aggregation const& agg,
 scan_type inclusive,
 null_policy null_handling = null_policy::EXCLUDE,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());

std::pair<std::unique_ptr<scalar>, std::unique_ptr<scalar>> minmax(
 column_view const& col,
 rmm::cuda_stream_view stream = cudf::get_default_stream(),
 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
 // 组结束

} // 命名空间 CUDF_EXPORT cudf