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

Switched to datumIsEqual for duplicate check in IVFFlat [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-11 16:37:34 -07:00
parent bbfe1e8b61
commit 000cc13c29
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/ivfkmeans.c
View File

@@ -5,6 +5,7 @@
#include "ivfflat.h" #include "ivfflat.h"
#include "miscadmin.h" #include "miscadmin.h"
#include "utils/datum.h"
#include "utils/memutils.h" #include "utils/memutils.h"
/* /*
@@ -123,11 +124,11 @@ QuickCenters(Relation index, VectorArray samples, VectorArray centers)
qsort(samples->items, samples->length, samples->itemsize, CompareVectors); qsort(samples->items, samples->length, samples->itemsize, CompareVectors);
for (int i = 0; i < samples->length; i++) for (int i = 0; i < samples->length; i++)
{ {
Vector *vec = (Vector *) VectorArrayGet(samples, i); Datum vec = PointerGetDatum(VectorArrayGet(samples, i));
if (i == 0 || CompareVectors(vec, VectorArrayGet(samples, i - 1)) != 0) if (i == 0 || !datumIsEqual(vec, PointerGetDatum(VectorArrayGet(samples, i - 1)), false, -1))
{ {
VectorArraySet(centers, centers->length, vec); VectorArraySet(centers, centers->length, DatumGetPointer(vec));
centers->length++; centers->length++;
} }
} }
@@ -509,7 +510,7 @@ CheckCenters(Relation index, VectorArray centers)
qsort(centers->items, centers->length, centers->itemsize, CompareVectors); qsort(centers->items, centers->length, centers->itemsize, CompareVectors);
for (int i = 1; i < centers->length; i++) for (int i = 1; i < centers->length; i++)
{ {
if (CompareVectors(VectorArrayGet(centers, i), VectorArrayGet(centers, i - 1)) == 0) if (datumIsEqual(PointerGetDatum(VectorArrayGet(centers, i)), PointerGetDatum(VectorArrayGet(centers, i - 1)), false, -1))
elog(ERROR, "Duplicate centers detected. Please report a bug."); elog(ERROR, "Duplicate centers detected. Please report a bug.");
} }