aboutsummaryrefslogtreecommitdiff
path: root/llama_internal.h
diff options
context:
space:
mode:
Diffstat (limited to 'llama_internal.h')
-rw-r--r--llama_internal.h12
1 files changed, 12 insertions, 0 deletions
diff --git a/llama_internal.h b/llama_internal.h
new file mode 100644
index 0000000..543eed9
--- /dev/null
+++ b/llama_internal.h
@@ -0,0 +1,12 @@
+// Internal header to be included by llama.cpp and tests/benchmarks only.
+
+#ifndef LLAMA_INTERNAL_H
+#define LLAMA_INTERNAL_H
+
+#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 // LLAMA_INTERNAL_H