summaryrefslogtreecommitdiff
path: root/llama.h
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2023-04-13 18:04:45 +0300
committerGeorgi Gerganov <ggerganov@gmail.com>2023-04-13 18:04:45 +0300
commit9190e8eac8bdc108c40d2d7505e9b45fa773251f (patch)
treef2f527a98d809c727e645769658065b5225f61e4 /llama.h
parentc85980acd04631a7c43d13676276f76ec72f5dfe (diff)
llama : merge llama_internal.h into llama.h
Hide it behind an #ifdef
Diffstat (limited to 'llama.h')
-rw-r--r--llama.h11
1 files changed, 11 insertions, 0 deletions
diff --git a/llama.h b/llama.h
index 7a258a1e..19221759 100644
--- a/llama.h
+++ b/llama.h
@@ -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