Update nvtx. (#10227)

dmlc · Apr 28, 2024 · 5e64276 · 5e64276
1 parent 837d44a
commit 5e64276
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 9 deletions.
diff --git a/cmake/Utils.cmake b/cmake/Utils.cmake
@@ -289,7 +289,7 @@ macro(xgboost_target_link_libraries target)
   endif()
 
   if(USE_NVTX)
-    target_link_libraries(${target} PRIVATE CUDA::nvToolsExt)
+    target_link_libraries(${target} PRIVATE CUDA::nvtx3)
   endif()
 
   if(MINGW)

diff --git a/src/common/timer.cc b/src/common/timer.cc
@@ -8,12 +8,10 @@
 #include "../collective/communicator-inl.h"
 
 #if defined(XGBOOST_USE_NVTX)
-#include <nvToolsExt.h>
+#include <nvtx3/nvToolsExt.h>
 #endif  // defined(XGBOOST_USE_NVTX)
 
-namespace xgboost {
-namespace common {
-
+namespace xgboost::common {
 void Monitor::Start(std::string const &name) {
   if (ConsoleLogger::ShouldLog(ConsoleLogger::LV::kDebug)) {
     auto &stats = statistics_map_[name];
@@ -66,6 +64,4 @@ void Monitor::Print() const {
   LOG(CONSOLE) << "======== Monitor (" << rank << "): " << label_ << " ========";
   this->PrintStatistics(stat_map);
 }
-
-}  // namespace common
-}  // namespace xgboost
+}  // namespace xgboost::common
diff --git a/tests/cpp/CMakeLists.txt b/tests/cpp/CMakeLists.txt
@@ -59,7 +59,6 @@ endif()
 target_sources(testxgboost PRIVATE ${TEST_SOURCES} ${xgboost_SOURCE_DIR}/plugin/example/custom_obj.cc)
 
 if(USE_CUDA AND PLUGIN_RMM)
-  find_package(CUDA)
   target_include_directories(testxgboost PRIVATE ${CUDA_INCLUDE_DIRS})
 endif()