@@ -179,15 +179,15 @@ int search_disk_index(diskann::Metric &metric, const std::string &index_path_pre
179179 std::string recall_string = " Recall@" + std::to_string (recall_at);
180180 diskann::cout << std::setw (6 ) << " L" << std::setw (12 ) << " Beamwidth" << std::setw (16 ) << " QPS" << std::setw (16 )
181181 << " Mean Latency" << std::setw (16 ) << " 99.9 Latency" << std::setw (16 ) << " Mean IOs" << std::setw (16 )
182- << " CPU (s)" ;
182+ << " Mean IO (us) " << std::setw ( 16 ) << " CPU (s)" ;
183183 if (calc_recall_flag)
184184 {
185185 diskann::cout << std::setw (16 ) << recall_string << std::endl;
186186 }
187187 else
188188 diskann::cout << std::endl;
189- diskann::cout << " ==============================================================="
190- " ======================================================="
189+ diskann::cout << " ================================================================== "
190+ " ================================================================= "
191191 << std::endl;
192192
193193 std::vector<std::vector<uint32_t >> query_result_ids (Lvec.size ());
@@ -270,6 +270,9 @@ int search_disk_index(diskann::Metric &metric, const std::string &index_path_pre
270270 auto mean_cpuus = diskann::get_mean_stats<float >(stats, query_num,
271271 [](const diskann::QueryStats &stats) { return stats.cpu_us ; });
272272
273+ auto mean_io_us = diskann::get_mean_stats<float >(stats, query_num,
274+ [](const diskann::QueryStats &stats) { return stats.io_us ; });
275+
273276 double recall = 0 ;
274277 if (calc_recall_flag)
275278 {
@@ -280,7 +283,7 @@ int search_disk_index(diskann::Metric &metric, const std::string &index_path_pre
280283
281284 diskann::cout << std::setw (6 ) << L << std::setw (12 ) << optimized_beamwidth << std::setw (16 ) << qps
282285 << std::setw (16 ) << mean_latency << std::setw (16 ) << latency_999 << std::setw (16 ) << mean_ios
283- << std::setw (16 ) << mean_cpuus;
286+ << std::setw (16 ) << mean_io_us << std::setw ( 16 ) << mean_cpuus;
284287 if (calc_recall_flag)
285288 {
286289 diskann::cout << std::setw (16 ) << recall << std::endl;
0 commit comments