From 929bf9d8e929a15c5d06dceeace791a26bc64b82 Mon Sep 17 00:00:00 2001 From: JayjeetAtGithub Date: Mon, 12 Aug 2024 10:52:06 -0700 Subject: [PATCH] Minor updates and docstrings --- cpp/examples/tpch/utils.hpp | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/cpp/examples/tpch/utils.hpp b/cpp/examples/tpch/utils.hpp index d5b8ff01a07..90ebb3277b6 100644 --- a/cpp/examples/tpch/utils.hpp +++ b/cpp/examples/tpch/utils.hpp @@ -527,7 +527,7 @@ cudf::size_type get_sf() */ void print_device_info() { - std::cout << "####### Device Information ##########" << std::endl; + std::cout << "######### Device Information ##########" << std::endl; auto num_devices = rmm::get_num_cuda_devices(); std::cout << "CUDA device count: " << num_devices << std::endl; @@ -599,10 +599,19 @@ void write_to_parquet_source(std::unique_ptr const& table, source.d_buffer.data(), h_buffer.data(), h_buffer.size(), cudaMemcpyDefault, stream.value())); } +/** + * @brief Populate the parquet data sources using the dataset + * directory information and the names of the tables to generate + * + * @param dataset_dir The directory containing the parquet files + * @param table_names The names of the tables to generate + * @param sources The parquet data sources to populate + */ void populate_parquet_sources(std::string const& dataset_dir, std::vector const& table_names, std::unordered_map& sources) { + CUDF_FUNC_RANGE(); std::for_each(table_names.begin(), table_names.end(), [&](auto const& table_name) { sources[table_name] = parquet_source(); });