thek0tyara/llama-cpp-turboquant
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 4

llama: print memory breakdown on exit (#15860)

Browse source 
* llama: print memory breakdown on exit
This commit is contained in:
Johannes Gäßler 2025-09-24 16:53:48 +02:00 • committed by GitHub
parent f2a789e334
commit e789095502
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
18 changed files with 243 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/llama-memory-recurrent.cpp
View file

@ -359,6 +359,14 @@ llama_pos llama_memory_recurrent::seq_pos_max(llama_seq_id seq_id) const {
return result;
}
std::map<ggml_backend_buffer_type_t, size_t> llama_memory_recurrent::memory_breakdown() const {
std::map<ggml_backend_buffer_type_t, size_t> ret;
for (const ggml_backend_buffer_ptr & buf_ptr : bufs) {
ret[ggml_backend_buffer_get_type(buf_ptr.get())] += ggml_backend_buffer_get_size(buf_ptr.get());
}
return ret;
}
llama_memory_context_ptr llama_memory_recurrent::init_batch(llama_batch_allocr & balloc, uint32_t n_ubatch, bool embd_all) {
do {
balloc.split_reset();