diff options
| author | Artem Belevich <tra@google.com> | 2025-07-31 17:21:40 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-07-31 17:21:40 -0700 |
| commit | 4e596fc2852eddc23fc0cd287eb5cd8e4664855b (patch) | |
| tree | 70a636372c4b1f4c9854ee82735e3053728aec01 /llvm/lib/Object/ELFObjectFile.cpp | |
| parent | 92ca087b456c6cb1b1005e7d225bdd4ad6f6eeec (diff) | |
| download | llvm-4e596fc2852eddc23fc0cd287eb5cd8e4664855b.zip llvm-4e596fc2852eddc23fc0cd287eb5cd8e4664855b.tar.gz llvm-4e596fc2852eddc23fc0cd287eb5cd8e4664855b.tar.bz2 | |
[ELF] handle new NVIDIA GPU variants. (#151604)
Diffstat (limited to 'llvm/lib/Object/ELFObjectFile.cpp')
| -rw-r--r-- | llvm/lib/Object/ELFObjectFile.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/llvm/lib/Object/ELFObjectFile.cpp b/llvm/lib/Object/ELFObjectFile.cpp index 0919c6a..aff047c 100644 --- a/llvm/lib/Object/ELFObjectFile.cpp +++ b/llvm/lib/Object/ELFObjectFile.cpp @@ -688,11 +688,20 @@ StringRef ELFObjectFileBase::getNVPTXCPUName() const { case ELF::EF_CUDA_SM100: return getPlatformFlags() & ELF::EF_CUDA_ACCELERATORS ? "sm_100a" : "sm_100"; + case ELF::EF_CUDA_SM101: + return getPlatformFlags() & ELF::EF_CUDA_ACCELERATORS ? "sm_101a" + : "sm_101"; + case ELF::EF_CUDA_SM103: + return getPlatformFlags() & ELF::EF_CUDA_ACCELERATORS ? "sm_103a" + : "sm_103"; // Rubin architecture. case ELF::EF_CUDA_SM120: return getPlatformFlags() & ELF::EF_CUDA_ACCELERATORS ? "sm_120a" : "sm_120"; + case ELF::EF_CUDA_SM121: + return getPlatformFlags() & ELF::EF_CUDA_ACCELERATORS ? "sm_121a" + : "sm_121"; default: llvm_unreachable("Unknown EF_CUDA_SM value"); } |