ccf80703ef
* Clang-tidy static analysis * Modernise checks * Google coding standard checks * Identifier renaming according to Google style
91 lines
2.3 KiB
C++
91 lines
2.3 KiB
C++
/*!
|
|
* Copyright by Contributors 2017
|
|
*/
|
|
#pragma once
|
|
#include <xgboost/logging.h>
|
|
#include <chrono>
|
|
#include <iostream>
|
|
#include <map>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
namespace xgboost {
|
|
namespace common {
|
|
struct Timer {
|
|
using ClockT = std::chrono::high_resolution_clock;
|
|
using TimePointT = std::chrono::high_resolution_clock::time_point;
|
|
using DurationT = std::chrono::high_resolution_clock::duration;
|
|
using SecondsT = std::chrono::duration<double>;
|
|
|
|
TimePointT start;
|
|
DurationT elapsed;
|
|
Timer() { Reset(); }
|
|
void Reset() {
|
|
elapsed = DurationT::zero();
|
|
Start();
|
|
}
|
|
void Start() { start = ClockT::now(); }
|
|
void Stop() { elapsed += ClockT::now() - start; }
|
|
double ElapsedSeconds() const { return SecondsT(elapsed).count(); }
|
|
void PrintElapsed(std::string label) {
|
|
char buffer[255];
|
|
snprintf(buffer, sizeof(buffer), "%s:\t %fs", label.c_str(),
|
|
SecondsT(elapsed).count());
|
|
LOG(CONSOLE) << buffer;
|
|
Reset();
|
|
}
|
|
};
|
|
|
|
/**
|
|
* \struct Monitor
|
|
*
|
|
* \brief Timing utility used to measure total method execution time over the
|
|
* lifetime of the containing object.
|
|
*/
|
|
|
|
struct Monitor {
|
|
bool debug_verbose = false;
|
|
std::string label = "";
|
|
std::map<std::string, Timer> timer_map;
|
|
Timer self_timer;
|
|
|
|
Monitor() { self_timer.Start(); }
|
|
|
|
~Monitor() {
|
|
if (!debug_verbose) return;
|
|
|
|
LOG(CONSOLE) << "======== Monitor: " << label << " ========";
|
|
for (auto &kv : timer_map) {
|
|
kv.second.PrintElapsed(kv.first);
|
|
}
|
|
self_timer.Stop();
|
|
self_timer.PrintElapsed(label + " Lifetime");
|
|
}
|
|
void Init(std::string label, bool debug_verbose) {
|
|
this->debug_verbose = debug_verbose;
|
|
this->label = label;
|
|
}
|
|
void Start(const std::string &name) { timer_map[name].Start(); }
|
|
void Start(const std::string &name, std::vector<int> dList) {
|
|
if (debug_verbose) {
|
|
#ifdef __CUDACC__
|
|
#include "device_helpers.cuh"
|
|
dh::SynchronizeNDevices(dList.size(), dList);
|
|
#endif
|
|
}
|
|
timer_map[name].Start();
|
|
}
|
|
void Stop(const std::string &name) { timer_map[name].Stop(); }
|
|
void Stop(const std::string &name, std::vector<int> dList) {
|
|
if (debug_verbose) {
|
|
#ifdef __CUDACC__
|
|
#include "device_helpers.cuh"
|
|
dh::SynchronizeNDevices(dList.size(), dList);
|
|
#endif
|
|
}
|
|
timer_map[name].Stop();
|
|
}
|
|
};
|
|
} // namespace common
|
|
} // namespace xgboost
|