Skip to content

[SPARK-40872][3.3] Fallback to original shuffle block when a push-merged shuffle chunk is zero-size#38751

Closed
gaoyajun02 wants to merge 1 commit intoapache:branch-3.3from
gaoyajun02:SPARK-40872-backport
Closed

[SPARK-40872][3.3] Fallback to original shuffle block when a push-merged shuffle chunk is zero-size#38751
gaoyajun02 wants to merge 1 commit intoapache:branch-3.3from
gaoyajun02:SPARK-40872-backport

Conversation

@gaoyajun02
Copy link
Contributor

@gaoyajun02 gaoyajun02 commented Nov 22, 2022

What changes were proposed in this pull request?

This is a backport PR of #38333.
When push-based shuffle is enabled, a zero-size buf error may occur when fetching shuffle chunks from bad nodes, especially when memory is full. In this case, we can fall back to original shuffle blocks.

Why are the changes needed?

When the reduce task obtains the shuffle chunk with a zero-size buf, we let it fall back to original shuffle block. After verification, these blocks can be read successfully without shuffle retry.

Does this PR introduce any user-facing change?

No.

How was this patch tested?

UT

…huffle chunk is zero-size

When push-based shuffle is enabled, a zero-size buf error may occur when fetching shuffle chunks from bad nodes, especially when memory is full. In this case, we can fall back to original shuffle blocks.

When the reduce task obtains the shuffle chunk with a zero-size buf, we let it fall back to original shuffle block. After verification, these blocks can be read successfully without shuffle retry.

No.

UT

Closes apache#38333 from gaoyajun02/SPARK-40872.

Authored-by: gaoyajun02 <gaoyajun02@meituan.com>
Signed-off-by: Mridul <mridul<at>gmail.com>
(cherry picked from commit 72cce5c)
@github-actions github-actions bot added the CORE label Nov 22, 2022
Copy link
Contributor

@mridulm mridulm 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 to me.
Will wait for the tests to pass

@HyukjinKwon
Copy link
Member

Seems like the test failure looks unrelated. I don't mind merging it as is.

Feel free to retrigger https://github.com/gaoyajun02/spark/runs/9633118279 @gaoyajun02

@AmplabJenkins
Copy link

Can one of the admins verify this patch?

@gaoyajun02
Copy link
Contributor Author

Seems like the test failure looks unrelated. I don't mind merging it as is.

Feel free to retrigger https://github.com/gaoyajun02/spark/runs/9633118279 @gaoyajun02

tests all passed @HyukjinKwon

@HyukjinKwon
Copy link
Member

Merged to branch-3.3.

HyukjinKwon pushed a commit that referenced this pull request Nov 27, 2022
…ged shuffle chunk is zero-size

### What changes were proposed in this pull request?
This is a backport PR of #38333.
When push-based shuffle is enabled, a zero-size buf error may occur when fetching shuffle chunks from bad nodes, especially when memory is full. In this case, we can fall back to original shuffle blocks.

### Why are the changes needed?
When the reduce task obtains the shuffle chunk with a zero-size buf, we let it fall back to original shuffle block. After verification, these blocks can be read successfully without shuffle retry.

### Does this PR introduce _any_ user-facing change?
No.

### How was this patch tested?
UT

Closes #38751 from gaoyajun02/SPARK-40872-backport.

Authored-by: gaoyajun02 <gaoyajun02@meituan.com>
Signed-off-by: Hyukjin Kwon <gurwls223@apache.org>
@mridulm
Copy link
Contributor

mridulm commented Nov 27, 2022

Thanks for the PR @gaoyajun02, and thanks for merging it @HyukjinKwon !

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

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants

Comments