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

This commit is contained in:
Justine Tunney 2024-04-26 10:05:33 -04:00 committed by GitHub
parent bbe3c6e761
commit 4b1c3c98b4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 87 additions and 89 deletions

View file

@ -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
}