label_bins.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/column/column_view.hpp>
#include <cudf/types.hpp>
#include <cudf/utilities/memory_resource.hpp>

#include <rmm/mr/device/device_memory_resource.hpp>

namespace CUDF_EXPORT cudf {

enum class inclusive { YES, NO };

std::unique_ptr<column> label_bins(
 column_view const& input,
 column_view const& left_edges,
 inclusive left_inclusive,
 column_view const& right_edges,
 inclusive right_inclusive,
 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 CUDF_EXPORT cudf