From 6c7859510d6635b44e10c9e136db283d87299bfa Mon Sep 17 00:00:00 2001 From: Yosshi999 Date: Wed, 15 Jun 2022 20:01:36 +0900 Subject: [PATCH] Support compute capability 87 archs such as Orin (#601) --- cmake/cuda.cmake | 3 +++ 1 file changed, 3 insertions(+) diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index 3e9a9bd98..ec16fbe9c 100644 --- a/cmake/cuda.cmake +++ b/cmake/cuda.cmake @@ -56,6 +56,9 @@ if (NOT CMAKE_CUDA_ARCHITECTURES) # cuda doesn't support `sm_86` until version 11.1 set(_NVCC_FLAGS "${_NVCC_FLAGS} -gencode arch=compute_86,code=sm_86") endif () + if (CUDA_VERSION_MINOR VERSION_GREATER_EQUAL "4") + set(_NVCC_FLAGS "${_NVCC_FLAGS} -gencode arch=compute_87,code=sm_87") + endif () endif () endif ()