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

CUDA: Introduce arm64 subarchitecture. #8622

Merged
merged 4 commits into from
May 14, 2024
Merged

CUDA: Introduce arm64 subarchitecture. #8622

merged 4 commits into from
May 14, 2024

Conversation

maleadt
Copy link
Contributor

@maleadt maleadt commented May 6, 2024

This PR introduces yet another CUDA-specific platform tag, "cuda_platform", to differentiate between SBSA and Jetson on aarch64. Jetson binaries are only available on the redist server starting with CUDA 11.8, so I've removed support for older versions.

@stemann Sorry for removing CUDA 10.2 support again, but I haven't seen any changes in CUDA.jl so I figured you weren't interested in that anymore. If you are, it would be relatively easy to get your 10.2 build support compatible with this change: It only requires adding a regular runfile path for SBSA in the 10.2 SDK JLL, keeping your apt-based ones for the Jetson platform.

Changes like this are always fragile and very hard to test, so I expect breakage...

@maleadt maleadt added the cuda 🕹️ Builders related to Nvidia CUDA label May 6, 2024
@maleadt maleadt force-pushed the tb/cuda_platform branch 2 times, most recently from c05a8a9 to f264e8f Compare May 6, 2024 20:29
@maleadt maleadt marked this pull request as draft May 7, 2024 07:49
@stemann
Copy link
Contributor

stemann commented May 7, 2024

@maleadt Thanks for the ping - highly appreciated. It would be nice to keep 10.2 functional to to be able build "ancient" stuff like Torch_jll v1.10.2 (using the original binary dependencies, CUDA 10.2 and 11.3): FluxML/Torch.jl#61

... and yeah, I know - 11.3 is also not available since CUDA_SDK... :-)

@maleadt maleadt marked this pull request as ready for review May 14, 2024 11:26
@maleadt
Copy link
Contributor Author

maleadt commented May 14, 2024

I've added the old SDKs etc back, just only providing SBSA. 10.2 is special, only providing Tegra. @stemann I'll leave the generalization to SBSA up to you, if you're interested.

@maleadt maleadt merged commit 3de8a79 into master May 14, 2024
95 checks passed
@maleadt maleadt deleted the tb/cuda_platform branch May 14, 2024 12:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cuda 🕹️ Builders related to Nvidia CUDA
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants