vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update ggml.h

Browse source
This commit is contained in:
John 2024-01-22 23:31:24 +01:00 • committed by GitHub
parent d85a629a6c
commit 607fbe99c7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ggml.h
View file

@ -1899,6 +1899,10 @@ extern "C" {
// dump the graph into a file using the dot format // 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); 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 // build gradient checkpointing backward graph gb for gf using provided checkpoints
// gb_tmp will contain original backward graph with rewritten backward process nodes, // gb_tmp will contain original backward graph with rewritten backward process nodes,
// but without the second forward pass nodes. // but without the second forward pass nodes.