# SPDX-FileCopyrightText: Copyright Contributors to the Kokkos project
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

# 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})
else()
  set(TPL_ENABLE_CUDA OFF)
endif()