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

Fixed locking for UpdateNeighborsInMemory

Browse Source
This commit is contained in:
Andrew Kane
2024-08-02 14:34:56 -07:00
parent 020d3edaa9
commit 3c99b55cf9
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/hnswbuild.c
View File

@@ -381,6 +381,18 @@ UpdateNeighborsInMemory(char *base, FmgrInfo *procinfo, Oid collation, HnswEleme
int lm = HnswGetLayerM(m, lc); int lm = HnswGetLayerM(m, lc);
HnswNeighborArray *neighbors = HnswGetNeighbors(base, e, lc); HnswNeighborArray *neighbors = HnswGetNeighbors(base, e, lc);
/* Copy neighbors to local memory if needed */
if (base != NULL)
{
Size neighborsSize = HNSW_NEIGHBOR_ARRAY_SIZE(HnswGetLayerM(m, lc));
HnswNeighborArray *neighborsData = palloc(neighborsSize);
LWLockAcquire(&e->lock, LW_SHARED);
memcpy(neighborsData, neighbors, neighborsSize);
LWLockRelease(&e->lock);
neighbors = neighborsData;
}
for (int i = 0; i < neighbors->length; i++) for (int i = 0; i < neighbors->length; i++)
{ {
HnswCandidate *hc = &neighbors->items[i]; HnswCandidate *hc = &neighbors->items[i];