Revision f01eafd79f3bd8a02cdce1422a4b2846b93bdc52 authored by Wenchen Fan on 05 August 2024, 06:57:25 UTC, committed by Wenchen Fan on 05 August 2024, 06:57:25 UTC
### What changes were proposed in this pull request? We missed the fact that submitting a shuffle or broadcast query stage can be heavy, as it needs to submit subqueries and wait for the results. This blocks the AQE loop and hurts the parallelism of AQE. This PR fixes the problem by using shuffle/broadcast's own thread pool to wait for subqueries and other preparations. This PR also re-implements https://github.com/apache/spark/pull/45234 to avoid submitting the shuffle job if the query is failed and all query stages need to be cancelled. ### Why are the changes needed? better parallelism for AQE ### Does this PR introduce _any_ user-facing change? no ### How was this patch tested? new test case ### Was this patch authored or co-authored using generative AI tooling? no Closes #47533 from cloud-fan/aqe. Authored-by: Wenchen Fan <wenchen@databricks.com> Signed-off-by: Wenchen Fan <wenchen@databricks.com>
1 parent 94f8872
File | Mode | Size |
---|---|---|
docs | ||
lib | ||
packaging | ||
pyspark | ||
test_coverage | ||
test_support | ||
.coveragerc | -rw-r--r-- | 872 bytes |
.gitignore | -rw-r--r-- | 97 bytes |
MANIFEST.in | -rw-r--r-- | 1.3 KB |
README.md | -rw-r--r-- | 2.1 KB |
mypy.ini | -rw-r--r-- | 3.9 KB |
run-tests | -rwxr-xr-x | 1.1 KB |
run-tests-with-coverage | -rwxr-xr-x | 2.8 KB |
run-tests.py | -rwxr-xr-x | 15.6 KB |
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...