From 512bbb804488089bef6a06a94b57861ce27eb6fb Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Sat, 21 Dec 2024 01:31:19 -0800
Subject: [PATCH] add log sink

---
 ml/backend/ggml/ggml/src/ggml.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/ml/backend/ggml/ggml/src/ggml.go b/ml/backend/ggml/ggml/src/ggml.go
index f8f490484..586c675c0 100644
--- a/ml/backend/ggml/ggml/src/ggml.go
+++ b/ml/backend/ggml/ggml/src/ggml.go
@@ -5,6 +5,7 @@ package ggml
 // #cgo CPPFLAGS: -I${SRCDIR}/../include -I${SRCDIR}/ggml-cpu
 // #include <stdlib.h>
 // #include "ggml-backend.h"
+// extern void sink(int level, char *text, void *user_data);
 import "C"
 import (
 	"log/slog"
@@ -18,6 +19,13 @@ import (
 	_ "github.com/ollama/ollama/ml/backend/ggml/ggml/src/ggml-cpu"
 )
 
+func init() {
+	C.ggml_log_set((C.ggml_log_callback)(C.sink), nil)
+}
+
+//export sink
+func sink(level C.int, text *C.char, _ unsafe.Pointer) {}
+
 var OnceLoad = sync.OnceFunc(func() {
 	var lib struct{ name, pattern, defaultValue string }
 	switch runtime.GOOS {