From 5971e28f8c7291eb49e88e2004f4346d82f26c74 Mon Sep 17 00:00:00 2001 From: Andrew Kane Date: Fri, 11 Feb 2022 16:05:45 -0800 Subject: [PATCH] Added more benchmarking [skip ci] --- src/ivfbuild.c | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ivfbuild.c b/src/ivfbuild.c index 8e8dd28..51c6a55 100644 --- a/src/ivfbuild.c +++ b/src/ivfbuild.c @@ -389,7 +389,7 @@ ComputeCenters(IvfflatBuildState * buildstate) /* Calculate centers */ UpdateProgress(PROGRESS_CREATEIDX_SUBPHASE, PROGRESS_IVFFLAT_PHASE_KMEANS); - IvfflatKmeans(buildstate->index, buildstate->samples, buildstate->centers); + IvfflatBench("k-means", IvfflatKmeans(buildstate->index, buildstate->samples, buildstate->centers)); /* Free samples before we allocate more memory */ pfree(buildstate->samples); @@ -522,7 +522,7 @@ BuildIndex(Relation heap, Relation index, IndexInfo *indexInfo, /* Create pages */ CreateMetaPage(index, buildstate->dimensions, buildstate->lists, forkNum); CreateListPages(index, buildstate->centers, buildstate->dimensions, buildstate->lists, forkNum, &buildstate->listInfo); - CreateEntryPages(buildstate, forkNum); + IvfflatBench("CreateEntryPages", CreateEntryPages(buildstate, forkNum)); FreeBuildState(buildstate); }