diff options
Diffstat (limited to 'llama.cpp/ggml/src/ggml-zdnn/utils.hpp')
| -rw-r--r-- | llama.cpp/ggml/src/ggml-zdnn/utils.hpp | 19 |
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 |
