Clean up logger.h
Change-Id: Ibcb58d2236b012d00c3fc421a425c03093de5d50
This commit is contained in:
@@ -99,14 +99,6 @@ MemoryPool* default_memory_pool = nullptr;
|
|||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
||||||
namespace roctracer {
|
|
||||||
|
|
||||||
// Logger routines and primitives
|
|
||||||
util::Logger::mutex_t util::Logger::mutex_;
|
|
||||||
std::atomic<util::Logger*> util::Logger::instance_{};
|
|
||||||
|
|
||||||
} // namespace roctracer
|
|
||||||
|
|
||||||
///////////////////////////////////////////////////////////////////////////////////////////////////
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
||||||
// Public library methods
|
// Public library methods
|
||||||
//
|
//
|
||||||
@@ -117,7 +109,7 @@ ROCTRACER_API uint32_t roctracer_version_minor() { return ROCTRACER_VERSION_MINO
|
|||||||
|
|
||||||
// Returns the last error
|
// Returns the last error
|
||||||
ROCTRACER_API const char* roctracer_error_string() {
|
ROCTRACER_API const char* roctracer_error_string() {
|
||||||
return strdup(util::Logger::LastMessage().c_str());
|
return strdup(util::Logger::Instance().LastMessage().c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
// Return Op string by given domain and activity/API codes
|
// Return Op string by given domain and activity/API codes
|
||||||
@@ -881,9 +873,6 @@ ROCTRACER_API roctracer_status_t roctracer_set_properties(roctracer_domain_t dom
|
|||||||
API_METHOD_SUFFIX
|
API_METHOD_SUFFIX
|
||||||
}
|
}
|
||||||
|
|
||||||
__attribute__((constructor)) void constructor() { util::Logger::Create(); }
|
|
||||||
__attribute__((destructor)) void destructor() { util::Logger::Destroy(); }
|
|
||||||
|
|
||||||
extern "C" {
|
extern "C" {
|
||||||
|
|
||||||
// The HSA_AMD_TOOL_PRIORITY variable must be a constant value type initialized by the loader
|
// The HSA_AMD_TOOL_PRIORITY variable must be a constant value type initialized by the loader
|
||||||
|
|||||||
+21
-74
@@ -43,11 +43,9 @@ namespace roctracer::util {
|
|||||||
|
|
||||||
class Logger {
|
class Logger {
|
||||||
public:
|
public:
|
||||||
typedef std::recursive_mutex mutex_t;
|
template <typename T> Logger& operator<<(T&& m) {
|
||||||
|
|
||||||
template <typename T> Logger& operator<<(const T& m) {
|
|
||||||
std::ostringstream oss;
|
std::ostringstream oss;
|
||||||
oss << m;
|
oss << std::forward<T>(m);
|
||||||
if (!streaming_)
|
if (!streaming_)
|
||||||
Log(oss.str());
|
Log(oss.str());
|
||||||
else
|
else
|
||||||
@@ -56,7 +54,7 @@ class Logger {
|
|||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
typedef void (*manip_t)();
|
using manip_t = void (*)();
|
||||||
Logger& operator<<(manip_t f) {
|
Logger& operator<<(manip_t f) {
|
||||||
f();
|
f();
|
||||||
return *this;
|
return *this;
|
||||||
@@ -65,59 +63,35 @@ class Logger {
|
|||||||
static void begm() { Instance().ResetStreaming(true); }
|
static void begm() { Instance().ResetStreaming(true); }
|
||||||
static void endl() { Instance().ResetStreaming(false); }
|
static void endl() { Instance().ResetStreaming(false); }
|
||||||
|
|
||||||
static const std::string& LastMessage() {
|
const std::string& LastMessage() {
|
||||||
Logger& logger = Instance();
|
std::lock_guard lock(mutex_);
|
||||||
std::lock_guard<mutex_t> lck(mutex_);
|
return message_[GetTid()];
|
||||||
return logger.message_[GetTid()];
|
|
||||||
}
|
|
||||||
|
|
||||||
static Logger* Create() {
|
|
||||||
std::lock_guard<mutex_t> lck(mutex_);
|
|
||||||
Logger* obj = instance_.load(std::memory_order_relaxed);
|
|
||||||
if (obj == NULL) {
|
|
||||||
obj = new Logger();
|
|
||||||
if (obj == NULL) {
|
|
||||||
std::cerr << "ROCTracer: log object creation failed" << std::endl << std::flush;
|
|
||||||
abort();
|
|
||||||
}
|
|
||||||
instance_.store(obj, std::memory_order_release);
|
|
||||||
}
|
|
||||||
return obj;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void Destroy() {
|
|
||||||
std::lock_guard<mutex_t> lck(mutex_);
|
|
||||||
if (instance_ != NULL) delete instance_.load();
|
|
||||||
instance_ = NULL;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static Logger& Instance() {
|
static Logger& Instance() {
|
||||||
Logger* obj = instance_.load(std::memory_order_acquire);
|
static Logger instance;
|
||||||
if (obj == NULL) obj = Create();
|
return instance;
|
||||||
return *obj;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static uint32_t GetPid() { return syscall(__NR_getpid); }
|
static uint32_t GetPid() { return syscall(__NR_getpid); }
|
||||||
static uint32_t GetTid() { return syscall(__NR_gettid); }
|
static uint32_t GetTid() { return syscall(__NR_gettid); }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
Logger() : file_(NULL), dirty_(false), streaming_(false), messaging_(false) {
|
Logger() : file_(nullptr), dirty_(false), streaming_(false), messaging_(false) {
|
||||||
const char* path = getenv("ROCTRACER_LOG");
|
const char* var = getenv("ROCTRACER_LOG");
|
||||||
if (path != NULL) {
|
if (var != nullptr) file_ = fopen("/tmp/roctracer_log.txt", "a");
|
||||||
file_ = fopen("/tmp/roctracer_log.txt", "a");
|
|
||||||
}
|
|
||||||
ResetStreaming(false);
|
ResetStreaming(false);
|
||||||
}
|
}
|
||||||
|
|
||||||
~Logger() {
|
~Logger() {
|
||||||
if (file_ != NULL) {
|
if (file_ != nullptr) {
|
||||||
if (dirty_) Put("\n");
|
if (dirty_) Put("\n");
|
||||||
fclose(file_);
|
fclose(file_);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void ResetStreaming(const bool messaging) {
|
void ResetStreaming(const bool messaging) {
|
||||||
std::lock_guard<mutex_t> lck(mutex_);
|
std::lock_guard lock(mutex_);
|
||||||
if (messaging) {
|
if (messaging) {
|
||||||
message_[GetTid()] = "";
|
message_[GetTid()] = "";
|
||||||
} else if (streaming_) {
|
} else if (streaming_) {
|
||||||
@@ -129,11 +103,11 @@ class Logger {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void Put(const std::string& m) {
|
void Put(const std::string& m) {
|
||||||
std::lock_guard<mutex_t> lck(mutex_);
|
std::lock_guard lock(mutex_);
|
||||||
if (messaging_) {
|
if (messaging_) {
|
||||||
message_[GetTid()] += m;
|
message_[GetTid()] += m;
|
||||||
}
|
}
|
||||||
if (file_ != NULL) {
|
if (file_ != nullptr) {
|
||||||
dirty_ = true;
|
dirty_ = true;
|
||||||
flock(fileno(file_), LOCK_EX);
|
flock(fileno(file_), LOCK_EX);
|
||||||
fprintf(file_, "%s", m.c_str());
|
fprintf(file_, "%s", m.c_str());
|
||||||
@@ -143,7 +117,7 @@ class Logger {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void Log(const std::string& m) {
|
void Log(const std::string& m) {
|
||||||
const time_t rawtime = time(NULL);
|
const time_t rawtime = time(nullptr);
|
||||||
tm tm_info;
|
tm tm_info;
|
||||||
localtime_r(&rawtime, &tm_info);
|
localtime_r(&rawtime, &tm_info);
|
||||||
char tm_str[26];
|
char tm_str[26];
|
||||||
@@ -158,8 +132,7 @@ class Logger {
|
|||||||
bool streaming_;
|
bool streaming_;
|
||||||
bool messaging_;
|
bool messaging_;
|
||||||
|
|
||||||
static mutex_t mutex_;
|
std::recursive_mutex mutex_;
|
||||||
static std::atomic<Logger*> instance_;
|
|
||||||
std::map<uint32_t, std::string> message_;
|
std::map<uint32_t, std::string> message_;
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -170,51 +143,25 @@ class Logger {
|
|||||||
roctracer::util::Logger::Instance() \
|
roctracer::util::Logger::Instance() \
|
||||||
<< "fatal: " << roctracer::util::Logger::begm << stream << roctracer::util::Logger::endl; \
|
<< "fatal: " << roctracer::util::Logger::begm << stream << roctracer::util::Logger::endl; \
|
||||||
abort(); \
|
abort(); \
|
||||||
} while (0)
|
} while (false)
|
||||||
|
|
||||||
#define ERR_LOGGING(stream) \
|
#define ERR_LOGGING(stream) \
|
||||||
do { \
|
do { \
|
||||||
roctracer::util::Logger::Instance() \
|
roctracer::util::Logger::Instance() \
|
||||||
<< "error: " << roctracer::util::Logger::begm << stream << roctracer::util::Logger::endl; \
|
<< "error: " << roctracer::util::Logger::begm << stream << roctracer::util::Logger::endl; \
|
||||||
} while (0)
|
} while (false)
|
||||||
|
|
||||||
#define INFO_LOGGING(stream) \
|
#define INFO_LOGGING(stream) \
|
||||||
do { \
|
do { \
|
||||||
roctracer::util::Logger::Instance() \
|
roctracer::util::Logger::Instance() \
|
||||||
<< "info: " << roctracer::util::Logger::begm << stream << roctracer::util::Logger::endl; \
|
<< "info: " << roctracer::util::Logger::begm << stream << roctracer::util::Logger::endl; \
|
||||||
} while (0)
|
} while (false)
|
||||||
|
|
||||||
#define WARN_LOGGING(stream) \
|
#define WARN_LOGGING(stream) \
|
||||||
do { \
|
do { \
|
||||||
std::cerr << "ROCProfiler: " << stream << std::endl; \
|
std::cerr << "ROCProfiler: " << stream << std::endl; \
|
||||||
roctracer::util::Logger::Instance() << "warning: " << roctracer::util::Logger::begm << stream \
|
roctracer::util::Logger::Instance() << "warning: " << roctracer::util::Logger::begm << stream \
|
||||||
<< roctracer::util::Logger::endl; \
|
<< roctracer::util::Logger::endl; \
|
||||||
} while (0)
|
} while (false)
|
||||||
|
|
||||||
#ifdef DEBUG
|
|
||||||
#define DBG_LOGGING(stream) \
|
|
||||||
do { \
|
|
||||||
roctracer::util::Logger::Instance() \
|
|
||||||
<< roctracer::util::Logger::begm << "debug: \"" << stream << "\"" \
|
|
||||||
<< " in " << __FUNCTION__ << " at " << __FILE__ << " line " << __LINE__ \
|
|
||||||
<< roctracer::util::Logger::endl; \
|
|
||||||
} while (0)
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if DEBUG_TRACE_ON
|
|
||||||
inline static void DEBUG_TRACE(const char* fmt, ...) {
|
|
||||||
constexpr int size = 256;
|
|
||||||
char buf[size];
|
|
||||||
|
|
||||||
va_list valist;
|
|
||||||
va_start(valist, fmt);
|
|
||||||
vsnprintf(buf, size, fmt, valist);
|
|
||||||
printf("%u:%u %s", roctracer::util::Logger::GetPid(), roctracer::util::Logger::GetTid(), buf);
|
|
||||||
fflush(stdout);
|
|
||||||
va_end(valist);
|
|
||||||
}
|
|
||||||
#else
|
|
||||||
#define DEBUG_TRACE(...)
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif // SRC_UTIL_LOGGER_H_
|
#endif // SRC_UTIL_LOGGER_H_
|
||||||
|
|||||||
مرجع در شماره جدید
Block a user