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

Use fused multiply-add for cosine distance

Browse Source
This commit is contained in:
Andrew Kane
2023-07-18 16:10:17 -07:00
parent 8a05debda1
commit b710dc68a0
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/vector.c
View File

@@ -627,9 +627,9 @@ cosine_distance(PG_FUNCTION_ARGS)
Vector *b = PG_GETARG_VECTOR_P(1); Vector *b = PG_GETARG_VECTOR_P(1);
float *ax = a->x; float *ax = a->x;
float *bx = b->x; float *bx = b->x;
double distance = 0.0; float distance = 0.0;
double norma = 0.0; float norma = 0.0;
double normb = 0.0; float normb = 0.0;
CheckDims(a, b); CheckDims(a, b);
@@ -642,7 +642,7 @@ cosine_distance(PG_FUNCTION_ARGS)
} }
/* Use sqrt(a * b) over sqrt(a) * sqrt(b) */ /* Use sqrt(a * b) over sqrt(a) * sqrt(b) */
PG_RETURN_FLOAT8(1 - (distance / sqrt(norma * normb))); PG_RETURN_FLOAT8(1.0 - ((double) distance / sqrt((double) norma * (double) normb)));
} }
/* /*