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

Turn log level into enum and some minor changes.

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

10
llama.cpp
View file

@ -57,7 +57,7 @@
#endif #endif
void llama_log_internal(int level, const char* format, ...); 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_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_WARN(...) llama_log_internal(LLAMA_LOG_LEVEL_WARN , __VA_ARGS__)
#define LLAMA_LOG_ERROR(...) llama_log_internal(LLAMA_LOG_LEVEL_ERROR, __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 #define vsnprintf _vsnprintf
#endif #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_list args_copy;
va_copy(args_copy, args); va_copy(args_copy, args);
char buffer[128]; char buffer[128];
@ -3776,15 +3776,15 @@ void llama_log_internal_v(int level, const char * format, va_list args) {
va_end(args_copy); 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_list args;
va_start(args, format); va_start(args, format);
llama_log_internal_v(level, format, args); llama_log_internal_v(level, format, args);
va_end(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) level;
(void) ctx; (void) user_data;
fprintf(stderr, "%s\n", text); 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_progress_callback)(float progress, void *ctx);
typedef void (*llama_log_callback)(int level, const char* text, void *ctx); enum llama_log_level {
#define LLAMA_LOG_LEVEL_ERROR 2 LLAMA_LOG_LEVEL_ERROR = 2,
#define LLAMA_LOG_LEVEL_WARN 3 LLAMA_LOG_LEVEL_WARN = 3,
#define LLAMA_LOG_LEVEL_INFO 4 LLAMA_LOG_LEVEL_INFO = 4
};
typedef void (*llama_log_callback)(llama_log_level level, const char* text, void *ctx);
struct llama_context_params { struct llama_context_params {
uint32_t seed; // RNG seed, -1 for random uint32_t seed; // RNG seed, -1 for random