FindCUDA: Cache CUDA_TOOLKIT_ROOT_DIR when cross-compiling

Previously we cached `CUDA_TOOLKIT_ROOT_DIR` when building for the host,
but not when cross-compiling.  Make it consistent.
This commit is contained in:
Jinzhe Zeng
2024-03-09 17:19:12 -05:00
committed by Brad King
parent 168672581d
commit 6e8f64950b

View File

@@ -908,7 +908,7 @@ if(CMAKE_CROSSCOMPILING)
if (EXISTS "${CUDA_TOOLKIT_ROOT}/targets/${CUDA_TOOLKIT_TARGET_NAME}")
set(CUDA_TOOLKIT_TARGET_DIR "${CUDA_TOOLKIT_ROOT}/targets/${CUDA_TOOLKIT_TARGET_NAME}" CACHE PATH "CUDA Toolkit target location.")
SET (CUDA_TOOLKIT_ROOT_DIR ${CUDA_TOOLKIT_ROOT})
SET (CUDA_TOOLKIT_ROOT_DIR ${CUDA_TOOLKIT_ROOT} CACHE PATH "Toolkit location." FORCE)
mark_as_advanced(CUDA_TOOLKIT_TARGET_DIR)
endif()