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

Improved memory estimation - #384 [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2023-12-20 11:10:27 -05:00
parent e88a425c9b
commit 2bff7ccaa2
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/hnswbuild.c
View File

@@ -349,8 +349,11 @@ HnswElementMemory(HnswElement e, int m)
elementSize += sizeof(HnswNeighborArray) * (e->level + 1); elementSize += sizeof(HnswNeighborArray) * (e->level + 1);
elementSize += sizeof(HnswCandidate) * (m * (e->level + 2)); elementSize += sizeof(HnswCandidate) * (m * (e->level + 2));
elementSize += sizeof(ItemPointerData); elementSize += sizeof(List);
elementSize += sizeof(ItemPointerData) + sizeof(uintptr_t);
elementSize += VARSIZE_ANY(DatumGetPointer(e->value)); elementSize += VARSIZE_ANY(DatumGetPointer(e->value));
/* Each allocation has a 64-bit header */
elementSize += (e->level + 7) * sizeof(uint64);
return elementSize; return elementSize;
} }
@@ -428,7 +431,7 @@ BuildCallback(Relation index, CALLBACK_ITEM_POINTER, Datum *values,
if (dup != NULL) if (dup != NULL)
{ {
HnswAddHeapTid(dup, tid); HnswAddHeapTid(dup, tid);
buildstate->memoryLeft -= sizeof(ItemPointerData); buildstate->memoryLeft -= sizeof(ItemPointerData) + sizeof(uintptr_t) + sizeof(uint64);
} }
/* Add to buildstate or free */ /* Add to buildstate or free */