diff options
Diffstat (limited to 'llama_internal.h')
-rw-r--r-- | llama_internal.h | 12 |
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 |