Skip to content

vllm_omni.model_executor.models.common.ops

Common operators shared across models and hardware backends.

This module contains fused operators that can be reused across multiple models and compiled for different hardware targets (CUDA, ROCm, etc.).

Modules:

Name Description
fused_adaptive_group_norm_silu

Fused Adaptive Group Normalization (AdaGN) + SiLU operator.

fused_group_norm_silu

Fused GroupNorm + SiLU operator.