Skip to content

[#2494] feat(spark): Overlapping compression to avoid blocking shuffle writing #2511

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

Merged
merged 1 commit into from
Jul 3, 2025

Conversation

zuston
Copy link
Member

@zuston zuston commented Jun 19, 2025

What changes were proposed in this pull request?

Support overlapping compression to avoid block shuffle writing

Why are the changes needed?

for #2494 .

After applying the proposed improvements, the client observed that shuffle performance doubled in a 100GB Terasort benchmark. Specifically, the overall shuffle speed improved by over 100%, significantly reducing the job’s runtime and demonstrating the effectiveness of the optimizations in large-scale sorting workloads.

image

image

Does this PR introduce any user-facing change?

Yes.

How was this patch tested?

Will be added later.

@zuston zuston changed the title [#2494] feat(spark): Overlapping compression to avoid block shuffle writing [#2494] feat(spark): Overlapping compression to avoid blocking shuffle writing Jun 19, 2025
Copy link

github-actions bot commented Jun 19, 2025

Test Results

 3 072 files  +23   3 072 suites  +23   6h 48m 50s ⏱️ +45s
 1 190 tests + 2   1 189 ✅ + 2   1 💤 ±0  0 ❌ ±0 
15 090 runs  +23  15 075 ✅ +23  15 💤 ±0  0 ❌ ±0 

Results for commit 6017641. ± Comparison against base commit f2d7d86.

♻️ This comment has been updated with latest results.

@zuston
Copy link
Member Author

zuston commented Jun 20, 2025

Maybe this is not a best strategy to reduce the compression cost, but this is acceptable. Actually the biggest cost of compression is the frequent memory allocation, if we could reuse the compression buffer in the final phase of rpc sending, I think the performance improvement will be significant.

@zuston zuston marked this pull request as draft June 23, 2025 02:23
@zuston zuston marked this pull request as ready for review June 25, 2025 04:08
@zuston zuston force-pushed the overlapCompression branch from a094902 to 49b380c Compare June 25, 2025 04:08
@zuston
Copy link
Member Author

zuston commented Jun 25, 2025

cc @jerqi

Copy link
Contributor

@jerqi jerqi left a comment

Choose a reason for hiding this comment

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

LGTM.

@jerqi
Copy link
Contributor

jerqi commented Jun 26, 2025

Could you give some data for this improvement?

@zuston
Copy link
Member Author

zuston commented Jun 26, 2025

Could you give some data for this improvement?

Still in evaluating and I will explore the compression improvements

@zuston zuston force-pushed the overlapCompression branch from 4309d2d to 6017641 Compare July 2, 2025 06:13
@zuston
Copy link
Member Author

zuston commented Jul 2, 2025

Could you give some data for this improvement?

Updated in the description.

@jerqi
Copy link
Contributor

jerqi commented Jul 2, 2025

Could you give some data for this improvement?

Updated in the description.

Got it.

@zuston zuston merged commit 77e6ab1 into apache:master Jul 3, 2025
41 checks passed
@zuston zuston deleted the overlapCompression branch July 3, 2025 03:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants