parquet_io_utils.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <cudf/io/datasource.hpp>
#include <cudf/io/text/byte_range_info.hpp>
 
#include <rmm/cuda_stream_view.hpp>
#include <rmm/device_buffer.hpp>
#include <rmm/mr/device_memory_resource.hpp>
 
#include <future>
#include <tuple>
#include <vector>
 
namespace CUDF_EXPORT cudf {
namespace io::parquet {
 
using cudf::io::text::byte_range_info;
 
std::unique_ptr<cudf::io::datasource::buffer> fetch_footer_to_host(
  cudf::io::datasource& datasource);
 
std::unique_ptr<cudf::io::datasource::buffer> fetch_page_index_to_host(
  cudf::io::datasource& datasource, byte_range_info const page_index_bytes);
 
std::tuple<std::vector<rmm::device_buffer>,
           std::vector<cudf::device_span<uint8_t const>>,
           std::future<void>>
fetch_byte_ranges_to_device_async(cudf::io::datasource& datasource,
                                  cudf::host_span<byte_range_info const> byte_ranges,
                                  rmm::cuda_stream_view stream,
                                  rmm::device_async_resource_ref mr);
  // end of group
}  // namespace io::parquet
}  // namespace CUDF_EXPORT cudf