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

Separate stats filtering helpers to reuse in page pruning #18034

Open
wants to merge 13 commits into
base: branch-25.04
Choose a base branch
from

Conversation

mhaseeb123
Copy link
Member

@mhaseeb123 mhaseeb123 commented Feb 19, 2025

Description

Contributes to #17896

This PR separates stats based filtering helpers for reuse in page pruning using stats in Parquet PageIndex.

Checklist

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

Copy link

copy-pr-bot bot commented Feb 19, 2025

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@github-actions github-actions bot added libcudf Affects libcudf (C++/CUDA) code. CMake CMake build issue labels Feb 19, 2025
@mhaseeb123 mhaseeb123 added 2 - In Progress Currently a work in progress improvement Improvement / enhancement to an existing function non-breaking Non-breaking change cuIO cuIO issue labels Feb 19, 2025
@@ -740,4 +462,157 @@ std::optional<std::vector<std::vector<size_type>>> collect_filtered_row_group_in
return {filtered_row_group_indices};
}

// All required templates for `stats_caster_base` and `stats_caster_base::host_column<T>`
Copy link
Member Author

@mhaseeb123 mhaseeb123 Feb 19, 2025

Choose a reason for hiding this comment

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

Moved definitions as-is. All required templates for row group or page pruning are instantiated here as well. Definitions moved to stats_filter_helpers.hpp as it seems like it doesn't instantiate all templates needed for page pruning

@mhaseeb123 mhaseeb123 marked this pull request as ready for review February 19, 2025 19:39
@mhaseeb123 mhaseeb123 requested review from a team as code owners February 19, 2025 19:39
@mhaseeb123 mhaseeb123 added the 3 - Ready for Review Ready for review by team label Feb 19, 2025
size_type total_row_groups;
std::vector<metadata> const& per_file_metadata;
host_span<std::vector<size_type> const> row_group_indices;

template <typename ToType, typename FromType>
Copy link
Member Author

@mhaseeb123 mhaseeb123 Feb 19, 2025

Choose a reason for hiding this comment

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

Moved to stats_filter_helpers.hpp

@@ -248,146 +110,6 @@ struct stats_caster {
}
};

/**
Copy link
Member Author

Choose a reason for hiding this comment

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

Moved completely to stats_filter_helpers.hpp/cpp

@mhaseeb123 mhaseeb123 changed the title Separate stats filtering helpers to reuse for page pruning Separate stats filtering helpers to reuse in page pruning Feb 19, 2025
@mhaseeb123 mhaseeb123 removed the 2 - In Progress Currently a work in progress label Feb 19, 2025
Copy link
Contributor

@davidwendt davidwendt left a comment

Choose a reason for hiding this comment

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

Looks good. Some minor comments.

Copy link
Contributor

@karthikeyann karthikeyann left a comment

Choose a reason for hiding this comment

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

LGTM 👍

@mhaseeb123
Copy link
Member Author

Any failing GTests are fixed by #18019


#include "io/parquet/parquet_common.hpp"

#include <cudf/ast/detail/expression_transformer.hpp>
Copy link
Member Author

Choose a reason for hiding this comment

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

Moved definitions for stats_caster_base here as not all templates instantiated at one place.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
3 - Ready for Review Ready for review by team CMake CMake build issue cuIO cuIO issue improvement Improvement / enhancement to an existing function libcudf Affects libcudf (C++/CUDA) code. non-breaking Non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants