ollama debug tensor

This commit is contained in:
Michael Yang
2025-03-09 14:45:36 -07:00
parent 6b0486c216
commit 9e4642e9b3
8 changed files with 186 additions and 1 deletions

View File

@@ -355,7 +355,7 @@ func New(r *os.File, params ml.BackendParams) (ml.Backend, error) {
if C.ggml_backend_is_cpu(b) {
// set number of threads for cpu backend
C.ggml_backend_cpu_set_n_threads(b, C.int(params.NumThreads))
C.ggml_backend_cpu_set_n_threads(b, C.int(Threads(params.NumThreads)))
}
}