diff --git a/cmake/CMakeLists.txt b/cmake/CMakeLists.txt index 304cf0c890..cb09051936 100644 --- a/cmake/CMakeLists.txt +++ b/cmake/CMakeLists.txt @@ -1801,6 +1801,9 @@ if (onnxruntime_USE_CUDA) set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode=arch=compute_53,code=sm_53") # TX1, Nano set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode=arch=compute_62,code=sm_62") # TX2 set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode=arch=compute_72,code=sm_72") # AGX Xavier, NX Xavier + if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 11) + set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -gencode=arch=compute_87,code=sm_87") # AGX Orin, NX Orin + endif() else() # the following compute capabilities are removed in CUDA 11 Toolkit if (CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 11)