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-model.cpp
View file

@ -6005,6 +6005,14 @@ size_t llama_model::n_devices() const {
return devices.size();
}
std::map<ggml_backend_buffer_type_t, size_t> llama_model::memory_breakdown() const {
std::map<ggml_backend_buffer_type_t, size_t> ret;
for (const ggml_backend_buffer_ptr & buf_ptr : pimpl->bufs) {
ret[ggml_backend_buffer_get_type(buf_ptr.get())] += ggml_backend_buffer_get_size(buf_ptr.get());
}
return ret;
}
uint64_t llama_model::n_elements() const {
return pimpl->n_elements;
}