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

Improved code [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-09-23 22:07:30 -07:00
parent 261ba4a0d9
commit 724bde5020
2 changed files with 1 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/minivec.c
View File

@@ -652,7 +652,6 @@ MinivecL1Distance(int dim, fp8 * ax, fp8 * bx)
{ {
float distance = 0.0; float distance = 0.0;
/* Auto-vectorized */
for (int i = 0; i < dim; i++) for (int i = 0; i < dim; i++)
distance += fabsf(Fp8ToFloat4(ax[i]) - Fp8ToFloat4(bx[i])); distance += fabsf(Fp8ToFloat4(ax[i]) - Fp8ToFloat4(bx[i]));
@@ -697,7 +696,6 @@ minivec_l2_norm(PG_FUNCTION_ARGS)
fp8 *ax = a->x; fp8 *ax = a->x;
double norm = 0.0; double norm = 0.0;
/* Auto-vectorized */
for (int i = 0; i < a->dim; i++) for (int i = 0; i < a->dim; i++)
{ {
double axi = (double) Fp8ToFloat4(ax[i]); double axi = (double) Fp8ToFloat4(ax[i]);
@@ -724,7 +722,6 @@ minivec_l2_normalize(PG_FUNCTION_ARGS)
result = InitMiniVector(a->dim); result = InitMiniVector(a->dim);
rx = result->x; rx = result->x;
/* Auto-vectorized */
for (int i = 0; i < a->dim; i++) for (int i = 0; i < a->dim; i++)
norm += (double) Fp8ToFloat4(ax[i]) * (double) Fp8ToFloat4(ax[i]); norm += (double) Fp8ToFloat4(ax[i]) * (double) Fp8ToFloat4(ax[i]);
@@ -766,15 +763,8 @@ minivec_add(PG_FUNCTION_ARGS)
result = InitMiniVector(a->dim); result = InitMiniVector(a->dim);
rx = result->x; rx = result->x;
/* Auto-vectorized */
for (int i = 0, imax = a->dim; i < imax; i++) for (int i = 0, imax = a->dim; i < imax; i++)
{
#ifdef FLT16_SUPPORT
rx[i] = ax[i] + bx[i];
#else
rx[i] = Float4ToFp8Unchecked(Fp8ToFloat4(ax[i]) + Fp8ToFloat4(bx[i])); rx[i] = Float4ToFp8Unchecked(Fp8ToFloat4(ax[i]) + Fp8ToFloat4(bx[i]));
#endif
}
/* Check for overflow */ /* Check for overflow */
for (int i = 0, imax = a->dim; i < imax; i++) for (int i = 0, imax = a->dim; i < imax; i++)
@@ -805,15 +795,8 @@ minivec_sub(PG_FUNCTION_ARGS)
result = InitMiniVector(a->dim); result = InitMiniVector(a->dim);
rx = result->x; rx = result->x;
/* Auto-vectorized */
for (int i = 0, imax = a->dim; i < imax; i++) for (int i = 0, imax = a->dim; i < imax; i++)
{
#ifdef FLT16_SUPPORT
rx[i] = ax[i] - bx[i];
#else
rx[i] = Float4ToFp8Unchecked(Fp8ToFloat4(ax[i]) - Fp8ToFloat4(bx[i])); rx[i] = Float4ToFp8Unchecked(Fp8ToFloat4(ax[i]) - Fp8ToFloat4(bx[i]));
#endif
}
/* Check for overflow */ /* Check for overflow */
for (int i = 0, imax = a->dim; i < imax; i++) for (int i = 0, imax = a->dim; i < imax; i++)
@@ -844,15 +827,8 @@ minivec_mul(PG_FUNCTION_ARGS)
result = InitMiniVector(a->dim); result = InitMiniVector(a->dim);
rx = result->x; rx = result->x;
/* Auto-vectorized */
for (int i = 0, imax = a->dim; i < imax; i++) for (int i = 0, imax = a->dim; i < imax; i++)
{
#ifdef FLT16_SUPPORT
rx[i] = ax[i] * bx[i];
#else
rx[i] = Float4ToFp8Unchecked(Fp8ToFloat4(ax[i]) * Fp8ToFloat4(bx[i])); rx[i] = Float4ToFp8Unchecked(Fp8ToFloat4(ax[i]) * Fp8ToFloat4(bx[i]));
#endif
}
/* Check for overflow and underflow */ /* Check for overflow and underflow */
for (int i = 0, imax = a->dim; i < imax; i++) for (int i = 0, imax = a->dim; i < imax; i++)

2
src/minivec.h
View File

@@ -139,7 +139,7 @@ Float4ToFp8(float num)
{ {
fp8 result = Float4ToFp8Unchecked(num); fp8 result = Float4ToFp8Unchecked(num);
if (unlikely(Fp8IsNan(result)) && !isnan(num)) if (unlikely(Fp8IsNan(result)) && !isinf(num))
{ {
char *buf = palloc(FLOAT_SHORTEST_DECIMAL_LEN); char *buf = palloc(FLOAT_SHORTEST_DECIMAL_LEN);