partially revert changes

This commit is contained in:
Cebtenzzre 2023-09-30 12:52:15 -04:00
parent f12863ed38
commit d7f06a2d17

View file

@ -435,24 +435,24 @@ void mul_vec_q_n_f32(device const void * src0, device const float * src1, device
const uint offset0 = first_row * nb + im/gqa*(nb*ne0);
device const block_q_type * x = (device const block_q_type *) src0 + offset0;
device const float * y = (device const float *) src1 + r1*ne10 + im*ne00*ne1;
float yl[QK4_0/2]; // src1 vector cache
float yl[16]; // src1 vector cache
float sumf[nr]={0.f};
const int ix = tiisg/2;
const int il = QK4_0/4*(tiisg%2);
const int il = 8*(tiisg%2);
device const float * yb = y + ix * QK4_0 + il;
// each thread in a SIMD group deals with half a block.
for (int ib = ix; ib < nb; ib += nw/2) {
float sumy = 0;
for (int i = 0; i < QK4_0/4; i += 2) {
for (int i = 0; i < 8; i += 2) {
sumy += yb[i] + yb[i+1];
yl[i+0] = yb[i+ 0];
yl[i+1] = yb[i+ 1]/256.f;
sumy += yb[i+QK4_0/2] + yb[i+QK4_0/2+1];
yl[i+QK4_0/4] = yb[i+QK4_0/2] /16.f;
yl[i+QK4_0/4+1] = yb[i+QK4_0/2+1]/4096.f;
sumy += yb[i+16] + yb[i+17];
yl[i+8] = yb[i+16]/16.f;
yl[i+9] = yb[i+17]/4096.f;
}
for (int row = 0; row < nr; row++) {