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

Benchmarks for Arrow IPC writer #7090

Merged
merged 6 commits into from
Feb 8, 2025
Merged

Conversation

alamb
Copy link
Contributor

@alamb alamb commented Feb 6, 2025

Which issue does this PR close?

Rationale for this change

We want to make the IPC writer faster by turning off validation but we currently don't have any benchmarks for this feature

What changes are included in this PR?

Add IPC writing benchmarks, based on @andygrove's work in #6972

I am working in IPC reading benchmarks as well

Are there any user-facing changes?

No, this is just benchmark code

@github-actions github-actions bot added the arrow Changes to the arrow crate label Feb 6, 2025
use criterion::{criterion_group, criterion_main, Criterion};
use std::sync::Arc;

fn criterion_benchmark(c: &mut Criterion) {
Copy link
Contributor Author

Choose a reason for hiding this comment

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

This benchmark benchmarks writing 10 batches (rather than the original benchmark from @andygrove that did both 1 and 10 batches). I profiled them both and the actual encoding of batches far dwarfs the writer creation time so I thought it would be best to focus on just writing

@alamb alamb changed the title Add benchmarks for Arrow IPC writer Benchmarks for Arrow IPC writer Feb 6, 2025
@alamb alamb marked this pull request as ready for review February 6, 2025 20:34
@alamb
Copy link
Contributor Author

alamb commented Feb 8, 2025

Thank you @tustvold

@alamb alamb merged commit 91aa6fd into apache:main Feb 8, 2025
30 checks passed
@alamb alamb deleted the alamb/ipc_benchmark branch February 8, 2025 16:31
phillipleblanc pushed a commit to spiceai/arrow-rs that referenced this pull request Feb 10, 2025
* Add benchmarks for Arrow IPC writer

* Add benchmarks for Arrow IPC writer

* reuse target buffer

* rename, etc

* Add compression type

* update

---------

Co-authored-by: Andy Grove <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
arrow Changes to the arrow crate
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants