1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
# - Find CuBlas library
#
# This module defines
# CUBBLAS_LIBRARY, libraries to link against to use Cublas.
# CUBLAS_FOUND, If false, do not try to use Cublas.
FIND_LIBRARY(CUBLAS_LIBRARY NAMES cublas libcublas PATHS /usr/local/cuda/lib64 )
FIND_LIBRARY(CUDART_LIBRARY NAMES cudart libcudart PATHS /usr/local/cuda/lib64 )
# handle the QUIETLY and REQUIRED arguments and set LOKI_FOUND to TRUE if
# all listed variables are TRUE
INCLUDE(FindPackageHandleStandardArgs)
FIND_PACKAGE_HANDLE_STANDARD_ARGS(CuBlas DEFAULT_MSG CUBLAS_LIBRARY CUDART_LIBRARY)
MARK_AS_ADVANCED(CUBLAS_LIBRARY CUDART_LIBRARY)
|