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

Allow selection of a particular GPU (via the mask). #670

Draft
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

bosilca
Copy link
Contributor

@bosilca bosilca commented Sep 10, 2024

No description provided.

@bosilca bosilca requested a review from a team as a code owner September 10, 2024 04:27
@bosilca
Copy link
Contributor Author

bosilca commented Sep 10, 2024

Added support for CUDA_VISIBLE_DEVICES.

Use the parsec_device_mask to select devices allowed for each process.
If unset then use CUDA_VISIBLE_DEVICES to set the mask of allowed
devices.

Signed-off-by: George Bosilca <gbosilca@nvidia.com>
@abouteiller
Copy link
Contributor

Why is this needed?

On master (w/o this PR) Setting the environment variable already lets one fine control the allotment of devices to the parsec processes.

What do we gain from having a parsec-level mask that is a subsetting of the cuda one?

@bosilca
Copy link
Contributor Author

bosilca commented Sep 18, 2024

I'm not sure what you are referring to. Are you saying that if CUDA_VISIBLE_DEVICES is set CUDA only make the specified devices visible to parsec so we can ignore the second part of the patch ?

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