data_chunk_source.hpp

查看此文件的文档。

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

#pragma once

#include <cudf/utilities/export.hpp>
#include <cudf/utilities/span.hpp>

#include <rmm/cuda_stream_pool.hpp>
#include <rmm/device_buffer.hpp>

namespace CUDF_EXPORT cudf {
namespace io {
namespace text {

class device_data_chunk {
 public
 virtual ~device_data_chunk() = default;
 [[nodiscard]] virtual char const* data() const = 0;
 [[nodiscard]] virtual std::size_t size() const = 0;
 virtual operator device_span<char const>() const = 0;
};

class data_chunk_reader {
 public
 virtual ~data_chunk_reader() = default;
 virtual void skip_bytes(std::size_t size) = 0;

 virtual std::unique_ptr<device_data_chunk> get_next_chunk(std::size_t size,
 rmm::cuda_stream_view stream) = 0;
};

class data_chunk_source {
 public
 virtual ~data_chunk_source() = default;

 [[nodiscard]] virtual std::unique_ptr<data_chunk_reader> create_reader() const = 0;
};
 // 组结束

} // namespace text
} // namespace io
} // namespace CUDF_EXPORT cudf