Update to remove not supported archs

triton-inference-server · Jan 13, 2025 · 3057190 · 3057190
1 parent 87b5771
commit 3057190
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -121,7 +121,7 @@ if(${TRITON_ENABLE_GPU})
 
   enable_language(CUDA)
   set_target_properties(kernel_library_new PROPERTIES LANGUAGE CUDA)
-  set_target_properties(kernel_library_new PROPERTIES CUDA_ARCHITECTURES "72-real;75-real;80-real;86-real;89-real;90-real;100-real;120-real")
+  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")
   set_target_properties(kernel_library_new PROPERTIES POSITION_INDEPENDENT_CODE ON)
   set_target_properties(kernel_library_new PROPERTIES LINKER_LANGUAGE CUDA)
   target_compile_features(kernel_library_new PUBLIC cxx_std_${TRITON_MIN_CXX_STANDARD})