From 607fbe99c7c3d93bed911ffe612d6c211f4bda80 Mon Sep 17 00:00:00 2001 From: John <78893154+cmp-nct@users.noreply.github.com> Date: Mon, 22 Jan 2024 23:31:24 +0100 Subject: [PATCH] Update ggml.h --- ggml.h | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/ggml.h b/ggml.h index dca7bd9ce..6c501e170 100644 --- a/ggml.h +++ b/ggml.h @@ -1899,6 +1899,10 @@ extern "C" { // dump the graph into a file using the dot format GGML_API void ggml_graph_dump_dot(const struct ggml_cgraph * gb, const struct ggml_cgraph * gf, const char * filename); + // visualize the tensor - extended adds more information - when printing sample content extended will also print src0 and src1 content + // example: ggml_tensor_printf(some_ggml_tensor,"function_name",0,true,true); + void ggml_tensor_printf(const struct ggml_tensor *tensor, char *prefix, int line, bool extended, bool print_sample); + // build gradient checkpointing backward graph gb for gf using provided checkpoints // gb_tmp will contain original backward graph with rewritten backward process nodes, // but without the second forward pass nodes.