small pathc

BlaiseMuhirwa · Dec 8, 2023 · 902900d · 902900d
1 parent c154a05
commit 902900d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/flatnav/Index.h b/flatnav/Index.h
@@ -65,7 +65,7 @@ template <typename dist_t, typename label_t> class Index {
    */
   Index(std::shared_ptr<DistanceInterface<dist_t>> dist,
         std::vector<std::vector<uint32_t>> &outdegree_table)
-      : _M(outdegree_table[0].size()), _max_node_count(outdegree_table.size()),
+      : _M(32), _max_node_count(outdegree_table.size()),
         _cur_num_nodes(0), _distance(dist),
         _visited_nodes(outdegree_table.size() + 1),
         _outdegree_table(std::move(outdegree_table)) {
@@ -90,13 +90,14 @@ template <typename dist_t, typename label_t> class Index {
     for (node_id_t node = 0; node < _outdegree_table.value().size(); node++) {
       node_id_t *links = getNodeLinks(node);
       for (int i = 0; i < _M; i++) {
-        if (_outdegree_table.value()[node].size() < _M) {
+        if (i >= _outdegree_table.value()[node].size()) {
           links[i] = node;
         }
         else {
           auto linkvalue = _outdegree_table.value()[node][i];
           links[i] = linkvalue;
         }
+
       }
     }
   }

diff --git a/tools/load_mtx.cpp b/tools/load_mtx.cpp
@@ -52,6 +52,15 @@ Graph loadGraphFromMatrixMarket(const char *filename) {
   return graph;
 }
 
+void printVector(float* vector) {
+  for (int i = 0; i < 128; i++) {
+    std::cout << vector[i] << " ";
+  }
+  std::cout << std::endl;
+}
+
+
+
 int main() {
   // Replace with your filename
   const char *ground_truth_file =
@@ -80,6 +89,7 @@ int main() {
   float *queries = queryfile.data<float>();
   int *gtruth = truthfile.data<int>();
 
+
   std::cout << "constructing the index" << std::endl;
   auto distance = std::make_shared<flatnav::SquaredL2Distance>(128);
   std::unique_ptr<flatnav::Index<flatnav::SquaredL2Distance, int>> index =