Update nvtx. (#10227)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user