Skip to content

Commit 9b82f8e

Browse files
Add support NVTX3 for CMake versions >3.25
1 parent 6117c1e commit 9b82f8e

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

cmake/FindNVTX.cmake

+19-2
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,25 @@ if(CMAKE_C_COMPILER_ID STREQUAL "PGI" OR CMAKE_C_COMPILER_ID STREQUAL "NVHPC" )
4040
)
4141
else()
4242
find_package(CUDAToolkit REQUIRED COMPONENTS CUDA::nvtx3)
43-
set (NVTX_LIBRARY "CUDA::nvtx3")
44-
#TODO: Implementation for NVTX3
43+
44+
find_path(NVTX_ROOT
45+
NAMES include/nvtx3/nvToolsExt.h
46+
HINTS ${CUDAToolkit_LIBRARY_DIR}/..
47+
)
48+
49+
find_path(NVTX_INCLUDE_DIRS
50+
NAMES nvToolsExt.h
51+
HINTS ${NVTX_ROOT}/include/nvtx3
52+
)
53+
54+
include(FindPackageHandleStandardArgs)
55+
find_package_handle_standard_args(NVTX DEFAULT_MSG
56+
NVTX_INCLUDE_DIRS
57+
)
58+
59+
mark_as_advanced(
60+
NVTX_INCLUDE_DIRS
61+
)
4562
endif()
4663

4764

0 commit comments

Comments
 (0)