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

Show memory from outer context [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-04-11 18:38:35 -07:00
parent e6a591275a
commit 94ee6b13c3
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/ivfkmeans.c
View File

@@ -176,13 +176,13 @@ QuickCenters(Relation index, VectorArray samples, VectorArray centers, IvfflatTy
* Show memory usage
*/
static void
ShowMemoryUsage(Size estimatedSize)
ShowMemoryUsage(MemoryContext context, Size estimatedSize)
{
#if PG_VERSION_NUM >= 130000
elog(INFO, "total memory: %zu MB",
MemoryContextMemAllocated(CurrentMemoryContext, true) / (1024 * 1024));
MemoryContextMemAllocated(context, true) / (1024 * 1024));
#else
MemoryContextStats(CurrentMemoryContext);
MemoryContextStats(context);
#endif
elog(INFO, "estimated memory: %zu MB", estimatedSize / (1024 * 1024));
}
@@ -280,7 +280,7 @@ ElkanKmeans(Relation index, VectorArray samples, VectorArray centers, IvfflatTyp
elog(ERROR, "Unsupported type");
#ifdef IVFFLAT_MEMORY
ShowMemoryUsage(totalSize);
ShowMemoryUsage(oldCtx, totalSize);
#endif
/* Pick initial centers */