Fix CUDA architectures

triton-inference-server · May 24, 2024 · d030469 · d030469
1 parent 42c6e17
commit d030469
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -120,7 +120,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 "53;60;61;62;70;72;75")
+  set_target_properties(kernel_library_new PROPERTIES CUDA_ARCHITECTURES 53 60-real 62-real 70-real 72-real 75-real 80-real 86-real 89-real 90-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})