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

fix: correct the logic of transform shuffle exchange #1384

Closed
wants to merge 1 commit into from

Conversation

wForget
Copy link
Member

@wForget wForget commented Feb 10, 2025

Which issue does this PR close?

minor fix

Rationale for this change

The child of CometShuffleExchangeExec should be columnar

What changes are included in this PR?

How are these changes tested?

@wForget wForget marked this pull request as draft February 10, 2025 15:15
@@ -862,7 +862,7 @@ class CometSparkSessionExtensions
newOp match {
case Some(nativeOp) =>
s.child match {
case n if n.isInstanceOf[CometNativeExec] || !n.supportsColumnar =>
case n if n.isInstanceOf[CometNativeExec] || n.supportsColumnar =>
Copy link
Contributor

Choose a reason for hiding this comment

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

If the child is not a CometNativeExec can be be a ShuffleExchangeExec? In that case we should not use CometColumnarShuffle.

Copy link
Member Author

Choose a reason for hiding this comment

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

I see, thank you. CometShuffleExchangeExec supports row base input.

@wForget wForget closed this Feb 11, 2025
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