Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# ==================================================================================================
# This file is part of the CodeVault project. The project is licensed under Apache Version 2.0.
# CodeVault is part of the EU-project PRACE-4IP (WP7.3.C).
#
# Author(s):
# Valeriu Codreanu <valeriu.codreanu@surfsara.nl>
#
# ==================================================================================================
cmake_minimum_required(VERSION 2.8.10 FATAL_ERROR)
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/../../../../cmake/Modules")
# Packages are optional: if they are not present, certain code samples are not compiled
find_package(OpenMP) # Built-in in CMake
find_package(MPI) # Built-in in CMake
find_package(CUDA) # Built-in in CMake
find_package(MKL)
include(${CMAKE_CURRENT_SOURCE_DIR}/../../../../cmake/common.cmake)
# ==================================================================================================
if ("${DWARF_PREFIX}" STREQUAL "")
set(DWARF_PREFIX 1_dense)
endif()
set(NAME ${DWARF_PREFIX}_lud_cusolver_mkl)
# ==================================================================================================
# C++ compiler settings
find_package(Common)
select_compiler_flags(cxx_flags
GNU "-march=native" # I suggest remove "-O3" as this is controlled by the CMAKE_BUILD_TYPE
CLANG "-march=native" # same here
Intel "-axavx2,avx")
set(CXX_FLAGS ${cxx_flags})
if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU")
set(CXX_FLAGS "${CXX_FLAGS} -Wall -Wno-comment")
if(APPLE)
set(CXX_FLAGS "${CXX_FLAGS} -Wa,-q")
endif()
endif()
if (OPENMP_FOUND)
set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
endif()
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${CXX_FLAGS}")
# NVCC compiler settings
if (CUDA_FOUND)
set(CUDA_PROPAGATE_HOST_FLAGS OFF)
set(CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -O3")
set(CUDA_HOST_COMPILER "g++")
endif()
# ==================================================================================================
# LUD with the CUDA cuSOLVER and MKL libraries
if (CUDA_FOUND)
if (NOT ${MKL_FOUND})
message("## Skipping '${NAME}': no MKL support found")
install(CODE "MESSAGE(\"${NAME} can only be built with MKL.\")")
else (NOT ${MKL_FOUND})
if (NOT CUDA_cusolver_LIBRARY)
message("## Skipping '${NAME}': no cuSOLVER support found")
dummy_install(${NAME} "CUDA CUSOLVER library")
return()
endif()
include_directories(${MKL_INCLUDE_DIR})
link_directories(${MKL_LIBRARY_DIR})
cuda_add_executable(${NAME} src/lud_cusolver_mkl.cpp)
target_link_libraries(${NAME} ${CUDA_cusolver_LIBRARY})
target_link_libraries(${NAME} mkl_intel_lp64 mkl_sequential mkl_core)
install(TARGETS ${NAME} DESTINATION bin)
if (OPENMP_FOUND)
message("** Enabling '${NAME}': with OpenMP")
else()
message("** Enabling '${NAME}': without OpenMP")
endif()
endif()
else(CUDA_FOUND)
message("## Skipping '${NAME}': no CUDA found")
install(CODE "MESSAGE(\"${NAME} can only be built with CUDA.\")")
endif()
unset(NAME)
# ==================================================================================================