summaryrefslogtreecommitdiff
path: root/llama.cpp/ggml/src/ggml-cpu/llamafile/sgemm.h
diff options
context:
space:
mode:
Diffstat (limited to 'llama.cpp/ggml/src/ggml-cpu/llamafile/sgemm.h')
-rw-r--r--llama.cpp/ggml/src/ggml-cpu/llamafile/sgemm.h25
1 files changed, 25 insertions, 0 deletions
diff --git a/llama.cpp/ggml/src/ggml-cpu/llamafile/sgemm.h b/llama.cpp/ggml/src/ggml-cpu/llamafile/sgemm.h
new file mode 100644
index 0000000..867b0c0
--- /dev/null
+++ b/llama.cpp/ggml/src/ggml-cpu/llamafile/sgemm.h
@@ -0,0 +1,25 @@
+#pragma once
+#include <stdint.h>
+#include <stdbool.h>
+
+#if defined(__VXE__) || defined(__VXE2__)
+#include <vecintrin.h>
+#endif
+
+#ifdef _MSC_VER
+#define NOINLINE __declspec(noinline)
+#else
+#define NOINLINE __attribute__((__noinline__))
+#endif
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+bool llamafile_sgemm(const struct ggml_compute_params * params, int64_t, int64_t, int64_t,
+ const void *, int64_t, const void *, int64_t, void *, int64_t,
+ int, int, int);
+
+#ifdef __cplusplus
+}
+#endif