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

Eliminate duplicate allocation of nested string columns #15061

Merged

Conversation

vuule
Copy link
Contributor

@vuule vuule commented Feb 15, 2024

Description

Issue #14965

Checklist

  • I am familiar with the Contributing Guidelines.
  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.

@vuule vuule added bug Something isn't working cuIO cuIO issue non-breaking Non-breaking change labels Feb 15, 2024
@vuule vuule self-assigned this Feb 15, 2024
@github-actions github-actions bot added the libcudf Affects libcudf (C++/CUDA) code. label Feb 15, 2024
@vuule vuule marked this pull request as ready for review February 26, 2024 18:43
@vuule vuule closed this Feb 26, 2024
@vuule vuule reopened this Feb 26, 2024
@vuule vuule requested a review from a team as a code owner February 26, 2024 18:48
vuule and others added 2 commits February 26, 2024 11:40
Co-authored-by: Nghia Truong <7416935+ttnghia@users.noreply.github.com>
@@ -281,7 +282,7 @@ void reader::impl::decode_page_data(size_t skip_rows, size_t num_rows)
out_buf.user_data |= PARQUET_COLUMN_BUFFER_FLAG_LIST_TERMINATED;
} else if (out_buf.type.id() == type_id::STRING) {
// need to cap off the string offsets column
size_type const sz = static_cast<size_type>(col_sizes[idx]);
size_type const sz = static_cast<size_type>(col_string_sizes[idx]);
Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Using size_type here for now. When we get to supporting larger string columns in the reader, we'll probably use an offsetalator (or something similar) here.

@vuule vuule added the 5 - Ready to Merge Testing and reviews complete, ready to merge label Feb 26, 2024
Copy link
Member

@harrism harrism left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Just one missed error handling macro. Otherwise looks good.

cpp/src/io/parquet/reader_impl.cpp Outdated Show resolved Hide resolved
@vuule vuule requested a review from harrism February 26, 2024 20:58
Copy link
Member

@harrism harrism left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@vuule
Copy link
Contributor Author

vuule commented Feb 29, 2024

/merge

@rapids-bot rapids-bot bot merged commit 08e3c96 into rapidsai:branch-24.04 Feb 29, 2024
76 checks passed
@vuule vuule deleted the bug-read_parquet-nested-str-alloc branch February 29, 2024 20:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
5 - Ready to Merge Testing and reviews complete, ready to merge bug Something isn't working cuIO cuIO issue libcudf Affects libcudf (C++/CUDA) code. non-breaking Non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants