fadb458e80
Diff: https://github.com/pytorch/pytorch/compare/v2.7.1...v2.8.0 Changelog: https://github.com/pytorch/pytorch/releases/tag/v2.8.0 Co-authored-by: Luna Nova <git@lunnova.dev>
27 lines
1.1 KiB
Diff
27 lines
1.1 KiB
Diff
diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake
|
|
index a93386c27f8..7c6b98006bf 100644
|
|
--- a/cmake/Dependencies.cmake
|
|
+++ b/cmake/Dependencies.cmake
|
|
@@ -967,7 +967,7 @@ endif()
|
|
|
|
# ---[ nvtx
|
|
if(USE_SYSTEM_NVTX)
|
|
- find_path(nvtx3_dir NAMES nvtx3 PATHS ${CUDA_INCLUDE_DIRS})
|
|
+ find_path(nvtx3_dir NAMES nvtx3/nvtx3.hpp PATHS ${CUDA_INCLUDE_DIRS})
|
|
find_package_handle_standard_args(nvtx3 DEFAULT_MSG nvtx3_dir)
|
|
if(NOT nvtx3_FOUND)
|
|
message(WARNING "Cannot find system NVTX3, find shipped NVTX3 instead")
|
|
@@ -977,9 +977,10 @@ if(NOT TARGET CUDA::nvtx3)
|
|
add_library(CUDA::nvtx3 INTERFACE IMPORTED)
|
|
endif()
|
|
if(NOT nvtx3_dir)
|
|
- find_path(nvtx3_dir NAMES nvtx3 PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH)
|
|
- target_include_directories(CUDA::nvtx3 INTERFACE "${nvtx3_dir}")
|
|
+ find_path(nvtx3_dir NAMES nvtx3/nvtx3.hpp PATHS "${PROJECT_SOURCE_DIR}/third_party/NVTX/c/include" NO_DEFAULT_PATH)
|
|
endif()
|
|
+target_include_directories(CUDA::nvtx3 INTERFACE "${nvtx3_dir}")
|
|
+message(STATUS "Using NVTX3 include directory: ${nvtx3_dir}")
|
|
|
|
|
|
# ---[ HIP
|