Skip to main content

Crate ailake_index

Crate ailake_index 

Source
Expand description

ailake-index — HNSW and IVF-PQ index lifecycle

Search backend priority (all detected at runtime, no build-time GPU SDK required):

  1. NVIDIA CUDA — cuBLAS SGEMM via libloading dlopen of libcudart + libcublas
  2. AMD ROCm — hipBLAS SGEMM via libloading dlopen of libamdhip64 + libhipblas
  3. CPU rayon — parallel brute-force, always available

Re-exports§

pub use hardware::detect_backend;
pub use hardware::detect_cuda;
pub use hardware::detect_rocm;
pub use hardware::HardwareBackend;
pub use hardware::HardwareProfile;
pub use hnsw::HnswBuilder;
pub use hnsw::HnswConfig;
pub use hnsw::HnswIndex;
pub use ivf_pq::find_valid_pq_m;
pub use ivf_pq::IvfPqCodebook;
pub use ivf_pq::IvfPqConfig;
pub use ivf_pq::IvfPqIndex;
pub use ivf_pq::IvfPqSerializer;
pub use mmap_loader::MmapLoader;
pub use serialize::HnswSerializer;

Modules§

gpu
GPU-accelerated vector search and k-means.
hardware
Hardware capability detection for adaptive index selection.
hnsw
ivf_pq
mmap_loader
serialize

Enums§

AnyIndex
Unified index type: dispatches search to HNSW or IVF-PQ.