diff options
| author | Sergey Kozub <skozub@nvidia.com> | 2025-01-21 11:00:24 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-01-21 11:00:24 +0100 |
| commit | 616979ebd7dc9ae63522788750ea3dc6a96aa69f (patch) | |
| tree | 470e63784f436e66c67d0efea82de31c3125b86e /clang/lib/Basic/Cuda.cpp | |
| parent | 9b6e8df89689c37821e0ec37c63ac3fd40ea5339 (diff) | |
| download | llvm-616979ebd7dc9ae63522788750ea3dc6a96aa69f.zip llvm-616979ebd7dc9ae63522788750ea3dc6a96aa69f.tar.gz llvm-616979ebd7dc9ae63522788750ea3dc6a96aa69f.tar.bz2 | |
[NVPTX] Add support for PTX 8.6 and CUDA 12.6 (12.8) (#123398)
Add CUDA versions 12.7, 12.8, 12.9 which support PTX8.6+ (enables using Blackwell-specific instructions).
Diffstat (limited to 'clang/lib/Basic/Cuda.cpp')
| -rw-r--r-- | clang/lib/Basic/Cuda.cpp | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/clang/lib/Basic/Cuda.cpp b/clang/lib/Basic/Cuda.cpp index d56609a..b146142 100644 --- a/clang/lib/Basic/Cuda.cpp +++ b/clang/lib/Basic/Cuda.cpp @@ -44,6 +44,9 @@ static const CudaVersionMapEntry CudaNameVersionMap[] = { CUDA_ENTRY(12, 4), CUDA_ENTRY(12, 5), CUDA_ENTRY(12, 6), + CUDA_ENTRY(12, 7), + CUDA_ENTRY(12, 8), + CUDA_ENTRY(12, 9), {"", CudaVersion::NEW, llvm::VersionTuple(std::numeric_limits<int>::max())}, {"unknown", CudaVersion::UNKNOWN, {}} // End of list tombstone. }; @@ -98,6 +101,7 @@ static const OffloadArchToStringMap arch_names[] = { SM(90), // Hopper SM(90a), // Hopper SM(100), // Blackwell + SM(100a), // Blackwell GFX(600), // gfx600 GFX(601), // gfx601 GFX(602), // gfx602 @@ -227,8 +231,8 @@ CudaVersion MinVersionForOffloadArch(OffloadArch A) { case OffloadArch::SM_90a: return CudaVersion::CUDA_120; case OffloadArch::SM_100: - return CudaVersion::NEW; // TODO: use specific CUDA version once it's - // public. + case OffloadArch::SM_100a: + return CudaVersion::CUDA_127; default: llvm_unreachable("invalid enum"); } |