KBaba7/llama.cpp

1#pragma once2 3#include "ggml.h"4#include "ggml-backend.h"5 6#ifdef  __cplusplus7extern "C" {8#endif9 10#define GGML_VK_NAME "Vulkan"11#define GGML_VK_MAX_DEVICES 1612 13GGML_BACKEND_API void ggml_vk_instance_init(void);14 15// backend API16GGML_BACKEND_API ggml_backend_t ggml_backend_vk_init(size_t dev_num);17 18GGML_BACKEND_API bool ggml_backend_is_vk(ggml_backend_t backend);19GGML_BACKEND_API int  ggml_backend_vk_get_device_count(void);20GGML_BACKEND_API void ggml_backend_vk_get_device_description(int device, char * description, size_t description_size);21GGML_BACKEND_API void ggml_backend_vk_get_device_memory(int device, size_t * free, size_t * total);22 23GGML_BACKEND_API ggml_backend_buffer_type_t ggml_backend_vk_buffer_type(size_t dev_num);24// pinned host buffer for use with the CPU backend for faster copies between CPU and GPU25GGML_BACKEND_API ggml_backend_buffer_type_t ggml_backend_vk_host_buffer_type(void);26 27GGML_BACKEND_API ggml_backend_reg_t ggml_backend_vk_reg(void);28 29#ifdef  __cplusplus30}31#endif32