neighbor buffer parsing

microsoft · May 17, 2023 · c72d7fb · c72d7fb
1 parent 60568f5
commit c72d7fb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/pq_flash_index.cpp b/src/pq_flash_index.cpp
@@ -1602,11 +1602,11 @@ std::vector<std::uint64_t> PQFlashIndex<T, LabelT>::get_neighbors(std::uint64_t
 #endif
 
     char *node_buf = OFFSET_TO_NODE(buf, VID);
-    char *node_coords = OFFSET_TO_NODE_NHOOD(node_buf);
-    std::uint64_t nnbrs = (std::uint64_t)(*node_buf);
+    std::uint32_t *nbr_buf = OFFSET_TO_NODE_NHOOD(node_buf);
+    std::uint64_t nnbrs = (std::uint64_t)(*nbr_buf);
 
     std::vector<std::uint64_t> out(nnbrs);
-    std::uint32_t *nbrs = (std::uint32_t *)(node_buf + sizeof(nnbrs));
+    std::uint32_t *nbrs = (std::uint32_t *)(nbr_buf + 1);
     for (size_t i = 0; i < nnbrs; i++)
     {
         out[i] = nbrs[i];