-
Notifications
You must be signed in to change notification settings - Fork 88
/
FindROCTX.cmake
55 lines (50 loc) · 1.54 KB
/
FindROCTX.cmake
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
#.rst:
# FindROCTX
# -------
#
# Find the ROCTX headers and library, usually provided by ROCm.
#
# Imported targets
# ^^^^^^^^^^^^^^^^
#
# This module defines the following :prop_tgt:`IMPORTED` target:
#
# ``roc::roctx``
# The ROCTX library, if found.
#
# Result variables
# ^^^^^^^^^^^^^^^^
#
# This module will set the following variables in your project:
#
# ``ROCTX_INCLUDE_DIRS``
# where to find roctx.h
#
# ``ROCTX_LIBRARIES``
# the libraries to link against in order to use the ROCTX library.
#
# ``ROCTX_FOUND``
# If false, do not try to use the ROCTX library.
find_path(ROCTX_INCLUDE_DIR NAMES roctx.h HINTS ${ROCTRACER_PATH}/include ${ROCM_PATH}/include/roctracer)
mark_as_advanced(ROCTX_INCLUDE_DIR)
if(NOT ROCTX_LIBRARY)
find_library(ROCTX_LIBRARY NAMES roctx64 HINTS ${ROCTRACER_PATH}/lib ${ROCM_PATH}/lib)
endif()
include(FindPackageHandleStandardArgs)
find_package_handle_standard_args(ROCTX REQUIRED_VARS ROCTX_LIBRARY ROCTX_INCLUDE_DIR)
if(ROCTX_FOUND)
set(ROCTX_LIBRARIES ${ROCTX_LIBRARY})
set(ROCTX_INCLUDE_DIRS ${ROCTX_INCLUDE_DIR})
unset(ROCTX_LIBRARY)
unset(ROCTX_INCLUDE_DIR)
if(NOT TARGET roc::roctx)
add_library(roc::roctx UNKNOWN IMPORTED)
set_target_properties(roc::roctx PROPERTIES
INTERFACE_INCLUDE_DIRECTORIES "${ROCTX_INCLUDE_DIRS}")
if(EXISTS "${ROCTX_LIBRARIES}")
set_target_properties(roc::roctx PROPERTIES
IMPORTED_LINK_INTERFACE_LANGUAGES "C"
IMPORTED_LOCATION "${ROCTX_LIBRARIES}")
endif()
endif()
endif()