From: Andrew Godfrey <redacted>
Date: Thu, 30 Nov 2023 21:56:19 +0000 (-0800)
Subject: main : pass LOG_TEE callback to llama.cpp log (#4033)
X-Git-Tag: upstream/0.0.4488~2900
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=8efa0f6ebed53c9453e6721da86fb294e5015909;p=pkg%2Fggml%2Fsources%2Fllama.cpp

main : pass LOG_TEE callback to llama.cpp log (#4033)

* main : Call llama_log_set to use LOG_TEE

* tabs to spaces
---

diff --git a/examples/main/main.cpp b/examples/main/main.cpp
index 31ec8cad..c5cdfbf2 100644
--- a/examples/main/main.cpp
+++ b/examples/main/main.cpp
@@ -100,6 +100,12 @@ static void sigint_handler(int signo) {
 }
 #endif
 
+static void llama_log_callback_logTee(ggml_log_level level, const char * text, void * user_data) {
+    (void) level;
+    (void) user_data;
+    LOG_TEE("%s", text);
+}
+
 int main(int argc, char ** argv) {
     gpt_params params;
     g_params = &params;
@@ -113,6 +119,7 @@ int main(int argc, char ** argv) {
     log_set_target(log_filename_generator("main", "log"));
     LOG_TEE("Log start\n");
     log_dump_cmdline(argc, argv);
+    llama_log_set(llama_log_callback_logTee, nullptr);
 #endif // LOG_DISABLE_LOGS
 
     // TODO: Dump params ?