ggml-opencl.h 639 B

123456789101112131415161718192021222324
  1. #pragma once
  2. #ifdef __cplusplus
  3. extern "C" {
  4. #endif
  5. void ggml_cl_init(void);
  6. enum ggml_blas_order {
  7. GGML_BLAS_ORDER_ROW_MAJOR = 101,
  8. GGML_BLAS_ORDER_COLUMN_MAJOR = 102,
  9. };
  10. enum ggml_blas_op {
  11. GGML_BLAS_OP_N = 111,
  12. GGML_BLAS_OP_T = 112,
  13. GGML_BLAS_OP_C = 113,
  14. };
  15. void ggml_cl_sgemm_wrapper(const enum ggml_blas_order order, const enum ggml_blas_op trans_a, const enum ggml_blas_op trans_b, const int m, const int n, const int k, const float alpha, const void *host_a, const int lda, const float *host_b, const int ldb, const float beta, float *host_c, const int ldc, const int btype);
  16. #ifdef __cplusplus
  17. }
  18. #endif