7 #ifndef TREELITE_LOGGING_H_ 8 #define TREELITE_LOGGING_H_ 11 #include <dmlc/thread_local.h> 15 class LogCallbackRegistry {
17 using Callback = void (*)(
const char*);
19 : log_callback_([] (const char* msg) { std::cerr << msg << std::endl; }) {}
20 inline void Register(Callback log_callback) {
21 this->log_callback_ = log_callback;
23 inline Callback Get()
const {
27 Callback log_callback_;
30 using LogCallbackRegistryStore = dmlc::ThreadLocalStore<LogCallbackRegistry>;
34 #endif // TREELITE_LOGGING_H_