ComponentsKVBM

KVBM Integrations

View as Markdown

KVBM Integrates with Inference frameworks (vLLM, TRTLLM, SGLang) via Connector APIs to influence KV caching behaviour, scheduling, and forward pass execution. There are two components of the interface, Scheduler and Worker. Scheduler(leader) is responsible for the orchestration of KV block offload/onboard, builds metadata specifying transfer data to the workers. It also maintains hooks for handling asynchronous transfer completion. Worker is responsible for reading metadata built by the scheduler(leader), does async onboarding/ offloading at the end of the forward pass.

Typical KVBM Integrations

The following figure shows the typical integration of KVBM with inference frameworks (vLLM used as an example)

vLLM KVBM Integration vLLM KVBM Integration

How to run KVBM with Frameworks

Onboarding

Onboarding blocks from Host to Device Onboarding blocks from Host to Device Onboarding blocks from Disk to Device Onboarding blocks from Disk to Device

Offloading

Offloading blocks from Device to Host&Disk Offloading blocks from Device to Host&Disk