CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-11-04 09:32:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

gguf : mmap tensor data example

Browse Source
This commit is contained in:
M. Yusuf Sarıgöz
2023-07-31 17:46:12 +03:00
parent b26f5b2e43
commit bb42aefaeb
1 changed files with 26 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
examples/gguf/gguf.cpp
View File

@@ -1,4 +1,5 @@
#include "ggml.h"
#include "llama-util.h"
#include <cstdio>
#include <cinttypes>
@@ -376,6 +377,30 @@ bool gguf_ex_read_2(const std::string & fname) {
// TODO: mmap based on tensor infos
struct llama_file file(fname.c_str(), "rb");
llama_mmap data_mmap(&file, 0, false);
const int n_tensors = gguf_get_n_tensors(ctx);
for (int i = 0; i < n_tensors; ++i) {
const char * name = gguf_get_tensor_name(ctx, i);
const size_t offset = gguf_get_data_offset(ctx) + gguf_get_tensor_offset(ctx, i);
struct ggml_tensor * cur = ggml_get_tensor(ctx_data, name);
cur->data = static_cast<char *>(data_mmap.addr) + offset;
// print first 10 elements
const float * data = (const float *) cur->data;
printf("%s data[:10] : ", name);
for (int j = 0; j < 10; ++j) {
printf("%f ", data[j]);
}
printf("\n\n");
}
fprintf(stdout, "%s: ctx_data size: %zu\n", __func__, ggml_get_mem_size(ctx_data));
ggml_free(ctx_data);