mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Support compute capability 87 archs such as Orin (#601)

pull/605/head
Yosshi999 2022-06-15 20:01:36 +09:00 committed by GitHub
parent 28beb9a481
commit 6c7859510d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cmake/cuda.cmake
View File

@ -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 ()