summaryrefslogtreecommitdiff
path: root/llama.cpp/ggml/src/ggml-zdnn/utils.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'llama.cpp/ggml/src/ggml-zdnn/utils.hpp')
-rw-r--r--llama.cpp/ggml/src/ggml-zdnn/utils.hpp19
1 files changed, 19 insertions, 0 deletions
diff --git a/llama.cpp/ggml/src/ggml-zdnn/utils.hpp b/llama.cpp/ggml/src/ggml-zdnn/utils.hpp
new file mode 100644
index 0000000..c1e2028
--- /dev/null
+++ b/llama.cpp/ggml/src/ggml-zdnn/utils.hpp
@@ -0,0 +1,19 @@
+#ifndef GGML_ZDNN_UTILITIES_HPP
+#define GGML_ZDNN_UTILITIES_HPP
+
+#include "common.hpp"
+
+zdnn_data_types ggml_zdnn_type_mapping(ggml_type type);
+
+void ggml_zdnn_create_tensor(zdnn_tensor_desc & pre_tfm_desc,
+ zdnn_tensor_desc & tfm_desc,
+ zdnn_ztensor & ztensor,
+ const ggml_tensor * src,
+ const int64_t * ne,
+ const zdnn_data_layouts layout);
+
+void ggml_zdnn_load_tensor(zdnn_ztensor & ztensor, void * buffer);
+
+void ggml_zdnn_init_tensor(ggml_backend_zdnn_buffer * buffer, const ggml_tensor * tensor);
+
+#endif // GGML_ZDNN_UTILITIES_HPP