feature: output math lib logs on all ranks.

Author: caic99

source/module_base/gather_math_lib_info.cpp

@@ -1,4 +1,11 @@
 // This file defines the math lib wrapper for output information before executing computations.
+
+/*
+When INFO is defined in cmake configure, a macro of GATHER_INFO will be defined.
+This macro will be used to output information before executing computations.
+Results will output to OUT/math_info.log, see ModuleBase::Global_File::make_dir_out .
+*/
+
 #include <ostream>
 #undef GATHER_INFO
 #include "module_base/blas_connector.h"

source/module_base/global_file.cpp

@@ -142,10 +142,12 @@ void ModuleBase::Global_File::make_dir_out(
     if(rank==0)
     {
         open_log(GlobalV::ofs_warning, "warning", calculation, restart);
+    }
+
 #ifdef GATHER_INFO
-        open_log(GlobalV::ofs_info, "math_info", calculation, restart);
+    open_log(GlobalV::ofs_info, "math_info_" + std::to_string(rank), calculation, restart);
 #endif
-    }
+
     return;
 }
 
@@ -224,10 +226,11 @@ void ModuleBase::Global_File::close_all_log(const int rank, const bool out_alllo
     if (rank==0)
     {
         close_log(GlobalV::ofs_warning, "warning.log");
+    }
+
 #ifdef GATHER_INFO
-        close_log(GlobalV::ofs_info, "math_info");
+    close_log(GlobalV::ofs_info, "math_info");
 #endif
-    }
     return;
 }
 }

source/module_base/test/CMakeLists.txt

@@ -2,7 +2,7 @@ remove_definitions(-D__MPI)
 AddTest(
   TARGET base_blas_connector
   LIBS ${math_libs}
-  SOURCES blas_connector_test.cpp ../gather_math_lib_info.cpp
+  SOURCES blas_connector_test.cpp
 )
 AddTest(
   TARGET base_timer