Skip to content

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>
@devreal devreal marked this pull request as draft September 10, 2024 13:21
@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