vllm_omni.diffusion.offloader.host_registration ¶

Platform-neutral lifecycle for registering existing host mappings.

HostRegistration ¶

Bases: Protocol

Transport-owned registration for one process-local host mapping.

region_count: int

total_bytes: int

close() -> tuple[str, ...]

Raised before mutation when the complete mapping exceeds its budget.

Raised when registered ranges cannot be released safely.

active_registration = active_registration

Raised when existing host ranges cannot be registered safely.

register_host_mappings(
    regions: Sequence[MappedHostRegion],
    *,
    device: device,
    max_bytes: int | None,
) -> HostRegistration

Register immutable ranges with the active platform implementation.