Skip to content

Commit 65642c7

Browse files
committed
[libc][NFC] Support sm_89 and sm_90 NVIDIA GPUs
Summary: These architectures were left out from the list and should be supported.
1 parent fd66195 commit 65642c7

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

libc/cmake/modules/LLVMLibCObjectRules.cmake

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,10 @@ function(get_nvptx_compile_options output_var gpu_arch)
109109
list(APPEND nvptx_options "--cuda-feature=+ptx72")
110110
elseif(${gpu_arch} STREQUAL "sm_86")
111111
list(APPEND nvptx_options "--cuda-feature=+ptx72")
112+
elseif(${gpu_arch} STREQUAL "sm_89")
113+
list(APPEND nvptx_options "--cuda-feature=+ptx72")
114+
elseif(${gpu_arch} STREQUAL "sm_90")
115+
list(APPEND nvptx_options "--cuda-feature=+ptx72")
112116
else()
113117
message(FATAL_ERROR "Unknown Nvidia GPU architecture '${gpu_arch}'")
114118
endif()

libc/cmake/modules/prepare_libc_gpu_build.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ set(all_amdgpu_architectures "gfx700;gfx701;gfx801;gfx803;gfx900;gfx902;gfx906"
99
"gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036"
1010
"gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151")
1111
set(all_nvptx_architectures "sm_35;sm_37;sm_50;sm_52;sm_53;sm_60;sm_61;sm_62"
12-
"sm_70;sm_72;sm_75;sm_80;sm_86")
12+
"sm_70;sm_72;sm_75;sm_80;sm_86;sm_89;sm_90")
1313
set(all_gpu_architectures
1414
"${all_amdgpu_architectures};${all_nvptx_architectures}")
1515
set(LIBC_GPU_ARCHITECTURES "all" CACHE STRING

0 commit comments

Comments
 (0)