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

Ensure order is deterministic for SelectNeighbors closer caching

Browse Source
This commit is contained in:
Andrew Kane
2023-10-06 12:26:53 -07:00
parent 62482e3760
commit cae162ffc6
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/hnswutils.c
View File

@@ -887,6 +887,12 @@ CompareCandidateDistances(const void *a, const void *b)
if (hca->distance > hcb->distance) if (hca->distance > hcb->distance)
return -1; return -1;
if (hca->element < hcb->element)
return 1;
if (hca->element > hcb->element)
return -1;
return 0; return 0;
} }
@@ -1048,6 +1054,10 @@ HnswInsertElement(HnswElement element, HnswElement entryPoint, Relation index, F
else else
lw = w; lw = w;
/* Ensure order is deterministic for SelectNeighbors closer caching */
if (index == NULL)
list_sort(lw, CompareCandidateDistances);
neighbors = SelectNeighbors(lw, lm, lc, procinfo, collation, element, NULL, NULL); neighbors = SelectNeighbors(lw, lm, lc, procinfo, collation, element, NULL, NULL);
AddConnections(element, neighbors, lm, lc); AddConnections(element, neighbors, lm, lc);