vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

gguf : add gguf_get_tensor_name()

Browse source
This commit is contained in:
Georgi Gerganov 2023-07-26 16:36:03 +03:00
parent 78b226a959
commit 860c9c63ce
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
2 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ggml.c
View file

@ -18770,6 +18770,10 @@ size_t gguf_get_tensor_offset(struct gguf_context * ctx, int i) {
return ctx->infos[i].offset; return ctx->infos[i].offset;
} }
char * gguf_get_tensor_name(struct gguf_context * ctx, int i) {
return ctx->infos[i].name.data;
}
//////////////////////////////////////////////////////////////////////////////// ////////////////////////////////////////////////////////////////////////////////
int ggml_cpu_has_avx(void) { int ggml_cpu_has_avx(void) {

1
ggml.h
View file

@ -1670,6 +1670,7 @@ extern "C" {
GGML_API int gguf_get_n_tensors (struct gguf_context * ctx); GGML_API int gguf_get_n_tensors (struct gguf_context * ctx);
GGML_API size_t gguf_get_tensor_offset(struct gguf_context * ctx, int i); GGML_API size_t gguf_get_tensor_offset(struct gguf_context * ctx, int i);
GGML_API char * gguf_get_tensor_name (struct gguf_context * ctx, int i);
// //
// system info // system info