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.
|
# Support Ampere GPUs.
|
||||||
list(APPEND CMAKE_CUDA_ARCHITECTURES "80")
|
list(APPEND CMAKE_CUDA_ARCHITECTURES "80")
|
||||||
endif(CUDAToolkit_VERSION VERSION_GREATER_EQUAL "11.0")
|
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.
|
# Support Hopper GPUs.
|
||||||
list(APPEND CMAKE_CUDA_ARCHITECTURES "90")
|
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}")
|
message("-- Setting CUDA Architecture to ${CMAKE_CUDA_ARCHITECTURES}")
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user