tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake
2index ef5c2fd4e97..6591296b704 100644
3--- a/cmake/Dependencies.cmake
4+++ b/cmake/Dependencies.cmake
5@@ -952,15 +952,16 @@ 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 else()
12- find_path(nvtx3_dir NAMES nvtx3 PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH)
13+ find_path(nvtx3_dir NAMES nvtx3/nvtx3.hpp PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH)
14 endif()
15 find_package_handle_standard_args(nvtx3 DEFAULT_MSG nvtx3_dir)
16 if(nvtx3_FOUND)
17 add_library(torch::nvtx3 INTERFACE IMPORTED)
18 target_include_directories(torch::nvtx3 INTERFACE "${nvtx3_dir}")
19 target_compile_definitions(torch::nvtx3 INTERFACE TORCH_CUDA_USE_NVTX3)
20+ message(STATUS "Using NVTX3 include directory: ${nvtx3_dir}")
21 else()
22 message(WARNING "Cannot find NVTX3, find old NVTX instead")
23 add_library(torch::nvtoolsext INTERFACE IMPORTED)