Jelajahi Sumber

add log sink

Michael Yang 4 bulan lalu
induk
melakukan
512bbb8044
1 mengubah file dengan 8 tambahan dan 0 penghapusan
  1. 8 0
      ml/backend/ggml/ggml/src/ggml.go

+ 8 - 0
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 {