42 lines
1.2 KiB
CMake
42 lines
1.2 KiB
CMake
#@HEADER
|
|
# ************************************************************************
|
|
#
|
|
# Kokkos v. 4.0
|
|
# Copyright (2022) National Technology & Engineering
|
|
# Solutions of Sandia, LLC (NTESS).
|
|
#
|
|
# Under the terms of Contract DE-NA0003525 with NTESS,
|
|
# the U.S. Government retains certain rights in this software.
|
|
#
|
|
# Part of Kokkos, under the Apache License v2.0 with LLVM Exceptions.
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
#
|
|
# ************************************************************************
|
|
# @HEADER
|
|
|
|
# Check for CUDA support
|
|
|
|
SET(_CUDA_FAILURE OFF)
|
|
|
|
# Have CMake find CUDA
|
|
IF(NOT _CUDA_FAILURE)
|
|
FIND_PACKAGE(CUDA 3.2)
|
|
IF (NOT CUDA_FOUND)
|
|
SET(_CUDA_FAILURE ON)
|
|
ENDIF()
|
|
ENDIF()
|
|
|
|
IF(NOT _CUDA_FAILURE)
|
|
# if we haven't met failure
|
|
macro(PACKAGE_ADD_CUDA_LIBRARY cuda_target)
|
|
TRIBITS_ADD_LIBRARY(${cuda_target} ${ARGN} CUDALIBRARY)
|
|
endmacro()
|
|
GLOBAL_SET(TPL_CUDA_LIBRARY_DIRS)
|
|
GLOBAL_SET(TPL_CUDA_INCLUDE_DIRS ${CUDA_TOOLKIT_INCLUDE})
|
|
GLOBAL_SET(TPL_CUDA_LIBRARIES ${CUDA_CUDART_LIBRARY} ${CUDA_cublas_LIBRARY} ${CUDA_cufft_LIBRARY})
|
|
KOKKOS_CREATE_IMPORTED_TPL_LIBRARY(CUSPARSE)
|
|
ELSE()
|
|
SET(TPL_ENABLE_CUDA OFF)
|
|
ENDIF()
|