From 6bb807ef74252d1cca1d69c0ec7680972ff5ff6f Mon Sep 17 00:00:00 2001 From: George Wu Date: Mon, 1 Aug 2022 09:45:58 -0700 Subject: [PATCH] add cuda compute 8.7 to Cmakelists.txt to support Nvidia Orin devices (#12377) * add cuda arch 8.7 to cmakelists.txt to support Nvidia Orin devices * add cuda version >= 11 check for orin support --- cmake/CMakeLists.txt | 3 +++ 1 file changed, 3 insertions(+) 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)