vllm_omni.diffusion.layers.qwen_select01_modulation ¶

can_use_qwen_select01_triton(x: Tensor) -> bool

fused_layernorm_select01(
    x: Tensor,
    mod_params: Tensor,
    index: Tensor,
    eps: float,
    weight: Tensor | None = None,
    bias: Tensor | None = None,
) -> tuple[Tensor, Tensor]

fused_residual_layernorm_select01(
    x: Tensor,
    residual: Tensor,
    residual_gate: Tensor,
    mod_params: Tensor,
    index: Tensor,
    eps: float,
    weight: Tensor | None = None,
    bias: Tensor | None = None,
) -> tuple[Tensor, Tensor, Tensor]

select01_modulation_native(
    mod_params: Tensor, index: Tensor
) -> tuple[Tensor, Tensor, Tensor]

split_select01_mod_params(
    mod_params: Tensor,
) -> tuple[Tensor, ...]