mirror of
https://github.com/ollama/ollama.git
synced 2025-05-10 18:06:33 +02:00
33 lines
1.1 KiB
Diff
33 lines
1.1 KiB
Diff
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
|
|
From: jmorganca <jmorganca@gmail.com>
|
|
Date: Tue, 8 Apr 2025 20:36:41 -0700
|
|
Subject: [PATCH] ollama debug tensor
|
|
|
|
---
|
|
ggml/src/ggml-cpu/ggml-cpu.c | 6 ++++++
|
|
1 file changed, 6 insertions(+)
|
|
|
|
diff --git a/ggml/src/ggml-cpu/ggml-cpu.c b/ggml/src/ggml-cpu/ggml-cpu.c
|
|
index 34624cca..59bd3c62 100644
|
|
--- a/ggml/src/ggml-cpu/ggml-cpu.c
|
|
+++ b/ggml/src/ggml-cpu/ggml-cpu.c
|
|
@@ -15,6 +15,8 @@
|
|
#include "ops.h"
|
|
#include "ggml.h"
|
|
|
|
+#include "ollama-debug.h"
|
|
+
|
|
#if defined(_MSC_VER) || defined(__MINGW32__)
|
|
#include <malloc.h> // using malloc.h with MSC/MINGW
|
|
#elif !defined(__FreeBSD__) && !defined(__NetBSD__) && !defined(__OpenBSD__)
|
|
@@ -2859,6 +2861,10 @@ static thread_ret_t ggml_graph_compute_thread(void * data) {
|
|
|
|
ggml_compute_forward(¶ms, node);
|
|
|
|
+#ifdef OLLAMA_DEBUG
|
|
+ ollama_debug(node, true);
|
|
+#endif
|
|
+
|
|
if (state->ith == 0 && cplan->abort_callback &&
|
|
cplan->abort_callback(cplan->abort_callback_data)) {
|
|
atomic_store_explicit(&tp->abort, node_n + 1, memory_order_relaxed);
|