From 273d985271580bce0684889d007bbac7b2484e66 Mon Sep 17 00:00:00 2001 From: ngxson Date: Mon, 26 Feb 2024 17:30:00 +0100 Subject: [PATCH] std::atomic_flag --- examples/server/server.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/examples/server/server.cpp b/examples/server/server.cpp index 8f9c86271c728..2f4acf54a6d0b 100644 --- a/examples/server/server.cpp +++ b/examples/server/server.cpp @@ -2768,12 +2768,11 @@ static void append_to_generated_text_from_generated_token_probs(llama_server_con } std::function shutdown_handler; -static bool is_terminating = false; +std::atomic_flag is_terminating = ATOMIC_FLAG_INIT; 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 } - is_terminating = true; shutdown_handler(signal); }