File tree Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -109,6 +109,10 @@ function(get_nvptx_compile_options output_var gpu_arch)
109
109
list (APPEND nvptx_options "--cuda-feature=+ptx72" )
110
110
elseif (${gpu_arch} STREQUAL "sm_86" )
111
111
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" )
112
116
else ()
113
117
message (FATAL_ERROR "Unknown Nvidia GPU architecture '${gpu_arch} '" )
114
118
endif ()
Original file line number Diff line number Diff line change @@ -9,7 +9,7 @@ set(all_amdgpu_architectures "gfx700;gfx701;gfx801;gfx803;gfx900;gfx902;gfx906"
9
9
"gfx1031;gfx1032;gfx1033;gfx1034;gfx1035;gfx1036"
10
10
"gfx1100;gfx1101;gfx1102;gfx1103;gfx1150;gfx1151" )
11
11
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 " )
13
13
set (all_gpu_architectures
14
14
"${all_amdgpu_architectures} ;${all_nvptx_architectures} " )
15
15
set (LIBC_GPU_ARCHITECTURES "all" CACHE STRING
You can’t perform that action at this time.
0 commit comments