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 9c3bfc1499
commit 405b9c77ad
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