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

Set length for newCenters and aggCenters [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-23 15:47:04 -07:00
parent 60ceaea4f2
commit 991743786a
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/ivfkmeans.c
View File

@@ -99,7 +99,7 @@ NormCenters(FmgrInfo *normalizeprocinfo, Oid collation, VectorArray centers)
ALLOCSET_DEFAULT_SIZES); ALLOCSET_DEFAULT_SIZES);
MemoryContext oldCtx = MemoryContextSwitchTo(normCtx); MemoryContext oldCtx = MemoryContextSwitchTo(normCtx);
for (int j = 0; j < centers->maxlen; j++) for (int j = 0; j < centers->length; j++)
{ {
Datum center = PointerGetDatum(VectorArrayGet(centers, j)); Datum center = PointerGetDatum(VectorArrayGet(centers, j));
Datum newCenter = IvfflatNormValue(normalizeprocinfo, collation, center); Datum newCenter = IvfflatNormValue(normalizeprocinfo, collation, center);
@@ -483,6 +483,9 @@ ElkanKmeans(Relation index, VectorArray samples, VectorArray centers, IvfflatTyp
else else
elog(ERROR, "Unsupported type"); elog(ERROR, "Unsupported type");
aggCenters->length = aggCenters->maxlen;
newCenters->length = newCenters->maxlen;
#ifdef IVFFLAT_MEMORY #ifdef IVFFLAT_MEMORY
ShowMemoryUsage(oldCtx, totalSize); ShowMemoryUsage(oldCtx, totalSize);
#endif #endif