mnist: fix segmentation fault (ggml/1227)

This commit is contained in:
Johannes Gäßler 2025-05-19 09:33:35 +02:00 committed by Georgi Gerganov
parent 8b5e19aea6
commit 6c35981a64
2 changed files with 7 additions and 0 deletions

View file

@ -128,6 +128,8 @@ extern "C" {
// set gradients to zero, initilize loss, and optionally reset the optimizer
GGML_API void ggml_opt_reset(ggml_opt_context_t opt_ctx, bool optimizer);
GGML_API bool ggml_opt_static_graphs(ggml_opt_context_t opt_ctx); // whether the graphs are allocated_statically
// get underlying tensors that store data
// if not using static graphs these pointers become invalid with the next call to ggml_opt_alloc
GGML_API struct ggml_tensor * ggml_opt_inputs( ggml_opt_context_t opt_ctx); // forward graph input tensor