CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-10-30 08:42:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

ggml : optimize non-SIMD Q4_0 vector dot product (#703)

Browse Source
This commit is contained in:
Stephan Walter
2023-04-13 14:59:50 +00:00
committed by GitHub
parent 6c248707f5
commit 6232f2d7fd
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ggml.c
View File

@@ -2160,18 +2160,20 @@ static void ggml_vec_dot_q4_0(const int n, float * restrict s, const void * rest
const uint8_t * restrict p0 = x[i].qs; const uint8_t * restrict p0 = x[i].qs;
const uint8_t * restrict p1 = y[i].qs; const uint8_t * restrict p1 = y[i].qs;
int sumi = 0;
for (int j = 0; j < QK/2; j++) { for (int j = 0; j < QK/2; j++) {
const uint8_t v0 = p0[j]; const uint8_t v0 = p0[j];
const uint8_t v1 = p1[j]; const uint8_t v1 = p1[j];
const float f0 = d0*((int8_t) (v0 & 0xf) - 8); const int8_t i0 = (int8_t) (v0 & 0xf) - 8;
const float f1 = d0*((int8_t) (v0 >> 4) - 8); const int8_t i1 = (int8_t) (v0 >> 4) - 8;
const float f2 = d1*((int8_t) (v1 & 0xf) - 8); const int8_t i2 = (int8_t) (v1 & 0xf) - 8;
const float f3 = d1*((int8_t) (v1 >> 4) - 8); const int8_t i3 = (int8_t) (v1 >> 4) - 8;
sumf += f0*f2 + f1*f3; sumi += i0*i2 + i1*i3;
} }
sumf += d0 * d1 * sumi;
} }
#endif #endif