liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-06-30 17:51:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added check for NaN

Browse Source
This commit is contained in:
Andrew Kane
2023-06-04 10:52:28 -07:00
parent e971fdd4fd
commit 482a5f8b66
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/ivfkmeans.c
View File

@@ -469,13 +469,16 @@ CheckCenters(Relation index, VectorArray centers)
if (centers->length != centers->maxlen) if (centers->length != centers->maxlen)
elog(ERROR, "Not enough centers. Please report a bug."); elog(ERROR, "Not enough centers. Please report a bug.");
/* Ensure no infinite values */ /* Ensure no NaN or infinite values */
for (i = 0; i < centers->length; i++) for (i = 0; i < centers->length; i++)
{ {
vec = VectorArrayGet(centers, i); vec = VectorArrayGet(centers, i);
for (j = 0; j < vec->dim; j++) for (j = 0; j < vec->dim; j++)
{ {
if (isnan(vec->x[j]))
elog(ERROR, "NaN detected. Please report a bug.");
if (isinf(vec->x[j])) if (isinf(vec->x[j]))
elog(ERROR, "Infinite value detected. Please report a bug."); elog(ERROR, "Infinite value detected. Please report a bug.");
} }