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

Add tests for GpuPartition #398

Merged
merged 1 commit into from
Jul 22, 2020
Merged

Conversation

jlowe
Copy link
Member

@jlowe jlowe commented Jul 21, 2020

This adds unit tests for GpuPartition. During testing I found a number of issues that are also addressed in this PR:

  • GpuShuffleEnv was a Scala object that didn't support reinitializing which is needed for unit tests to run in various shuffle configurations. Updated it to support reinitializing the state.
  • SparkSessionHolder doesn't handle cases where other tests tear down the Spark session and a new one needs to be created. Updated the session holder to handle this case.
  • HashSortOptimizeSuite assumed there was one and only one spark session, updated it to use the session handed to it.
  • GpuSemaphore would crash if there was no task context. I updated it to be a no-op when there's no task context (i.e.: during unit tests).

@jlowe jlowe added the test Only impacts tests label Jul 21, 2020
@jlowe jlowe added this to the Jul 20 - Jul 31 milestone Jul 21, 2020
@jlowe jlowe self-assigned this Jul 21, 2020
@jlowe
Copy link
Member Author

jlowe commented Jul 21, 2020

build

@revans2 revans2 merged commit 2dfbeaa into NVIDIA:branch-0.2 Jul 22, 2020
nartal1 pushed a commit to nartal1/spark-rapids that referenced this pull request Jun 9, 2021
nartal1 pushed a commit to nartal1/spark-rapids that referenced this pull request Jun 9, 2021
@jlowe jlowe deleted the partition-tests branch September 10, 2021 15:30
tgravescs pushed a commit to tgravescs/spark-rapids that referenced this pull request Nov 30, 2023
Signed-off-by: spark-rapids automation <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
test Only impacts tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants