Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion bindings/cpp/include/svs/runtime/vamana_index.h
Original file line number Diff line number Diff line change
Expand Up @@ -54,8 +54,11 @@ struct MemoryBreakdown {
size_t graph_bytes = 0;
size_t data_bytes = 0;
size_t metadata_bytes = 0;
size_t reverse_edges_bytes = 0;

size_t total() const { return graph_bytes + data_bytes + metadata_bytes; }
size_t total() const {
return graph_bytes + data_bytes + metadata_bytes + reverse_edges_bytes;
}
};

// Abstract interface for Vamana-based indices.
Expand Down
1 change: 1 addition & 0 deletions bindings/cpp/src/vamana_index_impl.h
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,7 @@ class VamanaIndexImpl {
out.graph_bytes = breakdown.graph_bytes;
out.data_bytes = breakdown.data_bytes;
out.metadata_bytes = breakdown.metadata_bytes;
out.reverse_edges_bytes = breakdown.reverse_edges_bytes;
}

size_t dimensions() const { return dim_; }
Expand Down
2 changes: 1 addition & 1 deletion bindings/python/src/vamana_common.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -145,7 +145,7 @@ prefetching during search.
.def(
py::init<svs::index::vamana::SearchBufferConfig, bool, size_t, size_t>(),
py::arg("buffer_config") = svs::index::vamana::SearchBufferConfig(),
py::arg("search_buffer_visited_set") = false,
py::arg("search_buffer_visited_set") = true,
py::arg("prefetch_lookahead") = 4,
py::arg("prefetch_step") = 1
)
Expand Down
4 changes: 2 additions & 2 deletions bindings/python/tests/test_vamana_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ def test_vamana_search_parameters(self):
# Default definitions.
x = svs.VamanaSearchParameters()
self.assertEqual(x.buffer_config, svs.SearchBufferConfig())
self.assertEqual(x.search_buffer_visited_set, False)
self.assertEqual(x.search_buffer_visited_set, True)
self.assertEqual(x, svs.VamanaSearchParameters())

x.buffer_config = svs.SearchBufferConfig(20)
Expand All @@ -57,7 +57,7 @@ def test_vamana_search_parameters(self):

x = svs.VamanaSearchParameters(svs.SearchBufferConfig(10, 20))
self.assertEqual(x.buffer_config, svs.SearchBufferConfig(10, 20))
self.assertEqual(x.search_buffer_visited_set, False)
self.assertEqual(x.search_buffer_visited_set, True)
self.assertNotEqual(x, svs.VamanaSearchParameters())
self.assertEqual(x, svs.VamanaSearchParameters(svs.SearchBufferConfig(10, 20)))

Expand Down
295 changes: 295 additions & 0 deletions include/svs/concurrent/README.md

Large diffs are not rendered by default.

Loading
Loading