diff --git a/cpp/benchmarks/io/parquet/parquet_reader_multithread.cpp b/cpp/benchmarks/io/parquet/parquet_reader_multithread.cpp index fbdcfb0ade9..bd80c4e0e88 100644 --- a/cpp/benchmarks/io/parquet/parquet_reader_multithread.cpp +++ b/cpp/benchmarks/io/parquet/parquet_reader_multithread.cpp @@ -25,25 +25,12 @@ #include #include -#include -#include -#include - #include #include #include -// TODO: remove this once pinned/pooled is enabled by default in cuIO -void set_cuio_host_pinned_pool() -{ - using host_pooled_mr = rmm::mr::pool_memory_resource; - static std::shared_ptr mr = std::make_shared( - std::make_shared().get(), 256ul * 1024 * 1024); - cudf::io::set_host_memory_resource(*mr); -} - size_t get_num_reads(nvbench::state const& state) { return state.get_int64("num_threads"); } size_t get_read_size(nvbench::state const& state) @@ -105,8 +92,6 @@ void BM_parquet_multithreaded_read_common(nvbench::state& state, size_t const data_size = state.get_int64("total_data_size"); auto const num_threads = state.get_int64("num_threads"); - set_cuio_host_pinned_pool(); - auto streams = cudf::detail::fork_streams(cudf::get_default_stream(), num_threads); cudf::detail::thread_pool threads(num_threads); @@ -186,8 +171,6 @@ void BM_parquet_multithreaded_read_chunked_common(nvbench::state& state, size_t const input_limit = state.get_int64("input_limit"); size_t const output_limit = state.get_int64("output_limit"); - set_cuio_host_pinned_pool(); - auto streams = cudf::detail::fork_streams(cudf::get_default_stream(), num_threads); cudf::detail::thread_pool threads(num_threads); auto [source_sink_vector, total_file_size, num_files] = write_file_data(state, d_types);