File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -4283,7 +4283,7 @@ static __device__ __forceinline__ float vec_dot_iq2_xxs_q8_1(
4283
4283
q8 += 8 ;
4284
4284
aux32 >>= 7 ;
4285
4285
}
4286
- const float d = (float )bq2->d * (0 .5f + aux32) * ( float ) bq8_1[ib32].ds . x * 0 .25f ;
4286
+ const float d = (float )bq2->d * (0 .5f + aux32) * __low2float ( bq8_1[ib32].ds ) * 0 .25f ;
4287
4287
return d * sumi;
4288
4288
#else
4289
4289
// iqs is 0...15
@@ -4294,7 +4294,7 @@ static __device__ __forceinline__ float vec_dot_iq2_xxs_q8_1(
4294
4294
const uint8_t * grid1 = (const uint8_t *)(iq2xxs_grid + aux8[2 *il+0 ]);
4295
4295
const uint8_t * grid2 = (const uint8_t *)(iq2xxs_grid + aux8[2 *il+1 ]);
4296
4296
const uint32_t aux32 = q2[2 ] | (q2[3 ] << 16 );
4297
- const float d = (float )bq2->d * (0 .5f + (aux32 >> 28 )) * ( float ) bq8_1[ib32].ds . x * 0 .25f ;
4297
+ const float d = (float )bq2->d * (0 .5f + (aux32 >> 28 )) * __low2float ( bq8_1[ib32].ds ) * 0 .25f ;
4298
4298
const uint8_t signs1 = ksigns_iq2xs[(aux32 >> 14 *il) & 127 ];
4299
4299
const uint8_t signs2 = ksigns_iq2xs[(aux32 >> (14 *il + 7 )) & 127 ];
4300
4300
const int8_t * q8 = bq8_1[ib32].qs + 16 *il;
@@ -4339,7 +4339,7 @@ static __device__ __forceinline__ float vec_dot_iq2_xs_q8_1(
4339
4339
}
4340
4340
q8 += 8 ;
4341
4341
}
4342
- const float d = (float )bq2->d * ( float ) bq8_1[ib32].ds . x * 0 .25f ;
4342
+ const float d = (float )bq2->d * __low2float ( bq8_1[ib32].ds ) * 0 .25f ;
4343
4343
return d * ((0 .5f + ls1) * sumi1 + (0 .5f + ls2) * sumi2);
4344
4344
#else
4345
4345
assert (false );
You can’t perform that action at this time.
0 commit comments