19 #include <rmm/detail/export.hpp>
20 #include <rmm/logger_macros.hpp>
22 #include <rapids_logger/logger.hpp>
24 namespace RMM_NAMESPACE {
34 inline rapids_logger::sink_ptr default_sink()
36 auto* filename = std::getenv(
"RMM_DEBUG_LOG_FILE");
37 if (filename !=
nullptr) {
38 return std::make_shared<rapids_logger::basic_file_sink_mt>(filename,
true);
40 return std::make_shared<rapids_logger::stderr_sink_mt>();
48 inline std::string default_pattern() {
return "[%6t][%H:%M:%S:%f][%-6l] %v"; }
55 inline rapids_logger::logger& default_logger()
57 static rapids_logger::logger logger_ = [] {
58 rapids_logger::logger logger_{
"RMM", {default_sink()}};
59 logger_.set_pattern(default_pattern());
60 #if RMM_LOG_ACTIVE_LEVEL <= RMM_LOG_LEVEL_DEBUG
61 #ifdef CUDA_API_PER_THREAD_DEFAULT_STREAM
62 logger_.debug(
"----- RMM LOG [PTDS ENABLED] -----");
64 logger_.debug(
"----- RMM LOG [PTDS DISABLED] -----");