Skip to content

Commit a830a75

Browse files
committed
Updated mlperf_logging to be compliant with new H2 logger format
1 parent 8e9fa85 commit a830a75

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

include/lbann/callbacks/mlperf_logging.hpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -131,11 +131,10 @@ class mlperf_logging : public callback_base {
131131
std::string m_sub_division;
132132
std::string m_sub_status;
133133
std::string m_sub_platform;
134-
//FIXME: Include option to create separate file with just logging data?
135134
/* @brief name of output file. Default = results.txt */
136135
//std::string m_output_filename;
137136
/* @brief DiHydrogen logger */
138-
h2::Logger m_logger{":::MLLOG"};
137+
h2::Logger m_logger{"mlperf_logger", "stdout", ":::MLLOG"};
139138

140139

141140
}; // class mlperf_logging

src/callbacks/mlperf_logging.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ namespace lbann {
4444
namespace callback {
4545

4646
namespace {
47+
4748
void print_value(std::ostringstream& os, int value)
4849
{
4950
os << value;
@@ -116,7 +117,7 @@ void mlperf_logging::print(std::ostringstream& os,
116117
else
117118
os << ", " << "\"epoch_num\": " << epoch << "}}";
118119

119-
H2_INFO(os.str());
120+
m_logger.get()->info(os.str());
120121
os.flush();
121122
}
122123

0 commit comments

Comments
 (0)