From d036fa23cde9d9457cd3b008d51ca342fc168593 Mon Sep 17 00:00:00 2001
From: Raghuveer Devulapalli <raghuveer.devulapalli@ibm.com>
Date: Wed, 20 May 2026 11:41:23 +0000
Subject: [PATCH] Record index build time in buildInMemory path

---
 .../src/main/java/io/github/jbellis/jvector/example/Grid.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/jvector-examples/src/main/java/io/github/jbellis/jvector/example/Grid.java b/jvector-examples/src/main/java/io/github/jbellis/jvector/example/Grid.java
index ce9d62c1b..8f45df2a0 100644
--- a/jvector-examples/src/main/java/io/github/jbellis/jvector/example/Grid.java
+++ b/jvector-examples/src/main/java/io/github/jbellis/jvector/example/Grid.java
@@ -574,12 +574,13 @@ private static Map<Set<FeatureId>, ImmutableGraphIndex> buildInMemory(List<? ext
                                                           FilteredForkJoinPool.createFilteredPool());
         start = System.nanoTime();
         var onHeapGraph = builder.build(floatVectors);
+        double buildTimeS = (System.nanoTime() - start) / 1_000_000_000.0;
         System.out.format("Build (%s) M=%d overflow=%.2f ef=%d in %.2fs%n",
                           "full res",
                           M,
                           neighborOverflow,
                           efConstruction,
-                          (System.nanoTime() - start) / 1_000_000_000.0);
+                          buildTimeS);
         for (int i = 0; i <= onHeapGraph.getMaxLevel(); i++) {
             System.out.format("  L%d: %d nodes, %.2f avg degree%n",
                               i,
@@ -603,6 +604,7 @@ private static Map<Set<FeatureId>, ImmutableGraphIndex> buildInMemory(List<? ext
             var index = OnDiskGraphIndex.load(ReaderSupplierFactory.open(graphPath));
             indexes.put(features, index);
         }
+        indexBuildTimes.put(ds.getName(), buildTimeS);
         return indexes;
     }