vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

std::atomic_flag

Browse source
This commit is contained in:
ngxson 2024-02-26 17:30:00 +01:00
parent 7547ef5c6d
commit 273d985271
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
examples/server/server.cpp
View file

@ -2768,12 +2768,11 @@ static void append_to_generated_text_from_generated_token_probs(llama_server_con
} }
std::function<void(int)> shutdown_handler; std::function<void(int)> shutdown_handler;
static bool is_terminating = false; std::atomic_flag is_terminating = ATOMIC_FLAG_INIT;
inline void signal_handler(int signal) { inline void signal_handler(int signal) {
if (is_terminating) { if (is_terminating.test_and_set()) {
exit(0); // force exit the process, in case it hangs exit(0); // force exit the process, in case it hangs
} }
is_terminating = true;
shutdown_handler(signal); shutdown_handler(signal);
} }