Skip to content

Commit 3d51ede

Browse files
committed
Update the CUDA_ARCHITECTURES
1 parent 3057190 commit 3d51ede

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ if(${TRITON_ENABLE_GPU})
121121

122122
enable_language(CUDA)
123123
set_target_properties(kernel_library_new PROPERTIES LANGUAGE CUDA)
124-
set_target_properties(kernel_library_new PROPERTIES CUDA_ARCHITECTURES "53;62-real;72-real;75-real;80-real;86-real;89-real;90-real;100-real;120-real")
124+
set_target_properties(kernel_library_new PROPERTIES CUDA_ARCHITECTURES "75-real;80-real;86-real;87-real;89-real;90-real;100-real;101-real;120-real")
125125
set_target_properties(kernel_library_new PROPERTIES POSITION_INDEPENDENT_CODE ON)
126126
set_target_properties(kernel_library_new PROPERTIES LINKER_LANGUAGE CUDA)
127127
target_compile_features(kernel_library_new PUBLIC cxx_std_${TRITON_MIN_CXX_STANDARD})

0 commit comments

Comments
 (0)