aboutsummaryrefslogtreecommitdiff
path: root/ggml-opencl.h
diff options
context:
space:
mode:
Diffstat (limited to 'ggml-opencl.h')
-rw-r--r--ggml-opencl.h24
1 files changed, 24 insertions, 0 deletions
diff --git a/ggml-opencl.h b/ggml-opencl.h
new file mode 100644
index 0000000..7bcc603
--- /dev/null
+++ b/ggml-opencl.h
@@ -0,0 +1,24 @@
+#pragma once
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+void ggml_cl_init(void);
+
+enum ggml_blas_order {
+ GGML_BLAS_ORDER_ROW_MAJOR = 101,
+ GGML_BLAS_ORDER_COLUMN_MAJOR = 102,
+};
+
+enum ggml_blas_op {
+ GGML_BLAS_OP_N = 111,
+ GGML_BLAS_OP_T = 112,
+ GGML_BLAS_OP_C = 113,
+};
+
+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);
+
+#ifdef __cplusplus
+}
+#endif