1 files changed, 7 insertions, 0 deletions
diff --git a/llama.h b/llama.h
index 04e2bf7..deb09fe 100644
--- a/llama.h
+++ b/llama.h
@@ -164,6 +164,13 @@ extern "C" {
 
 #ifdef __cplusplus
 }
+
+#include <string>
+#include <unordered_map>
+//
+// Internal function exposed for tests and benchmarks
+//
+std::unordered_map<std::string, struct ggml_tensor *>& llama_internal_get_tensor_map(struct llama_context * ctx);
 #endif
 
 #endif