nativesdk-cuda-environment: add settings for FindCUDAToolkit

cmake's FindCUDAToolkit module properly locates libraries in
our split host/target sysroot setup during bitbake builds, but
due to differences in the SDK environment, it does not.  Fix
this by explicitly setting CUDATookit_TARGET_DIR and priming
the settings for the cudart library in the toolchain
file we add.

Signed-off-by: Matt Madison <matt@madison.systems>
This commit is contained in:
Matt Madison
2023-04-11 06:40:42 -07:00
committed by Matt Madison
parent 8ea5b73309
commit bef4d30e05

View File

@@ -3,3 +3,5 @@ set(CUDA_TOOLKIT_TARGET_DIR "${CMAKE_SYSROOT}/usr/local/cuda-$ENV{CUDA_VERSION}"
set(CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES "${CUDA_TOOLKIT_ROOT_DIR}/include" "${CUDA_TOOLKIT_TARGET_DIR}/include" CACHE STRING "" FORCE)
set(CMAKE_CUDA_ARCHITECTURES "$ENV{CUDA_ARCHITECTURES}")
list(APPEND CMAKE_MODULE_PATH "${CMAKE_SYSROOT}/usr/share/cmake/Modules")
set(CUDAToolkit_TARGET_DIR "${CUDA_TOOLKIT_TARGET_DIR}" CACHE PATH "" FORCE)
find_library(CUDA_CUDART NAMES cudart HINTS ${CUDA_TOOLKIT_TARGET_DIR} PATH_SUFFIXES lib64 lib)