Turn log level into enum and some minor changes.

This commit is contained in:
Helmut 2023-07-23 19:01:37 +02:00
parent 671ec2c588
commit fc5586677e
2 changed files with 11 additions and 9 deletions

View file

@ -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);
}

10
llama.h
View file

@ -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