liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-14 16:46:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Started Neon intrinsics [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-27 22:50:47 -07:00
parent a72511db7f
commit 17855c9861
1 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/halfutils.c
View File

@@ -1,5 +1,7 @@
#include "postgres.h" #include "postgres.h"
#include <arm_neon.h>
#include "halfutils.h" #include "halfutils.h"
#include "halfvec.h" #include "halfvec.h"
@@ -28,9 +30,28 @@ static float
HalfvecL2SquaredDistanceDefault(int dim, half * ax, half * bx) HalfvecL2SquaredDistanceDefault(int dim, half * ax, half * bx)
{ {
float distance = 0.0; float distance = 0.0;
int i = 0;
/* TODO Improve */
#ifdef __ARM_FEATURE_FP16_VECTOR_ARITHMETIC
int count = (dim / 8) * 8;
float32x4_t dist = vmovq_n_f32(0);
for (; i < count; i += 8)
{
float16x8_t axs = vld1q_f16((const __fp16 *) (ax + i));
float16x8_t bxs = vld1q_f16((const __fp16 *) (bx + i));
float16x8_t diff = vsubq_f16(axs, bxs);
dist = vfmlalq_low_f16(dist, diff, diff);
dist = vfmlalq_high_f16(dist, diff, diff);
}
distance += vaddvq_f32(dist);
#endif
/* Auto-vectorized */ /* Auto-vectorized */
for (int i = 0; i < dim; i++) for (; i < dim; i++)
{ {
float diff = HalfToFloat4(ax[i]) - HalfToFloat4(bx[i]); float diff = HalfToFloat4(ax[i]) - HalfToFloat4(bx[i]);