feat: Add graph partition support for DiskANN backend

- Add GraphPartitioner class for advanced graph partitioning
- Add partition_graph_simple function for easy-to-use partitioning
- Add pybind11 dependency for C++ executable building
- Update __init__.py to export partition functions
- Include test scripts for partition functionality

The partition functionality allows optimizing disk-based indices
for better search performance and memory efficiency.
This commit is contained in:
yichuan520030910320
2025-08-05 23:11:09 -07:00
parent f94ce63d51
commit 77d7b60a61
5 changed files with 456 additions and 1 deletions

View File

@@ -1 +1,12 @@
from . import diskann_backend as diskann_backend
from . import graph_partition
# Export main classes and functions
from .graph_partition import GraphPartitioner, partition_graph
__all__ = [
"diskann_backend",
"graph_partition",
"GraphPartitioner",
"partition_graph"
]