Skip to content

[NVIDIA] Refactor Family Blackwell Support codegen #156176

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
Open
wants to merge 7 commits into
base: main
Choose a base branch
from
Open

[NVIDIA] Refactor Family Blackwell Support codegen #156176

wants to merge 7 commits into from
+4 −4

Conversation

johnnynunez
Copy link
Contributor

@johnnynunez johnnynunez commented Jun 17, 2025 •
edited
Loading

With the legacy driver (nvgpu) used for CUDA 12.9, Thor was operating with SM 10.1.
This changes to SM 11.0 when the newer driver model (OpenRM), which is intended for CUDA 13.0, is introduced.
Thor 10.1 --> 11.0
Spark 12.1

@pytorch-bot PyTorch Bot
Copy link

pytorch-bot bot commented Jun 17, 2025 •
edited
Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/156176

Note: Links to docs will display an error until the docs builds have been completed.

✅ No Failures

As of commit d2d2970 with merge base f990490 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@johnnynunez
Copy link
Contributor Author

@pytorchbot label "topic: not user facing"

@pytorch-bot pytorch-bot bot added the topic: not user facing topic category label Jun 17, 2025
@colesbury colesbury added the triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module label Jun 17, 2025
@colesbury colesbury requested a review from a team June 17, 2025 12:37
@ezyang
Copy link
Contributor

ezyang commented Jun 19, 2025

@cyyever, as author of #154595, I'll let you decide what to do about this one

@soumith soumith removed their request for review June 19, 2025 20:24
@cyyever
Copy link
Collaborator

cyyever commented Jun 22, 2025

I see the need to extend select_compute_arch.cmake even after the forked CUDA modules have been removed. Can you wait until I have fixed all build failures and merged #154595, then we can discuss how to extend?

@cyyever
Copy link
Collaborator

cyyever commented Jun 22, 2025 •
edited
Loading

@johnnynunez Could you try modifying torch_cuda_select_nvcc_arch_flags of cmake/public/utils.cmake and appending to CUDA_ARCHITECTURES?

@johnnynunez johnnynunez mentioned this pull request Jun 27, 2025
Open
@johnnynunez
Copy link
Contributor Author

johnnynunez commented Jul 2, 2025 •
edited
Loading

@johnnynunez Could you try modifying torch_cuda_select_nvcc_arch_flags of cmake/public/utils.cmake and appending to CUDA_ARCHITECTURES?

I think that i have to wait, right?

@johnnynunez johnnynunez requested a review from malfet as a code owner July 2, 2025 20:47
@johnnynunez johnnynunez mentioned this pull request Jul 13, 2025
Open
@rec rec temporarily deployed to upload-stats August 11, 2025 11:24 — with GitHub Actions Inactive
@johnnynunez johnnynunez reopened this Aug 12, 2025
@johnnynunez
Copy link
Contributor Author

johnnynunez commented Aug 12, 2025 •
edited
Loading

@tinglvv @malfet @atalman could you merge it? 10.1 not exists anymore in cuda 13. Thor is 11.0

it’s building for me https://pypi.jetson-ai-lab.io/sbsa/cu130/torch/2.9.0

@johnnynunez johnnynunez mentioned this pull request Aug 12, 2025
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@fmassa fmassa Awaiting requested review from fmassa fmassa is a code owner

@ezyang ezyang Awaiting requested review from ezyang ezyang is a code owner

@malfet malfet Awaiting requested review from malfet malfet is a code owner

Assignees
No one assigned
Labels
open source topic: not user facing topic category triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.