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

Improve SharedProcessPool tests performance #1950

Merged

Conversation

yczhang-nv
Copy link
Contributor

@yczhang-nv yczhang-nv commented Oct 16, 2024

Description

Some minor improvements to make SharedProcessPool related tests to run faster. Those are mostly combining multiple tests into one unit test function to reduce the number of times that the process pool being initialized and reset.

The total execution time of test_shared_process_pool.py and test_multi_processing_stage.py should be improved from ~60s to ~30s (when running locally).

Closes #1951

By Submitting this PR I confirm:

  • I am familiar with the Contributing Guidelines.
  • When the PR is ready for review, new or existing tests cover these changes.
  • When the PR is ready for review, the documentation is up to date with these changes.

@yczhang-nv yczhang-nv added non-breaking Non-breaking change improvement Improvement to existing functionality labels Oct 16, 2024
@yczhang-nv yczhang-nv self-assigned this Oct 16, 2024
@yczhang-nv yczhang-nv requested a review from a team as a code owner October 16, 2024 18:36
@yczhang-nv yczhang-nv marked this pull request as draft October 16, 2024 19:03
@yczhang-nv yczhang-nv changed the title Improve the SharedProcessPool tests to run faster Improve SharedProcessPool tests performance Oct 17, 2024
@yczhang-nv yczhang-nv marked this pull request as ready for review October 31, 2024 16:07
@yczhang-nv yczhang-nv changed the base branch from branch-24.10 to branch-25.02 October 31, 2024 16:19
@yczhang-nv yczhang-nv requested a review from a team as a code owner November 6, 2024 18:08
Copy link

copy-pr-bot bot commented Nov 6, 2024

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@yczhang-nv
Copy link
Contributor Author

/ok to test

@yczhang-nv
Copy link
Contributor Author

/ok to test

@yczhang-nv
Copy link
Contributor Author

/ok to test

@yczhang-nv
Copy link
Contributor Author

/ok to test

@yczhang-nv
Copy link
Contributor Author

/ok to test

@yczhang-nv
Copy link
Contributor Author

/ok to test

@yczhang-nv
Copy link
Contributor Author

/ok to test

@yczhang-nv
Copy link
Contributor Author

/ok to test

@yczhang-nv
Copy link
Contributor Author

/merge

@rapids-bot rapids-bot bot merged commit e90bafe into nv-morpheus:branch-25.02 Nov 22, 2024
11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
improvement Improvement to existing functionality non-breaking Non-breaking change
Projects
Status: Done
Development

Successfully merging this pull request may close these issues.

[BUG]: Improve the execution time of SharedProcessPool related unit tests
2 participants