Update nvtx. (#10227)

This commit is contained in:
Jiaming Yuan
2024-04-29 06:33:46 +08:00
committed by GitHub
parent 837d44a345
commit 5e64276a9b
3 changed files with 4 additions and 9 deletions

View File

@@ -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