Gabriele Cosmo
41 lines
1.1 KiB
CMake
41 lines
1.1 KiB
CMake
# Find the CUDA include directory and library.
|
|
#
|
|
# This module defines the `cuda` imported target that encodes all
|
|
# necessary information in its target properties.
|
|
#
|
|
# This package is necessary for GPU memory profiling
|
|
|
|
find_library(
|
|
Cuda_LIB
|
|
NAMES cudart
|
|
PATH_SUFFIXES lib lib32 lib64
|
|
DOC "Cuda Runtime Library required for GPU Memory usage info"
|
|
)
|
|
|
|
find_path(
|
|
Cuda_INCLUDE
|
|
NAMES cuda.h
|
|
PATH_SUFFIXES include
|
|
DOC "Cuda Include directory required for GPU Memory usage info"
|
|
)
|
|
|
|
find_path(
|
|
Cuda_Runtime_INCLUDE
|
|
NAMES cuda_runtime_api.h
|
|
PATH_SUFFIXES include
|
|
DOC "Cuda Runtime Include directory required for GPU Memory usage info"
|
|
)
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
find_package_handle_standard_args(
|
|
Cuda
|
|
REQUIRED_VARS Cuda_LIB Cuda_INCLUDE Cuda_Runtime_INCLUDE
|
|
)
|
|
|
|
add_library(Cuda SHARED IMPORTED)
|
|
set_property(TARGET Cuda PROPERTY IMPORTED_LOCATION ${Cuda_LIB})
|
|
set_property(TARGET Cuda PROPERTY INTERFACE_INCLUDE_DIRECTORIES ${Cuda_INCLUDE})
|
|
set_property(TARGET Cuda PROPERTY INTERFACE_INCLUDE_DIRECTORIES ${Cuda_Runtime_INCLUDE})
|
|
|
|
mark_as_advanced(Cuda_FOUND Cuda_LIB Cuda_INCLUDE Cuda_Runtime_INCLUDE)
|