]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commit
server : free draft/MTP resources on sleep to fix VRAM leak (#23461)
authorAman Gupta <redacted>
Thu, 21 May 2026 08:11:11 +0000 (16:11 +0800)
committerGitHub <redacted>
Thu, 21 May 2026 08:11:11 +0000 (16:11 +0800)
commit52fb93a2bd6b12673b9f4f225e61968e70443b11
treec119ca8dd7cecdeccc0316bc65aba9e626482e59
parentc9021714e843dd02670f18bd826be076bc58502c
server : free draft/MTP resources on sleep to fix VRAM leak (#23461)

The destroy() function in server_context_impl only cleaned up the main
model and context (via llama_init.reset()) but did not free the speculative
decoder (spec), draft context (ctx_dft), or draft model (model_dft).

For MTP (Multi-Token Prediction) models, ctx_dft holds GPU-allocated
resources (KV cache, compute buffers) that are not freed when entering
the sleeping state. On each sleep/resume cycle, new resources are
allocated without the old ones being freed, leading to a VRAM leak
that eventually crashes the server with out-of-memory errors.

Fix by explicitly resetting spec, ctx_dft, and model_dft in destroy()
before resetting llama_init, ensuring proper cleanup order to avoid
use-after-free.

ref: https://github.com/ggml-org/llama.cpp/issues/23395

Assisted-by: llama.cpp:local pi
tools/server/server-context.cpp