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 NVIDIA Blackwell #6991

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

Conversation

fabiendupont
Copy link

NVIDIA Blackwell GPU generation has number 10. The SM code and architecture should be 100, but the current code generates 1., because it expects a 2 characters string.

This change modifies the logic to consider it as a string that contains a ., hence splits the string and uses the array of strings.

@fabiendupont
Copy link
Author

@microsoft-github-policy-service agree company="Red Hat"

NVIDIA Blackwell GPU generation has number 10. The SM code and
architecture should be `100`, but the current code generates `1.`,
because it expects a 2 characters string.

This change modifies the logic to consider it as a string that
contains a `.`, hence splits the string and uses the array of strings.

Signed-off-by: Fabien Dupont <[email protected]>
@fabiendupont fabiendupont force-pushed the allow-nvidia-blackwell branch from 60fd1d9 to d63a324 Compare February 3, 2025 17:42
@tjruwase tjruwase added this pull request to the merge queue Feb 4, 2025
@github-merge-queue github-merge-queue bot removed this pull request from the merge queue due to failed status checks Feb 4, 2025
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.

3 participants