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.

Sign up for GitHub

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

Jump to bottom

Add gpuId proc on locale #25943

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Add gpuId proc on locale #25943

wants to merge 3 commits into from
+28 −0

Conversation

stonea
Copy link
Contributor

@stonea stonea commented Sep 13, 2024 •
edited
Loading

This PR adds a way to query gpuID (the location in the parent locale's here.gpus array) that a gpu locale is.

If used on a non gpu locale it will error out with this message:

Can not use 'gpuId' field on a non gpu locale

This should resolve #23650

TODO:

  • paratest
  • paratest nvidia

@stonea
add locale.gpuId function
a088cb7 
---
Signed-off-by: Andy Stone <[email protected]>
@stonea
add no gpu id test
febf76b 
---
Signed-off-by: Andy Stone <[email protected]>
@stonea stonea mentioned this pull request Sep 13, 2024
Open
@stonea
add doc for gpuId
c2378e6 
---
Signed-off-by: Andy Stone <[email protected]>
@stonea stonea self-assigned this Sep 13, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@stonea stonea

Labels
area: GPU Support
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

Querying sublocale ID with the GPU locale model
1 participant
@stonea