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

Added element-wise multiplication for vectors

Browse Source
This commit is contained in:
Andrew Kane
2023-07-15 20:19:51 -07:00
parent cd4ac17f9f
commit 08e7209810
7 changed files with 75 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/vector.c
View File

@@ -125,6 +125,14 @@ float_overflow_error(void)
(errcode(ERRCODE_NUMERIC_VALUE_OUT_OF_RANGE),
errmsg("value out of range: overflow")));
}
static pg_noinline void
float_underflow_error(void)
{
ereport(ERROR,
(errcode(ERRCODE_NUMERIC_VALUE_OUT_OF_RANGE),
errmsg("value out of range: underflow")));
}
#endif
/*
@@ -759,6 +767,42 @@ vector_sub(PG_FUNCTION_ARGS)
PG_RETURN_POINTER(result);
}
/*
* Multiply vectors
*/
PGDLLEXPORT PG_FUNCTION_INFO_V1(vector_mul);
Datum
vector_mul(PG_FUNCTION_ARGS)
{
Vector *a = PG_GETARG_VECTOR_P(0);
Vector *b = PG_GETARG_VECTOR_P(1);
float *ax = a->x;
float *bx = b->x;
Vector *result;
float *rx;
CheckDims(a, b);
result = InitVector(a->dim);
rx = result->x;
/* Auto-vectorized */
for (int i = 0, imax = a->dim; i < imax; i++)
rx[i] = ax[i] * bx[i];
/* Check for overflow and underflow */
for (int i = 0, imax = a->dim; i < imax; i++)
{
if (isinf(rx[i]))
float_overflow_error();
if (rx[i] == 0 && !(ax[i] == 0 || bx[i] == 0))
float_underflow_error();
}
PG_RETURN_POINTER(result);
}
/*
* Internal helper to compare vectors
*/