Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fix: add MPI_Finalize in QUIT function #5505

Open
wants to merge 2 commits into
base: develop
Choose a base branch
from
+70 −78
Open

Fix: add MPI_Finalize in QUIT function #5505

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
b0ddd17
add MPI_Finalize in QUIT function
kirk0830 Nov 15, 2024
660d8c2
Merge branch 'develop' into tool-quit-1
kirk0830 Nov 15, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
146 changes: 69 additions & 77 deletions source/module_base/tool_quit.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@
#endif

#ifdef __NORMAL

#else
#include "global_variable.h"
#include "module_parameter/parameter.h"
Expand All @@ -20,17 +19,14 @@ namespace ModuleBase
// NAME : WARNING( write information into GlobalV::ofs_warning)
// NAME : QUIT( exit the running program)
// NAME : WARNING_QUIT( write information into
// GlobalV::ofs_warning , and then quit)
// GlobalV::ofs_warning , and then quit)
//==========================================================
void WARNING(const std::string &file,const std::string &description)
{
#ifdef __NORMAL

#ifdef __NORMAL /* what is this??? */
#else

if (GlobalV::MY_RANK==0)
{
// std::cout << "\n "<<file<<" Warning : "<<description<<std::endl;
GlobalV::ofs_warning << " " << file <<" warning : "<< description<<std::endl;
}
#endif
Expand All @@ -39,69 +35,65 @@ void WARNING(const std::string &file,const std::string &description)

void QUIT()
{
QUIT(0);
/* it is quite strange to exit with code 0 by default... */
QUIT(0);
}

void QUIT(int ret)
void QUIT(const int ret)
{

#ifdef __NORMAL

#ifdef __NORMAL /* what is this??? */
#else
ModuleBase::timer::finish(GlobalV::ofs_running , !GlobalV::MY_RANK);

ModuleBase::Global_File::close_all_log(GlobalV::MY_RANK);

std::cout<<" See output information in : "<<PARAM.globalv.global_out_dir<<std::endl;
#endif

#ifdef __MPI /* if it is MPI run, finalize first, then exit */
MPI_Finalize();
#endif
exit(ret);
}


void WARNING_QUIT(const std::string &file,const std::string &description)
void WARNING_QUIT(const std::string &file, const std::string &description)
{
WARNING_QUIT(file, description, 0);
WARNING_QUIT(file, description, 0);
/* really? we return with 0? it is something like "we exit the program normally" */
}

void WARNING_QUIT(const std::string &file,const std::string &description,int ret)
void WARNING_QUIT(const std::string &file, const std::string &description, int ret)
{
#ifdef __NORMAL

std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " NOTICE " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;

#ifdef __NORMAL /* what is this??? */
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " NOTICE " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
#else
std::cout << " " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " NOTICE " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " " << std::endl;
std::cout << " " << description << std::endl;
std::cout << " CHECK IN FILE : " << PARAM.globalv.global_out_dir << "warning.log" << std::endl;
std::cout << " " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " NOTICE " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;


GlobalV::ofs_running << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
GlobalV::ofs_running << " NOTICE " << std::endl;
GlobalV::ofs_running << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
GlobalV::ofs_running << std::endl;
GlobalV::ofs_running << " " << description << std::endl;
GlobalV::ofs_running << " CHECK IN FILE : " << PARAM.globalv.global_out_dir << "warning.log" << std::endl;
GlobalV::ofs_running << std::endl;
GlobalV::ofs_running << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
GlobalV::ofs_running << " NOTICE " << std::endl;
GlobalV::ofs_running << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;

WARNING(file,description);
std::cout << " " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " NOTICE " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " " << std::endl;
std::cout << " " << description << std::endl;
std::cout << " CHECK IN FILE : " << PARAM.globalv.global_out_dir << "warning.log" << std::endl;
std::cout << " " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " NOTICE " << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;


GlobalV::ofs_running << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
GlobalV::ofs_running << " NOTICE " << std::endl;
GlobalV::ofs_running << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
GlobalV::ofs_running << std::endl;
GlobalV::ofs_running << " " << description << std::endl;
GlobalV::ofs_running << " CHECK IN FILE : " << PARAM.globalv.global_out_dir << "warning.log" << std::endl;
GlobalV::ofs_running << std::endl;
GlobalV::ofs_running << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
GlobalV::ofs_running << " NOTICE " << std::endl;
GlobalV::ofs_running << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;

WARNING(file,description);
GlobalV::ofs_running<<" Check in file : "<<PARAM.globalv.global_out_dir<<"warning.log"<<std::endl;

#endif

QUIT(ret);
}

Expand All @@ -112,34 +104,34 @@ void CHECK_WARNING_QUIT(const bool error_in, const std::string &file,const std::
#ifdef __NORMAL
// only for UT, do nothing here
#else
if(error_in)
{
//All cores will print inforamtion
std::cout.clear();
if(!GlobalV::ofs_running.is_open())
{
std::string logfile = PARAM.globalv.global_out_dir + "running_" + calculation + ".log";
GlobalV::ofs_running.open( logfile.c_str(), std::ios::app );
}
if(!GlobalV::ofs_warning.is_open())
{
std::string warningfile = PARAM.globalv.global_out_dir + "warning.log";
GlobalV::ofs_warning.open( warningfile.c_str(), std::ios::app );
}

//print error information
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " ERROR! " << description << std::endl;
std::cout << " CHECK IN FILE : " << PARAM.globalv.global_out_dir << "warning.log" << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
GlobalV::ofs_running << " ERROR! CHECK IN FILE : " << PARAM.globalv.global_out_dir << "warning.log" << std::endl;
GlobalV::ofs_warning << std::endl;
GlobalV::ofs_warning << " ERROR! " << file << ", core " << GlobalV::MY_RANK+1 << ": " << description << std::endl;
GlobalV::ofs_warning << std::endl;
exit(1);
}
if(error_in)
{
//All cores will print inforamtion
std::cout.clear();
if(!GlobalV::ofs_running.is_open())
{
std::string logfile = PARAM.globalv.global_out_dir + "running_" + calculation + ".log";
GlobalV::ofs_running.open( logfile.c_str(), std::ios::app );
}
if(!GlobalV::ofs_warning.is_open())
{
std::string warningfile = PARAM.globalv.global_out_dir + "warning.log";
GlobalV::ofs_warning.open( warningfile.c_str(), std::ios::app );
}

//print error information
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
std::cout << " ERROR! " << description << std::endl;
std::cout << " CHECK IN FILE : " << PARAM.globalv.global_out_dir << "warning.log" << std::endl;
std::cout << " !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!" << std::endl;
GlobalV::ofs_running << " ERROR! CHECK IN FILE : " << PARAM.globalv.global_out_dir << "warning.log" << std::endl;
GlobalV::ofs_warning << std::endl;
GlobalV::ofs_warning << " ERROR! " << file << ", core " << GlobalV::MY_RANK+1 << ": " << description << std::endl;
GlobalV::ofs_warning << std::endl;
exit(1);
}
#endif
return;
return;
}

}
2 changes: 1 addition & 1 deletion source/module_base/tool_quit.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ void WARNING(const std::string &file, const std::string &description);
* @brief Close .log files and exit
*
*/
[[noreturn]] void QUIT(int ret);
[[noreturn]] void QUIT(const int ret);

/**
* @brief Combine the functions of WARNING and QUIT
Expand Down