Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update nvtx. (#10227)

Browse Source
This commit is contained in:
Jiaming Yuan 2024-04-29 06:33:46 +08:00 committed by GitHub
parent 837d44a345
commit 5e64276a9b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmake/Utils.cmake
View File

@ -289,7 +289,7 @@ macro(xgboost_target_link_libraries target)
endif() endif()
if(USE_NVTX) if(USE_NVTX)
target_link_libraries(${target} PRIVATE CUDA::nvToolsExt) target_link_libraries(${target} PRIVATE CUDA::nvtx3)
endif() endif()
if(MINGW) if(MINGW)

10
src/common/timer.cc
View File

@ -8,12 +8,10 @@
#include "../collective/communicator-inl.h" #include "../collective/communicator-inl.h"
#if defined(XGBOOST_USE_NVTX) #if defined(XGBOOST_USE_NVTX)
#include <nvToolsExt.h> #include <nvtx3/nvToolsExt.h>
#endif // defined(XGBOOST_USE_NVTX) #endif // defined(XGBOOST_USE_NVTX)
namespace xgboost { namespace xgboost::common {
namespace common {
void Monitor::Start(std::string const &name) { void Monitor::Start(std::string const &name) {
if (ConsoleLogger::ShouldLog(ConsoleLogger::LV::kDebug)) { if (ConsoleLogger::ShouldLog(ConsoleLogger::LV::kDebug)) {
auto &stats = statistics_map_[name]; auto &stats = statistics_map_[name];
@ -66,6 +64,4 @@ void Monitor::Print() const {
LOG(CONSOLE) << "======== Monitor (" << rank << "): " << label_ << " ========"; LOG(CONSOLE) << "======== Monitor (" << rank << "): " << label_ << " ========";
this->PrintStatistics(stat_map); this->PrintStatistics(stat_map);
} }
} // namespace xgboost::common
} // namespace common
} // namespace xgboost

1
tests/cpp/CMakeLists.txt
View File

@ -59,7 +59,6 @@ endif()
target_sources(testxgboost PRIVATE ${TEST_SOURCES} ${xgboost_SOURCE_DIR}/plugin/example/custom_obj.cc) target_sources(testxgboost PRIVATE ${TEST_SOURCES} ${xgboost_SOURCE_DIR}/plugin/example/custom_obj.cc)
if(USE_CUDA AND PLUGIN_RMM) if(USE_CUDA AND PLUGIN_RMM)
find_package(CUDA)
target_include_directories(testxgboost PRIVATE ${CUDA_INCLUDE_DIRS}) target_include_directories(testxgboost PRIVATE ${CUDA_INCLUDE_DIRS})
endif() endif()