-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCMakeLists.txt
118 lines (94 loc) · 3.91 KB
/
CMakeLists.txt
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
cmake_minimum_required(VERSION 3.13)
project (libinternal_energy VERSION 0.0.1)
include(FeatureSummary)
include(CMakeDependentOption)
include(FindPackageHandleStandardArgs)
if(NOT INTERNAL_ENERGY_POSITION_INDEPENDENT_CODE)
set(INTERNAL_ENERGY_POSITION_INDEPENDENT_CODE OFF)
set(PERF_CPP_POSITION_INDEPENDENT_CODE OFF)
else()
set(PERF_CPP_POSITION_INDEPENDENT_CODE ON)
endif()
add_subdirectory(lib/perf-cpp)
set(CMAKE_MODULE_PATH "${CMAKE_SOURCE_DIR}/cmake;${CMAKE_MODULE_PATH}")
find_package(VEDA)
find_package(CUDAToolkit)
find_package(Threads)
find_library(ROCM_SMI_LIBRARY NAMES rocm_smi64
HINTS "/opt/rocm/lib")
find_path(ROCM_SMI_INCLUDE_DIR NAMES rocm_smi/rocm_smi.h
HINTS "/opt/rocm/include")
find_package_handle_standard_args(ROCM_SMI REQUIRED_VARS ROCM_SMI_LIBRARY)
if (ROCM_SMI_FOUND)
add_library(ROCM_SMI::ROCM_SMI UNKNOWN IMPORTED)
set_property(TARGET ROCM_SMI::ROCM_SMI PROPERTY IMPORTED_LOCATION ${ROCM_SMI_LIBRARY})
target_include_directories(ROCM_SMI::ROCM_SMI INTERFACE ${ROCM_SMI_INCLUDE_DIR})
endif()
option(INTERNAL_ENERGY_POSITION_INDEPENDENT_CODE "Whether to build libinternal_energy with position independent code" ${INTERNAL_ENERGY_POSITION_INDEPENDENT_CODE})
CMAKE_DEPENDENT_OPTION(USE_NVML "Use nvml to retrieve Nvidia GPU energy values" ON "CUDAToolkit_FOUND" OFF)
add_feature_info("USE_NVML" USE_NVML "Use nvml to retrieve Nvidia GPU energy values")
CMAKE_DEPENDENT_OPTION(USE_NEC "Use libveda to retrieve NEC Vector Engine energy values" ON "VEDA_FOUND" OFF)
add_feature_info("USE_NEC" USE_NEC "Use libveda to retrieve NEC Vector Engine energy values")
CMAKE_DEPENDENT_OPTION(USE_ROCM_SMI "Use rocm-smi to retrive AMDGPU energy values" ON "ROCM_SMI_FOUND" OFF)
add_feature_info("USE_ROCM_SMI" USE_ROCM_SMI "Use rocm-smi to retrieve AMDGPU energy values")
set(LIB_SRCS
src/perf.cpp
src/topology.cpp
src/energy.cpp
src/hwmon/hwmon.cpp)
add_library(internal_energy SHARED ${LIB_SRCS})
target_link_libraries(internal_energy PUBLIC perf-cpp)
target_link_libraries(internal_energy PUBLIC Threads::Threads)
target_link_libraries(internal_energy PUBLIC stdc++fs)
target_include_directories(internal_energy
PUBLIC
$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
)
set_property(TARGET internal_energy PROPERTY CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON) #...is required...
set(CMAKE_CXX_EXTENSIONS OFF) #...without compiler extensions like gnu++11
set_target_properties(internal_energy PROPERTIES
POSITION_INDEPENDENT_CODE ${INTERNAL_ENERGY_POSITION_INDEPENDENT_CODE}
)
if (${INTERNAL_ENERGY_POSITION_INDEPENDENT_CODE})
set(PERF_CPP_POSITION_INDEPENDENT_CODE)
endif()
if (USE_NVML)
if(CUDAToolkit_FOUND)
target_compile_definitions(internal_energy PUBLIC HAVE_NVML)
target_link_libraries(internal_energy PUBLIC CUDA::nvml)
else()
message(SEND_ERROR "NVML not found but requested")
endif()
endif()
if (USE_NEC)
if(VEDA_FOUND)
target_compile_definitions(internal_energy PUBLIC HAVE_NEC)
target_link_libraries(internal_energy PUBLIC VEDA::veda)
else()
message(SEND_ERROR "VEDA not found but requested")
endif()
endif()
if(USE_ROCM_SMI)
if(ROCM_SMI_FOUND)
target_compile_definitions(internal_energy PUBLIC HAVE_ROCM_SMI)
target_link_libraries(internal_energy PUBLIC ROCM_SMI::ROCM_SMI)
else()
message(SEND_ERROR "ROCM_SMI not found but requested")
endif()
endif()
add_executable(hwmon_test examples/hwmon_test.cpp)
target_link_libraries(hwmon_test PRIVATE internal_energy)
set_property(TARGET hwmon_test PROPERTY CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON) #...is required...
set(CMAKE_CXX_EXTENSIONS OFF) #...without compiler extensions like gnu++11
install(DIRECTORY
${CMAKE_CURRENT_SOURCE_DIR}/include/internal_energy
DESTINATION include)
install(
TARGETS
internal_energy
LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
INCLUDES DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}
)