at master 1.1 kB view raw
1diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake 2index a93386c27f8..7c6b98006bf 100644 3--- a/cmake/Dependencies.cmake 4+++ b/cmake/Dependencies.cmake 5@@ -967,7 +967,7 @@ endif() 6 7 # ---[ nvtx 8 if(USE_SYSTEM_NVTX) 9- find_path(nvtx3_dir NAMES nvtx3 PATHS ${CUDA_INCLUDE_DIRS}) 10+ find_path(nvtx3_dir NAMES nvtx3/nvtx3.hpp PATHS ${CUDA_INCLUDE_DIRS}) 11 find_package_handle_standard_args(nvtx3 DEFAULT_MSG nvtx3_dir) 12 if(NOT nvtx3_FOUND) 13 message(WARNING "Cannot find system NVTX3, find shipped NVTX3 instead") 14@@ -977,9 +977,10 @@ if(NOT TARGET CUDA::nvtx3) 15 add_library(CUDA::nvtx3 INTERFACE IMPORTED) 16 endif() 17 if(NOT nvtx3_dir) 18- find_path(nvtx3_dir NAMES nvtx3 PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH) 19- target_include_directories(CUDA::nvtx3 INTERFACE "${nvtx3_dir}") 20+ find_path(nvtx3_dir NAMES nvtx3/nvtx3.hpp PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH) 21 endif() 22+target_include_directories(CUDA::nvtx3 INTERFACE "${nvtx3_dir}") 23+message(STATUS "Using NVTX3 include directory: ${nvtx3_dir}") 24 25 26 # ---[ HIP