llama : merge llama_internal.h into llama.h

Hide it behind an #ifdef
This commit is contained in:
Georgi Gerganov 2023-04-13 18:04:45 +03:00
parent c85980acd0
commit 9190e8eac8
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
6 changed files with 14 additions and 16 deletions

11
llama.h
View file

@ -179,4 +179,15 @@ extern "C" {
}
#endif
// Internal API to be implemented by llama.cpp and used by tests/benchmarks only
#ifdef LLAMA_API_INTERNAL
#include <vector>
#include <string>
struct ggml_tensor;
std::vector<std::pair<std::string, struct ggml_tensor *>>& llama_internal_get_tensor_map(struct llama_context * ctx);
#endif
#endif // LLAMA_H