mirror of
https://github.com/ceres-solver/ceres-solver.git
synced 2026-08-29 08:34:37 +08:00
Expose Hopper (sm_90) support only for CUDA 12.0 and later (fixes #1165)
Change-Id: I1e0fdbabdd866494051ed21b7c456ece82a4d6a3
This commit is contained in:
+2
-2
@@ -345,10 +345,10 @@ if (USE_CUDA)
|
||||
# Support Ampere GPUs.
|
||||
list(APPEND CMAKE_CUDA_ARCHITECTURES "80")
|
||||
endif(CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.0")
|
||||
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8")
|
||||
if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.0")
|
||||
# Support Hopper GPUs.
|
||||
list(APPEND CMAKE_CUDA_ARCHITECTURES "90")
|
||||
endif(CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.8")
|
||||
endif(CUDAToolkit_VERSION VERSION_GREATER_EQUAL "12.0")
|
||||
message("-- Setting CUDA Architecture to ${CMAKE_CUDA_ARCHITECTURES}")
|
||||
endif()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user