gguf : add gguf_get_tensor_name()

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

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;
}
char * gguf_get_tensor_name(struct gguf_context * ctx, int i) {
return ctx->infos[i].name.data;
}
////////////////////////////////////////////////////////////////////////////////
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 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