root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-24 18:34:36 +00:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

llama : print timings on ctrl+c exit (#1021)

Browse Source 
* print timings on ctrl+c exit

* remove redundant free memory call.

* add global pointer to ctx.
This commit is contained in:
wbpxre150 2023-04-22 16:56:35 +08:00 committed by GitHub
parent 10f19c1121
commit 36b4f7e064
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
examples/main/main.cpp
View File

@ -25,6 +25,7 @@
#endif #endif
static console_state con_st; static console_state con_st;
static llama_context ** g_ctx;
static bool is_interacting = false; static bool is_interacting = false;
@ -36,6 +37,7 @@ void sigint_handler(int signo) {
if (!is_interacting) { if (!is_interacting) {
is_interacting=true; is_interacting=true;
} else { } else {
llama_print_timings(*g_ctx);
_exit(130); _exit(130);
} }
} }
@ -92,8 +94,9 @@ int main(int argc, char ** argv) {
// params.prompt = R"(// this function checks if the number n is prime // params.prompt = R"(// this function checks if the number n is prime
//bool is_prime(int n) {)"; //bool is_prime(int n) {)";
llama_context * ctx; llama_context * ctx;
g_ctx = &ctx;
// load the model // load the model
{ {