Skip to content

vllm_omni.diffusion.attention.backends.ring.ring_utils ¶

logger `module-attribute` ¶

logger = init_logger(__name__)

flatten_varlen_lse ¶

flatten_varlen_lse(lse, cu_seqlens)

unflatten_varlen_lse ¶

unflatten_varlen_lse(lse, cu_seqlens, max_seqlen: int)

update_out_and_lse ¶

update_out_and_lse(
    out: Tensor | None,
    lse: Tensor | None,
    block_out: Tensor,
    block_lse: Tensor,
    slice_=None,
) -> tuple[Tensor, Tensor]