# CMake script for finding AMGX
# Thomas Zwinger, CSC - IT Center for Science Ltd.
# 2020/03
#

#  AMGX_INCLUDE_DIR  - user modifiable choice of where to AMGX include dir
#  AMGX_LIBRARY      - user modifiable choice of where AMGX library is

# his module returns these variables for the rest of the project to use.
#
#  AMGX_FOUND              - True if AMGX found 
#  AMGX_LIBRARY           -  AMGX library
#  AMGX_INCLUDE_DIR       - AMGX include dir.
#  CUDA_LIBRARIES         - needed cuda libraries
#  CUDA_LIBDIR             - needed cuda library directory

#INCLUDE(${CMAKE_ROOT}/Modules/FindPackageHandleStandardArgs.cmake)

CMAKE_MINIMUM_REQUIRED(VERSION 2.8)

MESSAGE ("----------------------")
MESSAGE ("-- AMGX + CUDA:")
FIND_PACKAGE(CUDA)
MESSAGE("-- Cuda libraries: " ${CUDA_LIBRARIES})

SET(AMGX_FOUND FALSE)

FIND_PATH(AMGX_INCLUDE_DIR amgx_c.h
  HINTS 
  "${AMGXINCLUDE}" "${AMGX_ROOT}/include"
  )




FIND_LIBRARY(AMGX_LIBRARY
  NAMES amgx libamgx.a
  NAMES_PER_DIR
  HINTS "${AMGX_ROOT}/lib" "${AMGXLIB}"
  REQUIRED
  )

IF (AMGX_LIBRARY AND AMGX_INCLUDE_DIR)
  SET (AMGX_FOUND TRUE)
ENDIF()


IF (AMGX_FOUND)
  SET(AMGX_INCLUDE_DIRS "${AMGX_INCLUDE_DIR}")
  SET(CUDA_LIBRARIES ${CUDA_LIBRARIES} ${CUDA_cusparse_LIBRARY} ${CUDA_cublas_LIBRARY} ${CUDA_cusolver_LIBRARY} ${CUDA_nvToolsExt_LIBRARY})
  SET(AMGX_LIBRARIES ${AMGX_LIBRARY} ${CUDA_LIBRARIES})
  MESSAGE ("-- AMGX found")
  MESSAGE ("-- AMGX_ROOT= ${AMGX_ROOT}")
  MESSAGE ("----------------------")
ELSE()
    MESSAGE (FATAL_ERROR, "AMGX not found")
ENDIF()

MARK_AS_ADVANCED(
  AMGX_FOUND
  AMGX_INCLUDE_DIR
  AMGX_LIBRARY
  CUDA_LIBRARIES
  CUDA_LIBDIR)