vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix build issues in sgemm.cpp post rebase

Browse source
This commit is contained in:
Srihari-mcw 2024-07-07 19:29:04 -07:00
parent c480818d97
commit 5a6a235ac7
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
ggml/src/llamafile/sgemm.cpp
View file

@ -829,9 +829,8 @@ class tinyBLAS_Q0_B16_AVX {
: A(A), B(B), C(C), k(k), lda(lda), ldb(ldb), ldc(ldc), ith(ith), nth(nth) { : A(A), B(B), C(C), k(k), lda(lda), ldb(ldb), ldc(ldc), ith(ith), nth(nth) {
} }
void matmul(int64_t m, int64_t n, int task) { void matmul(int64_t m, int64_t n) {
if (task == GGML_TASK_TYPE_COMPUTE) mnpack(0, m, 0, n);
mnpack(0, m, 0, n);
} }
private: private:
@ -1398,7 +1397,7 @@ bool llamafile_sgemm(int64_t m, int64_t n, int64_t k, const void *A, int64_t lda
(const block_q8_0 *)B, ldb, (const block_q8_0 *)B, ldb,
(float *)C, ldc, (float *)C, ldc,
ith, nth}; ith, nth};
tb.matmul(m, n, task); tb.matmul(m, n);
return true; return true;
#else #else
return false; return false;
@ -1414,7 +1413,7 @@ bool llamafile_sgemm(int64_t m, int64_t n, int64_t k, const void *A, int64_t lda
(const block_q8_0 *)B, ldb, (const block_q8_0 *)B, ldb,
(float *)C, ldc, (float *)C, ldc,
ith, nth}; ith, nth};
tb.matmul(m, n, task); tb.matmul(m, n);
return true; return true;
#else #else
return false; return false;