modified LBANN_WARNING to call H2 logging LBANN_WARN

LLNL · Mar 20, 2023 · 6ff274c · 6ff274c
1 parent 3590a3d
commit 6ff274c
Showing 1 changed file with 2 additions and 16 deletions.
diff --git a/include/lbann/utils/exception.hpp b/include/lbann/utils/exception.hpp
@@ -28,6 +28,7 @@
 #define LBANN_UTILS_EXCEPTION_HPP_INCLUDED
 
 #include "lbann/comm.hpp"
+#include "lbann/utils/logging.hpp"
 
 #include <exception>
 #include <iostream>
@@ -50,22 +51,7 @@
   } while (0)
 
 // Macro to print a warning to standard error stream.
-#define LBANN_WARNING(...)                                                     \
-  do {                                                                         \
-    const int rank_LBANN_WARNING = lbann::get_rank_in_world();                 \
-    std::cerr << lbann::build_string(                                          \
-                   "LBANN warning",                                            \
-                   (rank_LBANN_WARNING >= 0                                    \
-                      ? " on rank " + std::to_string(rank_LBANN_WARNING)       \
-                      : std::string()),                                        \
-                   " (",                                                       \
-                   __FILE__,                                                   \
-                   ":",                                                        \
-                   __LINE__,                                                   \
-                   "): ",                                                      \
-                   __VA_ARGS__)                                                \
-              << std::endl;                                                    \
-  } while (0)
+#define LBANN_WARNING(...) LBANN_WARN(lbann::logging::LBANN_Logger_ID::LOG_RT, __VA_ARGS__)
 
 // Macro to print a message to standard cout stream.
 #define LBANN_MSG(...)                                                         \