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

Improved memory clean-up [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2023-08-06 14:09:42 -07:00
parent 72a8f68dc5
commit 0ffdefe138
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/hnswbuild.c
View File

@@ -183,6 +183,9 @@ CreateElementPages(HnswBuildState * buildstate)
UnlockReleaseBuffer(buf); UnlockReleaseBuffer(buf);
HnswUpdateMetaPage(index, true, buildstate->entryPoint, insertPage, forkNum); HnswUpdateMetaPage(index, true, buildstate->entryPoint, insertPage, forkNum);
pfree(etup);
pfree(ntup);
} }
/* /*
@@ -227,6 +230,8 @@ CreateNeighborPages(HnswBuildState * buildstate)
GenericXLogFinish(state); GenericXLogFinish(state);
UnlockReleaseBuffer(buf); UnlockReleaseBuffer(buf);
} }
pfree(ntup);
} }
/* /*
@@ -430,6 +435,7 @@ InitBuildState(HnswBuildState * buildstate, Relation heap, Relation index, Index
static void static void
FreeBuildState(HnswBuildState * buildstate) FreeBuildState(HnswBuildState * buildstate)
{ {
pfree(buildstate->normvec);
MemoryContextDelete(buildstate->tmpCtx); MemoryContextDelete(buildstate->tmpCtx);
} }

2
src/hnswutils.c
View File

@@ -611,7 +611,7 @@ HnswInitElement(ItemPointer heaptid, int m, double ml, int maxLevel)
void void
HnswFreeElement(HnswElement element) HnswFreeElement(HnswElement element)
{ {
list_free(element->heaptids); list_free_deep(element->heaptids);
for (int lc = 0; lc <= element->level; lc++) for (int lc = 0; lc <= element->level; lc++)
pfree(element->neighbors[lc].items); pfree(element->neighbors[lc].items);
pfree(element->neighbors); pfree(element->neighbors);