From e5da5cc5328209a54fd77f6c936d6bbe1c6f70bb Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Fri, 11 Feb 2022 03:41:54 -0800 Subject: [PATCH] Add one to error message to ceil [skip ci] --- src/ivfkmeans.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/ivfkmeans.c b/src/ivfkmeans.c index 763dc76..78b4f51 100644 --- a/src/ivfkmeans.c +++ b/src/ivfkmeans.c @@ -204,11 +204,12 @@ ElkanKmeans(Relation index, VectorArray samples, VectorArray centers) Size totalSize = centerCountsSize + closestCentersSize + lowerBoundSize + upperBoundSize + sSize + halfcdistSize + newcdistSize; /* Check memory requirements */ + /* Add one to error message to ceil */ if (totalSize / 1024 > maintenance_work_mem) ereport(ERROR, (errcode(ERRCODE_PROGRAM_LIMIT_EXCEEDED), 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 */ procinfo = index_getprocinfo(index, 1, IVFFLAT_KMEANS_DISTANCE_PROC);