llamafile : use 64-bit integers in sgemm (#6928)

1 files changed, 4 insertions, 2 deletions

diff --git a/sgemm.h b/sgemm.h
index da23b209..f29747d0 100644
--- a/sgemm.h
+++ b/sgemm.h
@@ -1,11 +1,13 @@
 #pragma once
+#include <stdint.h>
 #include <stdbool.h>
 #ifdef __cplusplus
 extern "C" {
 #endif
 
-bool llamafile_sgemm(int, int, int, const void *, int, const void *, int,
-                     void *, int, int, int, int, int, int, int);
+bool llamafile_sgemm(int64_t, int64_t, int64_t, const void *, int64_t,
+                     const void *, int64_t, void *, int64_t, int, int,
+                     int, int, int, int);
 
 #ifdef __cplusplus
 }