Exp: make changes to debug statements

microsoft · Sep 6, 2024 · 7a3822d · 7a3822d
1 parent ed26061
commit 7a3822d
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/src/index.cpp b/src/index.cpp
@@ -879,7 +879,6 @@ std::pair<uint32_t, uint32_t> Index<T, TagT, LabelT>::iterate_to_fixed_point(
 
     uint32_t hops = 0;
     uint32_t cmps = 0;
-    size_t iter_count = 1;
     while (best_L_nodes.has_unexpanded_node())
     {
         auto nbr = best_L_nodes.closest_unexpanded();
@@ -889,10 +888,9 @@ std::pair<uint32_t, uint32_t> Index<T, TagT, LabelT>::iterate_to_fixed_point(
             std::vector<uint32_t> id_scratch_temp = {n};
             std::vector<float> dist_scratch_temp = {0.0};
             compute_dists(id_scratch_temp, dist_scratch_temp);
-            diskann::cout<<"hop #"<<hops+1<<": "<<dist_scratch_temp[0]<<std::endl;
-            diskann::cout<<"Iteration: "<<iter_count<<std::endl;
-            diskann::cout<<"L size: "<<best_L_nodes.size()<<std::endl;
-            diskann::cout<<"Node expanded: "<<nbr.id<<std::endl;     
+            diskann::cout<<"Iteration/Hop: #"<<hops+1<<std::endl;
+            diskann::cout<<"Current L size: "<<best_L_nodes.size()<<std::endl;
+            diskann::cout<<"Node expanded(ID) : "<<nbr.id<<" Distance(ID,Query): "<<dist_scratch_temp[0]<<std::endl;     
         }
         hops++;
 
@@ -979,8 +977,8 @@ std::pair<uint32_t, uint32_t> Index<T, TagT, LabelT>::iterate_to_fixed_point(
 
         // Insert <id, dist> pairs into the pool of candidates
         if (search_invocation){
-            diskann::cout<<"Comparisons "<<(uint32_t)id_scratch.size()<<std::endl;
-            diskann::cout<<"Distances"<<std::endl;
+            diskann::cout<<"Comparisons(Neighbors of ID) "<<(uint32_t)id_scratch.size()<<std::endl;
+            diskann::cout<<"Distances of Neighbors: ID, Query"<<std::endl;
         }
         for (size_t m = 0; m < id_scratch.size(); ++m)
         {
@@ -992,10 +990,10 @@ std::pair<uint32_t, uint32_t> Index<T, TagT, LabelT>::iterate_to_fixed_point(
         }
     }
     if (search_invocation){
-        diskann::cout<<"Total hops: "<<hops<<std::endl;
+        diskann::cout<<"Total Iterations/Hops: "<<hops<<std::endl;
         diskann::cout<<"Total comparisons: "<<cmps<<std::endl;
         diskann::cout<<"L size: "<<best_L_nodes.size()<<std::endl;
-        diskann::cout<<"Expanded Nodes "<<scratch->expanded_nodes_vec().size()<<std::endl;
+        diskann::cout<<"Expanded Nodes List"<<scratch->expanded_nodes_vec().size()<<std::endl;
         for(auto neighbor : scratch->expanded_nodes_vec()){
             diskann::cout<<neighbor.id<<std::endl;
         }