From fc5586677e61eee60fad989a25888fb084e8b153 Mon Sep 17 00:00:00 2001 From: Helmut Date: Sun, 23 Jul 2023 19:01:37 +0200 Subject: [PATCH] Turn log level into enum and some minor changes. --- llama.cpp | 10 +++++----- llama.h | 10 ++++++---- 2 files changed, 11 insertions(+), 9 deletions(-) diff --git a/llama.cpp b/llama.cpp index 5d85234da..2d85f496c 100644 --- a/llama.cpp +++ b/llama.cpp @@ -57,7 +57,7 @@ #endif void llama_log_internal(int level, const char* format, ...); -void llama_log_callback_default(int level, const char * text, void * ctx); +void llama_log_callback_default(llama_log_level level, const char * text, void * user_data); #define LLAMA_LOG_INFO(...) llama_log_internal(LLAMA_LOG_LEVEL_INFO , __VA_ARGS__) #define LLAMA_LOG_WARN(...) llama_log_internal(LLAMA_LOG_LEVEL_WARN , __VA_ARGS__) #define LLAMA_LOG_ERROR(...) llama_log_internal(LLAMA_LOG_LEVEL_ERROR, __VA_ARGS__) @@ -3759,7 +3759,7 @@ void llama_log_set(llama_log_callback log_callback, void * user_data) { #define vsnprintf _vsnprintf #endif -void llama_log_internal_v(int level, const char * format, va_list args) { +static void llama_log_internal_v(llama_log_level level, const char * format, va_list args) { va_list args_copy; va_copy(args_copy, args); char buffer[128]; @@ -3776,15 +3776,15 @@ void llama_log_internal_v(int level, const char * format, va_list args) { va_end(args_copy); } -void llama_log_internal(int level, const char * format, ...) { +static void llama_log_internal(llama_log_level level, const char * format, ...) { va_list args; va_start(args, format); llama_log_internal_v(level, format, args); va_end(args); } -void llama_log_callback_default(int level, const char * text, void *ctx) { +static void llama_log_callback_default(llama_log_level level, const char * text, void * user_data) { (void) level; - (void) ctx; + (void) user_data; fprintf(stderr, "%s\n", text); } diff --git a/llama.h b/llama.h index 2dd7c84df..02c7e3c8f 100644 --- a/llama.h +++ b/llama.h @@ -82,10 +82,12 @@ extern "C" { typedef void (*llama_progress_callback)(float progress, void *ctx); - typedef void (*llama_log_callback)(int level, const char* text, void *ctx); - #define LLAMA_LOG_LEVEL_ERROR 2 - #define LLAMA_LOG_LEVEL_WARN 3 - #define LLAMA_LOG_LEVEL_INFO 4 + enum llama_log_level { + LLAMA_LOG_LEVEL_ERROR = 2, + LLAMA_LOG_LEVEL_WARN = 3, + LLAMA_LOG_LEVEL_INFO = 4 + }; + typedef void (*llama_log_callback)(llama_log_level level, const char* text, void *ctx); struct llama_context_params { uint32_t seed; // RNG seed, -1 for random