Turn log level into enum and some minor changes.
This commit is contained in:
parent
671ec2c588
commit
fc5586677e
2 changed files with 11 additions and 9 deletions
10
llama.cpp
10
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);
|
||||
}
|
||||
|
|
10
llama.h
10
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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue