Skip to content

[CELEBORN-2457][CIP-24] Adaptive partition write parallelism - #3843

Open
yew1eb wants to merge 1 commit into
apache:mainfrom
yew1eb:CELEBORN-2457
Open

[CELEBORN-2457][CIP-24] Adaptive partition write parallelism#3843
yew1eb wants to merge 1 commit into
apache:mainfrom
yew1eb:CELEBORN-2457

Conversation

@yew1eb

@yew1eb yew1eb commented Sep 8, 2026

Copy link
Copy Markdown
Contributor

What changes were proposed in this pull request?

Adaptive partition write parallelism (design doc: <[link]
(https://docs.google.com/document/d/1Kzn6_sEkzR4aDu2XofKBLUYxHyArSYdcYrlfZJuDBIw/edit?usp=sharing) >):

  • A hot reduce partition writes to multiple active locations in parallel; mappers are routed by mapId % activeCount.
  • The LifecycleManager measures per-location fill time from retire reports and ramps parallelism toward ceil(targetSplitInterval / fillTime) — closed-loop, no manual tuning. Every revive reply carries the full active set.
  • One additive proto field (PbChangeLocationPartitionInfo.additionalPartitions); three client-side configs, all off by default.

Why are the changes needed?

A reduce partition is written through a single active location at a time, concentrating all mappers' write pressure on one point. In one of
our production jobs (one reduce partition, 26,090 mappers), 90% of task time was spent waiting on shuffle writes; with this feature, write
throughput improved from 6.33 MB/s to 649 MB/s.

Does this PR resolve a correctness bug?

  • Yes

Does this PR introduce any user-facing change?

  • Yes

Three new client configs (enabled / maxLocations / targetSplitInterval), disabled by default — no behavior change unless explicitly enabled. Protocol change is additive and safe for mixed old/new client and LifecycleManager versions.

How was this patch tested?

  • New unit suites: PartitionLocationGroupSuiteJ, ReviveManagerSuiteJ, ShuffleClientSuiteJ, PartitionHotnessTrackerSuite, ChangePartitionManagerAdaptiveParallelismSuite, RequestLocationCallContextSuite.
  • Existing client/LM suites pass unchanged with the feature off.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant