Fixed and readded debug code for causes

ggml-org · Aug 16, 2024 · fe40950 · fe40950
1 parent d86f10a
commit fe40950
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ggml/src/ggml-backend.c b/ggml/src/ggml-backend.c
@@ -1120,8 +1120,15 @@ static int ggml_backend_sched_backend_from_buffer(ggml_backend_sched_t sched, co
     return -1;
 }
 
+#if 0
+#define GGML_SCHED_MAX_SPLITS_DEBUG 4096
+static char causes[GGML_DEFAULT_GRAPH_SIZE*16 + GGML_SCHED_MAX_SPLITS_DEBUG*GGML_SCHED_MAX_SPLIT_INPUTS][128]; // debug only
+#define SET_CAUSE(node, ...) sprintf(causes[hash_id(node)], __VA_ARGS__)
+#define GET_CAUSE(node) causes[hash_id(node)]
+#else
 #define SET_CAUSE(node, ...)
 #define GET_CAUSE(node) ""
+#endif
 
 // returns the backend that should be used for the node based on the current locations
 static int ggml_backend_sched_backend_id_from_cur(ggml_backend_sched_t sched, struct ggml_tensor * tensor) {