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

Add one to error message to ceil [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2022-02-11 03:41:54 -08:00
parent f59b1184fc
commit e5da5cc532
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/ivfkmeans.c
View File

@@ -204,11 +204,12 @@ ElkanKmeans(Relation index, VectorArray samples, VectorArray centers)
Size totalSize = centerCountsSize + closestCentersSize + lowerBoundSize + upperBoundSize + sSize + halfcdistSize + newcdistSize; Size totalSize = centerCountsSize + closestCentersSize + lowerBoundSize + upperBoundSize + sSize + halfcdistSize + newcdistSize;
/* Check memory requirements */ /* Check memory requirements */
/* Add one to error message to ceil */
if (totalSize / 1024 > maintenance_work_mem) if (totalSize / 1024 > maintenance_work_mem)
ereport(ERROR, ereport(ERROR,
(errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED), (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED),
errmsg("memory required is %zu MB, maintenance_work_mem is %zu MB", errmsg("memory required is %zu MB, maintenance_work_mem is %zu MB",
totalSize / (1024 * 1024), ((Size) maintenance_work_mem) / 1024))); totalSize / (1024 * 1024) + 1, ((Size) maintenance_work_mem) / 1024)));
/* Set support functions */ /* Set support functions */
procinfo = index_getprocinfo(index, 1, IVFFLAT_KMEANS_DISTANCE_PROC); procinfo = index_getprocinfo(index, 1, IVFFLAT_KMEANS_DISTANCE_PROC);