-
Notifications
You must be signed in to change notification settings - Fork 458
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
[CORE] ColumnarBroadcastExchangeExec should set/cancel with job tag for Spark3.5 #4882
Conversation
Thanks for opening a pull request! Could you open an issue for this pull request on Github Issues? https://github.com/oap-project/gluten/issues Then could you also rename commit message and pull request title in the following format?
See also: |
Run Gluten Clickhouse CI |
runId.toString, | ||
s"broadcast exchange (runId $runId)", | ||
interruptOnCancel = true) | ||
SparkShimLoader.getSparkShims.setJobDescriptionOrTagForBroadcastExchange(sparkContext, this) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This change is required, otherwise AQE can not cancel broadcast job. BroadcastQueryStageExec
use cancelJobsWithTag
to cancel broadcast.
Run Gluten Clickhouse CI |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Thanks.
===== Performance report for TPCH SF2000 with Velox backend, for reference only ====
|
…b tag for Spark3.5 (apache#4882)
…b tag for Spark3.5 (apache#4882)
…b tag for Spark3.5 (apache#4882)
What changes were proposed in this pull request?
Spark3.5 uses job tag to cancel broadcast exchange job. This pr adds two shim methods to be compatible with that:
See apache/spark#41440
How was this patch tested?
Pass CI