Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add compile time check to ensure the counting_iterator type in counting_transform_iterator fits in size_type #17118

Prev Previous commit
Next Next commit
style fix
  • Loading branch information
mhaseeb123 committed Oct 17, 2024
commit ac82d73d46942acc3ca21e6769e9d4f7aed1a2e4
3 changes: 2 additions & 1 deletion cpp/src/io/parquet/reader_impl_preprocess.cu
Original file line number Diff line number Diff line change
Expand Up @@ -1629,7 +1629,8 @@ void reader::impl::allocate_columns(read_mode mode, size_t skip_rows, size_t num
get_page_nesting_size{
d_cols_info.data(), max_depth, subpass.pages.size(), subpass.pages.device_begin()});

auto const reduction_keys = cudf::detail::make_counting_transform_iterator(key_start, get_reduction_key{subpass.pages.size()});
auto const reduction_keys = cudf::detail::make_counting_transform_iterator(
key_start, get_reduction_key{subpass.pages.size()});

// Find the size of each column
thrust::reduce_by_key(rmm::exec_policy_nosync(_stream),
Expand Down
Loading