aditya/llama.cpp
1
0
Fork 0
mirror of https://git.adityakumar.xyz/llama.cpp.git synced 2024-11-12 16:29:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix potential int8 overflow in non-SIMD vec_dot (#986)

Browse source
This commit is contained in:
Stephan Walter 2023-04-15 18:28:56 +00:00 committed by GitHub
parent 0ad964631f
commit 2f7c8e014e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
ggml.c
View file

@ -2373,11 +2373,11 @@ static void ggml_vec_dot_q4_0(const int n, float * restrict s, const void * rest
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 int8_t i0 = (int8_t) (v0 & 0xf) - 8; const int i0 = (v0 & 0xf) - 8;
const int8_t i1 = (int8_t) (v0 >> 4) - 8; const int i1 = (v0 >> 4) - 8;
const int8_t i2 = (int8_t) (v1 & 0xf) - 8; const int i2 = (v1 & 0xf) - 8;
const int8_t i3 = (int8_t) (v1 >> 4) - 8; const int i3 = (v1 >> 4) - 8;
sumi += i0*i2 + i1*i3; sumi += i0*i2 + i1*i3;
} }