ggml-sycl.h 712 B

123456789101112131415161718192021222324252627
  1. /*MIT license
  2. Copyright (C) 2024 Intel Corporation
  3. SPDX-License-Identifier: MIT
  4. */
  5. #pragma once
  6. #include "ggml.h"
  7. #include "ggml-backend.h"
  8. #ifdef __cplusplus
  9. extern "C" {
  10. #endif
  11. #define GGML_SYCL_MAX_DEVICES 16
  12. #define GGML_SYCL_NAME "SYCL"
  13. GGML_API void ggml_init_sycl(void);
  14. GGML_API bool ggml_sycl_compute_forward(struct ggml_compute_params * params, struct ggml_tensor * tensor);
  15. GGML_API ggml_backend_t ggml_backend_sycl_init(int device);
  16. GGML_API ggml_backend_buffer_type_t ggml_backend_sycl_buffer_type(int device);
  17. GGML_API ggml_backend_buffer_type_t ggml_backend_sycl_host_buffer_type(void);
  18. GGML_API void ggml_backend_sycl_print_sycl_devices(void);
  19. #ifdef __cplusplus
  20. }
  21. #endif