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

perf: Move row group decode off async thread for local streaming parquet scan #19828

Merged
merged 8 commits into from
Nov 17, 2024

Conversation

nameexhaustion
Copy link
Collaborator

@nameexhaustion nameexhaustion commented Nov 17, 2024

Ensure we don't block tokio threads on compute.

@github-actions github-actions bot added performance Performance issues or improvements python Related to Python Polars rust Related to Rust Polars labels Nov 17, 2024
/// Run a task on the rayon threadpool. To avoid deadlocks, if the current thread is already a
/// rayon thread, the task is executed on the current thread after tokio's `block_in_place` is
/// used to spawn another thread to poll futures.
pub async fn spawn_rayon<F, O>(&self, func: F) -> O
Copy link
Collaborator Author

@nameexhaustion nameexhaustion Nov 17, 2024

Choose a reason for hiding this comment

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

Made this a function, but I think we won't be using it much as we move to the new streaming engine.

Copy link

codecov bot commented Nov 17, 2024

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 79.34%. Comparing base (50c1135) to head (a4e367f).
Report is 11 commits behind head on main.

Additional details and impacted files
@@            Coverage Diff             @@
##             main   #19828      +/-   ##
==========================================
+ Coverage   79.33%   79.34%   +0.01%     
==========================================
  Files        1548     1548              
  Lines      214208   214192      -16     
  Branches     2460     2460              
==========================================
+ Hits       169934   169943       +9     
+ Misses      43716    43691      -25     
  Partials      558      558              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@nameexhaustion nameexhaustion marked this pull request as ready for review November 17, 2024 06:48
@ritchie46 ritchie46 merged commit 5210d99 into pola-rs:main Nov 17, 2024
21 checks passed
@nameexhaustion nameexhaustion deleted the parquet-async-nonblock branch November 18, 2024 08:22
@c-peters c-peters added the accepted Ready for implementation label Nov 18, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
accepted Ready for implementation performance Performance issues or improvements python Related to Python Polars rust Related to Rust Polars
Projects
Archived in project
Development

Successfully merging this pull request may close these issues.

3 participants