jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add log sink

Browse Source
This commit is contained in:
Michael Yang 2024-12-21 01:31:19 -08:00
parent bca6ed0ccc
commit 512bbb8044
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ml/backend/ggml/ggml/src/ggml.go
View File

@ -5,6 +5,7 @@ package ggml
// #cgo CPPFLAGS: -I${SRCDIR}/../include -I${SRCDIR}/ggml-cpu // #cgo CPPFLAGS: -I${SRCDIR}/../include -I${SRCDIR}/ggml-cpu
// #include <stdlib.h> // #include <stdlib.h>
// #include "ggml-backend.h" // #include "ggml-backend.h"
// extern void sink(int level, char *text, void *user_data);
import "C" import "C"
import ( import (
"log/slog" "log/slog"
@ -18,6 +19,13 @@ import (
_ "github.com/ollama/ollama/ml/backend/ggml/ggml/src/ggml-cpu" _ "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 OnceLoad = sync.OnceFunc(func() {
var lib struct{ name, pattern, defaultValue string } var lib struct{ name, pattern, defaultValue string }
switch runtime.GOOS { switch runtime.GOOS {