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

Improved code

Browse Source
This commit is contained in:
Andrew Kane
2024-01-07 18:30:51 -08:00
parent cb4c770df2
commit c7fe1571ee
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/hnswbuild.c
View File

@@ -353,9 +353,12 @@ InsertTupleInMemory(Relation index, Datum *values, ItemPointer heaptid, HnswBuil
/* Look for duplicate */ /* Look for duplicate */
dup = HnswFindDuplicate(element); dup = HnswFindDuplicate(element);
/* Update neighbors if needed */
if (dup == NULL) if (dup == NULL)
{ {
/* Add element */
slist_push_head(&buildstate->elements, &element->next);
/* Update neighbors */
for (int lc = element->level; lc >= 0; lc--) for (int lc = element->level; lc >= 0; lc--)
{ {
int lm = HnswGetLayerM(m, lc); int lm = HnswGetLayerM(m, lc);
@@ -364,14 +367,11 @@ InsertTupleInMemory(Relation index, Datum *values, ItemPointer heaptid, HnswBuil
for (int i = 0; i < neighbors->length; i++) for (int i = 0; i < neighbors->length; i++)
HnswUpdateConnection(element, &neighbors->items[i], lm, lc, NULL, NULL, procinfo, collation); HnswUpdateConnection(element, &neighbors->items[i], lm, lc, NULL, NULL, procinfo, collation);
} }
/* Update entry point if needed */
if (entryPoint == NULL || element->level > entryPoint->level)
buildstate->entryPoint = element;
} }
/* Update entry point if needed */
if (dup == NULL && (entryPoint == NULL || element->level > entryPoint->level))
buildstate->entryPoint = element;
if (dup == NULL)
slist_push_head(&buildstate->elements, &element->next);
else else
{ {
/* No need to free element since memory unlikely to be reallocated */ /* No need to free element since memory unlikely to be reallocated */