[Model] Minimize LTX-2.5 adapter changes

Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>

.buildkite/amd/scripts/bootstrap-amd-omni.sh

@@ -1,7 +1,4 @@
 #!/bin/bash
-# SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
-
 # vllm-omni customized version
 # Based on: https://github.com/vllm-project/ci-infra/blob/main/buildkite/bootstrap-amd.sh
 # Last synced: 2025-12-15
@@ -9,9 +6,6 @@
 
 set -euo pipefail
 
-# The bootstrap runs from the repository root; ShellCheck does not follow
-# sourced files unless invoked with -x.
-# shellcheck disable=SC1091
 source .buildkite/common/scripts/resolve_skip_ci.sh
 
 if [[ -z "${RUN_ALL:-}" ]]; then
@@ -69,8 +63,6 @@ upload_pipeline() {
     # Install minijinja
     ls .buildkite || buildkite-agent annotate --style error 'Please merge upstream main branch for buildkite CI'
     curl -sSfL https://github.com/mitsuhiko/minijinja/releases/download/2.3.1/minijinja-cli-installer.sh | sh
-    # Installed by the minijinja bootstrap above and only present on the CI agent.
-    # shellcheck disable=SC1091
     source /var/lib/buildkite-agent/.cargo/env
 
     if [[ $BUILDKITE_PIPELINE_SLUG == "fastcheck" ]]; then
@@ -98,82 +90,22 @@ upload_pipeline() {
     cd .buildkite/amd
 
     # Select test definition file: merge suite for main, ready suite for PRs.
-    # For debugging, DEBUG_TEST_YAML accepts a comma-separated list containing
+    # For debugging, DEBUG_TEST_YAML can override the selection — accepts
-    # "merge" and/or "ready" (case-insensitive). Multiple suites are combined
+    # "merge" or "ready" (case-insensitive).
-    # before rendering so they share one amd-build step.
     if [[ -n "${DEBUG_TEST_YAML:-}" ]]; then
-        declare -a DEBUG_TEST_SPECS=()
+        case "${DEBUG_TEST_YAML,,}" in
-        declare -A SEEN_DEBUG_TESTS=()
+            merge)
-        IFS=',' read -ra REQUESTED_DEBUG_TESTS <<< "${DEBUG_TEST_YAML,,}"
+                TEST_YAML="test-amd-merge.yml"
-
+                ;;
-        for requested_test in "${REQUESTED_DEBUG_TESTS[@]}"; do
+            ready)
-            # Permit readable values such as "ready, merge," and ignore the
+                TEST_YAML="test-amd-ready.yml"
-            # empty item produced by a trailing comma.
+                ;;
-            requested_test="${requested_test//[[:space:]]/}"
+            *)
-            [[ -z "$requested_test" ]] && continue
+                echo "ERROR: DEBUG_TEST_YAML must be 'merge' or 'ready', got '$DEBUG_TEST_YAML'" >&2
-
-            if [[ -n "${SEEN_DEBUG_TESTS[$requested_test]:-}" ]]; then
-                echo "ERROR: duplicate DEBUG_TEST_YAML suite '$requested_test'" >&2
                 exit 1
-            fi
+                ;;
-            SEEN_DEBUG_TESTS[$requested_test]=1
+        esac
-
+        echo "DEBUG_TEST_YAML override: using $TEST_YAML"
-            case "$requested_test" in
-                ready)
-                    DEBUG_TEST_SPECS+=("READY_TESTS:test-amd-ready.yml")
-                    ;;
-                merge)
-                    DEBUG_TEST_SPECS+=("MERGE_TESTS:test-amd-merge.yml")
-                    ;;
-                *)
-                    echo "ERROR: DEBUG_TEST_YAML entries must be 'merge' or 'ready', got '$requested_test'" >&2
-                    exit 1
-                    ;;
-            esac
-        done
-
-        if [[ ${#DEBUG_TEST_SPECS[@]} -eq 0 ]]; then
-            echo "ERROR: DEBUG_TEST_YAML did not contain a test suite" >&2
-            exit 1
-        elif [[ ${#DEBUG_TEST_SPECS[@]} -eq 1 ]]; then
-            TEST_YAML="${DEBUG_TEST_SPECS[0]#*:}"
-        else
-            TEST_YAML=$(mktemp "${TMPDIR:-/tmp}/amd-debug-tests.XXXXXX.yml")
-            python - "$TEST_YAML" "${DEBUG_TEST_SPECS[@]}" <<'PY'
-import sys
-
-import yaml
-
-
-output_path, *suite_specs = sys.argv[1:]
-combined = {"env": {}, "steps": []}
-
-for suite_spec in suite_specs:
-    group_name, input_path = suite_spec.split(":", 1)
-    with open(input_path, encoding="utf-8") as test_file:
-        suite = yaml.safe_load(test_file)
-
-    for name, value in (suite.get("env") or {}).items():
-        previous = combined["env"].get(name, value)
-        if previous != value:
-            raise ValueError(
-                f"Conflicting environment value for {name}: {previous!r} != {value!r}"
-            )
-        combined["env"][name] = value
-
-    suite_steps = []
-    for entry in suite.get("steps") or []:
-        if "group" in entry:
-            suite_steps.extend(entry.get("steps") or [])
-        else:
-            suite_steps.append(entry)
-    combined["steps"].append({"group": group_name, "steps": suite_steps})
-
-with open(output_path, "w", encoding="utf-8") as output_file:
-    yaml.safe_dump(combined, output_file, sort_keys=False)
-PY
-        fi
-        echo "DEBUG_TEST_YAML override: using ${DEBUG_TEST_SPECS[*]}"
     elif [[ $BUILDKITE_BRANCH == "main" ]]; then
         TEST_YAML="test-amd-merge.yml"
     else
@@ -198,20 +130,19 @@ PY
             > pipeline.yaml
     )
     cat pipeline.yaml
-    if [[ "$TEST_YAML" == "${TMPDIR:-/tmp}/amd-debug-tests."*.yml ]]; then
-        rm -f -- "$TEST_YAML"
-    fi
     buildkite-agent artifact upload pipeline.yaml
     buildkite-agent pipeline upload pipeline.yaml
     exit 0
 }
 
 get_diff() {
-    git diff --name-only --diff-filter=ACMDR "$(git merge-base origin/main HEAD)"
+    $(git add .)
+    echo $(git diff --name-only --diff-filter=ACMDR $(git merge-base origin/main HEAD))
 }
 
 get_diff_main() {
-    git diff --name-only --diff-filter=ACMDR HEAD~1
+    $(git add .)
+    echo $(git diff --name-only --diff-filter=ACMDR HEAD~1)
 }
 
 file_diff=$(get_diff)
@@ -250,7 +181,7 @@ for file in $file_diff; do
     # First check if file matches any pattern
     matches_pattern=0
     for pattern in "${patterns[@]}"; do
-        if [[ $file == "$pattern"* ]] || [[ $file == "$pattern" ]]; then
+        if [[ $file == $pattern* ]] || [[ $file == $pattern ]]; then
             matches_pattern=1
             break
         fi
@@ -260,7 +191,7 @@ for file in $file_diff; do
     if [[ $matches_pattern -eq 1 ]]; then
         matches_ignore=0
         for ignore in "${ignore_patterns[@]}"; do
-            if [[ $file == "$ignore"* ]] || [[ $file == "$ignore" ]]; then
+            if [[ $file == $ignore* ]] || [[ $file == $ignore ]]; then
                 matches_ignore=1
                 break
             fi

.buildkite/amd/scripts/run-amd-test.sh

@@ -1,103 +1,153 @@
 #!/bin/bash
+# vllm-omni customized version
+# Based on: vllm/.buildkite/scripts/hardware_ci/run-amd-test.sh
+# Last synced: 2025-12-15
+# Modifications: docker image name for vllm-omni
 
-# Run vLLM-Omni ROCm tests directly in the MI300 Kubernetes pod. The pod's
+# This script runs test inside the corresponding ROCm docker container.
-# container image is selected by test-template-amd-omni.j2; MI300 has no DinD.
+set -o pipefail
-set -euo pipefail
 
-: "${PYTHONFAULTHANDLER:=1}"
+# Export Python path
-: "${HF_HOME:=/home/buildkite-agent/huggingface}"
+export PYTHONPATH=".."
-: "${HF_HUB_DOWNLOAD_TIMEOUT:=300}"
-: "${HF_HUB_ETAG_TIMEOUT:=60}"
-: "${MIOPEN_DEBUG_CONV_DIRECT:=0}"
-: "${MIOPEN_DEBUG_CONV_GEMM:=0}"
-: "${VLLM_ROCM_USE_AITER:=0}"
-export PYTHONFAULTHANDLER HF_HOME HF_HUB_DOWNLOAD_TIMEOUT HF_HUB_ETAG_TIMEOUT
-export MIOPEN_DEBUG_CONV_DIRECT MIOPEN_DEBUG_CONV_GEMM VLLM_ROCM_USE_AITER
-export PYTORCH_ROCM_ARCH=""
-export PYTHONPATH="${PYTHONPATH:-..}"
-
-if [[ "${VLLM_CI_DOCKER_DISABLED:-0}" != "1" ]]; then
-    echo "Error: MI300 CI must run natively with Docker disabled." >&2
-    exit 1
-fi
-
-if [[ -n "${TEST_COMMANDS:-}" ]]; then
-    commands="${TEST_COMMANDS}"
-else
-    commands="$*"
-fi
-if [[ -z "${commands}" ]]; then
-    echo "Error: No test commands provided." >&2
-    exit 1
-fi
-
-echo "--- Native in-pod ROCm CI"
-
-job_id="${BUILDKITE_JOB_ID:-${BUILDKITE_PARALLEL_JOB:-local}}"
-job_id="${job_id//[^A-Za-z0-9_.-]/_}"
-native_cache_root="/tmp/vllm-omni-native-${job_id}"
-TMPDIR="${native_cache_root}/tmp"
-TORCHINDUCTOR_CACHE_DIR="${native_cache_root}/torchinductor"
-TRITON_CACHE_DIR="${native_cache_root}/triton"
-VLLM_CACHE_ROOT="${native_cache_root}/vllm"
-XDG_CACHE_HOME="${native_cache_root}/xdg"
-HF_DATASETS_CACHE="${native_cache_root}/huggingface/datasets"
-export TMPDIR TORCHINDUCTOR_CACHE_DIR TRITON_CACHE_DIR VLLM_CACHE_ROOT
-export XDG_CACHE_HOME HF_DATASETS_CACHE
-mkdir -p "${TMPDIR}" "${TORCHINDUCTOR_CACHE_DIR}" "${TRITON_CACHE_DIR}" \
-    "${VLLM_CACHE_ROOT}" "${XDG_CACHE_HOME}" "${HF_HOME}" \
-    "${HF_DATASETS_CACHE}"
-
-if [[ "${VLLM_CI_REQUIRE_PERSISTENT_HF_CACHE:-0}" == "1" ]]; then
-    if ! command -v findmnt >/dev/null 2>&1; then
-        echo "Error: findmnt is required to verify the Hugging Face cache mount." >&2
-        exit 1
-    fi
-    hf_mount=$(findmnt -n -T "${HF_HOME}" -o TARGET 2>/dev/null || true)
-    if [[ -z "${hf_mount}" || "${hf_mount}" == "/" ]]; then
-        echo "Error: MI300 CI requires a persistent volume at or above ${HF_HOME}." >&2
-        exit 1
-    fi
-fi
 
+# Print ROCm version
+echo "--- ROCm info"
 rocminfo
 
-expected_gpus="${VLLM_CI_EXPECTED_GPU_COUNT:-1}"
+# cleanup older docker images
-python3 - <<'PY'
+cleanup_docker() {
-import os
+  # Get Docker's root directory
+  docker_root=$(docker info -f '{{.DockerRootDir}}')
+  if [ -z "$docker_root" ]; then
+    echo "Failed to determine Docker root directory."
+    exit 1
+  fi
+  echo "Docker root directory: $docker_root"
+  # Check disk usage of the filesystem where Docker's root directory is located
+  disk_usage=$(df "$docker_root" | tail -1 | awk '{print $5}' | sed 's/%//')
+  # Define the threshold
+  threshold=70
+  if [ "$disk_usage" -gt "$threshold" ]; then
+    echo "Disk usage is above $threshold%. Cleaning up Docker images and volumes..."
+    # Remove dangling images (those that are not tagged and not used by any container)
+    docker image prune -f
+    # Remove unused volumes / force the system prune for old images as well.
+    docker volume prune -f && docker system prune --force --filter "until=72h" --all
+    echo "Docker images and volumes cleanup completed."
+  else
+    echo "Disk usage is below $threshold%. No cleanup needed."
+  fi
+}
 
-import torch
+# Call the cleanup docker function
+cleanup_docker
 
-expected = int(os.environ.get("VLLM_CI_EXPECTED_GPU_COUNT", "1"))
+echo "--- Pulling container"
-assert torch.version.hip, "PyTorch is not a ROCm build"
+## Temporary change to use AMD Docker Hub to store the vllm-omni image
-assert torch.cuda.is_available(), "ROCm GPU is not available to PyTorch"
+# to bypass the rate limit issue with ECR Public Gallery.
-actual = torch.cuda.device_count()
+# Images are now stored in a separate repository for vllm-omni, instead of vllm-ci.
-assert actual == expected, f"Expected {expected} ROCm GPU(s), found {actual}"
+# TODO: @tjtanaa point back to ECR Public Gallery
-PY
+# once the amd agents are configured to use ECR Public Gallery.
+# image_name="public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:${BUILDKITE_COMMIT}-rocm-omni"
+image_name="rocm/vllm-omni:${BUILDKITE_COMMIT}"
+container_name="rocm_${BUILDKITE_COMMIT}_$(tr -dc A-Za-z0-9 < /dev/urandom | head -c 10; echo)"
 
-echo "Commands:${commands}"
+# TODO: @tjtanaa uncomment this once the amd agents are configured to use ECR Public Gallery.
-if /bin/bash -o pipefail -c '
+# # Install AWS CLI to authenticate to ECR Public Gallery to get higher rate limit for pulling images
-set -E
+# sudo apt-get update && sudo apt-get install -y awscli
-test_status=0
+##  Use safe docker login helper to prevent race conditions
-trap '\''
+# source "$(dirname "${BASH_SOURCE[0]}")/../../common/scripts/docker_login_ecr_public.sh"
-    command_status=$?
+# safe_docker_login_ecr_public
-    # Keep running subsequent commands, but retain a failure for the final exit.
+
-    # Prefer a test failure over pytest "no tests collected" when both occur.
+## Pull the container from AMD Docker Hub
-    if (( test_status == 0 || (test_status == 5 && command_status != 5) )); then
+
-        test_status=${command_status}
+docker pull "${image_name}"
-    fi
+
-'\'' ERR
+remove_docker_container() {
-eval "$1"
+   docker ps -aq --filter "name=${container_name}" | xargs -r docker rm -f || true
-exit "${test_status}"
+   docker image rm -f "${image_name}" || true
-' _ "${commands}"; then
+}
-    exit 0
+trap remove_docker_container EXIT
+
+echo "--- Running container"
+
+HF_CACHE="$(realpath ~)/huggingface"
+mkdir -p "${HF_CACHE}"
+HF_MOUNT="/root/.cache/huggingface"
+
+if [[ -n "${TEST_COMMAND:-}" ]]; then
+    commands="$TEST_COMMAND"
 else
-    exit_code=$?
+    commands="$@"
+fi
+echo "Commands:$commands"
+
+PARALLEL_JOB_COUNT=8
+MYPYTHONPATH=".."
+
+# Test that we're launching on the machine that has
+# proper access to GPUs
+render_gid=$(getent group render | cut -d: -f3)
+if [[ -z "$render_gid" ]]; then
+  echo "Error: 'render' group not found. This is required for GPU access." >&2
+  exit 1
 fi
 
-if [[ ${exit_code} -eq 5 && "${VLLM_CI_ALLOW_NO_TESTS:-0}" == "1" ]]; then
+# check if the command contains shard flag, we will run all shards in parallel because the host have 8 GPUs.
-    echo "Pytest collected no tests; VLLM_CI_ALLOW_NO_TESTS=1, treating exit code 5 as success."
+# TODO: @tjtanaa reenable to run VLLM_ROCM_USE_AITER=1 when AITER is shipped with prebuilt kernels.
-    exit 0
+if [[ $commands == *"--shard-id="* ]]; then
+  # assign job count as the number of shards used
+  commands=$(echo "$commands" | sed -E "s/--num-shards[[:blank:]]*=[[:blank:]]*[0-9]*/--num-shards=${PARALLEL_JOB_COUNT} /g" | sed 's/ \\ / /g')
+  for GPU in $(seq 0 $(($PARALLEL_JOB_COUNT-1))); do
+    # assign shard-id for each shard
+    commands_gpu=$(echo "$commands" | sed -E "s/--shard-id[[:blank:]]*=[[:blank:]]*[0-9]*/--shard-id=${GPU} /g" | sed 's/ \\ / /g')
+    echo "Shard ${GPU} commands:$commands_gpu"
+    echo "Render devices: $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES"
+    docker run \
+        --device /dev/kfd $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES \
+        --network=host \
+        --shm-size=16gb \
+        --group-add "$render_gid" \
+        -e MIOPEN_DEBUG_CONV_DIRECT=0 \
+        -e MIOPEN_DEBUG_CONV_GEMM=0 \
+        -e VLLM_ROCM_USE_AITER=0 \
+        -e HIP_VISIBLE_DEVICES="${GPU}" \
+        -e HF_TOKEN \
+        -v "${HF_CACHE}:${HF_MOUNT}" \
+        -e "HF_HOME=${HF_MOUNT}" \
+        -e "PYTHONPATH=${MYPYTHONPATH}" \
+        --name "${container_name}_${GPU}" \
+        "${image_name}" \
+        /bin/bash -c "${commands_gpu}" \
+        |& while read -r line; do echo ">>Shard $GPU: $line"; done &
+    PIDS+=($!)
+  done
+  #wait for all processes to finish and collect exit codes
+  for pid in "${PIDS[@]}"; do
+    wait "${pid}"
+    STATUS+=($?)
+  done
+  for st in "${STATUS[@]}"; do
+    if [[ ${st} -ne 0 ]]; then
+      echo "One of the processes failed with $st"
+      exit "${st}"
+    fi
+  done
+else
+  echo "Render devices: $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES"
+  docker run \
+          --device /dev/kfd $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES \
+          --network=host \
+          --shm-size=16gb \
+          --group-add "$render_gid" \
+          -e MIOPEN_DEBUG_CONV_DIRECT=0 \
+          -e MIOPEN_DEBUG_CONV_GEMM=0 \
+          -e VLLM_ROCM_USE_AITER=0 \
+          -e HF_TOKEN \
+          -v "${HF_CACHE}:${HF_MOUNT}" \
+          -e "HF_HOME=${HF_MOUNT}" \
+          -e "PYTHONPATH=${MYPYTHONPATH}" \
+          --name "${container_name}" \
+          "${image_name}" \
+          /bin/bash -c "${commands}"
 fi
-
-exit "${exit_code}"

.buildkite/amd/test-amd-merge.yml

@@ -7,113 +7,84 @@ steps:
 
 - group: ":card_index_dividers: Simple Test"
   steps:
-    - label: "Simple · Model Executor Test · Shard %N/%t"
+    - label: "Simple · Diffusion & Model Executor Test (1/2)"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
-      parallelism: 2
       depends_on: amd-build
+      timeout_in_minutes: 70
       mirror_hardwares: [amdproduction]
       grade: Blocking
       commands:
         - export VLLM_ROCM_USE_AITER=0
         # ignore test_teacache_extractors.py because it use rocm gemm kernel from vLLM
         # that is not supported on CPU
-        - "pytest -sv tests/model_executor -m 'core_model and cpu' --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB"
+        # CPU tests on ROCm gpu machines are very slow, so we give them more time.
+        - "timeout 60m pytest -sv tests/diffusion -m 'core_model and cpu' --ignore=tests/diffusion/cache/test_teacache_extractors.py --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
 
-    - label: "Simple · Diffusion Test · Shard %N/%t"
+    - label: "Simple · Diffusion & Model Executor Test (2/2)"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
-      parallelism: 4
       depends_on: amd-build
+      timeout_in_minutes: 70
       mirror_hardwares: [amdproduction]
       grade: Blocking
       commands:
         - export VLLM_ROCM_USE_AITER=0
-        # ignore test_teacache_extractors.py because it use rocm gemm kernel from vLLM
+        # CPU tests on ROCm gpu machines are very slow, so we give them more time.
-        # that is not supported on CPU
+        - "timeout 60m pytest -sv tests/model_executor -m 'core_model and cpu' --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
-        - "pytest -sv tests/diffusion -m 'core_model and cpu' --ignore=tests/diffusion/cache/test_teacache_extractors.py --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB"
-
-    - label: "Simple · Engine&Entrypoints Test"
-      agent_pool: mi300_1
-      depends_on: amd-build
-      mirror_hardwares: [amdproduction]
-      grade: Blocking
-      commands:
-        - export VLLM_ROCM_USE_AITER=0
-        - "pytest -sv tests/entrypoints tests/engine -m 'core_model and cpu'"
 
     - label: "Simple · Other Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
+      timeout_in_minutes: 50
       mirror_hardwares: [amdproduction]
       grade: Blocking
       commands:
         - export VLLM_ROCM_USE_AITER=0
         # ignore test_teacache_extractors.py because it use rocm gemm kernel from vLLM
         # that is not supported on CPU
-        - "pytest -sv tests/ -m 'core_model and cpu' --ignore=tests/diffusion --ignore=tests/model_executor --ignore=tests/entrypoints --ignore=tests/engine  --ignore=tests/diffusion/cache/test_teacache_extractors.py"
+        - "timeout 40m pytest -sv tests/diffusion tests/model_executor -m 'core_model and cpu' --ignore=tests/diffusion/cache/test_teacache_extractors.py --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
 
 - group: ":card_index_dividers: Diffusion Test"
   steps:
     - label: "Diffusion · LoRA Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
-      artifact_paths:
-        - lora-server.log
       commands:
-        - pytest -sv tests/diffusion/lora/ -m 'advanced_model and cuda' --run-level "advanced_model" 2>&1 | tee lora-server.log
+        - timeout 20m pytest -s -v tests/diffusion/lora/test_images_generations_lora.py
 
     - label: "Diffusion Model CPU offloading Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
       commands:
         - export VLLM_LOGGING_LEVEL=DEBUG
         - export VLLM_WORKER_MULTIPROC_METHOD=spawn
-        - pytest -s -v tests/diffusion/offloader/test_diffusion_cpu_offload.py
+        - |
-        - >-
+          timeout 20m bash -c '
-          pytest -s -v tests/diffusion/offloader/test_diffusion_layerwise_offload.py
+            set +e
-          --deselect='tests/diffusion/offloader/test_diffusion_layerwise_offload.py::test_layerwise_offload_diffusion_model[stabilityai/stable-audio-open-1.0]'
+            pytest -s -v tests/diffusion/offloader/test_diffusion_cpu_offload.py
+            EXIT1=\$?
+            pytest -s -v tests/diffusion/offloader/test_diffusion_layerwise_offload.py
+            EXIT2=\$?
+            exit \$((EXIT1 | EXIT2))
+          '
 
-    # TODO: Diagnose unstable Stable Audio layerwise-offload memory savings on ROCm.
+    - label: "Diffusion Sequence Parallelism Test (Need 4 GPUs)"
-    - label: "Diffusion Stable Audio Layerwise Offload Test (quarantined)"
+      agent_pool: mi325_4
-      agent_pool: mi300_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
-      grade: NonBlocking
-      commands:
-        - export VLLM_LOGGING_LEVEL=DEBUG
-        - export VLLM_WORKER_MULTIPROC_METHOD=spawn
-        - >-
-          pytest -s -v
-          'tests/diffusion/offloader/test_diffusion_layerwise_offload.py::test_layerwise_offload_diffusion_model[stabilityai/stable-audio-open-1.0]'
-
-    - label: "Diffusion Sequence Parallelism Test (Need 4 GPUs) · Shard 1/2"
-      agent_pool: mi300_4
-      depends_on: amd-build
-      mirror_hardwares: [amdproduction]
-      timeout_in_minutes: 40
       grade: Blocking
       commands:
         - export VLLM_LOGGING_LEVEL=DEBUG
         - export VLLM_WORKER_MULTIPROC_METHOD=spawn
-        - pytest -s -v tests/diffusion/distributed/test_sequence_parallel.py
+        - timeout 20m pytest -s -v tests/diffusion/distributed/test_sequence_parallel.py
-
+        - timeout 20m pytest -s -v tests/diffusion/distributed/test_ulysses_uaa_perf.py
-    - label: "Diffusion Sequence Parallelism Test (Need 4 GPUs) · Shard 2/2"
-      agent_pool: mi300_4
-      depends_on: amd-build
-      mirror_hardwares: [amdproduction]
-      timeout_in_minutes: 40
-      grade: Blocking
-      commands:
-        - export VLLM_LOGGING_LEVEL=DEBUG
-        - export VLLM_WORKER_MULTIPROC_METHOD=spawn
-        - pytest -s -v tests/diffusion/distributed/test_ulysses_uaa_perf.py
 
     # merge-only tests
     - label: "Diffusion Tensor Parallelism Test"
-      agent_pool: mi300_2
+      agent_pool: mi325_2
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -121,7 +92,7 @@ steps:
         - pytest -s -v tests/diffusion/distributed/test_tensor_parallel.py
 
     - label: "Diffusion Model Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -129,7 +100,7 @@ steps:
         - timeout 30m pytest -s -v tests/e2e/offline_inference/test_z_image.py
 
     - label: "Diffusion GPU Worker Test"
-      agent_pool: mi300_2
+      agent_pool: mi325_2
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -137,7 +108,7 @@ steps:
         - timeout 20m pytest -s -v tests/diffusion/test_diffusion_worker.py
 
     - label: "Diffusion · Wan22 Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -150,7 +121,7 @@ steps:
 
 ## ISSUE depends on `diffusers` package: https://github.com/huggingface/diffusers/issues/13274
 # - label: "Audio Generation Model Test"
-#   agent_pool: mi300_1
+#   agent_pool: mi325_1
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking
@@ -161,7 +132,7 @@ steps:
 #     - timeout 20m pytest -s -v tests/e2e/offline_inference/test_stable_audio_expansion.py -m "advanced_model and diffusion and L4" --run-level advanced_model
 
 - label: "Engine Test"
-  agent_pool: mi300_1
+  agent_pool: mi325_1
   depends_on: amd-build
   mirror_hardwares: [amdproduction]
   grade: Blocking
@@ -176,7 +147,7 @@ steps:
 - group: ":card_index_dividers: E2E Test"
   steps:
     - label: "Omni Model Test Qwen2-5-Omni"
-      agent_pool: mi300_4
+      agent_pool: mi325_4
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       commands:
@@ -184,70 +155,48 @@ steps:
         - pytest -s -v tests/e2e/*/test_qwen2_5_omni_expansion.py --run-level "core_model"
 
     - label: "Omni Model Test Qwen3-Omni"
-      agent_pool: mi300_2
+      agent_pool: mi325_2
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
       commands:
         - export VLLM_WORKER_MULTIPROC_METHOD=spawn
-        - pytest -s -v tests/e2e/offline_inference/test_qwen3_omni.py tests/e2e/online_serving/test_qwen3_omni.py tests/e2e/offline_inference/test_qwen3_omni_colocate_async.py -m 'advanced_model and cuda' --run-level 'advanced_model'
+        - pytest -s -v tests/e2e/offline_inference/test_qwen3_omni.py tests/e2e/online_serving/test_qwen3_omni.py -m 'advanced_model and cuda' --run-level 'advanced_model'
 
-    - label: "Qwen3-TTS CustomVoice E2E Test · Shard %N/%t"
+    - label: "Qwen3-TTS CustomVoice E2E Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
-      parallelism: 3
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
-      timeout_in_minutes: 80
       commands:
         - |
-          timeout --signal=TERM --kill-after=2m 80m bash -c '
+          timeout 50m bash -c '
             export VLLM_LOGGING_LEVEL=DEBUG
             export VLLM_WORKER_MULTIPROC_METHOD=spawn
             export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
-            TESTS=(
+            pytest -s -v tests/e2e/online_serving/test_qwen3_tts_customvoice.py tests/e2e/offline_inference/test_qwen3_tts_customvoice.py -m "advanced_model" --run-level "advanced_model"
-              "tests/e2e/online_serving/test_qwen3_tts_customvoice.py::test_default_cuda_graph_startup[async_chunk]"
-              "tests/e2e/online_serving/test_qwen3_tts_customvoice.py::test_text_to_audio_002[async_chunk]"
-              "tests/e2e/offline_inference/test_qwen3_tts_customvoice.py::test_text_to_audio_001[no_cuda_graph]"
-            )
-            TEST="$${TESTS[$$BUILDKITE_PARALLEL_JOB]}"
-            pytest -s -v "$$TEST" -m "advanced_model" --run-level "advanced_model"
-          '
-
-    # TODO: Restore this case to the blocking sharded step once its long-running
-    # Whisper validation on AMD CI is resolved.
-    - label: "Qwen3-TTS CustomVoice async_chunk Test (quarantined)"
-      agent_pool: mi300_1
-      depends_on: amd-build
-      mirror_hardwares: [amdproduction]
-      grade: NonBlocking
-      timeout_in_minutes: 80
-      commands:
-        - |
-          timeout --signal=TERM --kill-after=2m 75m bash -c '
-            export VLLM_LOGGING_LEVEL=DEBUG
-            export VLLM_WORKER_MULTIPROC_METHOD=spawn
-            export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
-            pytest -s -v "tests/e2e/online_serving/test_qwen3_tts_customvoice.py::test_text_to_audio_001[async_chunk]" -m "advanced_model" --run-level "advanced_model"
           '
 
     - label: "Qwen3-TTS Base E2E Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
       timeout_in_minutes: 50
       commands:
-        - export VLLM_LOGGING_LEVEL=DEBUG
+        - |
-        - export VLLM_WORKER_MULTIPROC_METHOD=spawn
+          timeout 40m bash -c '
-        - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
+            export VLLM_LOGGING_LEVEL=DEBUG
-        - timeout 40m pytest -s -v tests/e2e/online_serving/test_qwen3_tts_base.py tests/e2e/offline_inference/test_qwen3_tts_base.py -m "advanced_model and cuda" --run-level "advanced_model"
+            export VLLM_WORKER_MULTIPROC_METHOD=spawn
+            export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
+            pytest -s -v tests/e2e/online_serving/test_qwen3_tts_base.py tests/e2e/offline_inference/test_qwen3_tts_base.py -m 'advanced_model and cuda' --run-level 'advanced_model'
+          '
 
 # TODO: Bagel test on ROCm is very unstable. @tjtanaa
 # Need to debug before reneable numerical changes across large PRs
 # # split Bagel Model Test with H100 (Real Weights) into three tests
 # - label: "Bagel Text2Img Model Test (1/3)"
-#   agent_pool: mi300_1
+#   agent_pool: mi325_1
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking
@@ -259,7 +208,7 @@ steps:
 #     - timeout 30m pytest -s -v tests/e2e/offline_inference/test_bagel.py -m "advanced_model" --run-level "advanced_model" -k "shared_memory" -k "rocm"
 
 # - label: "Bagel Img2Img Model Test (2/3)"
-#   agent_pool: mi300_1
+#   agent_pool: mi325_1
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking
@@ -271,7 +220,7 @@ steps:
 #     - timeout 30m pytest -s -v tests/e2e/offline_inference/test_bagel.py -m "advanced_model" --run-level "advanced_model" -k "rocm"
 
 # - label: "Bagel Online Serving Test (3/3)"
-#   agent_pool: mi300_1
+#   agent_pool: mi325_1
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking

.buildkite/amd/test-amd-ready.yml

@@ -7,9 +7,8 @@ steps:
 
 - group: ":card_index_dividers: Simple Test"
   steps:
-    - label: "Simple · Model Executor Test · Shard %N/%t"
+    - label: "Simple · Diffusion & Model Executor Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
-      parallelism: 2
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -17,30 +16,10 @@ steps:
         - export VLLM_ROCM_USE_AITER=0
         # ignore test_teacache_extractors.py because it use rocm gemm kernel from vLLM
         # that is not supported on CPU
-        - "pytest -sv tests/model_executor -m 'core_model and cpu' --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT --shard-id=$$BUILDKITE_PARALLEL_JOB"
+        - "timeout 40m pytest -sv tests/diffusion tests/model_executor -m 'core_model and cpu' --ignore=tests/diffusion/cache/test_teacache_extractors.py --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
-
-    - label: "Simple · Diffusion Test"
-      agent_pool: mi300_1
-      depends_on: amd-build
-      mirror_hardwares: [amdproduction]
-      grade: Blocking
-      commands:
-        - export VLLM_ROCM_USE_AITER=0
-        # ignore test_teacache_extractors.py because it use rocm gemm kernel from vLLM
-        # that is not supported on CPU
-        - "pytest -sv tests/diffusion -m 'core_model and cpu'  --ignore=tests/diffusion/cache/test_teacache_extractors.py"
-
-    - label: "Simple · Engine&Entrypoints Test"
-      agent_pool: mi300_1
-      depends_on: amd-build
-      mirror_hardwares: [amdproduction]
-      grade: Blocking
-      commands:
-        - export VLLM_ROCM_USE_AITER=0
-        - "pytest -sv tests/entrypoints tests/engine -m 'core_model and cpu'"
 
     - label: "Simple · Other Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -48,19 +27,19 @@ steps:
         - export VLLM_ROCM_USE_AITER=0
         # ignore test_teacache_extractors.py because it use rocm gemm kernel from vLLM
         # that is not supported on CPU
-        - "pytest -sv tests/ -m 'core_model and cpu' --ignore=tests/diffusion --ignore=tests/model_executor --ignore=tests/entrypoints --ignore=tests/engine  --ignore=tests/diffusion/cache/test_teacache_extractors.py"
+        - "timeout 40m pytest -sv tests/diffusion tests/model_executor -m 'core_model and cpu' --ignore=tests/diffusion/cache/test_teacache_extractors.py --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
 
 - label: "Custom Pipeline Test"
-  agent_pool: mi300_1
+  agent_pool: mi325_1
   depends_on: amd-build
   mirror_hardwares: [amdproduction]
   grade: Blocking
   commands:
-    - pytest -s -v tests/e2e/features/custom_pipeline/ -m "core_model"
+    - timeout 20m pytest -s -v tests/e2e/features/custom_pipeline/ -m "core_model"
 
 
 - label: "Voxtral TTS CUDA Unit Test"
-  agent_pool: mi300_1
+  agent_pool: mi325_1
   depends_on: amd-build
   mirror_hardwares: [amdproduction]
   grade: Blocking
@@ -70,25 +49,33 @@ steps:
 - group: ":card_index_dividers: Diffusion Test"
   steps:
     - label: "Diffusion · Batch Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
       commands:
-        - pytest -sv tests/diffusion/batching -m 'core_model and cuda' --run-level "core_model"
+        - timeout 20m pytest -sv tests/diffusion/batching -m 'core_model and cuda' --run-level "core_model"
 
     - label: "Diffusion · Offloader Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
       commands:
         - export VLLM_LOGGING_LEVEL=DEBUG
         - export VLLM_WORKER_MULTIPROC_METHOD=spawn
-        - pytest -s -v tests/diffusion/offloader -m "core_model and cuda" --run-level "core_model"
+        - |
+          timeout 20m bash -c '
+            set +e
+            pytest -s -v tests/diffusion/offloader/test_diffusion_cpu_offload.py
+            EXIT1=\$?
+            pytest -s -v tests/diffusion/offloader/test_diffusion_layerwise_offload.py
+            EXIT2=\$?
+            exit \$((EXIT1 | EXIT2))
+          '
 
     - label: "Diffusion · Cache Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -96,10 +83,10 @@ steps:
         - export GPU_ARCHS=gfx942
         - export VLLM_LOGGING_LEVEL=DEBUG
         - export VLLM_WORKER_MULTIPROC_METHOD=spawn
-        - pytest -sv tests/diffusion/cache -m 'core_model and cuda' --run-level "core_model"
+        - timeout 15m pytest -s -v -m "core_model and cache and diffusion and not distributed_cuda and L4"
 
     - label: "Diffusion Sequence Parallelism Test"
-      agent_pool: mi300_2
+      agent_pool: mi325_2
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -109,7 +96,7 @@ steps:
         - timeout 20m pytest -s -v tests/diffusion/distributed/test_sequence_parallel.py -m core_model
 
     - label: "Diffusion GPU Worker Test"
-      agent_pool: mi300_2
+      agent_pool: mi325_2
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -117,7 +104,7 @@ steps:
         - timeout 20m pytest -s -v tests/diffusion/test_diffusion_worker.py
 
     - label: "Diffusion · Model Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -125,7 +112,7 @@ steps:
         - timeout 15m pytest -sv tests/diffusion/models/ -m 'core_model and cuda' --run-level "core_model"
 
     - label: "Diffusion · Wan22 Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -139,7 +126,7 @@ steps:
 
 ## ISSUE depends on `diffusers` package: https://github.com/huggingface/diffusers/issues/13274
 # - label: "Audio Generation Model Test"
-#   agent_pool: mi300_1
+#   agent_pool: mi325_1
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking
@@ -150,7 +137,7 @@ steps:
 #     - timeout 20m pytest -s -v tests/e2e/offline_inference/test_stable_audio_expansion.py -m "advanced_model and diffusion and L4" --run-level advanced_model
 
 - label: "Engine Test"
-  agent_pool: mi300_1
+  agent_pool: mi325_1
   depends_on: amd-build
   mirror_hardwares: [amdproduction]
   grade: Blocking
@@ -162,7 +149,7 @@ steps:
       '
 
 # - label: "Omni Model Test Qwen3-Omni"
-#   agent_pool: mi300_2
+#   agent_pool: mi325_2
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking
@@ -175,7 +162,7 @@ steps:
 - group: ":card_index_dividers: E2E Test"
   steps:
     - label: "Qwen3-TTS E2E Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
@@ -186,17 +173,20 @@ steps:
         - timeout 30m pytest -s -v tests/e2e/online_serving/test_qwen3_tts_customvoice.py -m "core_model" --run-level "core_model"
 
     - label: "CosyVoice3-TTS E2E Test"
-      agent_pool: mi300_1
+      agent_pool: mi325_1
       depends_on: amd-build
       mirror_hardwares: [amdproduction]
       grade: Blocking
       commands:
-        - pytest -s -v tests/e2e/online_serving/test_cosyvoice3_tts_expansion.py -m "slow" --run-level "core_model"
+        - |
+          timeout 20m bash -c '
+            pytest -s -v tests/e2e/online_serving/test_cosyvoice3_tts_expansion.py -m "slow" --run-level "core_model"
+          '
 
 # TODO: Bagel test on ROCm is very unstable. @tjtanaa
 # Need to debug before reneable numerical changes across large PRs
 # - label: "Bagel Text2Img Model Test"
-#   agent_pool: mi300_1
+#   agent_pool: mi325_1
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking
@@ -208,7 +198,7 @@ steps:
 #     - timeout 30m pytest -s -v tests/e2e/offline_inference/test_bagel.py -m "core_model" --run-level "core_model" -k "rocm"
 
 # - label: "Bagel Img2Img Model Test"
-#   agent_pool: mi300_1
+#   agent_pool: mi325_1
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking
@@ -220,7 +210,7 @@ steps:
 #     - timeout 30m pytest -s -v tests/e2e/offline_inference/test_bagel.py -m "core_model" --run-level "core_model" -k "rocm"
 
 # - label: "Bagel Online Serving Test"
-#   agent_pool: mi300_1
+#   agent_pool: mi325_1
 #   depends_on: amd-build
 #   mirror_hardwares: [amdproduction]
 #   grade: Blocking

.buildkite/amd/test-template-amd-omni.j2

@@ -1,65 +1,11 @@
 {# vllm-omni customized version
    Based on: https://github.com/vllm-project/ci-infra/blob/main/buildkite/test-template-amd.j2
-   Last synced: 2026-08-04
+   Last synced: 2025-12-15
-   Modifications: Use the shared MI300 queues and Omni command-passing contract
+   Modifications: Removed unused CUDA/NVIDIA logic, keeping only AMD tests
 #}
 {% set docker_image_amd = "rocm/vllm-omni:$BUILDKITE_COMMIT" %}
 {% set default_working_dir = "/app/vllm-omni" %}
 
-{# MI300 runs tests natively in Kubernetes; Docker-in-Docker is unavailable. #}
-{% macro amd_native_k8s_plugin(indent, step) -%}
-{{ indent }}plugins:
-{{ indent }}  - kubernetes:
-{{ indent }}      podSpecPatch:
-{{ indent }}        automountServiceAccountToken: false
-{{ indent }}        securityContext:
-{{ indent }}          seccompProfile:
-{{ indent }}            type: RuntimeDefault
-{{ indent }}        imagePullSecrets:
-{{ indent }}          - name: docker-config
-{{ indent }}        containers:
-{{ indent }}          - name: container-0
-{{ indent }}            image: {{ docker_image_amd }}
-{{ indent }}            imagePullPolicy: Always
-{{ indent }}            securityContext:
-{{ indent }}              allowPrivilegeEscalation: false
-{{ indent }}              capabilities:
-{{ indent }}                add:
-{{ indent }}                  - IPC_LOCK
-{{ indent }}            resources:
-{{ indent }}              limits:
-{{ indent }}                amd.com/gpu: "{{ step.agent_pool[-1:] }}"
-{{ indent }}              requests:
-{{ indent }}                amd.com/gpu: "{{ step.agent_pool[-1:] }}"
-{{ indent }}            volumeMounts:
-{{ indent }}              - name: devshm
-{{ indent }}                mountPath: /dev/shm
-{{ indent }}            env:
-{{ indent }}              - name: AMD_CI_RUNTIME
-{{ indent }}                value: native
-{{ indent }}              - name: NATIVE_CI
-{{ indent }}                value: "true"
-{{ indent }}              - name: VLLM_CI_DOCKER_DISABLED
-{{ indent }}                value: "1"
-{{ indent }}              - name: VLLM_CI_EXPECTED_GPU_COUNT
-{{ indent }}                value: "{{ step.agent_pool[-1:] }}"
-{{ indent }}              - name: HF_HOME
-{{ indent }}                value: /home/buildkite-agent/huggingface
-{{ indent }}              - name: MIOPEN_DEBUG_CONV_DIRECT
-{{ indent }}                value: "0"
-{{ indent }}              - name: MIOPEN_DEBUG_CONV_GEMM
-{{ indent }}                value: "0"
-{{ indent }}              - name: VLLM_ROCM_USE_AITER
-{{ indent }}                value: "0"
-{{ indent }}              - name: PYTORCH_ROCM_ARCH
-{{ indent }}                value: ""
-{{ indent }}        volumes:
-{{ indent }}          - name: devshm
-{{ indent }}            emptyDir:
-{{ indent }}              medium: Memory
-{{ indent }}              sizeLimit: 16Gi
-{%- endmacro %}
-
 {% if env %}
 env:
 {% for name in env %}
@@ -107,31 +53,19 @@ steps:
           {% if step.agent_pool %}
           queue: amd_{{ step.agent_pool }}
           {% else %}
-          queue: amd_mi300_1
+          queue: amd_mi325_1
           {% endif %}
-{{ amd_native_k8s_plugin("        ", step) }}
 {% set cmd_body = (step.command or (step.commands | join("\n"))) | trim %}
 {% set indented_cmd = cmd_body | replace("\n", "\n            ") %}
         command: bash .buildkite/amd/scripts/run-amd-test.sh
         env:
-          AMD_CI_RUNTIME: "native"
+          DOCKER_BUILDKIT: "1"
-          NATIVE_CI: "true"
-          VLLM_CI_DOCKER_DISABLED: "1"
-          VLLM_CI_EXPECTED_GPU_COUNT: "{{ step.agent_pool[-1:] }}"
-          VLLM_CI_REQUIRE_PERSISTENT_HF_CACHE: "1"
-          HF_HOME: "/home/buildkite-agent/huggingface"
-          HF_HUB_DOWNLOAD_TIMEOUT: "300"
-          HF_HUB_ETAG_TIMEOUT: "60"
-          MIOPEN_DEBUG_CONV_DIRECT: "0"
-          MIOPEN_DEBUG_CONV_GEMM: "0"
-          VLLM_ROCM_USE_AITER: "0"
-          PYTORCH_ROCM_ARCH: ""
 {% if step.env %}
 {% for name in step.env %}
           {{ name }}: "{{ step.env[name] }}"
 {% endfor %}
 {% endif %}
-          TEST_COMMANDS: |-
+          TEST_COMMAND: |-
             (command rocm-smi || true) && cd {{ (step.working_dir or default_working_dir) | safe }}
 {% if env %}
 {% for name in env %}
@@ -148,15 +82,6 @@ steps:
 {% endif %}
             {{ indented_cmd | safe }}
         priority: 100
-{% if step.parallelism %}
-        parallelism: {{ step.parallelism }}
-{% endif %}
-{% if step.artifact_paths %}
-        artifact_paths:
-{% for path in step.artifact_paths %}
-          - "{{ path }}"
-{% endfor %}
-{% endif %}
         {% if step.timeout_in_minutes %}
         timeout_in_minutes: {{ step.timeout_in_minutes }}
         {% endif %}
@@ -175,31 +100,19 @@ steps:
       {% if step.agent_pool %}
       queue: amd_{{ step.agent_pool }}
       {% else %}
-      queue: amd_mi300_1
+      queue: amd_mi325_1
       {% endif %}
-{{ amd_native_k8s_plugin("    ", step) }}
 {% set cmd_body = (step.command or (step.commands | join("\n"))) | trim %}
 {% set indented_cmd = cmd_body | replace("\n", "\n        ") %}
     command: bash .buildkite/amd/scripts/run-amd-test.sh
     env:
-      AMD_CI_RUNTIME: "native"
+      DOCKER_BUILDKIT: "1"
-      NATIVE_CI: "true"
-      VLLM_CI_DOCKER_DISABLED: "1"
-      VLLM_CI_EXPECTED_GPU_COUNT: "{{ step.agent_pool[-1:] }}"
-      VLLM_CI_REQUIRE_PERSISTENT_HF_CACHE: "1"
-      HF_HOME: "/home/buildkite-agent/huggingface"
-      HF_HUB_DOWNLOAD_TIMEOUT: "300"
-      HF_HUB_ETAG_TIMEOUT: "60"
-      MIOPEN_DEBUG_CONV_DIRECT: "0"
-      MIOPEN_DEBUG_CONV_GEMM: "0"
-      VLLM_ROCM_USE_AITER: "0"
-      PYTORCH_ROCM_ARCH: ""
 {% if step.env %}
 {% for name in step.env %}
       {{ name }}: "{{ step.env[name] }}"
 {% endfor %}
 {% endif %}
-      TEST_COMMANDS: |-
+      TEST_COMMAND: |-
         (command rocm-smi || true) && cd {{ (step.working_dir or default_working_dir) | safe }}
 {% if env %}
 {% for name in env %}
@@ -216,15 +129,6 @@ steps:
 {% endif %}
         {{ indented_cmd | safe }}
     priority: 100
-{% if step.parallelism %}
-    parallelism: {{ step.parallelism }}
-{% endif %}
-{% if step.artifact_paths %}
-    artifact_paths:
-{% for path in step.artifact_paths %}
-      - "{{ path }}"
-{% endfor %}
-{% endif %}
     {% if step.timeout_in_minutes %}
     timeout_in_minutes: {{ step.timeout_in_minutes }}
     {% endif %}

.buildkite/common/ci_mirror_hardwares.yml

@@ -1,31 +1,14 @@
 # Uploader-only hardware presets for CUDA / NPU test YAML.
 # Referenced as ``mirror_hardwares: l4_1`` on a step; ``upload_pipeline.py`` expands to
-# ``agents`` (+ optional top-level ``image`` for NPU) + ``plugins`` (+ ``retry``
+# ``agents`` (+ optional top-level ``image`` for NPU) + ``plugins``, then strips
-# for ``l4_*``), then strips ``mirror_hardwares`` before Buildkite upload.
+# ``mirror_hardwares`` before Buildkite upload.
-#
-# Omit the key: compose {chip}_{n} from pytest -m SKU + cards_n (max of several
-# positives; not cards_1 with no positive cards_* → that chip's highest existing preset).
-# Unset MIRROR_HW matches H100 or L4 in -m (both → H100); no match skips the step.
-# MIRROR_HW=b200 matches B200 in -m (else the step is skipped).
-# MIRROR_HW must be empty or b200 (case-insensitive); unknown values fail the upload.
-# CUDA preset strings (h100_4, l4_1) are omitted when MIRROR_HW=b200.
-# Unset MIRROR_HW keeps string presets. NPU presets ignore MIRROR_HW.
 #
 # CUDA naming: {chip}_{count}
-#   l4_1, l4_2, l4_3, l4_4, h100_1 … h100_4, b200_1 … b200_4
+#   l4_1, l4_4, h100_1 … h100_4
 # NPU naming: {queue}_{resource_class}
 #   a2b3_npu_1  — ascend-a2b3, resource_class npu-1 (A2/B3 CI image)
 #   a2b3_npu_4  — ascend-a2b3, resource_class npu-4
-#   a3_npu_2    — ascend-a3, resource_class npu-2 (A3 CI image tag …:$BUILDKITE_COMMIT-a3)
+#   a3_npu_2    — ascend-a3, resource_class npu-2
-#
-# B200 (b200-k8s): matches vLLM device ``b200-k8s`` from ci-infra
-# (AgentQueue.B200_K8S + b200_plugin_template). Queue routes the K8s pool;
-# podSpec uses hostNetwork + /raid|/mnt/shared (no H100-style nodeSelector).
-#
-# The ``l4_*`` presets run on the EKS ``l4-k8s`` queue (agent-stack-k8s),
-# packed onto g6.12xlarge L4 nodes (``vllm.ci/gpu-pool=l4x4``). GPU / CPU /
-# memory / shm requests scale with card count (1 GPU = 10 CPU / 40Gi / 24Gi shm).
-# Retry covers lost agents / killed pods, not pytest or OOM (exit 1 / 137).
 
 x-npu-k8s-plugins: &npu_k8s_plugins
   - kubernetes:
@@ -33,224 +16,37 @@ x-npu-k8s-plugins: &npu_k8s_plugins
         imagePullSecrets:
           - name: swr-secret
 
-# Infra-only retry. Omit exit 1 (pytest failed) and 137 (cgroup OOM / SIGKILL
-# of the test process). ci-infra ``K8S_RETRY`` still retries exit 1.
-x-l4-k8s-retry: &l4_k8s_retry
-  automatic:
-    - exit_status: -1
-      limit: 1
-    - exit_status: 128
-      limit: 1
-    - signal_reason: agent_stop
-      limit: 1
-    - signal_reason: agent_refused
-      limit: 1
-
 
 mirror_hardwares:
   l4_1:
     agents:
-      queue: l4-k8s
+      queue: gpu_1_queue
-    retry: *l4_k8s_retry
     plugins:
-      - kubernetes:
+      - docker#v5.2.0:
-          metadata:
+          image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-            annotations:
+          always-pull: true
-              cluster-autoscaler.kubernetes.io/safe-to-evict: "false"
+          propagate-environment: true
-              karpenter.sh/do-not-disrupt: "true"
+          shm-size: "8gb"
-          podSpec:
+          environment:
-            priorityClassName: ci
+            - HF_HOME=/fsx/hf_cache
-            serviceAccountName: buildkite-gpu-jobs
+            - HF_TOKEN
-            tolerations:
+          volumes:
-              - key: nvidia.com/gpu
+            - /fsx/hf_cache:/fsx/hf_cache
-                operator: Exists
-                effect: NoSchedule
-            nodeSelector:
-              vllm.ci/gpu-pool: l4x4
-            containers:
-              - image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-                resources:
-                  limits:
-                    nvidia.com/gpu: 1
-                  requests:
-                    nvidia.com/gpu: 1
-                    cpu: "10"
-                    memory: 40Gi
-                volumeMounts:
-                  - name: devshm
-                    mountPath: /dev/shm
-                  - name: hf-cache
-                    mountPath: /fsx/hf_cache
-                env:
-                  - name: HF_HOME
-                    value: /fsx/hf_cache
-                  - name: HF_TOKEN
-                    valueFrom:
-                      secretKeyRef:
-                        name: hf-token-secret
-                        key: token
-            volumes:
-              - name: devshm
-                emptyDir:
-                  medium: Memory
-                  sizeLimit: 24Gi
-              - name: hf-cache
-                hostPath:
-                  path: /fsx/hf_cache
-                  type: DirectoryOrCreate
-
-  l4_2:
-    agents:
-      queue: l4-k8s
-    retry: *l4_k8s_retry
-    plugins:
-      - kubernetes:
-          metadata:
-            annotations:
-              cluster-autoscaler.kubernetes.io/safe-to-evict: "false"
-              karpenter.sh/do-not-disrupt: "true"
-          podSpec:
-            priorityClassName: ci
-            serviceAccountName: buildkite-gpu-jobs
-            tolerations:
-              - key: nvidia.com/gpu
-                operator: Exists
-                effect: NoSchedule
-            nodeSelector:
-              vllm.ci/gpu-pool: l4x4
-            containers:
-              - image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-                resources:
-                  limits:
-                    nvidia.com/gpu: 2
-                  requests:
-                    nvidia.com/gpu: 2
-                    cpu: "20"
-                    memory: 80Gi
-                volumeMounts:
-                  - name: devshm
-                    mountPath: /dev/shm
-                  - name: hf-cache
-                    mountPath: /fsx/hf_cache
-                env:
-                  - name: HF_HOME
-                    value: /fsx/hf_cache
-                  - name: HF_TOKEN
-                    valueFrom:
-                      secretKeyRef:
-                        name: hf-token-secret
-                        key: token
-            volumes:
-              - name: devshm
-                emptyDir:
-                  medium: Memory
-                  sizeLimit: 48Gi
-              - name: hf-cache
-                hostPath:
-                  path: /fsx/hf_cache
-                  type: DirectoryOrCreate
-
-  l4_3:
-    agents:
-      queue: l4-k8s
-    retry: *l4_k8s_retry
-    plugins:
-      - kubernetes:
-          metadata:
-            annotations:
-              cluster-autoscaler.kubernetes.io/safe-to-evict: "false"
-              karpenter.sh/do-not-disrupt: "true"
-          podSpec:
-            priorityClassName: ci
-            serviceAccountName: buildkite-gpu-jobs
-            tolerations:
-              - key: nvidia.com/gpu
-                operator: Exists
-                effect: NoSchedule
-            nodeSelector:
-              vllm.ci/gpu-pool: l4x4
-            containers:
-              - image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-                resources:
-                  limits:
-                    nvidia.com/gpu: 3
-                  requests:
-                    nvidia.com/gpu: 3
-                    cpu: "30"
-                    memory: 120Gi
-                volumeMounts:
-                  - name: devshm
-                    mountPath: /dev/shm
-                  - name: hf-cache
-                    mountPath: /fsx/hf_cache
-                env:
-                  - name: HF_HOME
-                    value: /fsx/hf_cache
-                  - name: HF_TOKEN
-                    valueFrom:
-                      secretKeyRef:
-                        name: hf-token-secret
-                        key: token
-            volumes:
-              - name: devshm
-                emptyDir:
-                  medium: Memory
-                  sizeLimit: 72Gi
-              - name: hf-cache
-                hostPath:
-                  path: /fsx/hf_cache
-                  type: DirectoryOrCreate
 
   l4_4:
     agents:
-      queue: l4-k8s
+      queue: gpu_4_queue
-    retry: *l4_k8s_retry
     plugins:
-      - kubernetes:
+      - docker#v5.2.0:
-          metadata:
+          image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-            annotations:
+          always-pull: true
-              cluster-autoscaler.kubernetes.io/safe-to-evict: "false"
+          propagate-environment: true
-              karpenter.sh/do-not-disrupt: "true"
+          shm-size: "8gb"
-          podSpec:
+          environment:
-            priorityClassName: ci
+            - HF_HOME=/fsx/hf_cache
-            serviceAccountName: buildkite-gpu-jobs
+            - HF_TOKEN
-            tolerations:
+          volumes:
-              - key: nvidia.com/gpu
+            - /fsx/hf_cache:/fsx/hf_cache
-                operator: Exists
-                effect: NoSchedule
-            nodeSelector:
-              vllm.ci/gpu-pool: l4x4
-            containers:
-              - image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-                resources:
-                  limits:
-                    nvidia.com/gpu: 4
-                  requests:
-                    nvidia.com/gpu: 4
-                    cpu: "40"
-                    memory: 160Gi
-                volumeMounts:
-                  - name: devshm
-                    mountPath: /dev/shm
-                  - name: hf-cache
-                    mountPath: /fsx/hf_cache
-                env:
-                  - name: HF_HOME
-                    value: /fsx/hf_cache
-                  - name: HF_TOKEN
-                    valueFrom:
-                      secretKeyRef:
-                        name: hf-token-secret
-                        key: token
-            volumes:
-              - name: devshm
-                emptyDir:
-                  medium: Memory
-                  sizeLimit: 96Gi
-              - name: hf-cache
-                hostPath:
-                  path: /fsx/hf_cache
-                  type: DirectoryOrCreate
 
   h100_1:
     agents:
@@ -392,211 +188,6 @@ mirror_hardwares:
                   path: /mnt/hf-cache
                   type: DirectoryOrCreate
 
-  # vLLM device: b200-k8s → agents.queue b200-k8s + Buildkite K8s agent-stack podSpec
-  b200_1:
-    agents:
-      queue: b200-k8s
-    plugins:
-      - kubernetes:
-          podSpec:
-            runtimeClassName: nvidia
-            hostNetwork: true
-            dnsPolicy: ClusterFirstWithHostNet
-            imagePullSecrets:
-              - name: k8s-ecr-login-renew-docker-secret
-            containers:
-              - image: 936637512419.dkr.ecr.us-west-2.amazonaws.com/vllm-ci-pull-through-cache/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-                resources:
-                  limits:
-                    nvidia.com/gpu: 1
-                securityContext:
-                  capabilities:
-                    add:
-                      - IPC_LOCK
-                      - SYS_RESOURCE
-                volumeMounts:
-                  - name: devshm
-                    mountPath: /dev/shm
-                  - name: raid
-                    mountPath: /raid
-                  - name: shared
-                    mountPath: /mnt/shared
-                env:
-                  - name: NCCL_CUMEM_HOST_ENABLE
-                    value: "0"
-                  - name: HF_HOME
-                    value: /raid/hf_cache
-                  - name: HF_TOKEN
-                    valueFrom:
-                      secretKeyRef:
-                        name: hf-token-secret
-                        key: token
-            volumes:
-              - name: devshm
-                emptyDir:
-                  medium: Memory
-              - name: raid
-                hostPath:
-                  path: /raid
-                  type: DirectoryOrCreate
-              - name: shared
-                hostPath:
-                  path: /mnt/shared
-                  type: DirectoryOrCreate
-
-  b200_2:
-    agents:
-      queue: b200-k8s
-    plugins:
-      - kubernetes:
-          podSpec:
-            runtimeClassName: nvidia
-            hostNetwork: true
-            dnsPolicy: ClusterFirstWithHostNet
-            imagePullSecrets:
-              - name: k8s-ecr-login-renew-docker-secret
-            containers:
-              - image: 936637512419.dkr.ecr.us-west-2.amazonaws.com/vllm-ci-pull-through-cache/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-                resources:
-                  limits:
-                    nvidia.com/gpu: 2
-                securityContext:
-                  capabilities:
-                    add:
-                      - IPC_LOCK
-                      - SYS_RESOURCE
-                volumeMounts:
-                  - name: devshm
-                    mountPath: /dev/shm
-                  - name: raid
-                    mountPath: /raid
-                  - name: shared
-                    mountPath: /mnt/shared
-                env:
-                  - name: NCCL_CUMEM_HOST_ENABLE
-                    value: "0"
-                  - name: HF_HOME
-                    value: /raid/hf_cache
-                  - name: HF_TOKEN
-                    valueFrom:
-                      secretKeyRef:
-                        name: hf-token-secret
-                        key: token
-            volumes:
-              - name: devshm
-                emptyDir:
-                  medium: Memory
-              - name: raid
-                hostPath:
-                  path: /raid
-                  type: DirectoryOrCreate
-              - name: shared
-                hostPath:
-                  path: /mnt/shared
-                  type: DirectoryOrCreate
-
-  b200_3:
-    agents:
-      queue: b200-k8s
-    plugins:
-      - kubernetes:
-          podSpec:
-            runtimeClassName: nvidia
-            hostNetwork: true
-            dnsPolicy: ClusterFirstWithHostNet
-            imagePullSecrets:
-              - name: k8s-ecr-login-renew-docker-secret
-            containers:
-              - image: 936637512419.dkr.ecr.us-west-2.amazonaws.com/vllm-ci-pull-through-cache/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-                resources:
-                  limits:
-                    nvidia.com/gpu: 3
-                securityContext:
-                  capabilities:
-                    add:
-                      - IPC_LOCK
-                      - SYS_RESOURCE
-                volumeMounts:
-                  - name: devshm
-                    mountPath: /dev/shm
-                  - name: raid
-                    mountPath: /raid
-                  - name: shared
-                    mountPath: /mnt/shared
-                env:
-                  - name: NCCL_CUMEM_HOST_ENABLE
-                    value: "0"
-                  - name: HF_HOME
-                    value: /raid/hf_cache
-                  - name: HF_TOKEN
-                    valueFrom:
-                      secretKeyRef:
-                        name: hf-token-secret
-                        key: token
-            volumes:
-              - name: devshm
-                emptyDir:
-                  medium: Memory
-              - name: raid
-                hostPath:
-                  path: /raid
-                  type: DirectoryOrCreate
-              - name: shared
-                hostPath:
-                  path: /mnt/shared
-                  type: DirectoryOrCreate
-
-  b200_4:
-    agents:
-      queue: b200-k8s
-    plugins:
-      - kubernetes:
-          podSpec:
-            runtimeClassName: nvidia
-            hostNetwork: true
-            dnsPolicy: ClusterFirstWithHostNet
-            imagePullSecrets:
-              - name: k8s-ecr-login-renew-docker-secret
-            containers:
-              - image: 936637512419.dkr.ecr.us-west-2.amazonaws.com/vllm-ci-pull-through-cache/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
-                resources:
-                  limits:
-                    nvidia.com/gpu: 4
-                securityContext:
-                  capabilities:
-                    add:
-                      - IPC_LOCK
-                      - SYS_RESOURCE
-                volumeMounts:
-                  - name: devshm
-                    mountPath: /dev/shm
-                  - name: raid
-                    mountPath: /raid
-                  - name: shared
-                    mountPath: /mnt/shared
-                env:
-                  - name: NCCL_CUMEM_HOST_ENABLE
-                    value: "0"
-                  - name: HF_HOME
-                    value: /raid/hf_cache
-                  - name: HF_TOKEN
-                    valueFrom:
-                      secretKeyRef:
-                        name: hf-token-secret
-                        key: token
-            volumes:
-              - name: devshm
-                emptyDir:
-                  medium: Memory
-              - name: raid
-                hostPath:
-                  path: /raid
-                  type: DirectoryOrCreate
-              - name: shared
-                hostPath:
-                  path: /mnt/shared
-                  type: DirectoryOrCreate
-
   a2b3_npu_1:
     agents:
       queue: "ascend-a2b3"

.buildkite/common/scripts/run_cov_split.sh

@@ -1,98 +0,0 @@
-#!/usr/bin/env bash
-# Run a model's E2E tests once per entry mode as separate coverage runs and upload
-# the reports, so per-mode attribution survives (see vllm-project/vllm-omni#5332).
-#
-# Usage: run_cov_split.sh --model-id <id> [--offline <paths>] [--online <paths>] \
-#                         -- <pytest args...>
-#
-#   --model-id  Names the artifacts: coverage-<id>-{offline,online}-<step id>.xml.gz.
-#               Use the model's directory name under vllm_omni/*/models/.
-#   --offline   Pytest path(s) for the offline half. Repeatable.
-#   --online    Pytest path(s) for the online half. Repeatable.
-#   --          Everything after this is forwarded verbatim to every pytest run,
-#               quoting preserved: the job's own -m expression, --run-level, and
-#               anything else it needs (e.g. --test-config-file).
-#
-# At least one of --offline/--online is required, so a job with only one mode runs
-# just that half. Total runtime is bounded by the step's timeout_in_minutes.
-#
-# Coverage + artifact upload run only on scheduled weekly E2E sweeps
-# (WEEKLY=1 and BUILDKITE_BRANCH=main — same gate as bootstrap --e2e).
-# All other invocations (PR merge-test, regular main merge, local) run the
-# same combined pytest as the job used before coverage was added.
-#
-# Exits non-zero if any half or the upload failed.
-set -uo pipefail
-
-MODEL_ID=""
-OFFLINE=()
-ONLINE=()
-PYTEST_ARGS=()
-
-while [[ $# -gt 0 ]]; do
-    case "$1" in
-        --model-id) MODEL_ID="$2"; shift 2 ;;
-        --offline)  OFFLINE+=("$2"); shift 2 ;;
-        --online)   ONLINE+=("$2"); shift 2 ;;
-        --)         shift; PYTEST_ARGS=("$@"); break ;;
-        *) echo "run_cov_split.sh: unknown argument: $1" >&2; exit 2 ;;
-    esac
-done
-
-if [[ -z "${MODEL_ID}" ]]; then
-    echo "run_cov_split.sh: missing --model-id" >&2
-    exit 2
-fi
-
-if ((${#OFFLINE[@]} == 0 && ${#ONLINE[@]} == 0)); then
-    echo "run_cov_split.sh: need at least one of --offline / --online" >&2
-    exit 2
-fi
-
-_should_run_coverage() {
-    [[ "${WEEKLY:-}" == "1" && "${BUILDKITE_BRANCH:-}" == "main" ]]
-}
-
-if ! _should_run_coverage; then
-    PATHS=("${OFFLINE[@]}" "${ONLINE[@]}")
-    echo "--- pytest (no coverage): ${MODEL_ID}"
-    pytest -s -v "${PATHS[@]}" "${PYTEST_ARGS[@]}"
-    exit $?
-fi
-
-# Two steps can test the same model, and artifacts are scoped to the build (a
-# nightly one carries the ready, merge and nightly tiers at once), so unsuffixed
-# reports would share a path -- a glob download then races to rename onto one file.
-# Not BUILDKITE_JOB_ID: the step id survives retries, so attempts collapse under
-# Buildkite's latest-attempt lookup instead of accumulating. A parallel step would
-# need BUILDKITE_PARALLEL_JOB as well; no pilot is one.
-REPORT_SUFFIX="${BUILDKITE_STEP_ID:-local}"
-OFFLINE_REPORT="coverage-${MODEL_ID}-offline-${REPORT_SUFFIX}.xml"
-ONLINE_REPORT="coverage-${MODEL_ID}-online-${REPORT_SUFFIX}.xml"
-
-run_half() {
-    local mode="$1"
-    shift
-    local report="coverage-${MODEL_ID}-${mode}-${REPORT_SUFFIX}.xml"
-    echo "--- coverage: ${MODEL_ID} ${mode}"
-    pytest -s -v "$@" "${PYTEST_ARGS[@]}" --cov=vllm_omni --cov-report="xml:${report}"
-    local status=$?
-    # Cobertura lists every statement in vllm_omni whether or not it ran, so the
-    # report is ~7MB no matter how little the job covers. The repetition is what
-    # dominates it, and gzip takes that to ~400KB.
-    gzip -9 -f "${report}" || status=1
-    return "${status}"
-}
-
-# Clear both modes, not just the ones being run: the upload glob matches both, so
-# a report left by an earlier run would otherwise be published as if the mode it
-# belongs to had run this time.
-rm -f "${OFFLINE_REPORT}" "${ONLINE_REPORT}" "${OFFLINE_REPORT}.gz" "${ONLINE_REPORT}.gz"
-
-EXIT=0
-((${#OFFLINE[@]})) && { run_half offline "${OFFLINE[@]}" || EXIT=1; }
-((${#ONLINE[@]})) && { run_half online "${ONLINE[@]}" || EXIT=1; }
-
-buildkite-agent artifact upload "coverage-${MODEL_ID}-*-${REPORT_SUFFIX}.xml.gz" || EXIT=1
-
-exit "${EXIT}"

.buildkite/common/scripts/skip_ci.py

@@ -1,6 +1,5 @@
 #!/usr/bin/env python3
 # SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
 """Shared Buildkite skip-ci and CI-YAML level targeting logic.
 
 ``resolve_ci_decision(changed_files, ...)`` buckets explicit file lists (tests, dry-run).
@@ -11,6 +10,12 @@ CLI (for bootstrap scripts; exit code is the signal):
   python3 skip_ci.py gate <platform> <level>
       Exit 0 when bootstrap should stop (skip-all, or that L2/L3 target is off).
       Prints ``skip-all`` or ``skip-l23`` on stdout; logs the decision once.
+  python3 skip_ci.py check-skip-all
+      Exit 0 when only docs/skip-mark changes → bootstrap skips entire CI upload.
+  python3 skip_ci.py check-skip-l2-l3 <platform> <level>
+      Exit 0 when that L2/L3 pipeline should be skipped (CI config YAML-only diffs).
+  python3 skip_ci.py print-annotate
+      Print decision message for Buildkite annotation.
 """
 
 from __future__ import annotations
@@ -253,10 +258,11 @@ class CiDecision:
 
 @dataclass(frozen=True)
 class CiContext:
-    """Skip-ci decision plus changed files used for pipeline filtering."""
+    """Skip-ci decision plus git diff inputs used to compute it."""
 
     decision: CiDecision
     changed_files: list[str] | None
+    diff_range: str | None
 
 
 def _log(message: str) -> None:
@@ -326,7 +332,7 @@ def resolve_ci_context_from_git() -> CiContext:
     diff_range = _resolve_diff_range()
     changed_files = _changed_files_for_diff_range(diff_range)
     decision = resolve_ci_decision(changed_files, diff_range=diff_range)
-    return CiContext(decision=decision, changed_files=changed_files)
+    return CiContext(decision=decision, changed_files=changed_files, diff_range=diff_range)
 
 
 # --- Doc / skip-mark helpers ---
@@ -565,6 +571,23 @@ def main() -> int:
     gate_parser.add_argument("platform", help="cuda, amd, intel, or npu")
     gate_parser.add_argument("level", choices=("l2", "l3"), help="test level")
 
+    subparsers.add_parser(
+        "check-skip-all",
+        help="exit 0 when docs/skip-mark-only diff should skip entire CI upload",
+    )
+
+    skip_parser = subparsers.add_parser(
+        "check-skip-l2-l3",
+        help="exit 0 when PLATFORM l2/l3 pipeline should be skipped (CI config YAML-only diffs)",
+    )
+    skip_parser.add_argument("platform", help="cuda, amd, intel, or npu")
+    skip_parser.add_argument("level", choices=("l2", "l3"), help="test level")
+
+    subparsers.add_parser(
+        "print-annotate",
+        help="print human-readable skip reason for Buildkite annotation",
+    )
+
     args = parser.parse_args()
     decision = resolve_ci_context_from_git().decision
 
@@ -577,6 +600,16 @@ def main() -> int:
             return 0
         return 1
 
+    if args.command == "check-skip-all":
+        return 0 if decision.skip_all else 1
+
+    if args.command == "check-skip-l2-l3":
+        return 0 if not decision.is_run(args.platform, args.level) else 1
+
+    if args.command == "print-annotate":
+        print(decision.message or "CI will run normally")
+        return 0
+
     return 1
 
 

.buildkite/common/scripts/upload_pipeline.py

@@ -1,6 +1,5 @@
 #!/usr/bin/env python3
 # SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
 """Render and optionally upload Buildkite pipeline YAML with diff-aware logic.
 
 Bootstrap mode (``bootstrap-upload-steps.yml``):
@@ -12,19 +11,8 @@ Bootstrap mode (``bootstrap-upload-steps.yml``):
 
 Test pipeline mode (e.g. test-merge.yml):
   - Drop steps whose ``source_file_dependencies`` do not match changed files.
-  - Expand uploader-only ``mirror_hardwares`` into ``agents`` (+ optional ``image``
+  - Expand uploader-only ``mirror_hardwares: l4_1`` into ``agents`` (+ optional ``image``
     for NPU) + ``plugins`` (see ci_mirror_hardwares.yml).
-  - Omit ``mirror_hardwares`` to compose ``{chip}_{n}`` from pytest ``-m`` SKU
-    markers plus ``cards_n`` (max of several positives; ``not cards_1`` with no
-    positive ``cards_*`` uses that chip's highest existing preset). An explicit
-    preset string (``h100_2``) is used as-is (marks are ignored). Names that
-    are not keys in ``ci_mirror_hardwares.yml`` fail the upload.
-
-    Inferred chip: unset ``MIRROR_HW`` matches ``H100`` or ``L4`` in ``-m``
-    (both → H100); no match skips the step. ``MIRROR_HW=b200`` matches
-    ``B200`` in ``-m`` (otherwise skipped). ``-m`` is not rewritten.
-    ``MIRROR_HW`` must be empty or ``b200``; unknown values fail the upload.
-    A CUDA preset string such as ``h100_4`` is omitted when ``MIRROR_HW=b200``.
 
 Usage:
   python3 upload_pipeline.py [--upload] [--all | --e2e] <pipeline.yml>
@@ -36,13 +24,11 @@ from __future__ import annotations
 
 import argparse
 import copy
-import os
-import re
 import subprocess
 import sys
 from functools import lru_cache
 from pathlib import Path
-from typing import Any, Literal
+from typing import Any
 
 try:
     import yaml
@@ -58,14 +44,6 @@ from skip_ci import (
     resolve_ci_context_from_git,
 )
 
-if str(ROOT) not in sys.path:
-    sys.path.insert(0, str(ROOT))
-
-from tests.helpers.mark import (  # noqa: E402
-    get_skus_for_platform,
-    get_supported_card_counts,
-)
-
 # --- Constants ---
 
 LOG = "upload_pipeline"
@@ -80,12 +58,11 @@ BOOTSTRAP_UPLOAD_IF_KEYS = {
 E2E_GROUP_MARKER = "E2E Test"
 CI_MIRROR_HARDWARES_PATH = ROOT / ".buildkite/common/ci_mirror_hardwares.yml"
 
-# Bootstrap Buildkite ``if`` expressions.
+CUDA_NIGHTLY_ONLY = (
-# ``*_MAIN_IF``: main + env schedule. ``*_LABEL_IF``: PR label (and/or composed with MAIN).
+    '(build.pull_request.labels includes "nightly-test") || (build.branch == "main" && build.env("NIGHTLY") == "1")'
-# ``*_UPLOAD_IF``: full gate for uploading that child pipeline.
+)
-NIGHTLY_MAIN_IF = 'build.branch == "main" && build.env("NIGHTLY") == "1"'
+NPU_NIGHTLY_ONLY = (
-NIGHTLY_LABEL_IF = (
+    '(build.branch == "main" && build.env("NIGHTLY") == "1") || '
-    f"({NIGHTLY_MAIN_IF}) || "
     '(build.branch != "main" && ('
     'build.pull_request.labels includes "nightly-test" || '
     'build.pull_request.labels includes "omni-test" || '
@@ -94,19 +71,6 @@ NIGHTLY_LABEL_IF = (
     'build.pull_request.labels includes "diffusion-x2v-test"'
     "))"
 )
-WEEKLY_E2E_IF = 'build.branch == "main" && build.env("WEEKLY") == "1"'
-WEEKLY_MAIN_IF = 'build.branch == "main" && (build.env("WEEKLY") == "1" || build.env("NON_CRITICAL") == "1")'
-WEEKLY_LABEL_IF = f'({WEEKLY_MAIN_IF}) || (build.branch != "main" && build.pull_request.labels includes "weekly-test")'
-READY_LABEL_IF = 'build.branch != "main" && build.pull_request.labels includes "ready"'
-MERGE_LABEL_IF = 'build.branch != "main" && build.pull_request.labels includes "merge-test"'
-MERGE_MAIN_IF = (
-    'build.branch == "main" && build.env("NIGHTLY") != "1" && '
-    'build.env("WEEKLY") != "1" && build.env("NON_CRITICAL") != "1"'
-)
-READY_UPLOAD_IF = f"({WEEKLY_E2E_IF}) || ({READY_LABEL_IF})"
-MERGE_UPLOAD_IF = f"({WEEKLY_E2E_IF}) || (({MERGE_MAIN_IF}) || ({MERGE_LABEL_IF}))"
-BOOTSTRAP_DISABLED_IF = "false"
-BOOTSTRAP_ENABLED_IF = "true"
 
 
 # --- Logging ---
@@ -138,47 +102,74 @@ def _format_bootstrap_if(expr: str) -> str:
 
 
 def _compute_bootstrap_if_exprs(*, decision, platform: str) -> dict[str, str]:
+    disabled = "false"
+    nightly_main = 'build.branch == "main" && build.env("NIGHTLY") == "1"'
+    nightly_only = NPU_NIGHTLY_ONLY if platform == "npu" else CUDA_NIGHTLY_ONLY
+
     if platform == "npu":
-        ready_upload = READY_LABEL_IF
+        ready_pr = (
-        merge_upload = BOOTSTRAP_DISABLED_IF
+            'build.branch != "main" && ('
-        weekly_label_if = BOOTSTRAP_DISABLED_IF
+            'build.pull_request.labels includes "npu-test" || '
+            'build.pull_request.labels includes "ready"'
+            ")"
+        )
+        nightly_label_if = nightly_only
+        weekly_label_if = disabled
+        merge_base = disabled
     else:
-        ready_upload = READY_UPLOAD_IF
+        ready_pr = 'build.branch != "main" && build.pull_request.labels includes "ready"'
-        merge_upload = MERGE_UPLOAD_IF
+        merge_main = 'build.branch == "main" && build.env("NIGHTLY") != "1" && build.env("WEEKLY") != "1"'
-        weekly_label_if = WEEKLY_LABEL_IF
+        merge_pr = 'build.branch != "main" && build.pull_request.labels includes "merge-test"'
+        nightly_label_if = (
+            '(build.branch == "main" && build.env("NIGHTLY") == "1") || '
+            '(build.branch != "main" && ('
+            'build.pull_request.labels includes "nightly-test" || '
+            'build.pull_request.labels includes "omni-test" || '
+            'build.pull_request.labels includes "tts-test" || '
+            'build.pull_request.labels includes "diffusion-x2iat-test" || '
+            'build.pull_request.labels includes "diffusion-x2v-test"'
+            "))"
+        )
+        weekly_label_if = (
+            '(build.branch == "main" && build.env("WEEKLY") == "1") || '
+            '(build.branch != "main" && build.pull_request.labels includes "weekly-test")'
+        )
+        merge_base = f"({nightly_main}) || (({merge_main}) || ({merge_pr}))"
+
+    ready_base = f"({nightly_main}) || ({ready_pr})"
+    weekly_main = 'build.branch == "main" && build.env("WEEKLY") == "1"'
 
     if decision.skip_all:
         # Docs / skip-mark only: no PR-label escape hatch. Main scheduled
-        # NIGHTLY=1 still runs L4; WEEKLY=1 / NON_CRITICAL=1 still run L5.
+        # NIGHTLY=1 still runs L4 + L2/L3 (--e2e); WEEKLY=1 still runs L5.
-        # main+WEEKLY=1 also uploads L2/L3 (those steps then pass --e2e).
+        image_expr = f"({nightly_main}) || ({weekly_main})" if platform == "cuda" else nightly_main
-        image_expr = f"({NIGHTLY_MAIN_IF}) || ({WEEKLY_MAIN_IF})" if platform == "cuda" else NIGHTLY_MAIN_IF
+        ready_expr = nightly_main
-        ready_expr = WEEKLY_E2E_IF if platform == "cuda" else BOOTSTRAP_DISABLED_IF
+        merge_expr = nightly_main if platform == "cuda" else disabled
-        merge_expr = WEEKLY_E2E_IF if platform == "cuda" else BOOTSTRAP_DISABLED_IF
+        nightly_expr = nightly_main
-        nightly_expr = NIGHTLY_MAIN_IF
+        weekly_expr = weekly_main if platform == "cuda" else disabled
-        weekly_expr = WEEKLY_MAIN_IF if platform == "cuda" else BOOTSTRAP_DISABLED_IF
     elif decision.skip_l2_l3:
         l2_enabled = decision.is_run("npu", "l2") if platform == "npu" else decision.is_run("cuda", "l2")
         l3_enabled = platform == "cuda" and decision.is_run("cuda", "l3")
 
-        ready_expr = ready_upload if l2_enabled else BOOTSTRAP_DISABLED_IF
+        ready_expr = ready_base if l2_enabled else disabled
-        merge_expr = merge_upload if l3_enabled else BOOTSTRAP_DISABLED_IF
+        merge_expr = merge_base if l3_enabled else disabled
-        nightly_expr = NIGHTLY_LABEL_IF
+        nightly_expr = nightly_label_if
-        weekly_expr = weekly_label_if if platform == "cuda" else BOOTSTRAP_DISABLED_IF
+        weekly_expr = weekly_label_if if platform == "cuda" else disabled
 
-        image_parts = [f"({NIGHTLY_LABEL_IF})"]
+        image_parts = [f"({nightly_label_if})"]
         if platform == "cuda":
             image_parts.append(f"({weekly_label_if})")
         if l2_enabled:
-            image_parts.insert(0, f"({ready_upload})")
+            image_parts.insert(0, f"({ready_base})")
         if l3_enabled:
-            image_parts.insert(1 if l2_enabled else 0, f"({merge_upload})")
+            image_parts.insert(1 if l2_enabled else 0, f"({merge_base})")
         image_expr = " || ".join(image_parts)
     else:
-        image_expr = BOOTSTRAP_ENABLED_IF
+        image_expr = "true"
-        ready_expr = ready_upload
+        ready_expr = ready_base
-        merge_expr = merge_upload if platform == "cuda" else BOOTSTRAP_DISABLED_IF
+        merge_expr = merge_base if platform == "cuda" else disabled
-        nightly_expr = NIGHTLY_LABEL_IF
+        nightly_expr = nightly_label_if
-        weekly_expr = weekly_label_if if platform == "cuda" else BOOTSTRAP_DISABLED_IF
+        weekly_expr = weekly_label_if if platform == "cuda" else disabled
 
     return {
         "image": _format_bootstrap_if(image_expr),
@@ -254,257 +245,32 @@ def _load_mirror_hardwares() -> dict[str, dict[str, Any]]:
     return presets
 
 
-@lru_cache(maxsize=1)
+def _expand_mirror_hardwares(step: dict[str, Any]) -> dict[str, Any]:
-def _cuda_mirror_chips() -> tuple[str, ...]:
+    """Replace uploader-only ``mirror_hardwares`` with preset fields from ci_mirror_hardwares.yml."""
-    """Lowercase CUDA SKUs that have at least one ``{chip}_{n}`` preset.
+    hardware = step.get("mirror_hardwares")
+    if hardware is None:
+        return step
 
-    Longer tokens first so ``h100`` is not matched as a prefix of ``h200``.
+    if not isinstance(hardware, str) or not hardware.strip():
-    """
-    skus = get_skus_for_platform("cuda")
-    known = _load_mirror_hardwares()
-    chips = {
-        sku.lower() for sku in skus if any(name == sku.lower() or name.startswith(f"{sku.lower()}_") for name in known)
-    }
-    return tuple(sorted(chips, key=lambda chip: (-len(chip), chip)))
-
-
-def _read_cards_marks(expr: str) -> tuple[set[int], bool]:
-    """Parse registered ``cards_n`` / ``not cards_n`` from a pytest ``-m`` expr.
-
-    Returns positive card counts and whether any ``not cards_*`` is present.
-    """
-    counts = get_supported_card_counts()
-    alt = "|".join(str(n) for n in sorted(counts, key=lambda n: (-len(str(n)), -n))) if counts else "(?!)"
-    positives: set[int] = set()
-    has_not_cards = False
-    for match in re.finditer(rf"\b(?:(not)\s+)?cards_({alt})\b", expr):
-        if match.group(1):
-            has_not_cards = True
-        else:
-            positives.add(int(match.group(2)))
-    return positives, has_not_cards
-
-
-def _read_hardware_marks(expr: str) -> set[str]:
-    """Return lowercase positive CUDA SKUs that have a mirror preset.
-
-    ``not H100`` is ignored. SKUs with no ``{chip}_*`` preset (e.g. H200) are dropped.
-    """
-    skus = sorted(get_skus_for_platform("cuda"), key=lambda s: (-len(s), s))
-    if not skus:
-        return set()
-    alt = "|".join(re.escape(sku) for sku in skus)
-    mirror = set(_cuda_mirror_chips())
-    chips: set[str] = set()
-    for match in re.finditer(rf"\b(?:(not)\s+)?({alt})\b", expr):
-        if match.group(1):
-            continue
-        chip = match.group(2).lower()
-        if chip in mirror:
-            chips.add(chip)
-    return chips
-
-
-# CI policy: not pytest-mark registry.
-_SUPPORTED_MIRROR_HW_SELECTORS = frozenset({"b200"})
-# Unset MIRROR_HW: match H100 then L4 in -m (skip if neither).
-_DEFAULT_INFER_CHIPS = ("h100", "l4")
-_PYTEST_MARKER_ARG = re.compile(r"-m\s+(?:\"([^\"]*)\"|'([^']*)'|(\S+))")
-_CARDS_CHIP_MAX: Literal["max"] = "max"
-
-
-def _get_mirror_hw_selector() -> str:
-    """Return lowercase ``MIRROR_HW``, or empty to keep string presets / marker chips.
-
-    A set value must be ``b200`` (case-insensitive). Unknown values fail closed
-    so a typo cannot silently drop the CUDA pipeline.
-    """
-    selector = os.environ.get("MIRROR_HW", "").strip().lower()
-    if not selector:
-        return ""
-    if selector not in _SUPPORTED_MIRROR_HW_SELECTORS:
-        allowed = " or ".join(repr(s) for s in sorted(_SUPPORTED_MIRROR_HW_SELECTORS))
-        raise ValueError(f"unsupported MIRROR_HW={selector!r}; expected empty or {allowed}")
-    return selector
-
-
-def _read_pytest_marks(commands: Any) -> tuple[set[str], int | Literal["max"] | None]:
-    """Read pytest ``-m`` from step commands: CUDA SKU chips and cards count.
-
-    *chips* are lowercase SKUs from positive markers (``not H100`` is ignored).
-    *cards* is the max positive ``cards_n``, ``"max"`` when only ``not cards_*``
-    is present, or ``None``.
-    """
-    chunks: list[str] = []
-
-    def _collect(value: Any) -> None:
-        if value is None:
-            return
-        if isinstance(value, str):
-            chunks.append(value)
-        elif isinstance(value, list):
-            for part in value:
-                _collect(part)
-        else:
-            chunks.append(str(value))
-
-    _collect(commands)
-    text = "\n".join(chunks)
-
-    chips: set[str] = set()
-    positives: set[int] = set()
-    has_not_cards = False
-    for match in _PYTEST_MARKER_ARG.finditer(text):
-        expr = match.group(1) or match.group(2) or match.group(3) or ""
-        chips |= _read_hardware_marks(expr)
-        card_counts, not_cards = _read_cards_marks(expr)
-        positives.update(card_counts)
-        has_not_cards = has_not_cards or not_cards
-
-    if positives:
-        cards: int | Literal["max"] | None = max(positives)
-    elif has_not_cards:
-        cards = _CARDS_CHIP_MAX
-    else:
-        cards = None
-    return chips, cards
-
-
-def _compose_mirror_hardware_name(
-    chips: set[str],
-    cards: int | Literal["max"] | None,
-    *,
-    step_label: str,
-) -> str | None:
-    """Assemble ``{chip}_{n}`` from pytest ``-m`` chips/cards, or None to skip.
-
-    Unset ``MIRROR_HW``: ``H100`` if in ``-m``, else ``L4``, else skip.
-    Set ``MIRROR_HW``: that chip if in ``-m``, else skip.
-    ``cards`` is a registered ``cards_n``, ``"max"`` for that chip's highest
-    existing preset, or None (error after a chip matched — no ``cards_*``).
-    """
-    selector = _get_mirror_hw_selector()
-    if selector:
-        chip = selector if selector in chips else None
-        skip_reason = f"MIRROR_HW={selector!r} not in -m SKUs {sorted(chips)}"
-    else:
-        chip = next((c for c in _DEFAULT_INFER_CHIPS if c in chips), None)
-        skip_reason = f"no H100 or L4 in -m SKUs {sorted(chips)}"
-    if chip is None:
-        _log(f"skip {step_label}: {skip_reason}")
-        return None
-    if cards is None:
         raise ValueError(
-            f"step {step_label!r} has a pytest -m SKU marker but no cards_* "
+            f"mirror_hardwares must be a non-empty string in step {_get_step_label(step)!r}",
-            f"(or not cards_*); add cards_n or set mirror_hardwares to force a preset",
         )
 
-    count_source = "not cards_*" if cards == _CARDS_CHIP_MAX else f"cards_{cards}"
+    preset = _load_mirror_hardwares().get(hardware)
-    count = None if cards == _CARDS_CHIP_MAX else cards
-    registered = get_supported_card_counts()
-    if count is not None and count not in registered:
-        supported = ", ".join(f"cards_{n}" for n in sorted(registered)) or "(none)"
-        raise ValueError(
-            f"{count_source} in step {step_label!r} is not a registered cards_* mark; supported: {supported}",
-        )
-    known = _load_mirror_hardwares()
-    if count is None:
-        available = [n for n in sorted(registered) if f"{chip}_{n}" in known]
-        if not available:
-            raise ValueError(
-                f"{count_source} in step {step_label!r} has no CUDA count presets for {chip}",
-            )
-        count = max(available)
-    chosen = f"{chip}_{count}"
-    if chosen not in known:
-        raise ValueError(
-            f"{count_source} in step {step_label!r} has no preset {chosen!r}",
-        )
-    _log(f"{step_label}: {count_source} → {chosen!r}")
-    return chosen
-
-
-def _resolve_mirror_hardware_name(hardware: Any, *, step_label: str) -> str | None:
-    """Map ``mirror_hardwares`` to a ``ci_mirror_hardwares.yml`` preset, or None to skip.
-
-    The value is a preset name (``h100_2``). Unknown names fail later when the
-    registry is loaded. CUDA names are omitted when ``MIRROR_HW`` names a
-    different chip; NPU strings ignore ``MIRROR_HW``.
-    """
-    if isinstance(hardware, bool):
-        raise ValueError(f"mirror_hardwares must not be a boolean in step {step_label!r}")
-    if isinstance(hardware, str):
-        name = hardware.strip()
-    elif isinstance(hardware, int):
-        name = str(hardware)
-    else:
-        raise ValueError(
-            f"mirror_hardwares must be a preset string in step {step_label!r}",
-        )
-    if not name:
-        raise ValueError(f"mirror_hardwares must be a non-empty string in step {step_label!r}")
-
-    selector = _get_mirror_hw_selector()
-    token = name.lower()
-    chip = next(
-        (c for c in _cuda_mirror_chips() if token == c or token.startswith(f"{c}_")),
-        None,
-    )
-    if selector and chip is not None and chip != selector:
-        _log(
-            f"skip {step_label}: preset {name!r} is {chip} hardware; MIRROR_HW={selector!r}",
-        )
-        return None
-    return name
-
-
-def _expand_mirror_hardwares(step: dict[str, Any]) -> dict[str, Any] | None:
-    """Replace ``mirror_hardwares`` (or infer it from ``-m``) with preset agents/plugins.
-
-    Explicit preset string: used as-is; pytest marks are ignored.
-    Omitted key: match H100/L4 (or ``MIRROR_HW``) in ``-m``, then compose
-    ``{chip}_{n}`` from ``cards_n``. No match skips the step. Steps with no
-    SKU/cards and no ``mirror_hardwares`` are left unchanged (CPU jobs).
-
-    Returns None when the current ``MIRROR_HW`` selector does not apply.
-    """
-    step_label = _get_step_label(step)
-    has_pool = step.get("agents") is not None or step.get("plugins") is not None or step.get("image") is not None
-
-    if "mirror_hardwares" not in step:
-        if has_pool:
-            return step
-        chips, cards = _read_pytest_marks(step.get("commands"))
-        if not chips and cards is None:
-            return step
-        preset_name = _compose_mirror_hardware_name(chips, cards, step_label=step_label)
-    else:
-        if has_pool:
-            raise ValueError(
-                f"step {step_label!r} sets mirror_hardwares together with agents/plugins/image; "
-                f"use mirror_hardwares only",
-            )
-        preset_name = _resolve_mirror_hardware_name(
-            step.get("mirror_hardwares"),
-            step_label=step_label,
-        )
-
-    if preset_name is None:
-        return None
-
-    preset = _load_mirror_hardwares().get(preset_name)
     if preset is None:
         known = ", ".join(sorted(_load_mirror_hardwares()))
         raise ValueError(
-            f"unknown mirror_hardwares {preset_name!r} in step {step_label!r}; known: {known}",
+            f"unknown mirror_hardwares {hardware!r} in step {_get_step_label(step)!r}; known: {known}",
+        )
+
+    if step.get("agents") is not None or step.get("plugins") is not None or step.get("image") is not None:
+        raise ValueError(
+            f"step {_get_step_label(step)!r} sets mirror_hardwares together with agents/plugins/image; "
+            "use mirror_hardwares only",
         )
 
     expanded = copy.deepcopy(preset)
-    merged = {key: value for key, value in step.items() if key != "mirror_hardwares"} | expanded
+    return {key: value for key, value in step.items() if key != "mirror_hardwares"} | expanded
-    # Preset retry (K8S_RETRY on l4_*) must not clobber a step that opted out.
-    if "retry" in step:
-        merged["retry"] = step["retry"]
-    return merged
 
 
 def _match_source_file(changed_files: list[str], prefixes: list[str]) -> bool:
@@ -543,7 +309,7 @@ def _process_test_steps(
         nested = step.get("steps")
         if nested is not None:
             kept_nested = _process_test_steps(nested, changed_files)
-            if not kept_nested:
+            if changed_files is not None and not kept_nested:
                 _log(f"omit empty group {_get_step_label(step)!r}")
                 continue
             new_step = {key: value for key, value in step.items() if key != "source_file_dependencies"}
@@ -551,11 +317,14 @@ def _process_test_steps(
             processed.append(new_step)
             continue
 
-        leaf = {key: value for key, value in step.items() if key != "source_file_dependencies"}
+        if deps is not None:
-        expanded = _expand_mirror_hardwares(leaf)
+            processed.append(
-        if expanded is None:
+                _expand_mirror_hardwares(
-            continue
+                    {key: value for key, value in step.items() if key != "source_file_dependencies"},
-        processed.append(expanded)
+                ),
+            )
+        else:
+            processed.append(_expand_mirror_hardwares(step))
 
     return processed
 
@@ -581,10 +350,6 @@ def _render_test_pipeline(
     e2e_only: bool = False,
 ) -> dict[str, Any]:
     """Filter steps by PR diff and strip uploader-only ``source_file_dependencies`` metadata."""
-    # Validate MIRROR_HW once up front. Per-step skip must not run for typos
-    # (e.g. b20o), or a pipeline can shrink to leftover CPU-only steps and
-    # still upload successfully.
-    _get_mirror_hw_selector()
     steps = doc.get("steps")
     if not isinstance(steps, list):
         return doc
@@ -614,7 +379,10 @@ def _render_pipeline(
 
     text = path.read_text(encoding="utf-8")
     ctx = resolve_ci_context_from_git()
-    changed_files = None if force_all or e2e_only else ctx.changed_files
+    if force_all or e2e_only:
+        changed_files = None
+    else:
+        changed_files = ctx.changed_files
 
     doc = yaml.safe_load(text)
     if not isinstance(doc, dict):

.buildkite/cuda/bootstrap-upload-steps.yml

@@ -8,28 +8,28 @@ steps:
     agents:
       queue: "cpu_queue_premerge"
 
-  # L2 Test — PR ready label (diff-aware) or main+WEEKLY=1 (--e2e, full E2E group)
+  # L2 Test — PR ready label (diff-aware) or main+NIGHTLY=1 (--e2e, full E2E group)
   - label: "Upload Ready Pipeline"
     depends_on: image-build
     key: upload-ready-pipeline
     commands:
       - |
         EXTRA=""
-        if [ "${WEEKLY:-}" = "1" ] && [ "${BUILDKITE_BRANCH}" = "main" ]; then
+        if [ "${NIGHTLY:-}" = "1" ] && [ "${BUILDKITE_BRANCH}" = "main" ]; then
           EXTRA="--e2e"
         fi
         python3 .buildkite/common/scripts/upload_pipeline.py --upload $$EXTRA .buildkite/cuda/test-ready.yml
     agents:
       queue: "cpu_queue_premerge"
 
-  # L3 Test — PR merge-test label (diff-aware) or main+WEEKLY=1 (--e2e, full E2E group)
+  # L3 Test — PR merge-test label (diff-aware) or main+NIGHTLY=1 (--e2e, full E2E group)
   - label: "Upload Merge Pipeline"
     depends_on: image-build
     key: upload-merge-pipeline
     commands:
       - |
         EXTRA=""
-        if [ "${WEEKLY:-}" = "1" ] && [ "${BUILDKITE_BRANCH}" = "main" ]; then
+        if [ "${NIGHTLY:-}" = "1" ] && [ "${BUILDKITE_BRANCH}" = "main" ]; then
           EXTRA="--e2e"
         fi
         python3 .buildkite/common/scripts/upload_pipeline.py --upload $$EXTRA .buildkite/cuda/test-merge.yml
@@ -45,7 +45,7 @@ steps:
     agents:
       queue: "cpu_queue_premerge"
 
-  # L5 Test — main+WEEKLY=1 / NON_CRITICAL=1 (scheduled), or PR with weekly-test label
+  # L5 Test — main+WEEKLY=1 (scheduled), or PR with weekly-test label
   - label: "Upload Weekly Pipeline"
     depends_on: image-build
     key: upload-weekly-pipeline

.buildkite/cuda/test-merge.yml

@@ -1,6 +1,3 @@
-# CUDA CI L3 (`advanced_model` e2e) plus L1 CPU Simple Test (`core_model and cpu`).
-# Uploaded by Upload Merge Pipeline: PR label `merge-test`, or post-merge on main.
-
 env:
   VLLM_WORKER_MULTIPROC_METHOD: spawn
   VLLM_USE_FLASHINFER_MOE_FP16: 0
@@ -12,27 +9,23 @@ steps:
     depends_on: upload-merge-pipeline
     steps:
       - label: "Simple · Model Executor Test"
-        timeout_in_minutes: 20
         commands:
-          - "pytest -sv tests/model_executor -m 'core_model and cpu'"
+          - "timeout 20m pytest -sv tests/model_executor -m 'core_model and cpu' --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
         mirror_hardwares: l4_1
 
       - label: "Simple · Diffusion Test"
-        timeout_in_minutes: 20
         commands:
-          - "pytest -sv tests/diffusion -m 'core_model and cpu'"
+          - "timeout 20m pytest -sv tests/diffusion -m 'core_model and cpu' --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
         mirror_hardwares: l4_1
 
       - label: "Simple · Engine&Entrypoints Test"
-        timeout_in_minutes: 20
         commands:
-          - "pytest -sv tests/entrypoints tests/engine -m 'core_model and cpu'"
+          - "timeout 20m pytest -sv tests/entrypoints tests/engine -m 'core_model and cpu' --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
         mirror_hardwares: l4_1
 
       - label: "Simple · Other Test"
-        timeout_in_minutes: 20
         commands:
-          - "pytest -sv tests/ -m 'core_model and cpu' --ignore=tests/diffusion --ignore=tests/model_executor --ignore=tests/entrypoints --ignore=tests/engine"
+          - "timeout 20m pytest -sv tests/ -m 'core_model and cpu' --ignore=tests/diffusion --ignore=tests/model_executor --ignore=tests/entrypoints --ignore=tests/engine --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
         mirror_hardwares: l4_1
 
   - group: ":card_index_dividers: Diffusion Test"
@@ -83,14 +76,8 @@ steps:
           - tests/e2e/online_serving/test_qwen3_tts_customvoice.py
           - tests/e2e/offline_inference/test_qwen3_tts_customvoice.py
           - vllm_omni/model_executor/models/qwen3_tts/
-          - vllm_omni/model_executor/models/common/qwen3_code_predictor.py
-          - vllm_omni/model_executor/models/common/snake_activation.py
-          - vllm_omni/model_executor/models/common/alias_free_activation.py
           - vllm_omni/model_executor/stage_input_processors/qwen3_tts.py
           - vllm_omni/deploy/qwen3_tts.yaml
-          - vllm_omni/entrypoints/openai/serving_speech.py
-          - vllm_omni/entrypoints/openai/tts_adapters/
-          - vllm_omni/engine/stage_pool.py
         commands:
           - |
             timeout 40m bash -c "
@@ -105,58 +92,49 @@ steps:
           - tests/e2e/online_serving/test_qwen3_tts_base.py
           - tests/e2e/offline_inference/test_qwen3_tts_base.py
           - vllm_omni/model_executor/models/qwen3_tts/
-          - vllm_omni/model_executor/models/common/qwen3_code_predictor.py
-          - vllm_omni/model_executor/models/common/snake_activation.py
-          - vllm_omni/model_executor/models/common/alias_free_activation.py
           - vllm_omni/model_executor/stage_input_processors/qwen3_tts.py
           - vllm_omni/deploy/qwen3_tts.yaml
-          - vllm_omni/entrypoints/openai/serving_speech.py
-          - vllm_omni/entrypoints/openai/tts_adapters/
-          - vllm_omni/engine/stage_pool.py
-          - .buildkite/common/scripts/run_cov_split.sh
-          - pyproject.toml
-        # 40 -> 60: the coverage split runs the model twice.
-        timeout_in_minutes: 60
         commands:
           - |
-            export VLLM_LOGGING_LEVEL=DEBUG
+            timeout 40m bash -c "
-            export VLLM_ALLOW_LONG_MAX_MODEL_LEN=1
+              export VLLM_LOGGING_LEVEL=DEBUG
-            .buildkite/common/scripts/run_cov_split.sh \
+              export VLLM_ALLOW_LONG_MAX_MODEL_LEN=1
-              --model-id qwen3_tts \
+              pytest -s -v tests/e2e/online_serving/test_qwen3_tts_base.py tests/e2e/offline_inference/test_qwen3_tts_base.py -m 'advanced_model and cuda' --run-level 'advanced_model'
-              --offline tests/e2e/offline_inference/test_qwen3_tts_base.py \
+            "
-              --online tests/e2e/online_serving/test_qwen3_tts_base.py \
+        mirror_hardwares: l4_1
-              -- -m 'advanced_model and cuda' --run-level 'advanced_model'
+
+      - label: "TTS · Higgs-Audio-V3 Test"
+        source_file_dependencies:
+          - tests/e2e/online_serving/test_higgs_audio_v3.py
+          - vllm_omni/model_executor/models/higgs_audio_v3/
+          - vllm_omni/model_executor/stage_input_processors/higgs_audio_v3.py
+          - vllm_omni/deploy/higgs_multimodal_qwen3.yaml
+        commands:
+          - |
+            timeout 20m bash -c "
+              export VLLM_WORKER_MULTIPROC_METHOD=spawn
+              export VLLM_USE_DEEP_GEMM=0
+              export VLLM_MOE_USE_DEEP_GEMM=0
+              pytest -s -v tests/e2e/online_serving/test_higgs_audio_v3.py -m 'advanced_model' --run-level 'advanced_model'
+            "
         mirror_hardwares: l4_1
 
       - label: "Omni · Qwen3-Omni Test"
         source_file_dependencies:
           - tests/e2e/offline_inference/test_qwen3_omni.py
-          - tests/e2e/offline_inference/test_qwen3_omni_colocate_async.py
           - tests/e2e/online_serving/test_qwen3_omni.py
-          - tests/helpers/runtime.py
-          - vllm_omni/entrypoints/openai/serving_chat.py
           - vllm_omni/model_executor/models/qwen3_omni/
-          - vllm_omni/model_executor/models/common/qwen3_code_predictor.py
-          - vllm_omni/model_executor/models/common/snake_activation.py
-          - vllm_omni/model_executor/models/qwen2_5_omni/qwen2_5_omni_thinker.py
-          - vllm_omni/model_executor/models/qwen3_tts/cuda_graph_decoder_wrapper.py
           - vllm_omni/model_executor/stage_input_processors/qwen3_omni.py
           - vllm_omni/deploy/qwen3_omni_moe.yaml
-          - vllm_omni/deploy/qwen3_omni_moe_thinking.yaml
         timeout_in_minutes: 50
         commands:
-          - pytest -s -v tests/e2e/offline_inference/test_qwen3_omni.py tests/e2e/online_serving/test_qwen3_omni.py tests/e2e/offline_inference/test_qwen3_omni_colocate_async.py -m 'advanced_model and cuda' --run-level 'advanced_model'
+          - pytest -s -v tests/e2e/offline_inference/test_qwen3_omni.py tests/e2e/online_serving/test_qwen3_omni.py -m 'advanced_model and cuda' --run-level 'advanced_model'
         mirror_hardwares: h100_2
 
       - label: "Omni · MiniCPM-o 4.5 Test"
         source_file_dependencies:
           - tests/e2e/offline_inference/test_minicpmo_4_5.py
           - tests/e2e/online_serving/test_minicpmo_4_5.py
-          - vllm_omni/model_executor/models/minicpmo_4_5/
-          - vllm_omni/model_executor/models/step_audio2/
-          - vllm_omni/model_executor/models/cosyvoice3/code2wav_core/
-          - vllm_omni/model_executor/models/cosyvoice3/utils.py
-          - vllm_omni/model_executor/stage_input_processors/minicpmo_4_5_omni.py
           - vllm_omni/deploy/minicpmo_4_5.yaml
         timeout_in_minutes: 50
         commands:
@@ -177,36 +155,63 @@ steps:
           - tests/e2e/online_serving/test_minicpmo_realtime_duplex_drivers.py
           - tests/e2e/online_serving/helpers/minicpmo_realtime_duplex_scenarios.py
           - tests/e2e/online_serving/run_minicpmo_realtime_duplex_multi_session.py
-          - tests/e2e/online_serving/run_minicpmo_realtime_duplex_server_vad.py
           - tests/e2e/online_serving/run_minicpmo_realtime_duplex_soft_interrupt.py
-          - vllm_omni/deploy/minicpmo_4_5.yaml
           - tests/e2e/online_serving/helpers/minicpmo_4_5_duplex.py
+          - vllm_omni/deploy/minicpmo_4_5_duplex.yaml
           - vllm_omni/experimental/fullduplex/
           - vllm_omni/model_executor/models/minicpmo_4_5/
-          - vllm_omni/model_executor/models/step_audio2/
-          - vllm_omni/model_executor/models/cosyvoice3/code2wav_core/
-          - vllm_omni/model_executor/models/cosyvoice3/utils.py
           - vllm_omni/model_executor/stage_input_processors/minicpmo_4_5_omni.py
           - vllm_omni/core/sched/
           - vllm_omni/outputs/multimodal_accumulation.py
           - vllm_omni/engine/stage_init_utils.py
           - vllm_omni/engine/async_omni_engine.py
-          - vllm_omni/engine/orchestrator.py
           - vllm_omni/entrypoints/cli/serve.py
-          - vllm_omni/entrypoints/openai/api_server.py
-          - vllm_omni/entrypoints/openai/duplex_capability.py
-          - vllm_omni/config/stage_config.py
         timeout_in_minutes: 50
         commands:
           - pytest -s -v tests/e2e/online_serving/test_minicpmo_4_5_duplex.py -m 'advanced_model and cuda' --run-level 'advanced_model'
         mirror_hardwares: h100_1
 
+      - label: "TTS · VoxCPM2 Test"
+        source_file_dependencies:
+          - tests/e2e/offline_inference/test_voxcpm2_tts.py
+          - tests/e2e/online_serving/test_voxcpm2_tts.py
+          - vllm_omni/model_executor/models/voxcpm2/
+          - vllm_omni/deploy/voxcpm2.yaml
+        timeout_in_minutes: 30
+        commands:
+          - |
+            timeout 30m bash -c "
+              export VLLM_LOGGING_LEVEL=DEBUG
+              export VLLM_WORKER_MULTIPROC_METHOD=spawn
+              pytest -s -v tests/e2e/offline_inference/test_voxcpm2_tts.py tests/e2e/online_serving/test_voxcpm2_tts.py -m 'advanced_model and cuda' --run-level 'advanced_model'
+            "
+        mirror_hardwares: l4_1
+
+      - label: "Diffusion · Qwen Image Edit Test"
+        source_file_dependencies:
+          - tests/e2e/online_serving/test_qwen_image_edit.py
+          - vllm_omni/diffusion/models/qwen_image/
+        timeout_in_minutes: 20
+        commands:
+          - pytest -s -v tests/e2e/online_serving/test_qwen_image_edit.py -m 'advanced_model and cuda' --run-level 'advanced_model'
+        mirror_hardwares: h100_1
+      - label: "Diffusion · Bagel Test"
+        source_file_dependencies:
+          - tests/e2e/offline_inference/test_bagel.py
+          - tests/e2e/online_serving/test_bagel.py
+          - vllm_omni/model_executor/models/bagel/
+          - vllm_omni/diffusion/models/bagel/
+          - vllm_omni/model_executor/stage_input_processors/bagel.py
+          - vllm_omni/deploy/bagel.yaml
+        timeout_in_minutes: 20
+        commands:
+          - pytest -s -v tests/e2e/offline_inference/test_bagel.py tests/e2e/online_serving/test_bagel.py -m 'advanced_model and cuda' --run-level 'advanced_model'
+        mirror_hardwares: h100_1
+
       - label: "Diffusion · Qwen Image Test"
         source_file_dependencies:
           - tests/e2e/online_serving/test_qwen_image.py
           - vllm_omni/diffusion/models/qwen_image/
-          - vllm_omni/diffusion/models/dmd2/
-          - vllm_omni/diffusion/distributed/autoencoders/autoencoder_kl_qwenimage.py
         commands:
           - |
             timeout 40m bash -c '
@@ -215,12 +220,23 @@ steps:
             '
         mirror_hardwares: h100_1
 
+      - label: "Diffusion · Qwen Image Layered Test"
+        source_file_dependencies:
+          - tests/e2e/online_serving/test_qwen_image_layered.py
+          - vllm_omni/diffusion/models/qwen_image/
+        commands:
+          - |
+            timeout 40m bash -c "
+              export VLLM_IMAGE_FETCH_TIMEOUT=60
+              pytest -s -v tests/e2e/online_serving/test_qwen_image_layered.py -m 'advanced_model and cuda' --run-level 'advanced_model'
+            "
+        mirror_hardwares: h100_1
+
       - label: "Diffusion · Wan22 Test"
         source_file_dependencies:
           - tests/e2e/offline_inference/test_wan22_t2v.py
           - tests/e2e/online_serving/test_wan22_t2v.py
           - vllm_omni/diffusion/models/wan2_2/
-          - vllm_omni/diffusion/models/dmd2/
         commands:
           - |
             timeout 40m bash -c "
@@ -229,49 +245,20 @@ steps:
             "
         mirror_hardwares: h100_1
 
-      - label: "Diffusion · HunyuanVideo-1.5 Test"
+      - label: "Diffusion · Z Image Test"
         source_file_dependencies:
-          - tests/e2e/online_serving/test_hunyuan_video_15_expansion.py
+          - tests/e2e/offline_inference/test_z_image.py
-          - vllm_omni/diffusion/models/hunyuan_video/
+          - tests/e2e/online_serving/test_z_image.py
+          - vllm_omni/diffusion/models/z_image/
+        timeout_in_minutes: 30
         commands:
-          - |
+          - pytest -s -v tests/e2e/offline_inference/test_z_image.py tests/e2e/online_serving/test_z_image.py -m 'advanced_model and cuda and diffusion' --run-level 'advanced_model'
-            timeout 40m bash -c "
+        mirror_hardwares: l4_1
-              export VLLM_IMAGE_FETCH_TIMEOUT=60
+      - label: "Diffusion · FLUX.2 Klein Test"
-              pytest -s -v tests/e2e/online_serving/test_hunyuan_video_15_expansion.py -m 'advanced_model and cuda and diffusion' --run-level 'advanced_model'
-            "
-        mirror_hardwares: h100_1
-
-      - label: "Diffusion · MiniMax-H3 L3 DLO DP2 and FastH3"
         source_file_dependencies:
-          - tests/e2e/online_serving/minimax_h3/
+          - tests/e2e/offline_inference/test_flux2_klein.py
-          - tests/helpers/assertions.py
+          - vllm_omni/diffusion/models/flux2_klein/
-          - tests/helpers/mark.py
+        timeout_in_minutes: 30
-          - tests/helpers/media.py
-          - tests/helpers/runtime.py
-          - vllm_omni/diffusion/models/minimax_h3/
-          - vllm_omni/diffusion/lora/
-          - vllm_omni/diffusion/offloader/
-          - vllm_omni/diffusion/distributed/
-          - vllm_omni/entrypoints/openai/
-        # Keep four server lifecycles in one Buildkite job, but never in one
-        # pytest process. DLO uses multiprocessing shared-memory queues; the
-        # parent's resource_tracker can retain a stale /psm_* entry during
-        # teardown and strand the next Ref2VA collective.
-        # A per-file process plus a hard timeout also makes a regression fail
-        # instead of leaving the merge queue stuck indefinitely.
-        timeout_in_minutes: 150
         commands:
-          - |
+          - pytest -s -v tests/e2e/offline_inference/test_flux2_klein.py -m 'advanced_model and cuda and diffusion' --run-level 'advanced_model'
-            set -eu
+        mirror_hardwares: l4_1
-            export VLLM_IMAGE_FETCH_TIMEOUT=60
-            if [ -f vllm_omni/diffusion/models/minimax_h3/lora.py ]; then
-              export VLLM_TEST_MINIMAX_H3_TURBO_LORA=$(python3 -c 'from huggingface_hub import hf_hub_download; print(hf_hub_download(repo_id="lightx2v/Minimax-h3-Turbo", filename="minimax_h3_fl2v_turbo_4step_v1.0_768p_bf16.safetensors"))')
-            fi
-            if [ -f vllm_omni/diffusion/models/minimax_h3/fasth3.py ]; then
-              export VLLM_TEST_MINIMAX_H3_FASTH3_LORA=$(python3 -c 'from huggingface_hub import hf_hub_download; print(hf_hub_download(repo_id="FastVideo/FastVideo-FastH3-4-step-Preview-v1-LoRA", filename="dense-datafree/adapter_model.safetensors"))')
-            fi
-            timeout 35m pytest -s -v tests/e2e/online_serving/minimax_h3/test_minimax_h3_dlo_dp2_fl2va.py -m 'advanced_model and cuda and diffusion' --run-level 'advanced_model'
-            timeout 35m pytest -s -v tests/e2e/online_serving/minimax_h3/test_minimax_h3_dlo_dp2_ref2va.py -m 'advanced_model and cuda and diffusion' --run-level 'advanced_model'
-            timeout 35m pytest -s -v tests/e2e/online_serving/minimax_h3/test_minimax_h3_turbo_lora.py -m 'advanced_model and cuda and diffusion' --run-level 'advanced_model'
-            timeout 35m pytest -s -v tests/e2e/online_serving/minimax_h3/test_minimax_h3_fasth3_lora.py -m 'advanced_model and cuda and diffusion' --run-level 'advanced_model'
-        mirror_hardwares: h100_2

.buildkite/cuda/test-nightly.yml

@@ -1,7 +1,3 @@
-# CUDA CI L4 (`full_model`): nightly and release share this file.
-# Uploaded by Upload Nightly Pipeline: NIGHTLY=1, or PR labels (nightly-test / omni-test / ...).
-# Hardware: omit mirror_hardwares to compose from -m SKU + cards_n; or set a preset string to force.
-
 env:
   VLLM_WORKER_MULTIPROC_METHOD: spawn
   VLLM_USE_FLASHINFER_MOE_FP16: 0
@@ -16,33 +12,29 @@ steps:
     if: build.env("NIGHTLY") == "1" || build.pull_request.labels includes "nightly-test" || build.pull_request.labels includes "omni-test"
     steps:
 
-      - label: ":full_moon: Omni · Function Test with H100 · Single-GPU"
+      - label: ":full_moon: Omni · Function Test with H100"
         timeout_in_minutes: 120
         commands:
-          - pytest -sv tests/e2e --run-level "full_model" -m "full_model and H100 and B200 and omni and cards_1" --ignore=tests/e2e/accuracy --ignore=tests/e2e/online_serving/test_minicpmo_4_5_duplex_expansion.py
+          - pytest -sv tests/e2e --run-level "full_model" -m "full_model and H100 and omni" --ignore=tests/e2e/accuracy --ignore=tests/e2e/online_serving/test_qwen3_omni_multi_replicas.py --ignore=tests/e2e/online_serving/test_minicpmo_4_5_duplex_expansion.py
-
+        mirror_hardwares: h100_2
-      - label: ":full_moon: Omni · Function Test with H100 · 2-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - pytest -sv tests/e2e --run-level "full_model" -m "full_model and H100 and B200 and omni and cards_2" --ignore=tests/e2e/accuracy --ignore=tests/e2e/online_serving/test_qwen3_omni_multi_replicas.py
 
       - label: ":full_moon: Omni · MiniCPM-o 4.5 Duplex Test"
         timeout_in_minutes: 50
         commands:
-          - pytest -s -v tests/e2e/online_serving/test_minicpmo_4_5_duplex_expansion.py -m "full_model and cuda and H100 and omni and cards_1" --run-level "full_model"
+          - pytest -s -v tests/e2e/online_serving/test_minicpmo_4_5_duplex_expansion.py -m "full_model and cuda and H100 and omni" --run-level "full_model"
         mirror_hardwares: h100_1
 
-      - label: ":full_moon: Omni · Doc Test with L4 · 4-GPU"
+      - label: ":full_moon: Omni · Doc Test with L4"
         timeout_in_minutes: 90
         commands:
           - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
-          - pytest -s -v tests/examples/ -m "full_model and omni and L4 and cards_4" --run-level "full_model"
+          - pytest -s -v tests/examples/ -m "full_model and omni and L4" --run-level "full_model"
         mirror_hardwares: l4_4
 
-      - label: ":full_moon: Omni · Doc Test with H100 · 2-GPU"
+      - label: ":full_moon: Omni · Doc Test with H100"
         timeout_in_minutes: 90
         commands:
-          - pytest -s -v tests/examples/ -m "full_model and omni and H100 and cards_2" --run-level "full_model"
+          - pytest -s -v tests/examples/ -m "full_model and omni and H100" --run-level "full_model"
         mirror_hardwares: h100_2
 
       - label: ":full_moon: Omni · Accuracy Test"
@@ -91,12 +83,38 @@ steps:
           - export BENCHMARK_DIR=tests/dfx/perf/results
           - |
             set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen3_omni_async_chunk.json -m "H100 and full_model"
+            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen3_omni_async_chunk.json
             EXIT=$$?
             buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
             exit $$EXIT
         mirror_hardwares: h100_2
 
+      - label: ":full_moon: Omni · Perf Test · vLLM Text"
+        key: nightly-omni-performance-vllm-text
+        timeout_in_minutes: 180
+        commands:
+          - export BENCHMARK_DIR=tests/dfx/perf/results
+          - |
+            set +e
+            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen3_omni_vllm_text.json
+            EXIT=$$?
+            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
+            exit $$EXIT
+        mirror_hardwares: h100_2
+
+      - label: ":full_moon: Omni · Perf Test · Multi-Replica"
+        key: nightly-omni-performance-multi-replicas
+        timeout_in_minutes: 180
+        commands:
+          - export BENCHMARK_DIR=tests/dfx/perf/results
+          - |
+            set +e
+            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen3_omni_multi_replicas.json
+            EXIT=$$?
+            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
+            exit $$EXIT
+        mirror_hardwares: h100_3
+
       - label: ":full_moon: Omni · MiniCPM-o 4.5 · Perf Test"
         key: nightly-omni-performance-minicpmo-4-5
         timeout_in_minutes: 180
@@ -141,36 +159,63 @@ steps:
           - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
           - export VLLM_USE_DEEP_GEMM="0"
           - export VLLM_MOE_USE_DEEP_GEMM="0"
-          - pytest -s -v tests/e2e/ -m "full_model and L4 and tts" --run-level "full_model" --ignore=tests/e2e/accuracy
+          - pytest -s -v tests/e2e/ -m "full_model and L4 and tts" --run-level "full_model" --ignore=tests/e2e/accuracy --ignore=tests/e2e/online_serving/test_voxcpm2_tts_expansion.py
         mirror_hardwares: l4_1
 
-      - label: ":full_moon: TTS · Perf Test · Single-GPU"
+      - label: ":full_moon: TTS · Accuracy Test · VoxCPM2"
-        key: nightly-tts-performance-single
+        timeout_in_minutes: 30
+        commands:
+          - |
+            timeout 30m bash -c "
+              export VLLM_LOGGING_LEVEL=DEBUG
+              export VLLM_WORKER_MULTIPROC_METHOD=spawn
+              export VLLM_ALLOW_LONG_MAX_MODEL_LEN=1
+              pytest -s -v tests/e2e/online_serving/test_voxcpm2_tts_expansion.py -m 'full_model and cuda and tts' --run-level 'full_model'
+            "
+        mirror_hardwares: l4_1
+
+      - label: ":full_moon: TTS · Perf Test"
+        key: nightly-tts-performance
         timeout_in_minutes: 180
         commands:
           - export BENCHMARK_DIR=tests/dfx/perf/results
           - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
           - |
             set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_tts.json -m "H100 and cards_1"
+            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_tts.json
+            EXIT=$$?
+            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
+            exit $$EXIT
+        mirror_hardwares: h100_2
+
+      - label: ":full_moon: TTS · VoxCPM2 · Perf Test"
+        key: nightly-tts-performance-voxcpm2
+        timeout_in_minutes: 180
+        commands:
+          - export BENCHMARK_DIR=tests/dfx/perf/results
+          - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
+          - |
+            set +e
+            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_voxcpm2.json
             EXIT=$$?
             buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
             exit $$EXIT
         mirror_hardwares: h100_1
 
-      - label: ":full_moon: TTS · Perf Test · 2-GPU"
+      - label: ":full_moon: TTS · Higgs-Audio-V3 · Perf Test"
-        key: nightly-tts-performance-2gpu
+        key: nightly-tts-performance-higgs-audio-v3
         timeout_in_minutes: 180
         commands:
           - export BENCHMARK_DIR=tests/dfx/perf/results
-          - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
+          - export VLLM_USE_DEEP_GEMM="0"
+          - export VLLM_MOE_USE_DEEP_GEMM="0"
           - |
             set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_tts.json -m "H100 and cards_2"
+            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_higgs_audio_v3.json
             EXIT=$$?
             buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
             exit $$EXIT
-        mirror_hardwares: h100_2
+        mirror_hardwares: h100_1
 
   - group: ":card_index_dividers: Tiny Model Tests [Multi-GPU]"
     key: nightly-tiny-model-test-group
@@ -180,7 +225,7 @@ steps:
       build.pull_request.labels includes "nightly-test"
     steps:
 
-      - label: ":full_moon: Tiny Model Tests · Diffusion (Multi-GPU accelerations) · 2-GPU"
+      - label: ":full_moon: Tiny Model Tests · Diffusion (Multi-GPU accelerations)"
         timeout_in_minutes: 30
         # NOTE: Single-GPU (core_model) tiny model tests run on the ready label,
         # so nightly only adds the multi-GPU parallelism configurations.
@@ -190,14 +235,7 @@ steps:
         # for the CI, since these tests should pass irrespective of whether or not we use
         # the full weights or not.
         commands:
-          - pytest -sv tests/model_tests/diffusion -m 'full_model and cuda and cards_2' --run-level "core_model"
+          - timeout 25m pytest -sv tests/model_tests/diffusion -m 'full_model and cuda' --run-level "core_model"
-        mirror_hardwares: l4_2
-
-      - label: ":full_moon: Tiny Model Tests · Diffusion (Multi-GPU accelerations) · 4-GPU"
-        timeout_in_minutes: 30
-        # CFG+TP extra_test_groups request 2×2=4 devices (see get_required_device_count).
-        commands:
-          - pytest -sv tests/model_tests/diffusion -m 'full_model and cuda and cards_4' --run-level "core_model"
         mirror_hardwares: l4_4
 
   - group: ":card_index_dividers: Diffusion X2I(&A&T) Model Test"
@@ -208,6 +246,8 @@ steps:
       build.pull_request.labels includes "nightly-test" ||
       build.pull_request.labels includes "diffusion-x2iat-test"
     steps:
+      # Use explicit file shards so single-card cases do not reserve 2+ GPUs,
+      # and Bagel feature coverage is not duplicated via a broad tests/e2e sweep.
 
       - label: ":full_moon: Diffusion X2I(&A&T) · Function Test with H100 · Single-GPU"
         timeout_in_minutes: 120
@@ -215,35 +255,66 @@ steps:
           - >-
             pytest -sv
             tests/e2e/online_serving/test_qwen_image_expansion.py
-            -m "full_model and diffusion and H100 and cards_1"
+            tests/e2e/online_serving/test_qwen_image_edit_expansion.py
+            tests/e2e/online_serving/test_qwen_image_layered_expansion.py
+            tests/e2e/online_serving/test_boogu_image.py
+            tests/e2e/online_serving/test_boogu_image_edit.py
+            tests/e2e/online_serving/test_boogu_image_expansion.py
+            tests/e2e/online_serving/test_boogu_image_edit_expansion.py
+            tests/e2e/online_serving/test_bagel_expansion.py
+            tests/e2e/online_serving/test_krea2_expansion.py
+            tests/e2e/offline_inference/test_krea2_expansion.py
+            -m "full_model and diffusion and H100 and not distributed_cuda"
             --run-level "full_model"
         mirror_hardwares: h100_1
 
-      - label: ":full_moon: Diffusion X2I(&A&T) · Function Test with H100 · 2-GPU"
+      - label: ":full_moon: Diffusion X2I(&A&T) · Function Test with H100 · Multi-GPU Qwen-Image"
         timeout_in_minutes: 120
         commands:
           - >-
             pytest -sv
             tests/e2e/online_serving/test_qwen_image_expansion.py
-            -m "full_model and diffusion and H100 and cards_2"
+            tests/e2e/online_serving/test_qwen_image_edit_expansion.py
+            tests/e2e/offline_inference/test_krea2_expansion.py
+            -m "full_model and diffusion and H100 and distributed_cuda"
             --run-level "full_model"
         mirror_hardwares: h100_2
 
-      - label: ":full_moon: Diffusion X2I(&A&T) · Boogu Edit CFG · 3-GPU"
+      - label: ":full_moon: Diffusion X2I(&A&T) · Function Test with H100 · Multi-GPU Layered/LongCat/FLUX"
+        timeout_in_minutes: 90
+        commands:
+          - >-
+            pytest -sv
+            tests/e2e/online_serving/test_qwen_image_layered_expansion.py
+            -m "full_model and diffusion and H100 and distributed_cuda"
+            --run-level "full_model"
+        mirror_hardwares: h100_2
+
+      - label: ":full_moon: Diffusion X2I(&A&T) · Function Test with H100 · Multi-GPU Bagel"
         timeout_in_minutes: 120
         commands:
           - >-
             pytest -sv
-            tests/e2e/online_serving/test_boogu_image_edit_expansion.py
+            tests/e2e/online_serving/test_bagel_expansion.py
-            -k "test_double_guidance_cfg_parallel and cfg_parallel_3_full"
+            -m "full_model and diffusion and H100 and distributed_cuda"
-            -m "slow and diffusion and cuda and H100 and cards_3"
             --run-level "full_model"
-        mirror_hardwares: h100_3
+        mirror_hardwares: h100_2
 
-      - label: ":full_moon: Diffusion X2I(&A&T) · Function Test with L4 · 4-GPU"
+      - label: ":full_moon: Diffusion X2I(&A&T) · GLM-Image Function Test with H100"
         timeout_in_minutes: 120
         commands:
-          - pytest -sv tests/e2e/ -k "not test_wan and not hunyuan" -m "full_model and diffusion and L4 and cards_4" --run-level "full_model" --ignore=tests/e2e/accuracy
+          - >-
+            pytest -sv
+            tests/e2e/online_serving/test_glm_image_expansion.py
+            tests/e2e/offline_inference/test_glm_image_autoround_w4a16_expansion.py
+            -m "full_model and diffusion and H100"
+            --run-level "full_model"
+        mirror_hardwares: h100_2
+
+      - label: ":full_moon: Diffusion X2I(&A&T) · Function Test with L4"
+        timeout_in_minutes: 120
+        commands:
+          - pytest -sv tests/e2e/ -k "not test_wan and not test_bagel_expansion and not hunyuan" -m "full_model and diffusion and L4" --run-level "full_model" --ignore=tests/e2e/accuracy
         mirror_hardwares: l4_4
 
       - label: ":full_moon: Diffusion X2I(&A&T) · Doc Test"
@@ -256,7 +327,7 @@ steps:
         timeout_in_minutes: 180
         commands:
           - export VLLM_HTTP_TIMEOUT_KEEP_ALIVE=120
-          - pytest -s -v tests/e2e/accuracy/test_qwen_image.py --run-level full_model
+          - pytest -s -v tests/e2e/accuracy/test_qwen_image*.py --run-level full_model
           - pytest -s -v tests/e2e/accuracy/test_diffusers_backend_similarity.py -k '2i_matches_diffusers' -m full_model --run-level full_model
         mirror_hardwares: h100_1
 
@@ -268,42 +339,85 @@ steps:
           - pytest -s -v tests/e2e/accuracy/test_hunyuan_image3_pixel_accuracy.py -m full_model --run-level full_model
         mirror_hardwares: h100_4
 
-      - label: ":full_moon: HiDream-O1-Image · Accuracy Test"
+      - label: ":full_moon: Diffusion X2I(&A&T) · Perf Test · Qwen-Image"
-        timeout_in_minutes: 180
+        key: nightly-diffusion-x2iat-performance-qwen-image
-        commands:
-          - pytest -s -v tests/e2e/accuracy/test_hidream_o1_image.py -m full_model --run-level full_model
-        mirror_hardwares: h100_2
-
-      - label: ":full_moon: Diffusion X2I(&A&T) · Perf Test · Qwen-Image · Single-GPU"
-        key: nightly-diffusion-x2iat-performance-qwen-image-single
         timeout_in_minutes: 180
         commands:
           - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
-          - export CACHE_DIT_VERSION=1.5.0
+          - export DIFFUSION_ATTENTION_BACKEND=FLASH_ATTN
+          - export CACHE_DIT_VERSION=1.3.0
           - |
             set +e
-            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen_image_vllm_omni.json -m "H100 and cards_1"
+            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen_image_vllm_omni.json
+            EXIT=$$?
+            buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
+            buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
+            exit $$EXIT
+        mirror_hardwares: h100_4
+
+      - label: ":full_moon: Diffusion X2I(&A&T) · Perf Test · Qwen-Image-Edit"
+        key: nightly-diffusion-x2iat-performance-qwen-image-edit
+        timeout_in_minutes: 180
+        commands:
+          - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
+          - export DIFFUSION_ATTENTION_BACKEND=FLASH_ATTN
+          - export CACHE_DIT_VERSION=1.3.0
+          - |
+            set +e
+            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen_image_edit_vllm_omni.json
+            EXIT=$$?
+            buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
+            buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
+            exit $$EXIT
+        mirror_hardwares: h100_4
+
+      - label: ":full_moon: Diffusion X2I(&A&T) · Perf Test · Qwen-Image-Edit-2511"
+        key: nightly-diffusion-x2iat-performance-qwen-image-edit-2511
+        timeout_in_minutes: 180
+        commands:
+          - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
+          - export DIFFUSION_ATTENTION_BACKEND=FLASH_ATTN
+          - export CACHE_DIT_VERSION=1.3.0
+          - |
+            set +e
+            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen_image_edit_2511_vllm_omni.json
+            EXIT=$$?
+            buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
+            buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
+            exit $$EXIT
+        mirror_hardwares: h100_4
+
+      - label: ":full_moon: Diffusion X2I(&A&T) · Perf Test · Qwen-Image-Layered"
+        key: nightly-diffusion-x2iat-performance-qwen-image-layered
+        timeout_in_minutes: 120
+        commands:
+          - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
+          - export DIFFUSION_ATTENTION_BACKEND=FLASH_ATTN
+          - export CACHE_DIT_VERSION=1.3.0
+          - |
+            set +e
+            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen_image_layered_vllm_omni.json
             EXIT=$$?
             buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
             buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
             exit $$EXIT
         mirror_hardwares: h100_1
 
-      - label: ":full_moon: Diffusion X2I(&A&T) · Perf Test · Qwen-Image · 4-GPU"
+      - label: ":full_moon: Diffusion X2I(&A&T) · Perf Test · BAGEL"
-        key: nightly-diffusion-x2iat-performance-qwen-image-4gpu
+        key: nightly-diffusion-x2iat-performance-bagel
         timeout_in_minutes: 180
         commands:
           - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
           - export DIFFUSION_ATTENTION_BACKEND=FLASH_ATTN
-          - export CACHE_DIT_VERSION=1.5.0
+          - export CACHE_DIT_VERSION=1.3.0
           - |
             set +e
-            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen_image_vllm_omni.json -m "H100 and cards_4"
+            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_bagel_vllm_omni.json
             EXIT=$$?
             buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
             buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
             exit $$EXIT
-        mirror_hardwares: h100_4
+        mirror_hardwares: h100_1
 
   - group: ":card_index_dividers: Diffusion X2V Model Test"
     key: nightly-diffusion-x2v-group
@@ -314,36 +428,24 @@ steps:
       build.pull_request.labels includes "diffusion-x2v-test"
     steps:
 
-      - label: ":full_moon: Diffusion X2V · Wan2.2 T2V Function Test · Single-GPU"
+      - label: ":full_moon: Diffusion X2V · Wan2.2 T2V Function Test"
         timeout_in_minutes: 90
         commands:
-          - pytest -s -v tests/e2e/online_serving/test_wan22_expansion.py -k "t2v" -m "full_model and cuda and H100 and cards_1" --run-level "full_model"
+          - pytest -s -v tests/e2e/online_serving/test_wan22_expansion.py -k "t2v" -m "full_model and cuda" --run-level "full_model"
+        mirror_hardwares: h100_2
+
+      - label: ":full_moon: Diffusion X2V · Wan2.2 I2V/TI2V Function Test"
+        timeout_in_minutes: 90
+        commands:
+          - pytest -s -v tests/e2e/online_serving/test_wan22_expansion.py -k "i2v" -m "full_model and cuda" --run-level "full_model"
+        mirror_hardwares: h100_2
+
+      - label: ":full_moon: Diffusion X2V · LingBot Video MoE Function Test"
+        timeout_in_minutes: 90
+        commands:
+          - pytest -s -v tests/e2e/online_serving/test_lingbot_video_moe.py -k "video_generation" -m "full_model and cuda" --run-level "full_model"
         mirror_hardwares: h100_1
 
-      - label: ":full_moon: Diffusion X2V · Wan2.2 T2V Function Test · 2-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - pytest -s -v tests/e2e/online_serving/test_wan22_expansion.py -k "t2v" -m "full_model and cuda and H100 and cards_2" --run-level "full_model"
-        mirror_hardwares: h100_2
-
-      - label: ":full_moon: Diffusion X2V · Wan2.2 I2V/TI2V Function Test · Single-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - pytest -s -v tests/e2e/online_serving/test_wan22_expansion.py -k "i2v" -m "full_model and cuda and H100 and cards_1" --run-level "full_model"
-        mirror_hardwares: h100_1
-
-      - label: ":full_moon: Diffusion X2V · Wan2.2 I2V/TI2V Function Test · 2-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - pytest -s -v tests/e2e/online_serving/test_wan22_expansion.py -k "i2v" -m "full_model and cuda and H100 and cards_2" --run-level "full_model"
-        mirror_hardwares: h100_2
-
-      - label: ":full_moon: Diffusion X2V · HunyuanVideo-1.5 Function Test"
-        timeout_in_minutes: 90
-        commands:
-          - pytest -s -v tests/e2e/online_serving/test_hunyuan_video_15_expansion.py -m "full_model and cuda" --run-level "full_model"
-        mirror_hardwares: h100_2
-
       - label: ":full_moon: Diffusion X2V · Doc Test"
         timeout_in_minutes: 60
         commands:
@@ -354,6 +456,7 @@ steps:
         timeout_in_minutes: 90
         commands:
           - pytest -s -v tests/e2e/offline_inference/test_wan22_autoround_w4a16_expansion.py -m "full_model and cuda" --run-level "full_model"
+          - pytest -s -v tests/e2e/online_serving/test_wan_2_1_vace_expansion.py tests/e2e/online_serving/test_ltx2_expansion.py -m "full_model and cuda" --run-level "full_model"
         mirror_hardwares: h100_2
 
       - label: ":full_moon: Diffusion X2V · Wan2.2 Accuracy Test"
@@ -374,50 +477,21 @@ steps:
         commands:
           - pytest -s -v tests/e2e/accuracy/hunyuanvideo15_t2v/test_hunyuanvideo15_t2v_video_similarity.py -m full_model --run-level full_model
           - pytest -s -v tests/e2e/accuracy/hunyuanvideo15_i2v/test_hunyuanvideo15_i2v_video_similarity.py -m full_model --run-level full_model
+          - pytest -s -v 'tests/e2e/accuracy/ltx/test_ltx_official_similarity.py::test_ltx_one_stage_matches_official[ltx2_3_i2v]' -m full_model --run-level full_model
+          - pytest -s -v tests/e2e/accuracy/test_video_streaming_output_similarity.py -m full_model --run-level full_model
         mirror_hardwares: h100_2
 
-      - label: ":full_moon: Diffusion X2V · Perf Test · Single-GPU"
+      - label: ":full_moon: Diffusion X2V · Perf Test"
-        key: nightly-diffusion-x2v-performance-single
+        key: nightly-diffusion-x2v-performance
-        timeout_in_minutes: 180
-        commands:
-          - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
-          - |
-            set +e
-            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_wan22_i2v_vllm_omni.json -m "H100 and cards_1"
-            EXIT=$$?
-            buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
-            buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
-            exit $$EXIT
-        mirror_hardwares: h100_1
-
-      - label: ":full_moon: Diffusion X2V · Perf Test · 2-GPU"
-        key: nightly-diffusion-x2v-performance-2gpu
         timeout_in_minutes: 180
         commands:
           - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
           - export DIFFUSION_ATTENTION_BACKEND=FLASH_ATTN
           - |
             set +e
-            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_wan22_i2v_vllm_omni.json -m "H100 and cards_2"
+            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_wan22_i2v_vllm_omni.json
             EXIT1=$$?
-            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_cosmos3_vllm_omni.json -m "H100 and cards_2"
+            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_cosmos3_vllm_omni.json
-            EXIT2=$$?
-            buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
-            buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
-            if [ $$EXIT1 -ne 0 ]; then exit $$EXIT1; fi
-            exit $$EXIT2
-        mirror_hardwares: h100_2
-
-      - label: ":full_moon: Diffusion X2V · Perf Test · HunyuanVideo-1.5"
-        key: nightly-diffusion-x2v-performance-hunyuanvideo15
-        timeout_in_minutes: 180
-        commands:
-          - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
-          - |
-            set +e
-            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_hunyuanvideo15_t2v_vllm_omni.json
-            EXIT1=$$?
-            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_hunyuanvideo15_i2v_vllm_omni.json
             EXIT2=$$?
             buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
             buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
@@ -430,6 +504,7 @@ steps:
         timeout_in_minutes: 180
         commands:
           - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
+          - export DIFFUSION_ATTENTION_BACKEND=FLASH_ATTN
           - |
             set +e
             pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_minimax_h3_vllm_omni.json
@@ -444,34 +519,16 @@ steps:
     depends_on: upload-nightly-pipeline
     if: build.env("NIGHTLY") == "1" || build.pull_request.labels includes "nightly-test"
     steps:
-      - label: ":full_moon: Diffusion · Distributed Test with L4 · 2-GPU"
+      - label: ":full_moon: Diffusion · Distributed Test with L4"
         timeout_in_minutes: 60
         commands:
-          - pytest -sv tests/diffusion/distributed/ -m 'full_model and cuda and L4 and cards_2' --run-level "full_model"
+          - pytest -sv tests/diffusion/distributed/ -m 'full_model and cuda and L4' --run-level "full_model"
-        mirror_hardwares: l4_2
-
-      - label: ":full_moon: Diffusion · Distributed Test with L4 · 3-GPU"
-        timeout_in_minutes: 60
-        commands:
-          - pytest -sv tests/diffusion/distributed/ -m 'full_model and cuda and L4 and cards_3' --run-level "full_model"
-        mirror_hardwares: l4_3
-
-      - label: ":full_moon: Diffusion · Distributed Test with L4 · 4-GPU"
-        timeout_in_minutes: 60
-        commands:
-          - pytest -sv tests/diffusion/distributed/ -m 'full_model and cuda and L4 and cards_4' --run-level "full_model"
         mirror_hardwares: l4_4
 
-      - label: ":full_moon: Diffusion · Distributed Test with H100 · 2-GPU"
+      - label: ":full_moon: Diffusion · Distributed Test with H100"
         timeout_in_minutes: 60
         commands:
-          - pytest -sv tests/diffusion/distributed/ -m 'full_model and cuda and H100 and cards_2' --run-level "full_model"
+          - pytest -sv tests/diffusion/distributed/ -m 'full_model and cuda and H100' --run-level "full_model"
-        mirror_hardwares: h100_2
-
-      - label: ":full_moon: Diffusion · Distributed Test with H100 · 4-GPU"
-        timeout_in_minutes: 60
-        commands:
-          - pytest -sv tests/diffusion/distributed/ -m 'full_model and cuda and H100 and cards_4' --run-level "full_model"
         mirror_hardwares: h100_4
 
       - label: ":full_moon: Diffusion · Offloader Test"
@@ -495,10 +552,20 @@ steps:
       - label: ":full_moon: Diffusion · Quantization Test with L4"
         timeout_in_minutes: 60
         commands:
-          - pip install "vllm-gguf-plugin==0.0.4"
+          - pip install "vllm-gguf-plugin>=0.0.3"
           - pytest -sv tests/diffusion/quantization -m 'full_model and cuda and L4' --run-level "full_model"
         mirror_hardwares: l4_1
 
+  - label: ":bar_chart: Testcase Statistics"
+    key: nightly-testcase-statistics
+    timeout_in_minutes: 120
+    depends_on: upload-nightly-pipeline
+    if: build.env("NIGHTLY") == "1"
+    commands:
+      - python tools/nightly/buildkite_testcase_statistics.py -o tests/dfx/perf/results/buildkite_testcase_statistics.html
+      - buildkite-agent artifact upload "tests/dfx/perf/results/*.html"
+    mirror_hardwares: h100_1
+
   # NOTE: ":email: Nightly Collection & Email" is deprecated and will be removed soon.
   # Do not add new depends_on entries or extend this step.
   - label: ":email: Nightly Collection & Email"
@@ -506,25 +573,35 @@ steps:
     depends_on:
       - nightly-omni-performance-no-async-chunk
       - nightly-omni-performance-async-chunk
-      - nightly-tts-performance-single
+      - nightly-omni-performance-vllm-text
-      - nightly-tts-performance-2gpu
+      - nightly-omni-performance-multi-replicas
-      - nightly-diffusion-x2iat-performance-qwen-image-single
+      - nightly-tts-performance
-      - nightly-diffusion-x2iat-performance-qwen-image-4gpu
+      - nightly-tts-performance-higgs-audio-v3
-      - nightly-diffusion-x2v-performance-single
+      - nightly-diffusion-x2iat-performance-qwen-image
-      - nightly-diffusion-x2v-performance-2gpu
+      - nightly-diffusion-x2iat-performance-qwen-image-edit
+      - nightly-diffusion-x2iat-performance-qwen-image-edit-2511
+      - nightly-diffusion-x2iat-performance-qwen-image-layered
+      - nightly-diffusion-x2iat-performance-bagel
+      - nightly-diffusion-x2v-performance
+      - nightly-testcase-statistics
     if: build.env("NIGHTLY") == "1" && build.env("EMAIL_DISTRIBUTION") == "1"
     commands:
       - pip install openpyxl
       - export DEFAULT_INPUT_DIR=tests/dfx/perf/results
       - export DEFAULT_OUTPUT_DIR=tests/dfx/perf/results
-      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-tts-performance-single
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-tts-performance
-      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-tts-performance-2gpu
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-tts-performance-higgs-audio-v3
       - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-omni-performance-no-async-chunk
       - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-omni-performance-async-chunk
-      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2iat-performance-qwen-image-single
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-omni-performance-vllm-text
-      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2iat-performance-qwen-image-4gpu
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-omni-performance-multi-replicas
-      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2v-performance-single
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2iat-performance-qwen-image
-      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2v-performance-2gpu
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2iat-performance-qwen-image-edit
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2iat-performance-qwen-image-edit-2511
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2iat-performance-qwen-image-layered
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2iat-performance-bagel
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.json" . --step nightly-diffusion-x2v-performance
+      - buildkite-agent artifact download "tests/dfx/perf/results/*.html" . --step nightly-testcase-statistics
       - python tools/nightly/generate_nightly_perf_excel.py
       - python tools/nightly/generate_nightly_perf_html.py
       - python tools/nightly/send_nightly_email.py --report-file "tests/dfx/perf/results/*.xlsx, tests/dfx/perf/results/*.html"

.buildkite/cuda/test-ready.yml

@@ -1,6 +1,3 @@
-# CUDA CI L2 (`core_model` e2e) plus L1 CPU Simple Test (`core_model and cpu`).
-# Uploaded by Upload Ready Pipeline: PR label `ready`, or main+WEEKLY=1 with --e2e.
-
 env:
   VLLM_WORKER_MULTIPROC_METHOD: spawn
   VLLM_USE_FLASHINFER_MOE_FP16: 0
@@ -12,27 +9,23 @@ steps:
     depends_on: upload-ready-pipeline
     steps:
       - label: "Simple · Model Executor Test"
-        timeout_in_minutes: 20
         commands:
-          - "pytest -sv tests/model_executor -m 'core_model and cpu'"
+          - "timeout 20m pytest -sv tests/model_executor -m 'core_model and cpu' --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
         mirror_hardwares: l4_1
 
       - label: "Simple · Diffusion Test"
-        timeout_in_minutes: 20
         commands:
-          - "pytest -sv tests/diffusion -m 'core_model and cpu'"
+          - "timeout 20m pytest -sv tests/diffusion -m 'core_model and cpu' --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
         mirror_hardwares: l4_1
 
       - label: "Simple · Engine&Entrypoints Test"
-        timeout_in_minutes: 20
         commands:
-          - "pytest -sv tests/entrypoints tests/engine -m 'core_model and cpu'"
+          - "timeout 20m pytest -sv tests/entrypoints tests/engine -m 'core_model and cpu' --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
         mirror_hardwares: l4_1
 
       - label: "Simple · Other Test"
-        timeout_in_minutes: 20
         commands:
-          - "pytest -sv tests/ -m 'core_model and cpu' --ignore=tests/diffusion --ignore=tests/model_executor --ignore=tests/entrypoints --ignore=tests/engine"
+          - "timeout 20m pytest -sv tests/ -m 'core_model and cpu' --ignore=tests/diffusion --ignore=tests/model_executor --ignore=tests/entrypoints --ignore=tests/engine --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml"
         mirror_hardwares: l4_1
 
   - group: ":card_index_dividers: Diffusion Test"
@@ -54,20 +47,17 @@ steps:
           - timeout 15m pytest -sv tests/diffusion/batching -m 'core_model and cuda' --run-level "core_model"
         mirror_hardwares: l4_1
 
-      - label: "Diffusion · Distributed Test · Single-GPU"
+      - label: "Diffusion · Distributed & Attention Test"
+        source_file_dependencies:
+          - tests/diffusion/distributed/
+          - tests/diffusion/attention/test_attention_sp.py
+          - vllm_omni/diffusion/distributed/
+          - vllm_omni/diffusion/attention/
+          - vllm_omni/diffusion/data.py
+          - vllm_omni/diffusion/models/
         commands:
-          - timeout 20m pytest -sv tests/diffusion/distributed -m 'core_model and cuda and L4 and cards_1' --run-level "core_model"
+          - timeout 20m pytest -sv tests/diffusion/distributed -m 'core_model and cuda' --run-level "core_model"
-        mirror_hardwares: l4_1
+          - timeout 20m pytest -sv tests/diffusion/attention/test_attention_sp.py -m 'core_model and cuda' --run-level "core_model"
-
-      - label: "Diffusion · Distributed Test · 2-GPU"
-        commands:
-          - timeout 20m pytest -sv tests/diffusion/distributed -m 'core_model and cuda and L4 and cards_2' --run-level "core_model"
-        mirror_hardwares: l4_2
-
-      - label: "Diffusion · Distributed & Attention Test · 4-GPU"
-        commands:
-          - timeout 20m pytest -sv tests/diffusion/distributed -m 'core_model and cuda and L4 and cards_4' --run-level "core_model"
-          - timeout 20m pytest -sv tests/diffusion/attention/test_attention_sp.py -m 'core_model and cuda and L4 and cards_4' --run-level "core_model"
         mirror_hardwares: l4_4
 
       - label: "Diffusion · Model Test"
@@ -86,7 +76,7 @@ steps:
 
       - label: "Tiny Model Tests · Diffusion (base)"
         commands:
-          - timeout 15m pytest -sv -n 4 tests/model_tests/diffusion -m 'core_model and cuda' --run-level "core_model"
+          - timeout 15m pytest -sv tests/model_tests/diffusion -m 'core_model and cuda' --run-level "core_model"
         mirror_hardwares: l4_1
 
   - label: "Engine&Model Executor Test"
@@ -131,8 +121,32 @@ steps:
           - vllm_omni/outputs/
         commands:
           - timeout 20m pytest -s -v tests/e2e/features/custom_pipeline/ -m "core_model"
-        mirror_hardwares: l4_1
+        agents:
-
+          queue: "gpu_1_queue"
+        plugins:
+          - docker#v5.2.0:
+              image: public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT
+              always-pull: true
+              propagate-environment: true
+              # vLLM 0.27 added a pre-flight capacity check in
+              # distributed/device_communicators/shm_broadcast.py. Without this
+              # the container gets Docker's 64 MiB default and every test in
+              # this step dies at engine start with
+              #   RuntimeError: Insufficient space in /dev/shm:
+              #                 240 MiB required, 64 MiB available
+              # (9 failed in 8.7s on build #2949). vLLM 0.26 had no such check
+              # and allocated regardless, so this step passed on main while
+              # being under-provisioned the whole time — 0.27 turned a silent
+              # risk into an explicit error.
+              #
+              # This is the only inline docker plugin block in .buildkite/cuda/;
+              # every other step inherits common/ci_mirror_hardwares.yml, which
+              # already sets 8gb. Matching that rather than inventing a value.
+              shm-size: "8gb"
+              environment:
+                - "HF_HOME=/fsx/hf_cache"
+              volumes:
+                - "/fsx/hf_cache:/fsx/hf_cache"
 
       - label: "RLHF · VeRL-Omni E2E Test"
         source_file_dependencies:
@@ -154,13 +168,7 @@ steps:
       - label: "Omni · Qwen3-Omni Test"
         source_file_dependencies:
           - tests/e2e/online_serving/test_qwen3_omni.py
-          - tests/helpers/runtime.py
-          - vllm_omni/entrypoints/openai/serving_chat.py
           - vllm_omni/model_executor/models/qwen3_omni/
-          - vllm_omni/model_executor/models/common/qwen3_code_predictor.py
-          - vllm_omni/model_executor/models/common/snake_activation.py
-          - vllm_omni/model_executor/models/qwen2_5_omni/qwen2_5_omni_thinker.py
-          - vllm_omni/model_executor/models/qwen3_tts/cuda_graph_decoder_wrapper.py
           - vllm_omni/model_executor/stage_input_processors/qwen3_omni.py
           - vllm_omni/deploy/qwen3_omni_moe.yaml
         commands:
@@ -173,11 +181,6 @@ steps:
       - label: "Omni · MiniCPM-o 4.5 Offline Test"
         source_file_dependencies:
           - tests/e2e/offline_inference/test_minicpmo_4_5.py
-          - vllm_omni/model_executor/models/minicpmo_4_5/
-          - vllm_omni/model_executor/models/step_audio2/
-          - vllm_omni/model_executor/models/cosyvoice3/code2wav_core/
-          - vllm_omni/model_executor/models/cosyvoice3/utils.py
-          - vllm_omni/model_executor/stage_input_processors/minicpmo_4_5_omni.py
           - vllm_omni/deploy/minicpmo_4_5.yaml
         commands:
           - |
@@ -189,11 +192,6 @@ steps:
       - label: "Omni · MiniCPM-o 4.5 Online Test"
         source_file_dependencies:
           - tests/e2e/online_serving/test_minicpmo_4_5.py
-          - vllm_omni/model_executor/models/minicpmo_4_5/
-          - vllm_omni/model_executor/models/step_audio2/
-          - vllm_omni/model_executor/models/cosyvoice3/code2wav_core/
-          - vllm_omni/model_executor/models/cosyvoice3/utils.py
-          - vllm_omni/model_executor/stage_input_processors/minicpmo_4_5_omni.py
           - vllm_omni/deploy/minicpmo_4_5.yaml
         commands:
           - |
@@ -211,25 +209,17 @@ steps:
           - tests/e2e/online_serving/test_minicpmo_realtime_duplex_drivers.py
           - tests/e2e/online_serving/helpers/minicpmo_realtime_duplex_scenarios.py
           - tests/e2e/online_serving/run_minicpmo_realtime_duplex_multi_session.py
-          - tests/e2e/online_serving/run_minicpmo_realtime_duplex_server_vad.py
           - tests/e2e/online_serving/run_minicpmo_realtime_duplex_soft_interrupt.py
-          - vllm_omni/deploy/minicpmo_4_5.yaml
           - tests/e2e/online_serving/helpers/minicpmo_4_5_duplex.py
+          - vllm_omni/deploy/minicpmo_4_5_duplex.yaml
           - vllm_omni/experimental/fullduplex/
           - vllm_omni/model_executor/models/minicpmo_4_5/
-          - vllm_omni/model_executor/models/step_audio2/
-          - vllm_omni/model_executor/models/cosyvoice3/code2wav_core/
-          - vllm_omni/model_executor/models/cosyvoice3/utils.py
           - vllm_omni/model_executor/stage_input_processors/minicpmo_4_5_omni.py
           - vllm_omni/core/sched/
           - vllm_omni/outputs/multimodal_accumulation.py
           - vllm_omni/engine/stage_init_utils.py
           - vllm_omni/engine/async_omni_engine.py
-          - vllm_omni/engine/orchestrator.py
           - vllm_omni/entrypoints/cli/serve.py
-          - vllm_omni/entrypoints/openai/api_server.py
-          - vllm_omni/entrypoints/openai/duplex_capability.py
-          - vllm_omni/config/stage_config.py
         commands:
           - |
             timeout 20m bash -c "
@@ -237,30 +227,27 @@ steps:
             "
         mirror_hardwares: h100_1
 
-      - label: "Omni · MiniCPM-o 4.5 Perf Test"
+      - label: "TTS · VoxCPM2 Test"
-        timeout_in_minutes: 60
         source_file_dependencies:
-          - tests/dfx/perf/tests/test_minicpmo_4_5_ready.json
+          - tests/e2e/online_serving/test_voxcpm2_tts.py
-          - vllm_omni/deploy/minicpmo_4_5.yaml
+          - vllm_omni/model_executor/models/voxcpm2/
-          - vllm_omni/model_executor/models/minicpmo_4_5/
+          - vllm_omni/deploy/voxcpm2.yaml
-          - vllm_omni/model_executor/stage_input_processors/minicpmo_4_5_omni.py
+        timeout_in_minutes: 30
         commands:
-          - export BENCHMARK_DIR=tests/dfx/perf/results
+          - |
-          - pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_minicpmo_4_5_ready.json -m 'core_model and cuda'
+            timeout 30m bash -c "
-        mirror_hardwares: h100_1
+              export VLLM_LOGGING_LEVEL=DEBUG
+              export VLLM_WORKER_MULTIPROC_METHOD=spawn
+              pytest -s -v tests/e2e/online_serving/test_voxcpm2_tts.py -m 'core_model' --run-level 'core_model'
+            "
+        mirror_hardwares: l4_1
 
       - label: "TTS · Qwen3-TTS CustomVoice Test"
         source_file_dependencies:
           - tests/e2e/online_serving/test_qwen3_tts_customvoice.py
           - vllm_omni/model_executor/models/qwen3_tts/
-          - vllm_omni/model_executor/models/common/qwen3_code_predictor.py
-          - vllm_omni/model_executor/models/common/snake_activation.py
-          - vllm_omni/model_executor/models/common/alias_free_activation.py
           - vllm_omni/model_executor/stage_input_processors/qwen3_tts.py
           - vllm_omni/deploy/qwen3_tts.yaml
-          - vllm_omni/entrypoints/openai/serving_speech.py
-          - vllm_omni/entrypoints/openai/tts_adapters/
-          - vllm_omni/engine/stage_pool.py
         commands:
           - |
             timeout 20m bash -c "
@@ -269,11 +256,68 @@ steps:
             "
         mirror_hardwares: l4_1
 
+      - label: "TTS · Qwen3-TTS Base Test"
+        source_file_dependencies:
+          - tests/e2e/online_serving/test_qwen3_tts_base.py
+          - vllm_omni/model_executor/models/qwen3_tts/
+          - vllm_omni/model_executor/stage_input_processors/qwen3_tts.py
+          - vllm_omni/deploy/qwen3_tts.yaml
+        commands:
+          - |
+            timeout 20m bash -c "
+              export VLLM_WORKER_MULTIPROC_METHOD=spawn
+              export VLLM_ALLOW_LONG_MAX_MODEL_LEN=1
+              export VLLM_OMNI_USE_V2_RUNNER=1
+              pytest -s -v tests/e2e/online_serving/test_qwen3_tts_base.py -m 'core_model' --run-level 'core_model'
+            "
+        mirror_hardwares: l4_1
+
+      - label: "TTS · Higgs-Audio-V3 Test"
+        source_file_dependencies:
+          - tests/e2e/online_serving/test_higgs_audio_v3.py
+          - vllm_omni/model_executor/models/higgs_audio_v3/
+          - vllm_omni/model_executor/stage_input_processors/higgs_audio_v3.py
+          - vllm_omni/deploy/higgs_multimodal_qwen3.yaml
+        commands:
+          - |
+            timeout 20m bash -c "
+              export VLLM_WORKER_MULTIPROC_METHOD=spawn
+              export VLLM_USE_DEEP_GEMM=0
+              export VLLM_MOE_USE_DEEP_GEMM=0
+              pytest -s -v tests/e2e/online_serving/test_higgs_audio_v3.py -m 'core_model' --run-level 'core_model'
+            "
+        mirror_hardwares: l4_1
+
+      - label: "Diffusion · Qwen Image Edit Test"
+        source_file_dependencies:
+          - tests/e2e/online_serving/test_qwen_image_edit.py
+          - vllm_omni/diffusion/models/qwen_image/
+        commands:
+          - |
+            timeout 20m bash -c "
+              pytest -s -v tests/e2e/online_serving/test_qwen_image_edit.py -m 'core_model' --run-level 'core_model'
+            "
+        mirror_hardwares: h100_1
+
+      - label: "Diffusion · Bagel Test"
+        source_file_dependencies:
+          - tests/e2e/online_serving/test_bagel.py
+          - vllm_omni/model_executor/models/bagel/
+          - vllm_omni/diffusion/models/bagel/
+          - vllm_omni/model_executor/stage_input_processors/bagel.py
+          - vllm_omni/deploy/bagel.yaml
+        commands:
+          - |
+            timeout 40m bash -c "
+              export VLLM_IMAGE_FETCH_TIMEOUT=60
+              pytest -s -v tests/e2e/online_serving/test_bagel.py -m 'core_model' --run-level 'core_model'
+            "
+        mirror_hardwares: h100_1
+
       - label: "Diffusion · Wan22 Test"
         source_file_dependencies:
           - tests/e2e/online_serving/test_wan22_t2v.py
           - vllm_omni/diffusion/models/wan2_2/
-          - vllm_omni/diffusion/models/dmd2/
         commands:
           - |
             timeout 40m bash -c "
@@ -286,9 +330,6 @@ steps:
         source_file_dependencies:
           - tests/e2e/online_serving/test_cosmos3.py
           - vllm_omni/diffusion/models/cosmos3/
-          - vllm_omni/diffusion/distributed/autoencoders/autoencoder_kl_wan.py
-          - vllm_omni/experimental/world_models/
-          - vllm_omni/model_extras/cosmos3.py
         commands:
           - |
             timeout 40m bash -c "
@@ -297,30 +338,10 @@ steps:
             "
         mirror_hardwares: h100_1
 
-      - label: "Diffusion · MiniMax-H3 DLO DP2 T2VA Test"
-        timeout_in_minutes: 90
-        source_file_dependencies:
-          - tests/e2e/online_serving/test_minimax_h3_dlo_dp2_t2va.py
-          - tests/helpers/assertions.py
-          - tests/helpers/runtime.py
-          - vllm_omni/diffusion/models/minimax_h3/
-          - vllm_omni/diffusion/offloader/
-          - vllm_omni/diffusion/distributed/
-          - vllm_omni/entrypoints/openai/
-        commands:
-          - |
-            timeout 80m bash -c "
-              export VLLM_IMAGE_FETCH_TIMEOUT=60
-              pytest -s -v tests/e2e/online_serving/test_minimax_h3_dlo_dp2_t2va.py -m 'core_model and cuda and diffusion' --run-level 'core_model'
-            "
-        mirror_hardwares: h100_2
-
       - label: "Diffusion · Qwen Image Test"
         source_file_dependencies:
           - tests/e2e/online_serving/test_qwen_image.py
           - vllm_omni/diffusion/models/qwen_image/
-          - vllm_omni/diffusion/models/dmd2/
-          - vllm_omni/diffusion/distributed/autoencoders/autoencoder_kl_qwenimage.py
         commands:
           - |
             timeout 40m bash -c "
@@ -329,14 +350,14 @@ steps:
             "
         mirror_hardwares: h100_1
 
-      - label: "Diffusion · HunyuanVideo-1.5 Test"
+      - label: "Diffusion · Qwen Image Layered Test"
         source_file_dependencies:
-          - tests/e2e/online_serving/test_hunyuan_video_15_expansion.py
+          - tests/e2e/online_serving/test_qwen_image_layered.py
-          - vllm_omni/diffusion/models/hunyuan_video/
+          - vllm_omni/diffusion/models/qwen_image/
         commands:
           - |
             timeout 40m bash -c "
               export VLLM_IMAGE_FETCH_TIMEOUT=60
-              pytest -s -v tests/e2e/online_serving/test_hunyuan_video_15_expansion.py -m 'core_model and diffusion' --run-level 'core_model'
+              pytest -s -v tests/e2e/online_serving/test_qwen_image_layered.py -m 'core_model' --run-level 'core_model'
             "
         mirror_hardwares: h100_1

.buildkite/cuda/test-weekly.yml

@@ -1,6 +1,3 @@
-# CUDA CI L5 (`slow` / reliability / extra coverage).
-# Uploaded by Upload Weekly Pipeline: WEEKLY=1 / NON_CRITICAL=1, or PR label `weekly-test`.
-
 env:
   VLLM_WORKER_MULTIPROC_METHOD: spawn
   VLLM_USE_FLASHINFER_MOE_FP16: 0
@@ -8,153 +5,85 @@ env:
   HF_HUB_ETAG_TIMEOUT: 60
 
 steps:
-  - label: "Simple · CPU Coverage Test"
+
-    key: weekly-simple-cpu-coverage
+  - label: "Reliability Test - qwen3-omni"
+    timeout_in_minutes: 180
     depends_on: upload-weekly-pipeline
-    if: build.env("WEEKLY") == "1"
+    if: build.env("WEEKLY") == "1" || build.pull_request.labels includes "weekly-test"
-    timeout_in_minutes: 60
     commands:
-      - |
+      - pytest -s -v tests/dfx/reliability/test_reliability_qwen3_omni.py -m "slow"
-        set +e
+    mirror_hardwares: h100_2
-        REPORT="coverage-core-model-cpu-$${BUILDKITE_STEP_ID:-local}.xml"
+
-        pytest -sv tests/ -m 'core_model and cpu' --cov=vllm_omni --cov-report=term-missing:skip-covered --cov-report=xml:$${REPORT}
+  - label: "Reliability Test - wan22"
-        EXIT=$$?
+    timeout_in_minutes: 180
-        gzip -9 -f "$${REPORT}"
+    depends_on: upload-weekly-pipeline
-        buildkite-agent artifact upload "$${REPORT}.gz"
+    if: build.env("WEEKLY") == "1" || build.pull_request.labels includes "weekly-test"
-        exit $$EXIT
+    commands:
+      - pytest -s -v tests/dfx/reliability/test_reliability_wan22.py -m "slow"
+    mirror_hardwares: h100_2
+
+  - label: "Reliability Test - hunyuan-image"
+    timeout_in_minutes: 180
+    depends_on: upload-weekly-pipeline
+    if: build.env("WEEKLY") == "1" || build.pull_request.labels includes "weekly-test"
+    commands:
+      - pytest -s -v tests/dfx/reliability/test_reliability_hunyuan_image.py -m "slow"
+    mirror_hardwares: h100_4
+
+  - label: "Reliability Test - voxcpm2"
+    timeout_in_minutes: 180
+    depends_on: upload-weekly-pipeline
+    if: build.env("WEEKLY") == "1" || build.pull_request.labels includes "weekly-test"
+    commands:
+      - pytest -s -v tests/dfx/reliability/test_reliability_voxcpm2.py -m "slow"
     mirror_hardwares: l4_1
 
-  - group: "Reliability Test"
+  - group: "Reliability Test - Invalid parameters Test"
     depends_on: upload-weekly-pipeline
     if: build.env("WEEKLY") == "1" || build.pull_request.labels includes "weekly-test"
     steps:
 
-      - label: "Reliability Test · qwen3-omni"
+      - label: "Invalid parameters Test · H100"
-        timeout_in_minutes: 180
+        timeout_in_minutes: 120
         commands:
-          - pytest -s -v tests/dfx/reliability/test_reliability_qwen3_omni.py -m "slow"
+          - pytest -s -v tests/dfx/reliability/invalid_param_test/ -m "slow and H100"
         mirror_hardwares: h100_2
 
-      - label: "Reliability Test · wan22"
+      - label: "Invalid parameters Test · L4"
-        timeout_in_minutes: 180
-        commands:
-          - pytest -s -v tests/dfx/reliability/test_reliability_wan22.py -m "slow"
-        mirror_hardwares: h100_2
-
-      - label: "Reliability Test · hunyuan-image"
-        timeout_in_minutes: 180
-        commands:
-          - pytest -s -v tests/dfx/reliability/test_reliability_hunyuan_image.py -m "slow"
-        mirror_hardwares: h100_4
-
-      - label: "Reliability Test · Invalid parameters · H100 · Single-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - pytest -s -v tests/dfx/reliability/invalid_param_test/ -m "slow and H100 and cards_1"
-        mirror_hardwares: h100_1
-
-      - label: "Reliability Test · Invalid parameters · H100 · 2-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - pytest -s -v tests/dfx/reliability/invalid_param_test/ -m "slow and H100 and cards_2"
-        mirror_hardwares: h100_2
-
-      - label: "Reliability Test · Invalid parameters · L4"
         timeout_in_minutes: 120
         commands:
           - pytest -s -v tests/dfx/reliability/invalid_param_test/ -m "slow and L4"
         mirror_hardwares: l4_1
 
-  - group: "Perf Test"
+  - group: ":card_index_dividers: E2E Tests"
     depends_on: upload-weekly-pipeline
     if: build.env("WEEKLY") == "1" || build.pull_request.labels includes "weekly-test"
     steps:
 
-      - label: ":full_moon: Omni · Perf Test · vLLM Text"
+      - label: "Omni · L4"
-        key: weekly-omni-performance-vllm-text
-        timeout_in_minutes: 180
-        commands:
-          - export BENCHMARK_DIR=tests/dfx/perf/results
-          - |
-            set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen3_omni_vllm_text.json
-            EXIT=$$?
-            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
-            exit $$EXIT
-        mirror_hardwares: h100_2
-
-      - label: ":full_moon: Omni · Perf Test · Async Chunk · Random"
-        key: weekly-omni-performance-async-chunk-random
-        timeout_in_minutes: 180
-        commands:
-          - export BENCHMARK_DIR=tests/dfx/perf/results
-          - |
-            set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen3_omni_async_chunk.json -m "H100 and slow"
-            EXIT=$$?
-            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
-            exit $$EXIT
-        mirror_hardwares: h100_2
-
-      - label: ":full_moon: Omni · Perf Test · Multi-Replica"
-        key: weekly-omni-performance-multi-replicas
-        timeout_in_minutes: 180
-        commands:
-          - export BENCHMARK_DIR=tests/dfx/perf/results
-          - |
-            set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_qwen3_omni_multi_replicas.json
-            EXIT=$$?
-            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
-            exit $$EXIT
-        mirror_hardwares: h100_3
-
-  - group: ":card_index_dividers: E2E Tests"
-    depends_on: upload-weekly-pipeline
-    if: build.env("NON_CRITICAL") == "1"
-    steps:
-
-      # Online Qwen2.5 expansion is 2-card; offline expansion + AutoRound are 4-card.
-      - label: "Omni · L4 · 2-GPU"
         timeout_in_minutes: 90
         commands:
           - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
-          - pytest -s -v tests/e2e/online_serving/test_qwen2_5_omni_expansion.py -m "slow and L4 and omni and cards_2" --run-level "core_model"
+          - pytest -s -v tests/e2e/*/test_qwen2_5_omni_expansion.py --run-level "core_model"
-        mirror_hardwares: l4_2
-
-      - label: "Omni · L4 · 4-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
-          - pytest -s -v tests/e2e/offline_inference/test_qwen2_5_omni_expansion.py -m "slow and L4 and omni and cards_4" --run-level "core_model"
           - >-
             pytest -s -v tests/e2e/
             --ignore=tests/e2e/online_serving/test_qwen2_5_omni_expansion.py
             --ignore=tests/e2e/offline_inference/test_qwen2_5_omni_expansion.py
-            -m "slow and L4 and omni and cards_4"
+            -m "slow and L4 and omni"
             --run-level "full_model"
         mirror_hardwares: l4_4
 
-      - label: "Omni · H100 · Single-GPU"
+      - label: "Omni · H100"
         timeout_in_minutes: 90
         commands:
           - >-
             pytest -sv
-            tests/e2e/ -m "slow and H100 and omni and cards_1"
+            tests/e2e/ -m "slow and H100 and omni"
             --run-level "full_model"
-        mirror_hardwares: h100_1
+        mirror_hardwares: h100_2
-
-      - label: "Omni · H100 · 4-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - >-
-            pytest -sv
-            tests/e2e/ -m "slow and H100 and omni and cards_4"
-            --run-level "full_model"
-        mirror_hardwares: h100_4
 
       - label: "TTS · L4"
-        timeout_in_minutes: 120
+        timeout_in_minutes: 90
         commands:
           - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
           - export VLLM_USE_DEEP_GEMM="0"
@@ -179,72 +108,29 @@ steps:
             --run-level "full_model"
         mirror_hardwares: h100_1
 
-      # Own step: the image has no NanoCodec, so this file import-skips out of
+      - label: "Diffusion · L4"
-      # the mark-selected TTS steps above. The overrides keep nemo_toolkit[tts]
-      # from downgrading transformers; see
-      # examples/offline_inference/text_to_speech/README.md.
-      - label: "TTS · Gepard-1.0 · L4"
-        timeout_in_minutes: 60
-        commands:
-          - export VLLM_ALLOW_LONG_MAX_MODEL_LEN="1"
-          - printf '%s\n' "transformers>=5.5.3" "huggingface-hub>=1.0" > /tmp/gepard-overrides.txt
-          - uv pip install --python "$(python3 -c 'import sys; print(sys.executable)')" "nemo_toolkit[tts]==2.7.3" --overrides /tmp/gepard-overrides.txt
-          - >-
-            pytest -s -v tests/e2e/offline_inference/test_gepard_tts.py
-            -m "tts and cuda"
-            --run-level "full_model"
-        mirror_hardwares: l4_1
-
-      - label: "Diffusion · L4 · Single-GPU"
         timeout_in_minutes: 60
         commands:
           - >-
             pytest -sv
-            tests/e2e/ -m "slow and diffusion and L4 and cards_1"
+            tests/e2e/ -m "slow and diffusion and L4"
-            --run-level "full_model"
-        mirror_hardwares: l4_1
-
-      - label: "Diffusion · L4 · 2-GPU"
-        timeout_in_minutes: 60
-        commands:
-          - >-
-            pytest -sv
-            tests/e2e/ -m "slow and diffusion and L4 and cards_2"
-            --run-level "full_model"
-        mirror_hardwares: l4_2
-
-      - label: "Diffusion · L4 · 4-GPU"
-        timeout_in_minutes: 60
-        commands:
-          - >-
-            pytest -sv
-            tests/e2e/ -m "slow and diffusion and L4 and cards_4"
             --run-level "full_model"
         mirror_hardwares: l4_4
 
       - label: "Diffusion · H100 · Single-GPU"
-        timeout_in_minutes: 150
+        timeout_in_minutes: 120
         commands:
           - >-
             pytest -sv
-            tests/e2e/ -m "slow and diffusion and H100 and cards_1"
+            tests/e2e/ -m "slow and diffusion and H100 and not distributed_cuda"
             --run-level "full_model"
         mirror_hardwares: h100_1
 
-      - label: "Diffusion · H100 · 2-GPU"
+      - label: "Diffusion · H100 · Multi-GPU"
         timeout_in_minutes: 90
         commands:
           - >-
             pytest -sv
-            tests/e2e/ -m "slow and diffusion and H100 and cards_2"
+            tests/e2e/ -m "slow and diffusion and H100 and distributed_cuda"
             --run-level "full_model"
         mirror_hardwares: h100_2
-
-      - label: "Diffusion · H100 · 4-GPU"
-        timeout_in_minutes: 90
-        commands:
-          - >-
-            pytest -sv
-            tests/e2e/ -m "slow and diffusion and H100 and cards_4"
-            --run-level "full_model"
-        mirror_hardwares: h100_4

.buildkite/intel/pipeline-intel.yml

@@ -10,7 +10,7 @@ steps:
           DOCKER_BUILDKIT: "1"
           # Buildkite will automatically replace this with the actual commit hash
           VLLM_IMAGE_TAG: "${BUILDKITE_COMMIT}"
-          VLLM_VERSION: "v0.28.0"
+          VLLM_VERSION: "v0.27.0"
         priority: 100
         timeout_in_minutes: 60
         soft_fail: false

.buildkite/intel/scripts/run-xpu-test.sh

@@ -5,12 +5,7 @@ set -ex
 
 omni_source_dir=$(git rev-parse --show-toplevel)
 
-: "${VLLM_VERSION:?VLLM_VERSION must be set}"
+base_image_name="xpu/vllm-omni-ci-base:${VLLM_VERSION:?VLLM_VERSION must be set}"
-# Official upstream XPU image. It is published from vllm's own
-# docker/Dockerfile.xpu, which is exactly what build_vllm_base() clones and
-# builds below, so pulling it is equivalent to building the fallback base.
-upstream_base_image="vllm/vllm-openai-xpu:${VLLM_VERSION}"
-local_base_image="xpu/vllm-omni-ci-base:${VLLM_VERSION}"
 image_name="xpu/vllm-omni-ci:${BUILDKITE_COMMIT:?BUILDKITE_COMMIT must be set}"
 container_name="xpu_${BUILDKITE_COMMIT}_$(
     tr -dc A-Za-z0-9 </dev/urandom | head -c 10
@@ -54,36 +49,9 @@ build_vllm_base() (
         "${vllm_source_dir}"
 )
 
-# Resolve the vLLM base image, in priority order:
+if [ -z "$(docker images -q "${base_image_name}")" ]; then
-#   1. the published upstream image for VLLM_VERSION, pulled fresh (so moving
+    build_vllm_base
-#      tags like nightly are refreshed rather than served stale from disk);
+fi
-#   2. a local copy of that same image, when the registry is unreachable;
-#   3. VLLM_BASE, if that image is already on disk;
-#   4. otherwise build VLLM_BASE from the matching vLLM source tag (~30min).
-# VLLM_BASE names the fallback base and defaults to ${local_base_image}; set it
-# to reuse or produce a base under a different tag.
-resolve_vllm_base() {
-    if docker pull "${upstream_base_image}"; then
-        base_image_name="${upstream_base_image}"
-        return
-    fi
-
-    # Registry unreachable, but a previous run may have left the image behind.
-    if [ -n "$(docker images -q "${upstream_base_image}")" ]; then
-        echo "WARNING: could not pull ${upstream_base_image}; using local copy" >&2
-        base_image_name="${upstream_base_image}"
-        return
-    fi
-
-    echo "WARNING: ${upstream_base_image} unavailable remotely and locally" >&2
-    base_image_name="${VLLM_BASE:-${local_base_image}}"
-    if [ -z "$(docker images -q "${base_image_name}")" ]; then
-        echo "WARNING: building ${base_image_name} from vLLM ${VLLM_VERSION} source" >&2
-        build_vllm_base
-    fi
-}
-
-resolve_vllm_base
 
 # Try building the docker image
 docker_build "${image_name}" --build-arg "VLLM_BASE=${base_image_name}" --build-arg "VLLM_VERSION=${VLLM_VERSION}"
@@ -119,9 +87,8 @@ time timeout -k 30 30m docker run \
     echo $ZE_AFFINITY_MASK
     pip install tblib==3.1.0
     cd /workspace/vllm-omni
-    XPU_TEST_PATHS="tests/diffusion tests/dfx tests/e2e"
+    pytest -v -s -m "core_model and xpu and B60"
-    pytest -v -s $XPU_TEST_PATHS -m "core_model and xpu and B60"
     pytest -v -s tests/diffusion/quantization/test_mxfp8_config.py
-    pytest -v -s $XPU_TEST_PATHS -m "advanced_model and xpu and B60"
+    pytest -v -s -m "advanced_model and xpu and B60"
-    pytest -v -s $XPU_TEST_PATHS -m "omni and xpu and B60"
+    pytest -v -s -m "omni and xpu and B60"
 '

.buildkite/npu/test-npu-nightly.yml

@@ -69,67 +69,12 @@ steps:
           VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"
           HF_TOKEN: "${HF_TOKEN}"
         commands:
-          # Pin the Seed-TTS scorer to the second card: _get_eval_device() otherwise
-          # auto-picks npu:0, which the omni server under test already fills with its
-          # KV cache, so Whisper OOMs on `.to(device)` and every clip reports
-          # "ASR / WER failed".
-          - export SEED_TTS_WER_EVAL=1
-          - export SEED_TTS_EVAL_DEVICE=npu:1
           - |
             set +e
             pytest -s -v tests/e2e/accuracy/minicpmo_4_5/test_minicpmo_4_5.py -m "full_model and npu and A3" --run-level full_model
             EXIT=$$?
             buildkite-agent artifact upload "tests/e2e/accuracy/minicpmo_4_5/results/*.json"
             exit $$EXIT
-      - label: ":full_moon: Omni · MiniCPM-o 4.5 · Perf Test"
-        key: nightly-omni-performance-minicpmo-4-5-npu
-        timeout_in_minutes: 180
-        mirror_hardwares: a3_npu_2
-        env:
-          VLLM_WORKER_MULTIPROC_METHOD: spawn
-          VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"
-          HF_TOKEN: "${HF_TOKEN}"
-        commands:
-          - export BENCHMARK_DIR=tests/dfx/perf/results
-          - |
-            set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py -m "npu and A3" --test-config-file tests/dfx/perf/tests/test_minicpmo_4_5.json
-            EXIT=$$?
-            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
-            exit $$EXIT
-      - label: ":full_moon: Omni · MiniCPM-o 4.5 · Perf Test (A2)"
-        key: nightly-omni-performance-minicpmo-4-5-npu-a2
-        timeout_in_minutes: 180
-        mirror_hardwares: a2b3_npu_1
-        env:
-          VLLM_WORKER_MULTIPROC_METHOD: spawn
-          VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"
-          HF_TOKEN: "${HF_TOKEN}"
-          HF_HUB_OFFLINE: "1"
-        commands:
-          - export BENCHMARK_DIR=tests/dfx/perf/results
-          - |
-            set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py -m "npu and A2" --test-config-file tests/dfx/perf/tests/test_minicpmo_4_5.json
-            EXIT=$$?
-            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
-            exit $$EXIT
-      - label: ":full_moon: Omni · MiniCPM-o 4.5 · Duplex Seed-TTS Perf Test"
-        key: nightly-omni-performance-minicpmo-4-5-duplex-seed-tts-npu
-        timeout_in_minutes: 180
-        mirror_hardwares: a3_npu_2
-        env:
-          VLLM_WORKER_MULTIPROC_METHOD: spawn
-          VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"
-          HF_TOKEN: "${HF_TOKEN}"
-        commands:
-          - export BENCHMARK_DIR=tests/dfx/perf/results
-          - |
-            set +e
-            pytest -s -v tests/dfx/perf/scripts/run_benchmark.py -m "npu and A3" --test-config-file tests/dfx/perf/tests/test_minicpmo_4_5_duplex_seed_tts.json
-            EXIT=$$?
-            buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
-            exit $$EXIT
   - group: ":card_index_dividers: TTS Model Test"
     key: nightly-tts-test-group
     depends_on: upload-nightly-pipeline
@@ -151,6 +96,33 @@ steps:
             EXIT=$$?
             buildkite-agent artifact upload "tests/dfx/perf/results/*.json"
             exit $$EXIT
+  - group: ":card_index_dividers: Diffusion X2I(&A&T) Model Test"
+    key: nightly-diffusion-x2iat-group
+    depends_on: upload-nightly-pipeline
+    if: >-
+      build.env("NIGHTLY") == "1" ||
+      build.pull_request.labels includes "nightly-test" ||
+      build.pull_request.labels includes "diffusion-x2iat-test"
+    steps:
+      - label: ":full_moon: Diffusion X2I(&A&T) · Perf Test · Qwen-Image-Edit-2511"
+        key: nightly-diffusion-x2iat-performance-qwen-image-edit-2511
+        timeout_in_minutes: 180
+        mirror_hardwares: a3_npu_8
+        env:
+          VLLM_WORKER_MULTIPROC_METHOD: spawn
+          VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"
+          HF_TOKEN: "${HF_TOKEN}"
+          DIFFUSION_ATTENTION_BACKEND: TORCH_SDPA
+        commands:
+          - export DIFFUSION_BENCHMARK_DIR=tests/dfx/perf/results
+          - export CACHE_DIT_VERSION=1.3.0
+          - |
+            set +e
+            pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py -m npu --test-config-file tests/dfx/perf/tests/test_qwen_image_edit_2511_vllm_omni.json
+            EXIT=$$?
+            buildkite-agent artifact upload "tests/dfx/perf/results/diffusion_result_*.json"
+            buildkite-agent artifact upload "tests/dfx/perf/results/logs/*.log"
+            exit $$EXIT
   - group: ":card_index_dividers: Diffusion I2I Model Test"
     key: nightly-HunyuanImage3-x2i-test-group
     depends_on: upload-nightly-pipeline

.buildkite/npu/test-npu-ready.yml

@@ -1,27 +1,4 @@
 steps:
-  - label: "Platforms Test"
-    key: ready-platforms-test
-    depends_on: upload-ready-pipeline
-    mirror_hardwares: a3_npu_2
-    commands:
-      - >-
-        pytest -s -v
-        tests/platforms/npu/test_minicpmo_code2wav_npugraph.py
-        -m 'core_model and npu and A3 and omni'
-        --run-level 'core_model'
-
-  - label: "Diffusion NPU layer tests"
-    key: ready-diffusion-npu-layer-tests
-    depends_on: upload-ready-pipeline
-    mirror_hardwares: a3_npu_2
-    commands:
-      - >-
-        pytest -s -v
-        tests/diffusion/layers/test_activation.py
-        tests/diffusion/layers/test_fused_qk_norm_rope_npu.py
-        -m 'core_model and npu and A3'
-        --run-level 'core_model'
-
   - group: ":card_index_dividers: Omni Model Test"
     key: ready-omni-group
     depends_on: upload-ready-pipeline
@@ -40,22 +17,13 @@ steps:
           VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"
         commands:
           - pytest -s -v tests/e2e/online_serving/test_minicpmo_4_5_duplex.py -m 'core_model and npu and A3 and omni' --run-level 'core_model'
-      - label: "Omni · MiniCPM-o 4.5 Perf Test"
-        timeout_in_minutes: 60
-        mirror_hardwares: a3_npu_2
-        env:
-          VLLM_WORKER_MULTIPROC_METHOD: spawn
-          VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"
-        commands:
-          - export BENCHMARK_DIR=tests/dfx/perf/results
-          - pytest -s -v tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_minicpmo_4_5_ready.json -m 'core_model and npu and A3 and omni'
 
   - group: ":card_index_dividers: TTS Model Test"
     key: ready-tts-group
     depends_on: upload-ready-pipeline
     steps:
       - label: ":sound: TTS · Qwen3-TTS CustomVoice Test"
-        mirror_hardwares: a3_npu_2
+        mirror_hardwares: a2b3_npu_1
         env:
           VLLM_WORKER_MULTIPROC_METHOD: spawn
           VLLM_ALLOW_LONG_MAX_MODEL_LEN: "1"

.buildkite/release/release-pipeline.yml

@@ -40,11 +40,9 @@ steps:
     commands:
       - "bash .buildkite/release/scripts/publish-release-wheels-pypi.sh"
 
-  # Auto-skip when NIGHTLY=1 (scheduled); otherwise require manual unblock.
   - block: "Unblock to build release Docker images"
     depends_on: ~
     key: block-build-release-images
-    if: build.env("NIGHTLY") != "1"
 
   - group: "Build release Docker images"
     key: "build-release-images"
@@ -65,7 +63,7 @@ steps:
               --progress plain \
               -f docker/Dockerfile.cuda .
           - "docker push public.ecr.aws/q9t5s3a7/vllm-omni-release-repo:$BUILDKITE_COMMIT-$(uname -m)"
-          # Fallback default tag (no arch suffix) if arm64 build fails.
+          # re-tag to default image tag and push, just in case arm64 build fails
           - "docker tag public.ecr.aws/q9t5s3a7/vllm-omni-release-repo:$BUILDKITE_COMMIT-$(uname -m) public.ecr.aws/q9t5s3a7/vllm-omni-release-repo:$BUILDKITE_COMMIT"
           - "docker push public.ecr.aws/q9t5s3a7/vllm-omni-release-repo:$BUILDKITE_COMMIT"
         env:
@@ -88,26 +86,6 @@ steps:
         env:
           DOCKER_BUILDKIT: "1"
 
-  # NIGHTLY=1 only: promote ECR images to DockerHub nightly tags, then prune.
-  - label: "Publish nightly multi-arch image to DockerHub"
-    depends_on:
-      - build-release-image-x86
-      - build-release-image-arm64
-    if: build.env("NIGHTLY") == "1"
-    key: publish-nightly-images-dockerhub
-    agents:
-      queue: small_cpu_queue_release
-    commands:
-      - "bash .buildkite/release/scripts/publish-release-images.sh nightly"
-      - "bash .buildkite/release/scripts/cleanup-nightly-builds.sh"
-    plugins:
-      - docker-login#v3.0.0:
-          username: vllmbot
-          password-env: DOCKERHUB_TOKEN
-    env:
-      DOCKER_BUILDKIT: "1"
-      DOCKERHUB_USERNAME: "vllmbot"
-
   - block: "Publish release images to DockerHub"
     key: block-publish-release-images
     depends_on:

.buildkite/release/scripts/cleanup-nightly-builds.sh

@@ -1,147 +0,0 @@
-#!/bin/bash
-# SPDX-License-Identifier: Apache-2.0
-#
-# Clean up old nightly builds from DockerHub (vllm/vllm-omni), keeping the
-# newest 14 tags with the nightly- prefix.
-
-set -ex
-
-TAG_PREFIX="nightly-"
-REPO="vllm/vllm-omni"
-REPO_API_URL="https://hub.docker.com/v2/repositories/${REPO}/tags"
-
-echo "Cleaning up tags with prefix: $TAG_PREFIX in repository: $REPO"
-
-if [ -z "$DOCKERHUB_TOKEN" ]; then
-  echo "Error: DOCKERHUB_TOKEN environment variable is not set"
-  exit 1
-fi
-
-if [ -z "$DOCKERHUB_USERNAME" ]; then
-  echo "Error: DOCKERHUB_USERNAME environment variable is not set"
-  exit 1
-fi
-
-echo "Getting DockerHub bearer token..."
-set +x
-BEARER_TOKEN=$(curl -s -X POST \
-  -H "Content-Type: application/json" \
-  -d "{\"username\": \"$DOCKERHUB_USERNAME\", \"password\": \"$DOCKERHUB_TOKEN\"}" \
-  "https://hub.docker.com/v2/users/login" | jq -r '.token')
-set -x
-
-if [ -z "$BEARER_TOKEN" ] || [ "$BEARER_TOKEN" = "null" ]; then
-  echo "Error: Failed to get DockerHub bearer token"
-  exit 1
-fi
-
-get_all_tags() {
-  local page=1
-  local all_tags=""
-  # Hub `name` is a partial match (not a strict prefix). Use it to narrow the
-  # list, then enforce startswith() client-side. Exhaust API pages; do not
-  # stop merely because a page has no matching tags after filtering.
-  local name_query
-  name_query=$(printf '%s' "$TAG_PREFIX" | jq -sRr @uri)
-
-  while true; do
-    set +x
-    local response
-    response=$(curl -s -H "Authorization: Bearer $BEARER_TOKEN" \
-      "$REPO_API_URL?page=$page&page_size=100&name=${name_query}")
-    set -x
-
-    local page_count
-    page_count=$(echo "$response" | jq -r '.results | length')
-    if [ -z "$page_count" ] || [ "$page_count" = "0" ] || [ "$page_count" = "null" ]; then
-      break
-    fi
-
-    local tags
-    tags=$(echo "$response" | jq -r --arg prefix "$TAG_PREFIX" \
-      '.results[] | select(.name | startswith($prefix)) | "\(.last_updated)|\(.name)"')
-
-    if [ -n "$tags" ]; then
-      all_tags="$all_tags$tags"$'\n'
-    fi
-    page=$((page + 1))
-  done
-
-  echo "$all_tags" | sort -r | cut -d'|' -f2
-}
-
-delete_tag() {
-  local tag_name="$1"
-  echo "Deleting tag: $tag_name"
-
-  local delete_url="https://hub.docker.com/v2/repositories/${REPO}/tags/$tag_name"
-  set +x
-  local response http_code body
-  response=$(curl -s -w "\n%{http_code}" -X DELETE \
-    -H "Authorization: Bearer $BEARER_TOKEN" "$delete_url")
-  set -x
-  http_code=$(printf '%s\n' "$response" | tail -n1)
-  body=$(printf '%s\n' "$response" | sed '$d')
-
-  if [ "$http_code" -ge 200 ] && [ "$http_code" -lt 300 ]; then
-    echo "Successfully deleted tag: $tag_name (HTTP ${http_code})"
-    return 0
-  fi
-
-  local detail=""
-  if echo "$body" | jq -e '.detail' > /dev/null 2>&1; then
-    detail=$(echo "$body" | jq -r '.detail')
-  elif [ -n "$body" ]; then
-    detail="$body"
-  else
-    detail="(empty response body)"
-  fi
-  echo "Error: Failed to delete tag $tag_name (HTTP ${http_code}): ${detail}"
-  return 1
-}
-
-echo "Fetching all tags from DockerHub..."
-all_tags=$(get_all_tags)
-
-if [ -z "$all_tags" ]; then
-  echo "No tags found to clean up"
-  exit 0
-fi
-
-total_tags=$(echo "$all_tags" | wc -l)
-echo "Found $total_tags tags"
-
-tags_to_keep=14
-tags_to_delete=$((total_tags - tags_to_keep))
-
-if [ $tags_to_delete -le 0 ]; then
-  echo "No tags need to be deleted (only $total_tags tags found, keeping $tags_to_keep)"
-  exit 0
-fi
-
-echo "Will delete $tags_to_delete old tags, keeping the newest $tags_to_keep"
-
-tags_to_delete_list=$(echo "$all_tags" | tail -n +$((tags_to_keep + 1)))
-
-if [ -z "$tags_to_delete_list" ]; then
-  echo "No tags to delete"
-  exit 0
-fi
-
-echo "Deleting old tags..."
-delete_failures=0
-while IFS= read -r tag; do
-  if [ -n "$tag" ]; then
-    if ! delete_tag "$tag"; then
-      delete_failures=$((delete_failures + 1))
-    fi
-    sleep 1
-  fi
-done <<< "$tags_to_delete_list"
-
-if [ "$delete_failures" -gt 0 ]; then
-  echo "Cleanup finished with ${delete_failures} deletion failure(s)"
-  exit 1
-fi
-
-echo "Cleanup completed successfully"

.buildkite/release/scripts/publish-release-images.sh

@@ -1,87 +1,50 @@
 #!/bin/bash
 # SPDX-License-Identifier: Apache-2.0
 #
-# Publish Docker images from ECR staging to DockerHub.
+# Publish release Docker images from ECR to DockerHub.
-#
+# Pulls per-arch images, tags with `latest` and versioned tags, pushes them,
-# Usage:
+# then creates and pushes a multi-arch manifest.
-#   publish-release-images.sh            # release mode (default)
-#   publish-release-images.sh release
-#   publish-release-images.sh nightly
-#
-# Release tags: latest, v<version>
-# Nightly tags: nightly, nightly-<commit>
 
 set -euo pipefail
 
-MODE="${1:-release}"
+RELEASE_VERSION=$(buildkite-agent meta-data get release-version --default "" | sed 's/^v//')
+if [ -z "${RELEASE_VERSION}" ]; then
+  echo "ERROR: release-version metadata not set"
+  exit 1
+fi
 
+COMMIT="$BUILDKITE_COMMIT"
 DOCKERHUB_REPO="vllm/vllm-omni"
 ECR_REPO="public.ecr.aws/q9t5s3a7/vllm-omni-release-repo"
-COMMIT="$BUILDKITE_COMMIT"
-PRIMARY_TAGS=()
-EXTRA_MANIFEST_TAGS=()
-
-case "${MODE}" in
-  release)
-    RELEASE_VERSION=$(buildkite-agent meta-data get release-version --default "" | sed 's/^v//')
-    if [ -z "${RELEASE_VERSION}" ]; then
-      echo "ERROR: release-version metadata not set"
-      exit 1
-    fi
-    PRIMARY_TAGS=("latest" "v${RELEASE_VERSION}")
-    ;;
-  nightly)
-    # Commit-pinned manifest reuses nightly-{x86_64,aarch64} arch images.
-    PRIMARY_TAGS=("nightly")
-    EXTRA_MANIFEST_TAGS=("nightly-${COMMIT}")
-    ;;
-  *)
-    echo "Usage: $0 {release|nightly}"
-    exit 2
-    ;;
-esac
 
 echo "========================================"
-echo "Publishing ${MODE} images to ${DOCKERHUB_REPO}"
+echo "Publishing release images v${RELEASE_VERSION}"
 echo "  Commit: ${COMMIT}"
-echo "  Primary tags: ${PRIMARY_TAGS[*]}"
-if [ ${#EXTRA_MANIFEST_TAGS[@]} -gt 0 ]; then
-  echo "  Extra manifests: ${EXTRA_MANIFEST_TAGS[*]}"
-fi
 echo "========================================"
 
+# Login to ECR to pull staging images
 aws ecr-public get-login-password --region us-east-1 | \
   docker login --username AWS --password-stdin public.ecr.aws/q9t5s3a7
 
-SRC_X86="${ECR_REPO}:${COMMIT}-x86_64"
+docker pull "${ECR_REPO}:${COMMIT}-x86_64"
-SRC_ARM="${ECR_REPO}:${COMMIT}-aarch64"
+docker pull "${ECR_REPO}:${COMMIT}-aarch64"
 
-docker pull "${SRC_X86}"
+docker tag "${ECR_REPO}:${COMMIT}-x86_64" "${DOCKERHUB_REPO}:latest-x86_64"
-docker pull "${SRC_ARM}"
+docker tag "${ECR_REPO}:${COMMIT}-x86_64" "${DOCKERHUB_REPO}:v${RELEASE_VERSION}-x86_64"
+docker push "${DOCKERHUB_REPO}:latest-x86_64"
+docker push "${DOCKERHUB_REPO}:v${RELEASE_VERSION}-x86_64"
 
-publish_manifest() {
+docker tag "${ECR_REPO}:${COMMIT}-aarch64" "${DOCKERHUB_REPO}:latest-aarch64"
-  local tag="$1"
+docker tag "${ECR_REPO}:${COMMIT}-aarch64" "${DOCKERHUB_REPO}:v${RELEASE_VERSION}-aarch64"
-  local arch_tag_base="$2"
+docker push "${DOCKERHUB_REPO}:latest-aarch64"
-  docker manifest rm "${DOCKERHUB_REPO}:${tag}" || true
+docker push "${DOCKERHUB_REPO}:v${RELEASE_VERSION}-aarch64"
-  docker manifest create \
-    "${DOCKERHUB_REPO}:${tag}" \
-    "${DOCKERHUB_REPO}:${arch_tag_base}-x86_64" \
-    "${DOCKERHUB_REPO}:${arch_tag_base}-aarch64"
-  docker manifest push "${DOCKERHUB_REPO}:${tag}"
-}
 
-for tag in "${PRIMARY_TAGS[@]}"; do
+docker manifest rm "${DOCKERHUB_REPO}:latest" || true
-  docker tag "${SRC_X86}" "${DOCKERHUB_REPO}:${tag}-x86_64"
+docker manifest rm "${DOCKERHUB_REPO}:v${RELEASE_VERSION}" || true
-  docker tag "${SRC_ARM}" "${DOCKERHUB_REPO}:${tag}-aarch64"
+docker manifest create "${DOCKERHUB_REPO}:latest" "${DOCKERHUB_REPO}:latest-x86_64" "${DOCKERHUB_REPO}:latest-aarch64"
-  docker push "${DOCKERHUB_REPO}:${tag}-x86_64"
+docker manifest create "${DOCKERHUB_REPO}:v${RELEASE_VERSION}" "${DOCKERHUB_REPO}:v${RELEASE_VERSION}-x86_64" "${DOCKERHUB_REPO}:v${RELEASE_VERSION}-aarch64"
-  docker push "${DOCKERHUB_REPO}:${tag}-aarch64"
+docker manifest push "${DOCKERHUB_REPO}:latest"
-  publish_manifest "${tag}" "${tag}"
+docker manifest push "${DOCKERHUB_REPO}:v${RELEASE_VERSION}"
-done
-
-ARCH_BASE="${PRIMARY_TAGS[0]}"
-for tag in "${EXTRA_MANIFEST_TAGS[@]}"; do
-  publish_manifest "${tag}" "${ARCH_BASE}"
-done
 
 echo ""
-echo "Successfully published ${MODE} images"
+echo "Successfully published release images for v${RELEASE_VERSION}"

.claude/skills/add-diffusion-model/SKILL.md

@@ -105,7 +105,7 @@ For custom loop structures, follow `vllm_omni/diffusion/models/progress_bar.py`
 
 ## Path B: Custom/Private Repo Model
 
-For models without a diffusers pipeline — weights in custom format, model code in a private repo. Real examples: BAGEL, HunyuanImage3. See [references/custom-model-patterns.md](references/custom-model-patterns.md) for concrete custom-model integration patterns.
+For models without a diffusers pipeline — weights in custom format, model code in a private repo. Real examples: DreamID-Omni, BAGEL, HunyuanImage3. See [references/custom-model-patterns.md](references/custom-model-patterns.md) for concrete custom-model integration patterns.
 
 ### B1. Understand the reference repo
 
@@ -126,9 +126,9 @@ This is the key design decision for custom models. Follow these placement rules:
 | **Pipeline orchestration** (init, forward, denoise loop) | `vllm_omni/diffusion/models/<name>/pipeline_<name>.py` | Always required |
 | **Custom transformer/backbone** (ported and adapted to vllm-omni) | `vllm_omni/diffusion/models/<name>/<name>_transformer.py` or similar | `wan2_2.py`, `fusion.py`, `bagel_transformer.py` |
 | **Custom sub-models** (VAE, fusion, autoencoder) | `vllm_omni/diffusion/models/<name>/` as separate files | `autoencoder.py`, `fusion.py` |
-| **External dependency code** (original repo utilities) | **External repo**, installed via download script or pip | original-repo package via git clone |
+| **External dependency code** (original repo utilities) | **External repo**, installed via download script or pip | `dreamid_omni` package via git clone |
 | **Hardcoded model configs** | Module-level dicts in pipeline file | `VIDEO_CONFIG`, `AUDIO_CONFIG` dicts |
-| **Download/setup script** | `examples/offline_inference/<name>/download_<name>.py` | `download_<name>.py` |
+| **Download/setup script** | `examples/offline_inference/<name>/download_<name>.py` | `download_dreamid_omni.py` |
 | **Custom `model_index.json`** | Generated by download script, placed at model root | Minimal: `{"_class_name": "YourPipeline", ...}` |
 
 ### B3. Handle external dependencies
@@ -157,7 +157,7 @@ Copy the essential model files into `vllm_omni/diffusion/models/<name>/` and ada
 
 Custom models have two common patterns for weight loading:
 
-**Pattern 1: Bypass standard loader** (eager custom init)
+**Pattern 1: Bypass standard loader** (DreamID-Omni style)
 
 When the original model has complex custom init functions that load weights in `__init__`:
 
@@ -338,7 +338,7 @@ Classify the model's **CI priority** first:
 
 **L2 & L3 online — same file, dual marks on the baseline smoke:** The **first / simplest** case in `test_{slug}.py` (default deploy, minimal steps, single prompt) should carry **both** `@pytest.mark.core_model` and `@pytest.mark.advanced_model` on the **same** function so L2 (`test-ready.yml`) and L3 (`test-merge.yml`) share one smoke test. Heavier deploy variants or API paths in the same file use **`advanced_model` only**. When L3 moves to nightly, migrate those heavier cases into `test_{slug}_expansion.py` with `full_model` and remove the dedicated `test-merge.yml` job (see `test_longcat_image_expansion.py`, `test_qwen_image_expansion.py`).
 
-**L4 design (high priority):** Combine multiple supported features (Cache-DiT, TP, USP, CFG, HSDP, CPU offload, quantization) into **few parametrized** `OmniServerParams` rows so each feature appears in at least one case without exploding GPU jobs. Shard single-GPU vs multi-GPU cases across the nightly X2I/X2V function steps (`cards_1` vs `not cards_1`).
+**L4 design (high priority):** Combine multiple supported features (Cache-DiT, TP, USP, CFG, HSDP, CPU offload, quantization) into **few parametrized** `OmniServerParams` rows so each feature appears in at least one case without exploding GPU jobs. Shard single-GPU vs multi-GPU cases across the nightly X2I/X2V function steps (1-GPU vs `distributed_cuda`).
 
 **L4 design (medium / low):** One or two parametrized rows covering the best quality/perf trade-off; skip perf JSON unless the model is high priority.
 
@@ -610,24 +610,6 @@ See the [Profiling Single-Stage Diffusion](../../../docs/contributing/profiling.
 
 ---
 
-## Pre-commit conventions
-
-New library files must pass the local gates in
-[docs/contributing/README.md](../../../docs/contributing/README.md#linting).
-That page is the full hook list (SPDX, forbidden imports including Hugging Face
-Hub / Triton / pickle, `torch.cuda`, mypy, test marks, markdownlint, Buildkite,
-shellcheck). In particular:
-
-- SPDX copyright is `vLLM-Omni project` (stale `vLLM project` is rewritten).
-- Use `import regex as re` and `pybase64` in `vllm_omni/`; do not import stdlib
-  `re` or `base64`. Hugging Face Hub downloads go through
-  `vllm.transformers_utils.repo_utils`.
-- Do not add `torch.cuda.*` call sites; use `current_omni_platform`.
-- New `tests/**/test_*.py` files need a CI level mark and a hardware mark.
-- Do not expand `CHECK_IMPORTS[*].allowed_files` or `ALLOWED_FILES` without review.
-- GitHub Actions skips SPDX/shellcheck/mypy-3.10/test-marks/markdownlint; run
-  `pre-commit` locally.
-
 ## Iterative Development Tips
 
 1. **Start minimal**: Basic generation first, no parallelism/caching

.claude/skills/add-diffusion-model/references/custom-model-patterns.md

@@ -16,19 +16,20 @@ vllm_omni/diffusion/models/wan2_2/
 
 The transformer is loaded separately via `weights_sources` + `load_weights()`. Non-transformer components (VAE, text encoder) are loaded in `__init__` via `from_pretrained()`.
 
-### Custom model with external deps
+### Custom model with external deps (e.g., DreamID-Omni)
 
 ```
-vllm_omni/diffusion/models/<name>/
+vllm_omni/diffusion/models/dreamid_omni/
 ├── __init__.py                    # Exports pipeline only
-├── pipeline_<name>.py             # Pipeline: loads ALL weights in __init__ via custom helpers
+├── pipeline_dreamid_omni.py       # Pipeline: loads ALL weights in __init__ via custom helpers
-└── <backbone>.py                  # Custom transformer / fusion architecture
+├── fusion.py                      # Custom fusion architecture (video + audio cross-attention)
+└── wan2_2.py                      # Re-implemented Wan backbone with split API
 
-examples/offline_inference/<task>/
+examples/offline_inference/x_to_video_audio/
-└── download_<name>.py             # Downloads weights and/or clones code repo
+└── download_dreamid_omni.py       # Downloads weights from 3 HF repos + clones code repo
 ```
 
-All weights loaded eagerly in `__init__`. `load_weights()` is a no-op. External dependency imported with try/except.
+All weights loaded eagerly in `__init__`. `load_weights()` is a no-op. External dependency (`dreamid_omni` package) imported with try/except.
 
 ### Custom model with ported code (e.g., BAGEL)
 
@@ -75,7 +76,7 @@ self.weights_sources = [
 ]
 ```
 
-### Pattern 3: Fully custom loading
+### Pattern 3: Fully custom loading (DreamID-Omni)
 
 ```
 init → load ALL weights eagerly via custom helpers → load_weights() = no-op
@@ -107,8 +108,8 @@ Standard diffusers `model_index.json`:
 Custom model `model_index.json` (minimal):
 ```json
 {
-    "_class_name": "YourCustomPipeline",
+    "_class_name": "DreamIDOmniPipeline",
-    "fusion": "weights/model.safetensors"
+    "fusion": "DreamID-Omni/dreamid_omni.safetensors"
 }
 ```
 
@@ -116,7 +117,7 @@ The only **required** field is `_class_name` — it must match a key in `_DIFFUS
 
 ## External Dependency Management
 
-### Git clone + .pth injection
+### Git clone + .pth injection (DreamID-Omni pattern)
 
 ```python
 def download_dependency():
@@ -162,7 +163,7 @@ The engine checks `isinstance(pipeline, SupportImageInput)` at startup to config
 
 Diffusers models use `config.json` in each subfolder. Custom models often use:
 
-**Module-level config dicts**:
+**Module-level config dicts** (DreamID-Omni):
 ```python
 VIDEO_CONFIG = {
     "patch_size": [1, 2, 2], "model_type": "ti2v",
@@ -180,7 +181,7 @@ vae_cfg = bagel_cfg.get("vae_config", {})
 
 ## Custom Architecture Patterns
 
-### Split forward API
+### Split forward API (DreamID-Omni)
 
 When a fusion model needs to interleave blocks from two backbones:
 

.claude/skills/add-tts-model/SKILL.md

@@ -138,12 +138,9 @@ A shared buffer silently corrupts audio across concurrent requests — the sympt
    - Load codec weights (may need lazy loading from separate checkpoint)
    - Implement `forward()`: codec codes -> audio waveform
    - Return `OmniOutput` with `multimodal_outputs`
-5. **Define and register the pipeline topology** in
+5. **Create stage config YAML** defining both stages, memory allocation, and model paths
-   `vllm_omni/model_executor/models/<model>/pipeline.py`
+6. **Create stage input processor** for prompt building
-6. **Create the deploy YAML** under `vllm_omni/deploy/` for placement,
+7. **Write end2end.py** test script
-   memory sizing, connectors, and runtime overrides
-7. **Create stage input processor** for prompt building and inter-stage handoff
-8. **Write end2end.py** test script
 
 ### Critical Parameters to Get Right
 
@@ -177,7 +174,7 @@ These bugs appear in almost every new TTS PR. Check all before the first push. S
 - **All return paths must emit `model_outputs`** — if any early-return branch skips setting `model_outputs`, the serving layer silently drops that step's audio (fish speech: `fix: ensure ALL return paths emit model_outputs`)
 - **Per-request state isolation** — for batched concurrent requests, key all state by request ID; a shared buffer corrupts audio across requests (fish speech: `fix: per-request vocode + delta emission`)
 - **Codec tensor device** — move codec codes to the codec decoder's device before calling decode; mismatches cause silent CPU fallback or crashes (fish speech: `fix: use model device for CUDA stream`)
-- **AR stage `max_num_seqs`** — set to at least 4 in production deploy configs; for single-stage models this is the only stage. For two-stage models, Stage 0 (AR) needs `max_num_seqs ≥ 4` to pipeline concurrent requests; Stage 1 (codec decoder) is model-specific and may intentionally use `max_num_seqs: 1`. Defaulting the AR stage to 1 causes audio gaps under concurrency because the codec window round-robins across requests (RFC #2568)
+- **AR stage `max_num_seqs`** — set to at least 4 in production configs; for single-stage models this is the only stage. For two-stage models, Stage 0 (AR) needs `max_num_seqs ≥ 4` to pipeline concurrent requests; Stage 1 (codec decoder) typically uses `max_num_seqs: 1` intentionally. Default of 1 everywhere causes audio gaps under concurrency because the codec window round-robins across requests (RFC #2568)
 
 ### Optional Dependency Handling
 
@@ -193,18 +190,18 @@ pattern, signature constraints, and MOSS-TTS-Nano reference.
 When the upstream model cannot be cleanly split into an AR stage and a
 separate decoder, run the full pipeline inside a single AR worker and
 stream audio through a per-request `inference_stream()` generator keyed by
-`_omni_req_id`. Define one `StagePipelineConfig` with
+`_omni_req_id`. Stage config must set `worker_type: ar`,
-`execution_type=StageExecutionType.LLM_AR`, `engine_output_type="audio"`,
+`engine_output_type: audio`, `final_output: true`, `is_comprehension: true`,
-`final_output=True`, and `owns_tokenizer=True`. Set `async_chunk: false` in
+and `async_chunk: false` at the top level. Only extract params from
-the deploy YAML. Only extract params from
 `additional_information` that you actually forward, or pre-commit fails
 `ruff F841`.
 
 Full walkthrough with the complete `forward()` / `_create_stream_gen()`
-skeleton plus pipeline/deploy definitions:
+skeleton and stage-config fields:
 [references/single-stage-ar.md](references/single-stage-ar.md). For an
 in-tree reference, look for any single-stage AR model under
-`vllm_omni/model_executor/models/`, such as MOSS-TTS-Nano.
+`vllm_omni/model_executor/models/` — e.g. the MOSS-TTS-Nano integration when
+it lands.
 
 **VoxCPM2 is a different pattern** and should not reuse this skeleton — it
 runs the base LM under vLLM PagedAttention with external side-computation.
@@ -213,8 +210,7 @@ See `plan/voxcpm2_native_ar_design.md`.
 ### Deliverables
 
 - Model files in `vllm_omni/model_executor/models/<model_name>/`
-- Registered `pipeline.py` topology
+- Stage config YAML
-- Deploy YAML under `vllm_omni/deploy/`
 - Working `end2end.py` at `examples/offline_inference/text_to_speech/<model>/end2end.py`
 - New section in `examples/offline_inference/text_to_speech/README.md` (table row + per-model section). Do **not** create a top-level `examples/offline_inference/<model>/` dir or a per-model `README.md` inside `text_to_speech/<model>/` — the hub README is the documented surface and the mkdocs `generate_examples` hook only descends one level into `examples/<category>/`.
 
@@ -345,8 +341,8 @@ Heavier scenarios in the same file use **`advanced_model` only** (streaming, ext
 @pytest.mark.tts
 @hardware_test(res={"cuda": "L4"}, num_cards=1)
 @pytest.mark.parametrize("omni_server", tts_server_params, indirect=True)
-def test_voice_clone_en_non_streaming_001(omni_server, online_client) -> None:
+def test_voice_clone_en_non_streaming_001(omni_server, openai_client) -> None:
-    online_client.send_audio_speech_request({...})
+    openai_client.send_audio_speech_request({...})
 ```
 
 **L4 consolidation:** Prefer parametrized `OmniServerParams` rows (`default`, `async_chunk`, feature flags) in one expansion module rather than many merge-only files ([#1832](https://github.com/vllm-project/vllm-omni/issues/1832)).
@@ -363,8 +359,8 @@ def test_voice_clone_en_non_streaming_001(omni_server, online_client) -> None:
 
 | Path | Fixture | Call |
 |------|---------|------|
-| Online `/v1/*` | `online_client` | `online_client.send_*_request(request_config)` |
+| Online `/v1/*` | `openai_client` | `openai_client.send_*_request(request_config)` |
-| Offline inference | `offline_client` | `offline_client.send_*_request(request_config)` |
+| Offline inference | `omni_runner_handler` | `omni_runner_handler.send_*_request(request_config)` |
 
 1. **Grep `runtime.py` first** — reuse `send_omni_request`, `send_diffusion_request`, `send_audio_speech_request` (online + offline Qwen-style TTS), `send_single_stage_tts_request` (Coqui XTTS / MOSS-TTS-Nano offline), etc.
 2. **No matching helper** → add `send_<feature>_request` (or `send_<route>_http_request` for negative/dfx) in **`runtime.py`** with general `assert_*` bundled inside, **then** call it from the test.
@@ -396,7 +392,7 @@ See [vllm-omni-test skill](../vllm-omni-test/SKILL.md) § **Runtime send helpers
   `/health` return non-200 until you're actually ready.
 - **Mark tests per the CI Levels table** — baseline smoke: `core_model` + `advanced_model`; heavier cases: `advanced_model` only; L4 expansion: `full_model`
 - **No per-model helper modules** — do not create `tests/helpers/{slug}.py`; keep constants and `request_config` payloads in the test file
-- **Online and offline e2e go through `runtime.py`** — `online_client.send_audio_speech_request` (online); `offline_client.send_audio_speech_request` (Qwen-style offline) or `send_single_stage_tts_request` (single-stage offline). Add a new `send_*_request` in `runtime.py` when none fits; do not embed `omni.generate` or HTTP in tests
+- **Online and offline e2e go through `runtime.py`** — `openai_client.send_audio_speech_request` (online); `omni_runner_handler.send_audio_speech_request` (Qwen-style offline) or `send_single_stage_tts_request` (single-stage offline). Add a new `send_*_request` in `runtime.py` when none fits; do not embed `omni.generate` or HTTP in tests
 
 ## Phase 4: Async Chunk (Streaming)
 
@@ -404,29 +400,21 @@ See [vllm-omni-test skill](../vllm-omni-test/SKILL.md) § **Runtime send helpers
 
 ### Steps
 
-1. **Update the pipeline topology** so the producing stage declares its async
+1. **Update stage config YAML**:
-   handoff processor with `async_chunk_process_next_stage_input_func`.
-2. **Update the deploy YAML** to enable async chunk and configure the connector:
    ```yaml
    async_chunk: true
-
+   codec_chunk_frames: 25      # frames per chunk
-   connectors:
+   codec_left_context_frames: 25  # overlap for smooth boundaries
-     connector_of_shared_memory:
-       name: SharedMemoryConnector
-       extra:
-         codec_streaming: true
-         codec_chunk_frames: 25
-         codec_left_context_frames: 25
    ```
-3. **Implement chunk handling in Stage 1**:
+2. **Implement chunk handling in Stage 1**:
    - Accept partial input (chunk of codec codes)
    - Handle left context for smooth audio boundaries
    - Return partial audio in `OmniOutput`
-4. **Test streaming**:
+3. **Test streaming**:
    - Verify audio quality matches non-streaming output
    - Check for artifacts at chunk boundaries
    - Measure TTFA (time to first audio)
-5. **Update online serving** to support `stream=true` with PCM output
+4. **Update online serving** to support `stream=true` with PCM output
 
 ### Streaming Architecture
 
@@ -447,8 +435,7 @@ Stage 0 (AR)                    Stage 1 (Decoder)
 
 ### Deliverables
 
-- Updated pipeline async handoff processor and deploy config with
+- Updated stage config with async_chunk enabled
-  `async_chunk: true`
 - Smooth streaming audio without boundary artifacts
 - TTFA metrics
 
@@ -520,15 +507,14 @@ Use this checklist when integrating a new TTS model:
 - [ ] Config classes created with `model_type` registration
 - [ ] Stage 0 (AR) implemented and generates correct tokens
 - [ ] Stage 1 (Decoder) produces correct audio from tokens — dtype float32 for codec decoder
-- [ ] AR stage `max_num_seqs` ≥ 4 in the production deploy config unless the model has a tested lower limit
+- [ ] Stage 1 `max_num_seqs` ≥ 4 in production config (default 1 causes gaps under concurrency)
 - [ ] Optional dependency fallbacks handled at `load_weights()` time (torchaudio/soundfile/etc.)
 - [ ] Streaming: codec codes accumulated across AR steps (not reset per step)
 - [ ] Streaming: delta audio emitted per chunk, not full re-decoded waveform
 - [ ] Streaming: all `forward()` return paths emit `model_outputs`
 - [ ] Streaming: per-request state keyed by request ID (not shared across requests)
 - [ ] Streaming: codec tensors moved to codec decoder device before decode
-- [ ] Pipeline topology defined in `pipeline.py` and registered in `OMNI_PIPELINES`
+- [ ] Stage config YAML created
-- [ ] Deploy YAML created under `vllm_omni/deploy/`
 - [ ] `end2end.py` produces audio matching reference quality
 - [ ] README.md written
 
@@ -545,8 +531,7 @@ Use this checklist when integrating a new TTS model:
 - [ ] Documentation added (offline + online docs, nav, supported models)
 
 ### Phase 4: Async Chunk
-- [ ] Pipeline declares the async handoff processor
+- [ ] Stage config updated with `async_chunk: true`
-- [ ] Deploy config sets `async_chunk: true` and connector chunk parameters
 - [ ] Stage 1 handles partial chunks correctly
 - [ ] No audio artifacts at chunk boundaries
 - [ ] Streaming via API (`stream=true`) works
@@ -563,7 +548,7 @@ Use this checklist when integrating a new TTS model:
 - [ ] `pre-commit run --files <changed>` passes before every push
 - [ ] Every commit has `Signed-off-by` matching the author email (`git commit -s`)
 - [ ] `git config user.email` matches the email registered on your GitHub account
-- [ ] Hook list, failure table, and DCO recovery: [references/precommit-dco.md](references/precommit-dco.md)
+- [ ] Details and failure-recovery commands: [references/precommit-dco.md](references/precommit-dco.md)
 
 ## References
 

.claude/skills/add-tts-model/references/precommit-dco.md

@@ -30,18 +30,6 @@ but the changes are correct — stage the modified files and re-commit.
 | `ruff F841` | Variable extracted but never forwarded to model call | Remove the extraction or wire it through |
 | `ruff E402` | Import added below function definitions | Move to top-level import block |
 | `ruff format` | Line length, spacing, quote style | Accept auto-fix, stage, re-commit |
-| `check-spdx-header` | Missing header, or copyright still says `vLLM project` | Two-line Omni header (`vLLM-Omni project`); restage the rewrite |
-| `check-forbidden-imports` | Stdlib `re`/`base64`, pickle, Hugging Face Hub API, or direct Triton/TileLang | `import regex as re` and `pybase64`; use `vllm.transformers_utils.repo_utils`. Do **not** add the file to `allowed_files` without review |
-| `check-torch-cuda-call` | New `torch.cuda.*` call site | Use `current_omni_platform` / `OmniPlatform`; do not grow `ALLOWED_FILES` without review |
-| `check-tts-adapter-migration` | New `self._tts_model_type` branch in `serving_speech.py` | Put per-model logic in `tts_adapters/`. Lower `MAX_MODEL_TYPE_BRANCHES` when removing branches; do not raise it without review |
-| `check-mark` | New `tests/**/test_*.py` missing level/hardware mark, or direct `pytest.mark.H100` | Add `core_model`/`advanced_model`/… plus `cpu`/`cuda`/`hardware_test(`; SKU marks only via helpers |
-| `mypy-3.10` | Type error on changed `vllm_omni/` files | Fix the types; model trees are excluded. Extra versions: `pre-commit run --hook-stage manual mypy-3.12` |
-| `markdownlint-cli2` | Docs/README markdown lint | Accept auto-fix under `docs/` / `recipes/` / root README |
-| `check-buildkite` | Invalid `.buildkite/*.yml` | Fix the YAML; do not grow `SKIP_FILES` without review |
-| `shellcheck` | No `shellcheck` on PATH, or script warning | `apt-get`/`dnf`/`brew install shellcheck`, or `shellcheck.exe` on Windows PATH |
-
-Canonical policy, CI `SKIP` list, and allowlist locations:
-[docs/contributing/README.md](../../../docs/contributing/README.md#linting).
 
 ## DCO sign-off
 

.claude/skills/add-tts-model/references/single-stage-ar.md

@@ -86,51 +86,12 @@ class YourModelForCausalLM(nn.Module):
 - Handle both `"audio"` (incremental) and `"result"` (final combined) event
   types from upstream models.
 
-## Pipeline and deploy config
+## Stage config
 
-Define the fixed single-stage topology in `pipeline.py`:
+Single stage with `worker_type: ar`, `engine_output_type: audio`,
-
+`final_output: true`, `is_comprehension: true`, and `async_chunk: false` at
-```python
+the top level. Omitting any of these causes silent misclassification in the
-from vllm_omni.config.stage_config import (
+serving layer.
-    PipelineConfig,
-    StageExecutionType,
-    StagePipelineConfig,
-)
-
-YOUR_TTS_PIPELINE = PipelineConfig(
-    model_type="your_model_name",
-    default_deploy_config_name="your_model_name.yaml",
-    model_arch="YourModelForCausalLM",
-    stages=(
-        StagePipelineConfig(
-            stage_id=0,
-            model_stage="your_model_name",
-            execution_type=StageExecutionType.LLM_AR,
-            input_sources=(),
-            owns_tokenizer=True,
-            engine_output_type="audio",
-            final_output=True,
-            final_output_type="audio",
-        ),
-    ),
-)
-```
-
-Register it in `vllm_omni/config/pipeline_registry.py`. Keep placement and
-runtime sizing in `vllm_omni/deploy/your_model_name.yaml`:
-
-```yaml
-async_chunk: false
-
-stages:
-  - stage_id: 0
-    devices: "0"
-    max_num_seqs: 4
-    gpu_memory_utilization: 0.8
-```
-
-Do not move `execution_type`, output ownership, tokenizer ownership, or model
-architecture into the deploy YAML; those are pipeline topology.
 
 ## Lint discipline
 
@@ -140,5 +101,8 @@ pre-commit.
 
 ## Reference implementation
 
-Use `vllm_omni/model_executor/models/moss_tts_nano/pipeline.py` and
+Look for any single-stage AR model under
-`vllm_omni/deploy/moss_tts_nano.yaml` as the in-tree reference.
+`vllm_omni/model_executor/models/` — e.g. `moss_tts_nano/` when its
+integration lands. If none is in tree yet, follow the skeleton above and
+cross-check against the `is_comprehension: true` / `async_chunk: false`
+dispatch in `vllm_omni/entrypoints/openai/serving_speech.py`.

.claude/skills/find-simplifications/SKILL.md

@@ -1,135 +0,0 @@
----
-name: find-simplifications
-description: Find evidence-backed simplification candidates in vLLM-Omni and, when requested, turn them into focused proposals or code changes. Use for audits of dead, duplicated, speculative, over-generalized, unnecessarily defensive, or hand-rolled code. Use review-pr for ordinary correctness review and diffusion-perf-opt for performance-first optimization.
----
-
-# Find vLLM-Omni Simplifications
-
-Find a small number of well-proven ways to reduce code, state, APIs, or maintenance cost without weakening supported behavior. Treat this as an architecture and ownership audit, not a dead-code search or style cleanup.
-
-The default deliverable is a read-only report. Do not add TODOs, design documents, issues, or code unless the user asks for those changes.
-
-## Establish the Contract
-
-Before judging a candidate:
-
-1. Read the repository instructions supplied by the host.
-2. Read [`docs/design/architecture_overview.md`](../../../docs/design/architecture_overview.md).
-3. Select only the relevant active module and feature documents from [`docs/design/index.md`](../../../docs/design/index.md). Archived design pages are historical context, not current contracts.
-4. Identify the comparison base with `git merge-base HEAD origin/main`; do not assume the current branch represents the intended baseline.
-5. If tests or CI coverage may change, use [`vllm-omni-test`](../vllm-omni-test/SKILL.md) for test placement, markers, and commands.
-
-Separate an implementation accident from an intentional boundary. In particular:
-
-- vLLM-Omni extends and sometimes overrides upstream vLLM. A similar upstream implementation is evidence for reuse only after comparing API, lifecycle, and multi-stage semantics.
-- Registries, import strings, model configuration, deployment YAML, plugins, and optional dependencies create dynamic consumers that a static reference count can miss.
-- Hardware backends, model families, execution modes, and connector implementations may be intentionally parallel. Do not propose collapsing one only because it is inactive in the local environment.
-- Stage, replica, request, cache, and transport state can look redundant while representing different owners. Prove that two fields or barriers encode the same fact before merging them.
-- A shorter hot path is not a simplification if it adds synchronization, tensor copies, host materialization, recompilation, or a measurable quality/performance regression.
-
-## Strong Candidates
-
-A strong candidate removes a meaningful surface and has a clear replacement or deletion boundary. Examples include:
-
-- A public method, configuration option, registry entry, route, message field, or event has no production consumer and no compatibility obligation.
-- Tests or documentation are the only consumers of behavior that is not part of a supported contract.
-- Multiple layers mirror the same request, lifecycle, cache-validity, readiness, or topology state.
-- AR, diffusion, platform, or model-specific implementations duplicate a stable common operation and can share it without hiding performance-critical differences.
-- An inherited upstream vLLM primitive is reimplemented locally without an Omni-specific semantic difference.
-- Compatibility or fallback branches protect environments the project no longer supports.
-- Serialization, IPC, output formatting, or connector code materializes the same payload more than once or converts between equivalent representations.
-- A helper, abstraction, package split, or extension point has only one real implementation and adds more indirection than policy.
-- Hand-written infrastructure can be replaced by an existing project dependency, Python standard-library feature, PyTorch primitive, or upstream vLLM facility with net deletion and equal observability.
-- Tests maintain large mocks, snapshots, or expected data solely for an unused API or representation.
-
-Do not elevate typo fixes, formatting, isolated renames, or vague complexity complaints into simplification candidates. Bundle small cleanups only when they support one substantive removal.
-
-## Survey by Ownership Boundary
-
-For broad audits, cover the largest or riskiest production deltas first. Useful domains are:
-
-- `vllm_omni/engine/`: orchestration, stage pools, request state, output convergence, cancellation, and replica lifecycle.
-- `vllm_omni/worker/` and AR model execution: upstream vLLM overlays, scheduler/cache state, collective RPC, adapters, and weight loading.
-- `vllm_omni/diffusion/`: request/step execution, batching, output materialization, IPC, offload, parallelism, and model pipelines.
-- `vllm_omni/entrypoints/`: duplicated validation/defaulting, offline/online parity, streaming, and route compatibility.
-- Connectors and distributed code: ownership, copies, barriers, handles, cleanup, and failure propagation across process/device/node boundaries.
-- Platform backends: genuine vendor differences versus copied control flow that has drifted.
-- Configuration, registries, recipes, and examples: duplicate sources of truth and options that never reach a runtime consumer.
-- Tests and CI: redundant fixtures, mocks of obsolete contracts, and expensive coverage that can move to a cheaper level without losing the guarded behavior.
-
-When the user explicitly requests a repository-wide or many-candidate audit and parallel agent work is available and authorized, divide these domains among agents and require the same evidence format from each. Otherwise survey them sequentially.
-
-## Prove or Reject Each Candidate
-
-Use `rg` first, then read every relevant caller and implementation. Search exact Python symbols plus serialized names, route paths, configuration keys, registry strings, YAML values, and CLI spellings.
-
-Classify evidence:
-
-- **Production:** `vllm_omni/`, runtime configuration, registries, serving entrypoints, connectors, recipes used for supported deployments, and executable loader paths.
-- **Validation/support:** `tests/`, benchmarks, examples, docs, CI, and developer tooling.
-- **External/dynamic:** upstream vLLM contracts, third-party plugins, model repositories, optional backends, and string-loaded objects. State what was checked and what remains unknown.
-
-For each candidate, answer:
-
-1. What exact surface would disappear or merge?
-2. Who owns the state or behavior today?
-3. Which production consumers exist, including dynamic consumers?
-4. What supported behavior, compatibility, performance, memory, or observability could change?
-5. What is the smallest validation that would catch a bad simplification?
-6. Is the result net deletion after tests, adapters, migration glue, and documentation are counted?
-
-Reject or downgrade the candidate when a live consumer exists and removal is actually a feature decision; when an active design contract explains the separation; when the change only moves complexity; or when hardware/model evidence is unavailable for a performance-sensitive claim.
-
-## Lifecycle and Data-Movement Audits
-
-For asynchronous or distributed code, map the ownership graph before recommending deletion:
-
-```text
-request admission -> orchestrator -> stage/replica -> worker/device
-  -> connector or IPC boundary -> output aggregation -> terminal cleanup
-```
-
-Associate every lock, event, sentinel, queue, readiness flag, timeout, abort path, and cache marker with an owner and transition. Coalesce mechanisms only when they represent the same state and have the same failure boundary. Preserve separate mechanisms when they protect publication versus rollback, local versus remote ownership, first-terminal-outcome arbitration, or cleanup after partial failure.
-
-For tensors and media, also trace device, process, and ownership transitions. Count GPU-to-CPU copies, shared-memory/object-store handles, serialization, decode/encode steps, and unlink/release responsibility. Prefer eliminating a transfer or representation over merely hiding it behind a helper.
-
-## Dependency Replacements
-
-Prefer facilities already available in the repository before adding a dependency. For a proposed replacement:
-
-- Name the exact local behavior it covers and any semantics that remain in glue code.
-- Check maintenance, license, optional-dependency boundaries, platform support, and transitive footprint.
-- Compare net deletion, test burden, import/startup cost, and runtime behavior.
-- Benchmark when the replaced code is on a model, scheduler, connector, serialization, or serving hot path.
-
-A wrapper that retains the same state machine or conversion logic is not a simplification.
-
-## Deliver the Result
-
-Report only candidates supported by concrete evidence. For each one include:
-
-- **Surface:** files and symbols involved.
-- **Evidence:** production consumers found, dynamic paths checked, and relevant design contract.
-- **Simplification:** exactly what to remove, merge, inline, or reuse.
-- **Benefit:** deleted API/state/code, reduced data movement, or reduced maintenance/test burden.
-- **Risk and validation:** behavior that could change and the narrowest useful tests or measurements.
-- **Confidence:** high or medium, with the unresolved evidence named for medium-confidence items.
-
-Also list representative ideas that were rejected when that prevents repeated investigation. Prefer three high-confidence candidates over a long speculative inventory.
-
-If the user asks for a durable design proposal, place it under `docs/design/feature/` and register it in `docs/design/index.md`; use current design-document conventions and keep implementation details proportional to the decision. If the user asks for implementation, make the smallest coherent change, update affected tests/docs, and preserve public compatibility unless the user explicitly accepts a breaking change.
-
-## Folding Work from Another Branch
-
-Diff the source branch against its own merge base with `origin/main`, not against the current feature branch. Port only independently justified candidates. Consolidate overlaps into the candidate with stronger evidence, and do not preserve a candidate count by carrying weaker duplicates.
-
-## Validation
-
-Match validation to the outgoing diff:
-
-- Documentation or skill changes: run the repository skill validator when available, check links/paths, search for stale source-specific terminology, and run `git diff --check`.
-- Python changes: run the narrowest relevant pytest coverage through the repository environment, then lint/format the changed files.
-- Distributed, model, hardware, performance, or accuracy changes: run available CPU/static checks and state the exact GPU/NPU/model gap; never present simulated evidence as device validation.
-- Before submission, use [`precheck-pr`](../precheck-pr/SKILL.md) for the branch-level self-review.
-
-When summarizing, distinguish checks that passed from checks that were not runnable, and state the snapshot or branch against which the audit was performed.

.claude/skills/precheck-pr/SKILL.md

@@ -1,6 +1,6 @@
 ---
 name: precheck-pr
-description: Self-check your branch before creating a PR — catch dead code, prevent new model-specific Python examples, verify accuracy/perf claims, validate PR title format, and confirm merge readiness. Use when the user says "precheck", "self review", "pre-submit check", or "check my PR before I open it." Never posts to GitHub.
+description: Self-check your branch before creating a PR — catch dead code, verify accuracy/perf claims, validate PR title format, and confirm merge readiness. Use when the user says "precheck", "self review", "pre-submit check", or "check my PR before I open it." Never posts to GitHub.
 ---
 
 # PR Pre-Check
@@ -65,12 +65,6 @@ Ask: "Quick mode or full mode?" Then walk the checklist for the detected PR type
 
 **Also run the Code-Quality sweep on every PR**, regardless of type or mode: the five diff-scoped checks in [references/code-quality.md](references/code-quality.md) — kwargs fragility, broad-except swallow, `Any`/wrong type hints, hot-path `.clone()`/`deepcopy`, and event-loop blocking — plus the advisory conventions (log level, structured logging, synchronization, cleanup, dependencies, naming) in [checklists.md](references/checklists.md). These count only lines the PR *adds* — the pre-existing backlog across the repo is out of scope.
 
-**Also run the Examples-Policy check on every PR** using [references/examples-policy.md](references/examples-policy.md). Inspect only Python paths introduced by the diff. Treat a new model-, checkpoint-, vendor-, or family-specific Python example as blocking; do not report pre-existing example debt.
-
-**Also run the Simplification pass on every PR** using [find-simplifications](../find-simplifications/SKILL.md). Keep it diff-scoped: inspect added or modified public APIs, state, abstractions, fallback and compatibility paths, data movement, and only the adjacent ownership needed to prove a candidate. Do not expand the pass into a repository-wide audit. Zero candidates is a valid pass; report only evidence-backed opportunities. Mark a candidate as blocking only when it proves a correctness or merge-readiness problem, such as newly unreachable code. Otherwise report it as a warning.
-
-**Also confirm local pre-commit gates** from [docs/contributing/README.md](../../../docs/contributing/README.md#linting). GitHub Actions `SKIP` does not prove they passed. Full new-hook list: Omni SPDX (`vLLM-Omni project`); forbidden imports (stdlib `re`/`base64`, pickle, Hugging Face Hub API, Triton/TileLang); no new `torch.cuda` call sites; shellcheck (macOS/Windows must install the binary); mypy-3.10; test files have CI level + hardware marks; TTS `_tts_model_type` branches in `serving_speech.py` must not increase; markdownlint on `docs/`; Buildkite YAML schema. ✗ for growing `allowed_files` / `ALLOWED_FILES` or raising `MAX_MODEL_TYPE_BRANCHES` without review.
-
 ### Step 5: Print Report
 
 ```
@@ -83,8 +77,6 @@ Pre-check report for <branch>
   ─────────────────  ──────
   PR title format    ✓
   Code quality       ⚠ 1 broad except, 2 Any hints
-  Examples policy    ✗ new model-specific Python example
-  Simplification     ⚠ helper duplicates an existing owner
   PR desc integrity  ✓
   Registry/config    ✓
   Dead code          ⚠ 2 warnings

.claude/skills/precheck-pr/references/checklists.md

@@ -47,19 +47,6 @@ Diff-scoped sweep for five fragility patterns that are pervasive in the existing
 
 Roll the ⚠/✗ counts into the report as a single **Code quality** dimension row.
 
----
-
-## All PRs: Examples Policy (run on every PR, quick & full)
-
-Inspect Python files introduced under `examples/` using
-[examples-policy.md](examples-policy.md). This is a diff-scoped ratchet: existing
-model-specific examples may be edited or removed without triggering the check.
-
-- [ ] **No new model-specific Python examples:** the diff adds, copies, or renames no Python file whose path or behavior is specific to one model, checkpoint, vendor, or model family. ✗ for any violation. Route inference through a shared task/protocol entrypoint, model request behavior through `vllm_omni/model_extras`, and commands through task docs or `recipes/`.
-- [ ] **New generic example code is justified:** any new task-neutral Python file under `examples/` is an actual user-facing entrypoint rather than reusable production logic, a benchmark, an app, a setup tool, or a reproducer. ⚠ when the placement is questionable.
-
-Roll the result into the report as a separate **Examples policy** dimension.
-
 ### Conventions (eyeball — no grep)
 
 Judgment calls that don't grep cleanly — apply them while reading the diff, not as a mechanical check:
@@ -164,7 +151,6 @@ Diffusion models live under `vllm_omni/diffusion/`. In addition to the New Model
 - [ ] **PR body matches diff:** description doesn't claim changes not in the diff
 - [ ] **Branch is rebased:** `git merge-base HEAD origin/main` is recent, no merge conflicts
 - [ ] **CI gates passing:** DCO, pre-commit, build — check `gh pr view --json statusCheckRollup` if a PR already exists
-- [ ] **Local pre-commit gates:** full list in [Linting](../../../../docs/contributing/README.md#linting). Blocking when the diff introduces: stale `vLLM project` SPDX; stdlib `re`/`base64` or pickle in library code; new `torch.cuda` call sites; `tests/**/test_*.py` without level+hardware marks; new `_tts_model_type` branches in `serving_speech.py`; invalid `.buildkite` YAML. macOS/Windows need a native `shellcheck`. ✗ for growing `CHECK_IMPORTS[*].allowed_files`, `ALLOWED_FILES`, or `MAX_MODEL_TYPE_BRANCHES` without review.
 
 ### Full (adds)
 

.claude/skills/precheck-pr/references/examples-policy.md

@@ -1,68 +0,0 @@
-# Examples Python Policy
-
-Prevent new model-specific Python examples without forcing cleanup of the
-existing backlog. Run this check on every PR in quick and full modes.
-
-## Scope
-
-Set the merge base as in the main workflow, then list Python paths introduced
-by the PR, including additions, copies, and renames:
-
-```bash
-BASE="$(git merge-base HEAD origin/main)"
-git diff --name-status --diff-filter=ACR --find-renames --find-copies \
-  "${BASE}"...HEAD -- 'examples/**/*.py'
-```
-
-Inspect destination paths for copies and renames. Existing model-specific files
-that are only modified or deleted are grandfathered and must not be reported.
-
-## Blocking classification
-
-Mark ✗ when a newly introduced Python file does any of the following:
-
-- Names or scopes a path to one model, checkpoint, vendor, or model family.
-- Hard-codes a model identifier, model-specific prompt, request shape, output
-  conversion, or launch configuration behind an otherwise generic filename.
-- Duplicates an existing task/protocol runner to demonstrate one model.
-- Adds a per-model helper below a generic task hub such as `text_to_speech/`.
-
-Do not infer safety from a generic filename alone. For example, a file named
-`text_to_audio.py` is still model-specific if it only implements one model's
-contract.
-
-## Acceptable additions
-
-A new Python example may pass when it is a genuinely model-neutral,
-user-facing task or protocol entrypoint. It must accept model selection through
-configuration and keep model-specific behavior out of the script.
-
-Use ⚠ when the file is model-neutral but reusable logic appears misplaced under
-`examples/`. Route code by purpose:
-
-| Code purpose | Preferred location |
-|---|---|
-| Model prompt, defaults, request/output adaptation | `vllm_omni/model_extras/` or production model modules |
-| Runnable model command and validation evidence | Task documentation or `recipes/` |
-| UI or interactive application | `apps/` |
-| Benchmark or evaluator | `benchmarks/` |
-| Regression reproducer | `tests/` |
-| Download, conversion, or setup utility | `tools/` |
-
-If a model cannot use an existing shared runner, report the missing generic
-capability. Do not solve the gap by adding a model-specific Python example.
-
-## Report format
-
-Add one row to the pre-check report:
-
-```text
-Examples policy    ✓ no new Python example paths
-Examples policy    ⚠ generic helper belongs in tools/
-Examples policy    ✗ examples/offline_inference/new_model/end2end.py is model-specific
-Examples policy    ✗ examples/offline_inference/x_to_y_model_name.py is model-specific
-```
-
-For a blocker, name the path, the model-specific behavior, and the appropriate
-shared runner or destination. Do not propose deleting unrelated existing
-examples as part of the PR.

.claude/skills/production-add-diffusion-model/SKILL.md

@@ -1,458 +0,0 @@
----
-name: production-add-diffusion-model
-description: Productionize a vLLM-Omni diffusion model after its Day-0 vertical slice works. Use when work requires official API parity and input limits, feature-combination evidence, online FP8, distributed layerwise offload, per-request Cache-DiT, CUDA/ROCm/NPU/XPU recipes, faster or sparse attention, fused operators, USP or disaggregation analysis, continuous batching and abort handling, long-running RPS stability, or accuracy/performance/reliability CI. For initial architecture porting, registry wiring, and basic weight loading, use add-diffusion-model first.
----
-
-# Productionizing a Diffusion Model
-
-## Scope
-
-Use this skill to move a working model integration from Day-0 support to a
-measured, fail-closed production deployment. It is deliberately separate from
-[`add-diffusion-model`](../add-diffusion-model/SKILL.md); do not modify or copy
-that skill when productionizing a model. If registry, pipeline, or basic loader
-work is incomplete, read that skill first and finish the Day-0 vertical slice.
-
-MiniMax-H3 [PR #5691](https://github.com/vllm-project/vllm-omni/pull/5691)
-is the merged Day-0 case study. Its open optimization
-[issue #5700](https://github.com/vllm-project/vllm-omni/issues/5700) is an
-actively maintained roadmap and evidence map, not a list of already supported
-features. It can lag merged implementation: read its current revision, cited
-PRs, target source, and maintained recipes, then leave every unverified row
-`not tested` until the target revision has scoped evidence.
-
-Read these references before the corresponding gate:
-
-- [API and hardware recipes](references/api-and-recipes.md)
-- [Production feature patterns](references/feature-patterns.md)
-- [Performance patterns and code examples](references/performance-patterns.md)
-- [Serving, soak, and CI validation](references/production-validation.md)
-
-## Non-negotiable evidence contract
-
-Create a matrix whose row key is at least:
-
-```text
-task x API mode x execution mode/capacity x shape/schedule x attention backend x
-packed layout x cache policy x quantization x offload mode x topology x
-hardware x dtype x output representation/transport
-```
-
-Use exactly these states:
-
-| State | Meaning |
-|---|---|
-| `validated` | The exact row passed correctness and measured deployment tests, with reproducible artifacts. |
-| `limited` | A bounded subset passed; the limitation and rejection/fallback behavior are explicit. |
-| `unsupported` | A stable architectural/platform restriction is proven and fails early with an actionable error. |
-| `not tested` | No adequate evidence exists. This is the default. |
-
-Never infer compatibility from import success, server startup, another task,
-another card, or another topology. Record the model/checkpoint revision,
-vLLM-Omni commit, commands, request assets and hashes, seed/schedule/shape,
-raw output artifacts, environment, and result for every `validated` row.
-
-## Workflow
-
-Follow the gates in order. Keep a dense BF16 single-device path as the
-correctness oracle until every advertised fast path has parity evidence.
-
-### Gate 0: Freeze the official contract
-
-Pin the official implementation and checkpoint revision. Build a matrix for:
-
-- constructor/call arguments, defaults, scheduler and sigma schedule;
-- task-to-source modality/count/order rules;
-- prompt, MIME, bytes, pixels, frames, duration, FPS, steps, seed, guidance,
-  output count, and output format limits;
-- offline API and every public sync/async serving endpoint;
-- checkpoint partitions and task selection.
-
-Normalize offline, sync, and async requests through one validation contract.
-Reject an invalid task/source combination before downloads, upload persistence,
-temporary files, async job creation, or engine submission. Use 400 for invalid
-semantics and 413 for payload limits; clean partial resources on every failure.
-See [API and hardware recipes](references/api-and-recipes.md).
-
-### Gate 1: Re-prove correctness under production shapes
-
-1. Make loading strict: track expected, loaded, unexpected, and intentionally
-   ignored tensors. A fused destination is complete only after every source
-   shard arrives. Missing retained parameters abort startup.
-2. Fix reference revision, asset hashes, seed, schedule, and shape. Compare
-   component outputs, intermediate denoise latents, and final artifacts.
-3. Test unequal packed samples. Verify `cu_seqlens`, padding exclusion,
-   modality/CFG ownership, split/gather boundaries, and one output per request.
-4. Test partial process groups with non-member ranks and group size smaller
-   than world size. Do not construct a private group per component or request.
-5. Run negative API cases before positive E2E tests to prove side-effect-free
-   rejection and cleanup.
-
-### Gate 2: Wire shared fast primitives
-
-Prefer explicit, testable vLLM-Omni/vLLM operators with native fallbacks.
-`torch.compile` remains a separately validated regional optimization; it is
-not a substitute for correct operator selection or lifecycle-safe boundaries.
-
-#### RMSNorm + RoPE baseline
-
-Use the shared cross-platform operators, preserving the official Q/K norm and
-RoPE order:
-
-```python
-from vllm_omni.diffusion.layers.norm import RMSNorm
-from vllm_omni.diffusion.layers.rope import (
-    RotaryEmbedding,
-    apply_rope_to_qk,
-)
-
-self.norm_q = RMSNorm(head_dim, eps=eps)
-self.norm_k = RMSNorm(head_dim, eps=eps)
-self.rope = RotaryEmbedding(is_neox_style=False)
-
-query = self.norm_q(query)  # [B, S, Hq, D] or packed [T, Hq, D]
-key = self.norm_k(key)      # [B, S, Hkv, D] or packed [T, Hkv, D]
-query, key = apply_rope_to_qk(
-    self.rope,
-    query,
-    key,
-    (cos, sin),
-)
-```
-
-Verify `is_neox_style`, `half_head_dim`, partial rotary dimensions, cos/sin
-layout, and packed row ownership against the official model. `False` means
-interleaved/GPT-J style; do not assume the default is NeoX. Shared RMSNorm and
-RoPE are two dispatched operators, not automatically one fused kernel.
-
-For packed non-interleaved RoPE, use the shared fused boundary when its contract
-matches the model:
-
-```python
-from vllm_omni.diffusion.layers.fused_qk_norm_rope import fused_qk_norm_rope
-
-query, key = fused_qk_norm_rope(
-    query,                    # [T, Hq, D]
-    key,                      # [T, Hkv, D]
-    self.norm_q.weight,       # [D]
-    self.norm_k.weight,       # [D]
-    rope_table,               # [T, rotary_dim] = [cos | sin]
-    self.norm_q.variance_epsilon,
-)
-```
-
-The current CUDA fast path specializes BF16 `head_dim=128`, `rotary_dim=96`;
-the public function keeps an eager fallback for unsupported inputs. Verify the
-packed row/frequency contract and trace the actual fast path. Do not reshape an
-incompatible official RoPE layout merely to enter this kernel.
-
-MiniMax-H3-style partial NeoX RoPE rotates a prefix and passes the suffix
-through:
-
-```python
-self.q_norm = RMSNorm(head_dim, eps=eps, dtype=torch.bfloat16)
-self.k_norm = RMSNorm(head_dim, eps=eps, dtype=torch.bfloat16)
-self.rope = RotaryEmbedding(is_neox_style=True, half_head_dim=False)
-
-def apply_partial_rope(self, x, freqs, rot_dim):
-    x_rot, x_pass = x[..., :rot_dim], x[..., rot_dim:]
-    cos, sin = torch.cos(freqs).to(x.dtype), torch.sin(freqs).to(x.dtype)
-    return torch.cat((self.rope(x_rot, cos, sin), x_pass), dim=-1)
-```
-
-The partial-RoPE snippet assumes `import torch` and lives on the owning
-attention module; adapt names and validated frequency shapes rather than
-copying it at module scope.
-
-Call the modules normally so `CustomOp` selects CUDA/HIP/NPU/XPU/native
-implementations. Do not call `forward_cuda()` directly. Compare the shared
-path with native BF16 at operator, block, denoise-trajectory, and artifact
-levels before claiming either accuracy or speed.
-
-#### Faster BF16 attention and fused projections
-
-Use local TP head counts and role-aware shared attention:
-
-```python
-from vllm.model_executor.layers.linear import QKVParallelLinear
-from vllm_omni.diffusion.attention.layer import Attention
-
-self.qkv_proj = QKVParallelLinear(
-    hidden_size=hidden_size,
-    head_size=head_dim,
-    total_num_heads=num_heads,
-    total_num_kv_heads=num_kv_heads,
-    bias=False,
-    quant_config=quant_config,
-    prefix=f"{prefix}.qkv_proj",
-)
-self.attn = Attention(
-    num_heads=self.qkv_proj.num_heads,
-    num_kv_heads=self.qkv_proj.num_kv_heads,
-    head_size=head_dim,
-    softmax_scale=head_dim**-0.5,
-    causal=False,
-    qkv_layout="BSND",
-    role="self",
-    prefix=prefix,
-)
-```
-
-Prove the selected backend from runtime logs/trace and compare it with dense
-SDPA. Packed metadata must express real sample boundaries; do not build masks
-that the selected backend ignores. A backend that is faster for one shape or
-platform is not a global default. See [performance patterns](references/performance-patterns.md)
-for QKV, SwiGLU, AdaLN, sparse attention, redundancy, and fusion examples.
-
-### Gate 3: Add online FP8 without breaking the loader
-
-Route the active config to each component, then pass a stable runtime prefix
-and config into every quantizable vLLM linear:
-
-```python
-def resolve_component_quant_config(quant_config, component):
-    return quant_config.resolve(component) if hasattr(quant_config, "resolve") else quant_config
-
-transformer_quant_config = resolve_component_quant_config(
-    od_config.quantization_config,
-    "transformer",
-)
-self.transformer = MyDiT(
-    od_config,
-    quant_config=transformer_quant_config,
-    prefix="transformer",
-)
-
-self.qkv_proj = QKVParallelLinear(
-    hidden_size=hidden_size,
-    head_size=head_dim,
-    total_num_heads=num_heads,
-    total_num_kv_heads=num_kv_heads,
-    quant_config=quant_config,
-    prefix=f"{prefix}.qkv_proj",
-)
-```
-
-Use `build_quant_config()` for direct construction/testing:
-
-```python
-from vllm_omni.quantization import build_quant_config
-
-quant_config = build_quant_config({
-    "transformer": {"method": "fp8"},
-    "text_encoder": None,
-    "vae": None,
-    "default": None,
-})
-```
-
-Preserve each parameter's vLLM `weight_loader`; perform checkpoint layout
-conversion before invoking it. Account for every QKV/gate-up source shard.
-Keep precision-sensitive norm/modulation/embedders in BF16 unless separately
-proven, using the current `safe_quant_config` pattern where applicable.
-
-Validate resident DiT FP8 first: prove which modules are quantized, HBM saved,
-BF16-vs-FP8 trajectory/artifact quality, latency, and throughput. Text encoder,
-VAE, other hardware, TP/cache/DLO combinations are separate rows.
-Pre-quantized, pruned, rotated, or adapter-modified checkpoints are separate
-model/loading lanes; compare them with the released BF16 model without calling
-the result a pure quantization ablation. Read
-[`../quantization/SKILL.md`](../quantization/SKILL.md) if present and prefer the
-target revision's implementation/docs when sibling guidance differs.
-
-### Gate 4: Make every component offloadable
-
-Declare topology instead of relying on heuristic discovery:
-
-```python
-from typing import ClassVar
-
-import torch.nn as nn
-
-from vllm_omni.diffusion.models.interface import SupportsComponentDiscovery
-from vllm_omni.diffusion.offloader import OffloadPlan
-
-class MyPipeline(nn.Module, SupportsComponentDiscovery):
-    _dit_modules: ClassVar[list[str]] = ["transformer"]
-    _encoder_modules: ClassVar[list[str]] = ["text_encoder"]
-    _vae_modules: ClassVar[list[str]] = ["vae"]
-    _resident_modules: ClassVar[list[str]] = []
-    _offload_plan: ClassVar[OffloadPlan] = OffloadPlan(
-        on_demand_component_paths=frozenset({"text_encoder", "vae"}),
-        block_attrs={"transformer": ("blocks",)},
-        offload_submodules={"token_refiner": "blocks"},
-        resident_dit_paths=frozenset({"transformer"}),
-        encoder_block_attrs={"text_encoder": ("encoder.layers",)},
-    )
-```
-
-Add a test that resolves every dotted component/block path and fails if it is
-missing or is not the expected module/container; discovery may otherwise warn
-and skip. Validate ordinary layerwise offload and distributed layerwise
-offload (DLO) independently.
-
-Keep component lifecycle and host-weight storage separate. `OffloadPlan`
-declares what can be streamed; the diffusion loader owns any `HostWeightPlan`
-and hands the exact prevalidated plan to DLO. Do not make the offloader rescan
-checkpoint files or duplicate loader name/shape/dtype decisions. Direct mmap is
-a fail-closed optimization: current preflight requires TP1, no HSDP, no online
-quantization, complete bindings, and represented transforms. Otherwise the
-ordinary loader remains authoritative.
-
-If the target revision supports Host Weight Runtime (HWR), treat it as a
-separate, exact-identity startup/cache path rather than generic offload or
-zero-copy execution. Qualify `preferred` population/fallback and `required`
-consume-only failure independently; verify immutable manifests, final-layout
-restore, lease cleanup, corruption/quarantine, and node-local sharing. Current
-model contracts may be BF16 no-AllGather-only, so reject AllGather, online
-quantization, HSDP, LoRA/adapted weights, and unrecognized load formats unless
-the exact target revision explicitly adds them.
-
-On small-HBM cards, record load/materialization peak, resident peak, encode,
-each denoise wave, decode, transient per-rank HBM, host PSS, H2D, and collective
-time. Test DLO AllGather and `--dlo-no-use-allgather` as different deployments.
-Sweep the supported resident-block policies and publish the non-dominated
-latency-HBM frontier; a single minimum-memory or fastest point is not the DLO
-trade-off. Keep request-specific preparation replica-local. Only weight
-materialization may enter the declared DLO collective group; never use WORLD
-for per-request work in a DP deployment.
-Direct checkpoint mmap preflight is limited to TP1 without HSDP or online
-quantization; TP>1 falls back to the ordinary TP-aware loader and can still feed
-DLO AllGather or no-AllGather after scoped E2E. HSDP remains incompatible with
-DLO AllGather. A target revision may allow finalized per-tensor online FP8
-through the ordinary loader and AllGather; keep every other online quantizer
-fail-closed, and require model/card/topology E2E before promotion.
-Use deploy YAML for DP under `--omni`; vLLM DP CLI flags are rejected.
-
-### Gate 5: Add request-scoped acceleration safely
-
-Server-wide Cache-DiT support does not imply a per-request quality API. Only
-add request policy after defining calibrated quality tiers and a safe hook
-lifecycle using `SupportsRequestScopedCacheDiT`, `CacheDiTRequestSpec`, and
-`RequestScopedCacheDiTRuntime`. Validate alternating and concurrent quality
-tiers, real cache hits, refresh/reset, success, error, disconnect, abort, and
-the next uncached request. Do not batch different quality policies unless the
-batch compatibility key and hook ownership make that safe.
-
-Then test Cache-DiT against FP8, DLO, TP/SP, compile, and batching one
-combination at a time. See [production feature patterns](references/feature-patterns.md)
-for the lifecycle skeleton and compatibility matrix.
-
-### Gate 6: Optimize only from profiles
-
-Freeze one canonical workload manifest before profiling. Keep lossless runtime
-A/Bs, accelerated paths with numerical/quality trade-offs, and production
-topology studies in separate result lanes. Reconcile the client boundary as
-queue + encode + denoise + video/audio decode + output transport/codec +
-residual; do not hide a material residual in another stage.
-
-Profile stage time, GPU kernels, CPU/GPU synchronization, allocations, H2D,
-output payload/copies, and collectives before choosing work. Apply and A/B one
-change at a time:
-
-1. Remove synchronization and redundant denoise-loop work: hoist static prompt
-   conditioning/token refinement, RoPE tables, masks/metadata, bounded AdaLN
-   schedule projections, and reference scans; remove per-layer `.item()` and
-   unused masks/allocations.
-2. Select the fastest correct BF16 attention backend for each role and shape.
-3. Prefer explicit fused QKV, gate-up + `SiluAndMul`, RMSNorm/RoPE, AdaLN, and
-   layout/residual operators with guarded native fallbacks. Preserve every
-   materialized dtype/rounding boundary consumed by the unfused graph.
-4. Enable sparse attention only when platform, shape, topology, realized
-   sparsity, quality, fallback, and end-to-end speedup are all proven.
-5. Profile USP all-to-all count/bytes, packing, contiguous copies, overlap, and
-   scaling efficiency. Audit replicated cross-attention before using
-   `skip_sequence_parallel`. Qualify regular and accelerated Ulysses transport
-   independently; activation logs, JIT/readiness time, workspace growth,
-   stream ownership, maximum-shape warmup, and numerical drift are gates.
-   Preserve the checkpoint's Q:KV head ratio when padding GQA for Ulysses, and
-   validate strided QKV staging rather than assuming packed contiguity.
-6. Evaluate text-encoder or VAE disaggregation only if stage share, transfer
-   volume, reuse/concurrency, and independent scaling justify it. Tiling,
-   offload, or patch parallelism is not disaggregation.
-
-Do not combine independent optimizations into one performance claim. Preserve
-raw before/after runs with fixed workload and warmup policy.
-
-### Gate 7: Make serving interruptible and load-stable
-
-Treat request-mode batching and step-wise execution as separate capabilities:
-
-- request batching requires `supports_request_batch = True` and
-  `DiffusionRequestBatch -> list[DiffusionOutput]`, exactly one output per
-  request;
-- step execution requires the complete `SupportsStepExecution` contract:
-  `prepare_encode`, `denoise_step`, `step_scheduler`, and `post_decode`.
-
-Keep all mutable scheduler, RNG, latent, mask, cache, and step state request
-scoped. First validate `--step-execution --max-num-seqs 1` and step-boundary
-abort; only then test heterogeneous multi-request waves. Step continuous
-batching is experimental and is not automatically compatible with Cache-DiT.
-Before recommending it, predeclare a useful case and success threshold, then
-A/B it against request mode under the same arrival process. Structural support
-or a generic bridge is not latency, throughput, or cancellation evidence for a
-model.
-
-Test queued and in-flight abort, client disconnect, timeout, OOM, worker error,
-and restart. Require one terminal result, idempotent cleanup, no post-decode
-after abort, no cache/temp/VRAM/request-ID leak, and a successful next request.
-Inject late worker results/exceptions after cancellation and prove the result
-pump stays alive instead of raising `InvalidStateError`. For large batched or
-long-window video outputs, freeze dtype, range, layout, contiguity, ownership,
-payload size, serializer/codec limits, and offline-versus-HTTP semantics.
-Validate device-side output preparation, D2H/IPC or shared handles, remote
-encoding, event-loop responsiveness, and client materialization as separate
-stages, including abort and consumer failure cleanup.
-A model/VAE callback that publishes ordered chunks is only a producer contract;
-it is not transport, backpressure, cancellation, public streaming, or
-time-to-first-frame evidence. Distinguish complete-response faster-than-playback
-(`client E2E / output duration <= 1`) from streaming and report first-chunk,
-steady cadence, finalization, and complete-artifact latency separately.
-Run below-, near-, and above-saturation mixed-RPS soaks and report success/error
-rate, throughput, queue time, memory slope, temp growth, and worker health.
-Report p50/p95/p99 only from a declared arrival model with enough measured
-requests for the claimed percentile. See
-[serving and CI validation](references/production-validation.md).
-
-### Gate 8: Publish recipes and four independent CI tracks
-
-For every target vendor/card/topology, publish exact environment, serve/deploy
-command, one complete JSON or multipart curl per supported task, expected
-output checks, benchmark command, warmup/repeat/concurrency policy, raw metrics,
-per-rank HBM/host PSS, accuracy artifact, and limitations. Mark each
-CUDA/ROCm/NPU/XPU row `validated`, `unsupported`, or `not tested`; never inherit
-support from the platform abstraction or another card. Re-run DLO on each
-small-HBM recipe.
-
-Maintain separate CI gates:
-
-1. Function: strict load, API/offline positive and negative contracts.
-2. Accuracy: fixed reference revision/assets/seed/schedule and scoped tolerance.
-3. Performance: fixed best deployment plus memory-constrained/DLO row, raw JSON,
-   explicit warmup, enough samples for each claimed statistic, and owned
-   regression thresholds.
-4. Reliability: long mixed-RPS soak, abort/disconnect/fault cleanup, memory and
-   temporary-resource trend.
-
-## Production Definition of Done
-
-A model is production-ready only when:
-
-- every official advertised task passes API parity, offline/online parity,
-  strict loading, fixed-reference quality, and negative validation;
-- the recommended deployment row passes correctness, isolation, abort/error
-  cleanup, benchmark, and soak gates on named hardware;
-- online FP8 and DLO are each implemented and either validated in scoped rows
-  or explicitly limited/unsupported with fail-fast behavior;
-- per-request cache quality tiers have measured speed/quality frontiers and
-  request isolation, if advertised;
-- CUDA/ROCm/NPU/XPU recipes describe evidence rather than inferred support;
-- Function, Accuracy, Performance, and Reliability CI have owners, artifacts,
-  and actionable failure output;
-- accelerated paths identify every precision/quality trade-off, and the raw
-  offline plus online encoded output contracts are versioned and tested;
-- all unverified task/backend/cache/quant/offload/topology/hardware combinations
-  remain `not tested` and are not presented as supported.
-
-Day-0 support is still a valid milestone, but title, docs, and matrix must say
-which representative task passed and which production gates remain.

.claude/skills/production-add-diffusion-model/agents/openai.yaml

@@ -1,4 +0,0 @@
-interface:
-  display_name: "Production Add Diffusion Model"
-  short_description: "Productionize vLLM-Omni diffusion models"
-  default_prompt: "Use $production-add-diffusion-model to take this integrated diffusion model from Day-0 support to evidence-backed production readiness."

.claude/skills/production-add-diffusion-model/references/api-and-recipes.md

@@ -1,315 +0,0 @@
-# API Contract and Hardware Recipes
-
-## Contents
-
-1. [Freeze the upstream contract](#freeze-the-upstream-contract)
-2. [Validate before side effects](#validate-before-side-effects)
-3. [Task request templates](#task-request-templates)
-4. [Serving benchmark template](#serving-benchmark-template)
-5. [Hardware recipe contract](#hardware-recipe-contract)
-6. [Best deployment selection](#best-deployment-selection)
-
-## Freeze the upstream contract
-
-Pin three revisions independently:
-
-- official source implementation commit or package version;
-- checkpoint/model-card revision and partition;
-- vLLM-Omni target commit.
-
-Build a table before editing code. One row is one official task/partition.
-
-| Field | Required content |
-|---|---|
-| Task and partition | Official task name, checkpoint subdirectory, task selection rules |
-| Call contract | Constructor and call arguments, types, required/optional, official defaults |
-| Prompt/input | Prompt count/length, modality, MIME, order, count, decoded shape |
-| Size/time | Width/height alignment, pixels, frames, duration, FPS, start offset |
-| Denoise | Scheduler, sigma schedule, steps, CFG/guidance, seed behavior |
-| Output | Count, format, codec/sample rate, return type |
-| API | Offline mapping, sync endpoint, async endpoint/job lifecycle |
-| Rejection | Status code and stable error class/message for every invalid boundary |
-
-Official API parity includes limits and defaults, not just names. Do not add a
-generic default that changes a distilled checkpoint's fixed schedule, or accept
-an input combination the official task rejects.
-
-For every accepted boundary, add the neighboring rejected values. Examples:
-`max_images` and `max_images + 1`, smallest/largest legal duration, aligned and
-misaligned dimensions, omitted and illegal steps, one and too many outputs.
-
-## Validate before side effects
-
-Use one normalization/validation path for offline, sync, and async requests.
-The ordering must be:
-
-```text
-parse -> normalize aliases -> validate task/source matrix -> validate declared
-limits -> stream with byte caps -> inspect/decode with frame/pixel/duration caps
--> persist only if needed -> submit to engine
-```
-
-Fail before download, persistence, job creation, or engine submit whenever the
-available metadata is sufficient. For streamed uploads, enforce:
-
-- total reference count and per-modality count before reading bodies;
-- per-file bytes while streaming;
-- aggregate bytes across all files;
-- decoded frames, duration, pixels, and media type before engine submission.
-
-On rejection, close streams and remove every partial temporary resource. Use
-HTTP 400 for invalid task/field combinations and 413 for payload limits. Test
-both sync and async paths; an invalid async request must not create a job ID.
-
-Do not trust client MIME alone. Sniff/validate content and keep ordered repeated
-multipart fields ordered. Avoid logging prompts, raw references, signed URLs,
-or base64 payloads.
-
-## Task request templates
-
-These are endpoint-shape templates, not model recipes. Replace placeholders
-with the exact official defaults and validated assets for the target model.
-Every production recipe must include all advertised tasks and an expected
-output check.
-
-### Text to image: JSON
-
-```bash
-curl --fail-with-body -sS -X POST http://127.0.0.1:8091/v1/images/generations \
-  -H 'Content-Type: application/json' \
-  -d '{
-    "model": "<model>",
-    "prompt": "<validated prompt>",
-    "size": "1024x1024",
-    "num_inference_steps": 30,
-    "seed": 42,
-    "response_format": "b64_json"
-  }' | jq -er '.data[0].b64_json' | base64 -d > output.png
-file output.png
-```
-
-### Image edit: multipart
-
-```bash
-curl --fail-with-body -sS -X POST http://127.0.0.1:8091/v1/images/edits \
-  -F 'model=<model>' \
-  -F 'prompt=<validated edit prompt>' \
-  -F 'image=@/path/to/reference.png;type=image/png' \
-  -F 'size=1024x1024' \
-  -F 'num_inference_steps=30' \
-  -F 'seed=42' \
-  -F 'output_format=png' \
-  | jq -er '.data[0].b64_json' | base64 -d > edited.png
-file edited.png
-```
-
-Use repeated `image` fields only if the model's official input matrix allows
-multiple ordered sources. Never describe a file upload as JSON-only.
-
-### Public request fields versus internal sampling fields
-
-Use the field exposed by the target revision's endpoint schema; do not rename
-it from the pipeline's internal attribute. For example, the current video JSON
-and multipart contracts expose `extra_params`, while serving translates that
-object into `DiffusionSamplingParams.extra_args`. Other request paths may
-expose canonical `extra_args` or nest it under `extra_body`. Inspect the exact
-Pydantic model/FastAPI form dependency and exercise the generated curl before
-publishing a recipe. A deprecation warning in a different endpoint is not
-evidence that an unrecognized multipart form field works here.
-
-Current endpoint map (re-check it at the target commit):
-
-| Endpoint | Public model-specific field |
-|---|---|
-| `/v1/videos`, `/v1/videos/sync` | Multipart `extra_params` JSON string |
-| `/v1/images/generations` | JSON `extra_params` object |
-| `/v1/chat/completions` diffusion | `extra_body.extra_args` |
-| `/v1/images/edits` | Declared typed form fields only |
-| `/v1/audio/generate` | Declared typed JSON fields only |
-
-Prefer declared top-level fields. Use a generic model-specific envelope only
-when the endpoint actually exposes it.
-
-### Text to video: multipart
-
-```bash
-curl --fail-with-body -sS -X POST http://127.0.0.1:8091/v1/videos/sync \
-  -H 'Accept: video/mp4' \
-  -F 'model=<model>' \
-  -F 'prompt=<validated prompt>' \
-  -F 'size=1280x720' \
-  -F 'num_frames=81' \
-  -F 'fps=16' \
-  -F 'num_inference_steps=30' \
-  -F 'seed=42' \
-  -F 'extra_params={"task":"<official-task>"}' \
-  -o output.mp4
-ffprobe -v error -show_entries stream=codec_name,width,height,r_frame_rate \
-  -of json output.mp4
-```
-
-### Image/video to video: multipart
-
-```bash
-curl --fail-with-body -sS -X POST http://127.0.0.1:8091/v1/videos/sync \
-  -H 'Accept: video/mp4' \
-  -F 'model=<model>' \
-  -F 'prompt=<validated prompt>' \
-  -F 'num_frames=81' \
-  -F 'fps=16' \
-  -F 'num_inference_steps=30' \
-  -F 'seed=42' \
-  -F 'extra_params={"task":"<official-task>","start_time_seconds":0}' \
-  -F 'input_reference=@/path/to/reference.png;type=image/png' \
-  -o i2v.mp4
-
-curl --fail-with-body -sS -X POST http://127.0.0.1:8091/v1/videos/sync \
-  -H 'Accept: video/mp4' \
-  -F 'model=<model>' \
-  -F 'prompt=<validated prompt>' \
-  -F 'num_frames=81' \
-  -F 'num_inference_steps=30' \
-  -F 'seed=43' \
-  -F 'extra_params={"task":"<official-task>","start_time_seconds":0}' \
-  -F 'input_reference=@/path/to/reference.mp4;type=video/mp4' \
-  -o v2v.mp4
-```
-
-For multiple references, repeat the model's accepted multipart field in
-official order. Include complete downloadable or repository fixtures plus
-SHA256 hashes. A placeholder-only recipe is not runnable evidence.
-
-### Text to audio: JSON
-
-```bash
-curl --fail-with-body -sS -X POST http://127.0.0.1:8091/v1/audio/generate \
-  -H 'Content-Type: application/json' \
-  -d '{
-    "model": "<model>",
-    "input": "<validated audio prompt>",
-    "audio_length": 10.0,
-    "num_inference_steps": 100,
-    "seed": 42,
-    "response_format": "wav"
-  }' -o output.wav
-ffprobe -v error -show_entries stream=codec_name,sample_rate,channels \
-  -of json output.wav
-```
-
-Use `/v1/audio/generate` for general diffusion audio. Do not substitute the
-speech API unless the model is actually served through the speech contract.
-
-### Async video lifecycle
-
-Alongside `/v1/videos/sync`, include `POST /v1/videos`, status polling,
-content retrieval, cancellation/abort if public, and cleanup assertions. The
-same request must normalize to the same engine sampling parameters in both
-modes.
-
-## Serving benchmark template
-
-Run a smoke first, then fixed-concurrency and arrival-rate tests. Video task
-names for the harness are `t2v`, `i2v`, `ti2v`, or `v2v`; model-specific task
-selection belongs in `--extra-body` when required.
-
-```bash
-python benchmarks/diffusion/diffusion_benchmark_serving.py \
-  --base-url http://127.0.0.1:8091 \
-  --endpoint /v1/videos \
-  --model '<model>' \
-  --dataset random \
-  --task t2v \
-  --num-prompts 100 \
-  --width 1280 --height 720 \
-  --num-frames 81 --fps 16 \
-  --num-inference-steps 30 \
-  --request-rate 0.2 \
-  --max-concurrency 4 \
-  --warmup-requests 4 \
-  --warmup-concurrency 4 \
-  --extra-body '{"extra_params":"{\"task\":\"<official-task>\"}"}' \
-  --output-file result.json
-```
-
-The video backend currently stringifies each multipart value. Pre-serialize
-the nested `extra_params` object as above; passing it as a nested object would
-produce Python dict syntax rather than valid JSON. If the target revision
-changes the backend to JSON-encode dict/list values, update and smoke-test this
-command against that revision.
-
-For i2v/v2v, use the harness's validated dataset/trace path or provide the
-documented media source; do not silently benchmark t2v and label it i2v. Keep
-prompt, assets, seed, dimensions, frames, steps, guidance, warmup, concurrency,
-and arrival process identical across A/B runs.
-
-Keep three benchmark lanes distinct:
-
-1. lossless runtime A/B with the released schedule/precision/attention policy;
-2. accelerated-path A/B that changes one adapter, quantization, cache, or
-   sparsity policy and adds same-seed quality evidence;
-3. production-topology studies that vary placement or parallelism and report a
-   latency/throughput/memory frontier rather than a kernel speedup.
-
-For a fixed single-request A/B, retain every measured run and report median or
-mean with range after an explicit warmup; do not infer p95/p99 from a few
-repetitions. Tail latency requires a declared arrival process, request count,
-concurrency, and enough successful samples. Reconcile client E2E with queue,
-encoder, denoise, video/audio decode, D2H/IPC, codec, and residual timings.
-
-For generated media, define `real-time` rather than relying on the label. A
-complete-response claim uses `client E2E / validated output duration <= 1` and
-does not imply streaming or low time-to-first-frame. Report first chunk/fragment
-and cadence separately when a public streaming route exists. Never add gains
-across benchmark lanes with different schedules, prompts, seeds, artifacts, or
-topologies.
-
-The diffusion serving harness currently covers image/video task families, not
-`/v1/audio/generate`. For text-to-audio, use the target model's task-specific
-audio benchmark if one exists; otherwise add a serving harness that records
-request rate, concurrency, latency, generated-audio seconds per second, error
-rate, and output validation. Do not relabel the TTS harness or a video workload
-as diffusion-audio evidence.
-
-## Hardware recipe contract
-
-Create a separate row and runnable section for each vendor/card/topology:
-
-| Evidence | Required detail |
-|---|---|
-| Identity | Vendor, exact SKU, device count, HBM, interconnect |
-| Software | Driver, CUDA/ROCm/CANN/oneAPI, framework/container, dependency versions |
-| Source | vLLM-Omni commit, official/checkpoint revision and hashes |
-| Deployment | dtype, attention backend per role, TP/SP/CFG/HSDP/DP, cache, quant, offload |
-| Commands | Complete deploy YAML/serve command, every task curl, benchmark command |
-| Method | Warmup exclusion, repetitions or request count, concurrency, arrival distribution, duration, statistic definitions |
-| Results | Success rate, p50/p95/p99, RPS/throughput per device, stage latency |
-| Memory | Load/materialize, resident, encode, denoise, decode, transient per-rank HBM and host PSS |
-| Quality | Metric/tolerance, artifact links/hashes, temporal/audio checks |
-| Output | Raw/offline dtype, range, layout and ownership; online codec, payload bytes, transport route and client boundary |
-| Limits | `validated`/`limited`/`unsupported`/`not tested`, fallback/rejection behavior |
-
-Use current paths such as `recipes/<vendor>/`, `recipes/README.md`, supported
-models, and feature compatibility tables. Top-level documentation must link to
-the scoped matrix; it must not use an unqualified checkmark that hides a ROCm,
-NPU, XPU, task, topology, or dtype limitation.
-
-DLO evidence is per card, especially on small-HBM devices. A DLO result on one
-CUDA card does not establish another CUDA SKU, ROCm, NPU, or XPU support.
-
-## Best deployment selection
-
-Benchmark a correctness-approved candidate matrix, not every theoretical
-combination. Choose the recommended row by the production objective—latency,
-throughput, cost, or small-HBM fit—and document that objective.
-
-Do not recommend a row until it passes:
-
-- all advertised task/API contracts;
-- fixed-reference quality;
-- request isolation and abort/error cleanup;
-- sustained arrival-rate tests;
-- per-rank HBM and host PSS limits;
-- hardware-specific accuracy and performance CI.
-
-Keep the most memory-constrained DLO row even when it is not the fastest; it is
-a separate production recipe, not a failed latency candidate.

.claude/skills/production-add-diffusion-model/references/feature-patterns.md

@@ -1,462 +0,0 @@
-# Production Feature Patterns
-
-## Contents
-
-1. [Compatibility matrix](#compatibility-matrix)
-2. [Online FP8](#online-fp8)
-3. [Strict fused-weight loading](#strict-fused-weight-loading)
-4. [Distributed layerwise offload](#distributed-layerwise-offload)
-5. [Per-request Cache-DiT](#per-request-cache-dit)
-6. [Combination gates](#combination-gates)
-
-## Compatibility matrix
-
-Create rows at the granularity below. Add evidence links and a reason for every
-non-validated state.
-
-| Task/shape | Execution mode/capacity | Backend/layout | Quant | Cache policy | Offload | Topology | Hardware/dtype | Output contract/transport | State | Evidence |
-|---|---|---|---|---|---|---|---|---|---|---|
-| `<task>/<shape>/<schedule>` | serial request / 1 | SDPA, padded | BF16 | none | resident | 1 device | `<card>` BF16 | raw float / local | `not tested` | — |
-| `<task>/<shape>/<schedule>` | step / N | `<fast>`, packed | BF16 | none | resident | `<TP/SP>` | `<card>` BF16 | encoded / subprocess | `not tested` | — |
-| `<task>/<shape>/<schedule>` | step / N | `<fast>`, packed | FP8 | high | DLO no-AG | `<TP/SP>` | `<card>` | device uint8 / SHM | `not tested` | — |
-
-Initialize every row as `not tested`; the table is a work queue, not evidence.
-Promote the dense BF16 row to the oracle only after its scoped parity passes.
-Add one axis at a time. If the runtime cannot
-enforce a known-incompatible combination, add construction/admission
-validation before writing a recipe.
-
-## Online FP8
-
-### Component routing
-
-The public builder accepts one method or a longest-prefix component map:
-
-```python
-from vllm_omni.quantization import build_quant_config
-
-config = build_quant_config({
-    "transformer": {
-        "method": "fp8",
-        "ignored_layers": ["transformer.final_layer"],
-    },
-    "text_encoder": None,
-    "vae": None,
-    "default": None,
-})
-
-assert config.resolve("transformer.blocks.0.attn.qkv_proj") is not None
-assert config.resolve("text_encoder.layers.0.mlp") is None
-assert config.resolve("vae.decoder.conv_out") is None
-```
-
-Use actual runtime prefixes. Multi-DiT or nested pipelines may not use the
-generic names `transformer` and `vae`. A prefix that falls through to a default
-is not component validation. Add tests for every included and ignored prefix.
-
-For CLI serving, first validate resident DiT-only FP8:
-
-```bash
-vllm serve '<model>' --omni --quantization fp8 --max-num-seqs 1
-```
-
-For scoped routing:
-
-```bash
-vllm serve '<model>' --omni \
-  --diffusion-quantization-config \
-  '{"<dit-runtime-prefix>":{"method":"fp8","ignored_layers":["<exact-linear-prefix>"]},"<vae-runtime-prefix>":null,"default":null}' \
-  --max-num-seqs 1
-```
-
-Do not publish these placeholders. A recipe must contain the target model's
-verified prefixes.
-
-### Quantizable module wiring
-
-Pass `quant_config` and a stable prefix through every vLLM linear. Keep
-precision-sensitive modulation/norm layers safe unless proven otherwise.
-
-```python
-from vllm.model_executor.layers.activation import SiluAndMul
-from vllm.model_executor.layers.linear import (
-    MergedColumnParallelLinear,
-    QKVParallelLinear,
-    RowParallelLinear,
-)
-
-self.qkv_proj = QKVParallelLinear(
-    hidden_size=hidden_size,
-    head_size=head_dim,
-    total_num_heads=num_heads,
-    total_num_kv_heads=num_kv_heads,
-    bias=False,
-    quant_config=quant_config,
-    prefix=f"{prefix}.attn.qkv_proj",
-)
-self.gate_up_proj = MergedColumnParallelLinear(
-    hidden_size,
-    [intermediate_size, intermediate_size],
-    bias=False,
-    quant_config=quant_config,
-    prefix=f"{prefix}.mlp.gate_up_proj",
-)
-self.act_fn = SiluAndMul()
-self.down_proj = RowParallelLinear(
-    intermediate_size,
-    hidden_size,
-    bias=False,
-    input_is_parallel=True,
-    quant_config=quant_config,
-    prefix=f"{prefix}.mlp.down_proj",
-)
-```
-
-Set required checkpoint parameters to fail rather than silently initialize if
-the current loader supports `missing_param_init = "error"`.
-
-### FP8 evidence
-
-For each task/hardware row, capture:
-
-- startup proves the FP8 quant method is attached to intended linear modules;
-- no unexpected BF16 fallback and no meta/uninitialized parameters;
-- same-seed component, trajectory, and final-artifact comparison with BF16;
-- HBM at load, resident, encode, denoise, decode, and peak;
-- cold load time; fixed-work warm-latency raw runs; and serving p50/p95/p99 plus
-  throughput only from a declared arrival load with enough samples;
-- named ignored layers and why their BF16 retention is required.
-
-DiT FP8 evidence does not cover the text encoder or VAE. CUDA evidence does not
-cover ROCm/NPU/XPU. A pre-quantized checkpoint is a different loading path from
-online FP8 and needs its own state. If it is also pruned, rotated, distilled, or
-adapter-modified, report that model delta explicitly; BF16-versus-artifact
-quality and speed are deployment comparisons, not pure quantization ablations.
-
-## Strict fused-weight loading
-
-Never use `param.data.copy_()` for quantizable vLLM parameters. Transform the
-checkpoint layout before invoking the parameter loader so the online quant
-loader remains outermost:
-
-```python
-from vllm.model_executor.model_loader.weight_utils import default_weight_loader
-
-param = params[target_name]
-weight_loader = getattr(param, "weight_loader", default_weight_loader)
-
-if target_name.endswith(".qkv_proj.weight"):
-    packed_qkv = reorder_checkpoint_qkv(checkpoint_weight)
-    weight_loader(param, packed_qkv)
-elif target_name.endswith(".gate_up_proj.weight"):
-    gate, up = checkpoint_weight.chunk(2, dim=0)
-    weight_loader(param, gate, 0)
-    weight_loader(param, up, 1)
-else:
-    weight_loader(param, checkpoint_weight)
-```
-
-Track source shards per fused destination:
-
-```python
-expected = {
-    "blocks.0.attn.qkv_proj.weight": {"q", "k", "v"},
-    "blocks.0.mlp.gate_up_proj.weight": {0, 1},
-}
-seen = {name: set() for name in expected}
-
-# After mapping/loading each source:
-seen[target_name].add(shard_id)
-
-incomplete = {
-    name: sorted(shards - seen[name], key=str)
-    for name, shards in expected.items()
-    if shards - seen[name]
-}
-if incomplete:
-    raise RuntimeError(f"incomplete fused checkpoint parameters: {incomplete}")
-```
-
-Also compare retained model parameters with the returned loaded set. Explicit
-checkpoint ignores need a reason and test; an unknown or missing weight is not
-an ignore policy.
-
-Load-time adapter fusion creates a dedicated model, not a request-switchable
-LoRA. Pin the base and adapter revisions, validate every low-rank and full-rank
-delta target, freeze the adapter-declared schedule/task, and fuse before
-sharding only when the loader contract proves identical rank-local weights.
-Reject offload or host-weight paths that bypass that fusion rather than serving
-the unfused base model silently.
-
-## Distributed layerwise offload
-
-### Declare component topology
-
-Use `SupportsComponentDiscovery` and `OffloadPlan`. Validate all dotted paths
-at construction/test time because discovery may otherwise warn and skip.
-
-Treat a future/requested `--layerwise-offload-components` selector as a public
-API only if it exists in the target revision. Today, express component topology
-with `OffloadPlan` (`on_demand_component_paths`, `block_attrs`,
-`encoder_block_attrs`, `resident_dit_paths`, and `offload_submodules`). Do not
-invent a CLI flag or claim component-selective DLO from discovery declarations
-alone; validate the selector separately when it lands.
-
-```python
-from operator import attrgetter
-
-import torch
-
-def assert_offload_path(root, path):
-    value = attrgetter(path)(root)
-    if not isinstance(value, torch.nn.Module):
-        raise TypeError(f"offload path {path!r} is not an nn.Module")
-    return value
-
-for path in (
-    *pipeline._dit_modules,
-    *pipeline._encoder_modules,
-    *pipeline._vae_modules,
-    *pipeline._offload_plan.on_demand_component_paths,
-):
-    assert_offload_path(pipeline, path)
-```
-
-For `block_attrs`, resolve the DiT first and then each declared block attribute.
-Require an indexable block container with at least one `nn.Module`. Validate
-`offload_submodules`, `resident_dit_paths`, and `encoder_block_attrs` similarly.
-
-### Keep host storage loader-owned
-
-Current DLO accepts a loader-produced `HostWeightPlan`. Pipeline authors should
-not instantiate this plan or add ad hoc mmap capability flags. The loader must
-preflight complete runtime/checkpoint bindings, names, shapes, dtypes, persistent
-buffers, and represented transforms before it skips ordinary materialization.
-The exact accepted plan is consumed once by DLO; a rejected plan records an
-observable fallback reason and continues through the ordinary loader.
-
-Direct checkpoint mmap currently requires TP1 without HSDP or online
-quantization. TP>1 falls back to the ordinary TP-aware loader and may feed DLO
-AllGather or no-AllGather; the current design records a bounded DP2xTP2
-AllGather smoke, but that path does not receive direct-mmap host-page sharing.
-Per-tensor online FP8 must use the ordinary loader. Current revisions can admit
-its finalized weight/scale layout to DLO AllGather, including transposed runtime
-weights, while unsupported online quantizers still fail closed. Direct mmap
-remains unavailable, so capture the temporary full-model host materialization
-before rank sharding.
-
-Host Weight Runtime is a distinct opt-in path. Current final-layout consumers
-use no-AllGather DLO and exact model-declared BF16 representations. In
-`preferred` mode, consume an exact local hit or canonically load and publish for
-a future startup; in `required` mode, consume only and fail on a miss or unusable
-artifact. HWR must not interact with DLO AllGather, online quantization, HSDP,
-LoRA/adapted weights, or an unrecognized load format unless the target revision
-adds an exact producer/restore contract. It is immutable node-local CPU backing,
-not zero-copy GPU execution; the DLO transport still owns staging or direct H2D.
-Qualify identity inputs, cold population, warm reuse, atomic concurrent
-publication, source digests, corruption/quarantine, lease cleanup, capacity,
-node-local PSS, H2D, and E2E latency. A host-memory saving that materially
-regresses transfer or request latency is a separate memory-first deployment.
-
-### DLO deployment modes
-
-Treat these paths separately:
-
-| Path | Weight/runtime behavior | Required boundaries |
-|---|---|---|
-| DLO AllGather | Rank-sharded pinned host tensors; reconstruct layer through collective | TP>1 uses ordinary TP-aware loader output; HSDP and unsupported online quantizers are rejected; finalized per-tensor online FP8 is eligible only when recognized by the target revision; concurrent ranks must participate consistently |
-| DLO no-AllGather | Loader-approved checkpoint mmap or ordinary runtime tensors; each rank streams a complete block | Direct mmap is TP1/non-HSDP/non-online-quant only; TP/HSDP/online quant use ordinary runtime tensors and require scoped E2E |
-| SP + DLO | SP group can supply the DLO collective group | Packed boundaries and collective order require E2E |
-
-Single-stage examples:
-
-```bash
-# SP + DLO AllGather candidate; validate exact model/card before publishing.
-vllm serve '<model>' --omni \
-  --enable-distributed-layerwise-offload \
-  --usp 4 \
-  --max-num-seqs 1
-
-# TP + DLO rank-local candidate; not a support statement.
-vllm serve '<model>' --omni \
-  --tensor-parallel-size 2 \
-  --enable-distributed-layerwise-offload \
-  --dlo-no-use-allgather \
-  --max-num-seqs 1
-```
-
-Under `--omni`, configure DP in a deploy YAML rather than passing vLLM DP CLI
-flags:
-
-```yaml
-pipeline: <registered-pipeline>
-async_chunk: false
-data_parallel_size: 4
-stages:
-  - stage_id: 0
-    devices: "0,1,2,3"
-    max_num_seqs: 1
-    enable_distributed_layerwise_offload: true
-    dlo_use_allgather: true
-    parallel_config:
-      tensor_parallel_size: 1
-      sequence_parallel_size: 1
-```
-
-```bash
-vllm serve '<model>' --omni --deploy-config /path/to/dlo_dp4.yaml
-```
-
-Keep the YAML next to the validated recipe. Match its schema to the target
-revision; do not assume a model accepts another pipeline's deploy config.
-
-### DLO validation sequence
-
-1. Resident BF16 reference.
-2. Ordinary layerwise offload parity.
-3. DLO enable, warmup, same-seed generation, disable/restart.
-4. AllGather and no-AllGather separately.
-5. Concurrent requests with different prompts and identical collective-affecting
-   parameters; then reject or schedule incompatible steps/shapes safely.
-6. Queued/in-flight abort, error, idle rank, worker exit, and next request.
-7. TP, SP, cache, compile, online FP8 one axis at a time.
-8. Loader-plan selection/fallback reason, cold/warm storage lifecycle, and
-   corruption/stale-entry recovery when mmap/cache backing is advertised.
-9. Replica-local request preparation plus DLO-group-only weight collectives in
-   DP; exercise concurrent replicas to catch accidental WORLD rendezvous.
-10. Sweep resident block counts; publish raw latency/HBM pairs, dominated
-    points, and the non-dominated Pareto frontier.
-11. Per-rank HBM, host PSS for the full process tree, H2D and collective trace.
-
-On a small-HBM card, measure transient encode/VAE peaks as well as resident DiT
-weights. If a non-DiT component still exceeds HBM, combine only independently
-validated on-demand component staging, VAE tiling/patching, or another topology;
-do not hide the remaining peak behind the phrase “DLO enabled.”
-
-## Per-request Cache-DiT
-
-### Policy and lifecycle skeleton
-
-First verify that the target revision contains the shared protocol/runtime.
-If not, follow the nearest current model's lifecycle without inventing a new
-public API.
-
-The following is a **serial/exclusive request-mode sketch only**. Pipeline-wide
-hooks are mutable; concurrent requests need an admission/batching key and
-runner-owned transition serialization before this shape is safe.
-
-```python
-import torch.nn as nn
-
-from vllm_omni.diffusion.cache.cachedit import (
-    CacheDiTBackend,
-    CacheDiTRequestSpec,
-    RequestScopedCacheDiTRuntime,
-)
-
-class MyPipeline(nn.Module):
-    def __init__(self, *args, **kwargs):
-        super().__init__()
-        # ... components ...
-        self._cache_dit_runtime = RequestScopedCacheDiTRuntime(self)
-
-    def adopt_cache_dit_backend(self, backend: CacheDiTBackend) -> None:
-        self._cache_dit_runtime.adopt(
-            backend,
-            installation_key="my_model.generic",
-        )
-
-    def is_cache_dit_enabled(self) -> bool:
-        return self._cache_dit_runtime.is_enabled
-
-    def _cache_spec(self, quality, steps):
-        if quality == "lossless":
-            return None
-        if quality == "high":
-            return CacheDiTRequestSpec(
-                installation_key="my_model.high.v1",
-                cache_config=self._validated_high_config,
-                num_inference_steps=steps,
-            )
-        raise ValueError(f"unsupported quality tier: {quality!r}")
-
-    def forward(self, req):
-        quality, steps = validate_and_resolve_request(req)
-        self._cache_dit_runtime.prepare(self._cache_spec(quality, steps))
-        return self._run_validated_request(req)
-```
-
-The snippet shows the model-owned transition for exclusive execution. Put
-disable/restore handling in the runtime/runner lifecycle that owns success,
-exception, disconnect, and asynchronous abort; a `try/finally` inside
-`forward()` alone cannot observe every server cancellation boundary.
-
-The `installation_key` must include every policy dimension that changes hook
-installation; changes in step count can use runtime refresh. `None` disables
-installed hooks. Do not mutate the global startup cache config for a request.
-
-Centralize cleanup in the owning request/runner boundary. An exception, abort,
-disconnect, or admission failure must not leave a changed hook policy visible
-to the next request. The exact owner depends on the target revision; add a
-failure-injection test rather than relying on `forward()` alone to observe
-asynchronous abort.
-
-Example model-specific request form after calibration:
-
-```bash
-curl --fail-with-body -sS -X POST http://127.0.0.1:8091/v1/videos/sync \
-  -F 'model=<model>' \
-  -F 'prompt=<prompt>' \
-  -F 'quality=lossless' \
-  -F 'num_inference_steps=<validated-steps>' \
-  -o lossless.mp4
-
-curl --fail-with-body -sS -X POST http://127.0.0.1:8091/v1/videos/sync \
-  -F 'model=<model>' \
-  -F 'prompt=<prompt>' \
-  -F 'quality=high' \
-  -F 'num_inference_steps=<validated-steps>' \
-  -o high.mp4
-```
-
-Do not expose these tier names merely because another model uses them.
-
-### Cache-DiT evidence
-
-For every tier/task/shape/schedule:
-
-- show real cache hits/skipped blocks, not only “backend enabled” logs;
-- measure fixed-work speed and quality against lossless/native;
-- publish a speed/quality frontier and chosen tier mapping;
-- alternate lossless/high/lossless and vary steps to prove restore/refresh;
-- interleave tasks and shapes; run concurrent requests only when scheduling
-  guarantees exclusive compatible hook state;
-- inject validation error, denoise error, disconnect, and abort, then prove the
-  next request starts with its requested policy;
-- bound any cached tensors and clear request-local state.
-
-## Combination gates
-
-Start every combination as `not tested`.
-
-| Combination | Default production posture |
-|---|---|
-| Direct checkpoint mmap + TP>1/HSDP/online quant | Preflight falls back to the ordinary loader; do not claim direct-mmap savings |
-| Per-tensor online FP8 + DLO AllGather | Generic compatibility exists for finalized weights/scales; each model/card/topology remains a candidate until E2E, and other online quantizers fail closed |
-| Online FP8 + DLO no-AllGather | Candidate only after model/card/topology E2E |
-| TP>1 + DLO AllGather | Ordinary TP-aware loader only; bounded DP2xTP2 smoke exists, but each new model/card/topology still needs E2E |
-| TP>1 + DLO no-AllGather | Ordinary TP-aware loader only; candidate with limited generic coverage |
-| HSDP + DLO AllGather | Reject; it would double-shard HSDP-managed parameters |
-| HSDP + DLO no-AllGather | Configuration-compatible candidate with limited E2E coverage |
-| Cache-DiT + online FP8 | Separate trajectory, hit-rate, HBM, latency test |
-| Cache-DiT + DLO | Verify hooks, streamed blocks, memory peaks, abort cleanup |
-| Cache-DiT + step execution | Do not advertise until lifecycle/batching support is explicit |
-| Sparse attention + Ring | Verify backend execution; reject silently ignored sparse settings |
-| Load-time fused adapter + offload/HWR | Reject unless the selected loader path applies and validates the same fusion before sharding/storage |
-| Packed attention + any topology | Unequal-sample boundary parity is mandatory |
-| Device-side uint8 preparation + offline caller | Version the dtype/range/layout change; HTTP MP4 parity does not preserve the raw tensor contract |
-| Device-side preparation + remote codec | Validate route selection, strided/contiguous layouts, payload bytes, ordering, fallback, and encoded-media parity |
-
-A startup-only test never upgrades a combination. Require request output,
-parity/quality, actual fast-path evidence, resource cleanup, and a benchmark on
-the named hardware.

.claude/skills/production-add-diffusion-model/references/performance-patterns.md

@@ -1,481 +0,0 @@
-# Performance Patterns and Code Examples
-
-## Contents
-
-1. [Measurement loop](#measurement-loop)
-2. [Faster BF16 attention](#faster-bf16-attention)
-3. [Sparse attention](#sparse-attention)
-4. [Fusion-first operator patterns](#fusion-first-operator-patterns)
-5. [Remove redundant computation](#remove-redundant-computation)
-6. [USP communication optimization](#usp-communication-optimization)
-7. [Text encoder and VAE disaggregation](#text-encoder-and-vae-disaggregation)
-8. [Performance acceptance](#performance-acceptance)
-
-## Measurement loop
-
-Do not start with a list of optimizations. Start with a fixed correctness and
-performance workload:
-
-```text
-model/checkpoint revision + prompt/media hashes + seed + dimensions/frames +
-scheduler/sigma/steps/guidance + task + dtype + backend + topology + hardware
-```
-
-Capture:
-
-- stage time: load, encode, conditioning, denoise, VAE decode/postprocess;
-- GPU trace: kernel duration/count, launch gaps, allocations, H2D, collectives;
-- CPU trace: preprocessing, per-layer sync, Python dispatch, temporary files;
-- memory: per-rank HBM peaks and process-tree host PSS;
-- output: device preparation, D2H/IPC, payload bytes, codec wall/process CPU,
-  event-loop wait, client materialization, and signed residual;
-- serving: queue time, throughput, errors, request mix, and distribution samples.
-
-For a fixed-work single-request A/B, run one explicit warmup policy and at
-least three measured repetitions, keep every raw run, and report median or mean
-with range. Do not derive p95/p99 from three samples. For tail latency, declare
-the arrival model and collect enough successful requests for the percentile and
-confidence claimed. Implement one optimization at a time, re-run parity, and
-report both local kernel and end-to-end deltas. A kernel win that moves no E2E
-metric is not a production speed claim.
-
-Use the pipeline profiler already exposed by the target revision when
-available, and use PyTorch/CUDA/ROCm/NPU/XPU profiling tools appropriate to the
-platform. Do not force every platform through a CUDA-only workflow.
-
-## Faster BF16 attention
-
-### Shared attention wiring
-
-Use the shared role-aware attention layer and give it the true local head
-counts and QKV layout:
-
-```python
-import torch
-
-from vllm.model_executor.layers.linear import (
-    QKVParallelLinear,
-    RowParallelLinear,
-)
-from vllm_omni.diffusion.attention.layer import Attention
-
-class SelfAttention(torch.nn.Module):
-    def __init__(
-        self,
-        hidden_size,
-        head_dim,
-        num_heads,
-        num_kv_heads,
-        quant_config=None,
-        prefix="",
-    ):
-        super().__init__()
-        self.head_dim = head_dim
-        self.qkv_proj = QKVParallelLinear(
-            hidden_size=hidden_size,
-            head_size=head_dim,
-            total_num_heads=num_heads,
-            total_num_kv_heads=num_kv_heads,
-            bias=False,
-            quant_config=quant_config,
-            prefix=f"{prefix}.qkv_proj",
-        )
-        self.attention = Attention(
-            num_heads=self.qkv_proj.num_heads,
-            num_kv_heads=self.qkv_proj.num_kv_heads,
-            head_size=head_dim,
-            softmax_scale=head_dim**-0.5,
-            causal=False,
-            qkv_layout="BSND",
-            role="self",
-            prefix=prefix,
-        )
-        self.out_proj = RowParallelLinear(
-            num_heads * head_dim,
-            hidden_size,
-            bias=False,
-            input_is_parallel=True,
-            quant_config=quant_config,
-            prefix=f"{prefix}.out_proj",
-        )
-
-    def forward(self, x, attn_metadata=None):
-        qkv, _ = self.qkv_proj(x)
-        q_size = self.qkv_proj.num_heads * self.head_dim
-        kv_size = self.qkv_proj.num_kv_heads * self.head_dim
-        q, k, v = qkv.split([q_size, kv_size, kv_size], dim=-1)
-        q = q.unflatten(-1, (self.qkv_proj.num_heads, self.head_dim))
-        k = k.unflatten(-1, (self.qkv_proj.num_kv_heads, self.head_dim))
-        v = v.unflatten(-1, (self.qkv_proj.num_kv_heads, self.head_dim))
-        out = self.attention(q, k, v, attn_metadata)
-        out = out.flatten(-2)
-        out, _ = self.out_proj(out)
-        return out
-```
-
-Add Q/K norm and RoPE in the official order before the `Attention` call. Keep
-checkpoint mapping for the fused QKV and output projection complete.
-
-### Backend qualification
-
-For each role (`self`, cross-attention category, token refiner, etc.), record:
-
-- actual selected backend from logs/trace;
-- supported dtype, head size, QKV layout, mask/metadata, packed varlen behavior;
-- platform and compute capability;
-- TP/SP/Ring/AllGather-KV restrictions;
-- fallback behavior and whether fallback changes the claimed performance.
-
-Use unequal packed samples and compare with dense SDPA. Verify `cu_seqlens`,
-max sequence length, padding exclusion, modality boundaries, CFG ownership, and
-output slicing. Do not pass an `attn_mask` and assume it is consumed. Inspect
-the backend contract and trace.
-
-TRTLLM, Flash Attention, cuDNN, NPU varlen, and other backends are independent
-qualification rows. The shared `Attention` class does not make every backend
-packed-safe or faster automatically. Ring and hybrid paths require their own
-boundary parity. AllGather-KV and specific backend combinations may fail fast;
-keep that behavior visible in docs.
-
-## Sparse attention
-
-Sparse attention is a backend/model/platform optimization, not a synonym for
-low-precision dense attention. Enable it only after dense BF16 parity.
-
-For every candidate task/shape/topology:
-
-1. State the sparsity policy, selected backend/kernel, block/head dimensions,
-   supported hardware, external kernel/package revision, and fallback.
-2. Prove the sparse kernel executed and report realized sparsity; an enabled
-   flag is insufficient.
-3. Compare same-seed per-block outputs, denoise trajectory, and final artifact
-   with dense attention.
-4. Add modality-specific metrics: temporal consistency and audio/video sync for
-   video, spectral/audio quality for audio, and prompt/image metrics as relevant.
-5. Report attention time and full E2E time. Include preprocessing/index-build
-   overhead, peak memory, and the exact dense control.
-6. Validate each task and shape. Do not extrapolate a T2V result to I2V/Ref2V,
-   or NPU sparse support to CUDA/ROCm/XPU.
-
-If Ring bypasses the selected backend or a sparse setting would be silently
-ignored, reject the combination. Prefer a visible dense fallback only when the
-recipe labels the row `limited` and does not claim sparse speed.
-
-## Fusion-first operator patterns
-
-Prefer current shared/custom operators with explicit guards and native
-fallbacks. Prove parity at operator, block, trajectory, and artifact levels,
-then use a trace to prove fewer launches/bytes or lower latency.
-
-### Q/K RMSNorm + RoPE
-
-```python
-from vllm_omni.diffusion.layers.norm import RMSNorm
-from vllm_omni.diffusion.layers.rope import RotaryEmbedding, apply_rope_to_qk
-
-self.norm_q = RMSNorm(head_dim, eps=eps)
-self.norm_k = RMSNorm(head_dim, eps=eps)
-# Example only: set these booleans from the official checkpoint contract.
-use_neox_layout = False
-cos_sin_are_half_dim = True
-self.rope = RotaryEmbedding(
-    is_neox_style=use_neox_layout,
-    half_head_dim=cos_sin_are_half_dim,
-)
-
-q = self.norm_q(q)
-k = self.norm_k(k)
-q, k = apply_rope_to_qk(self.rope, q, k, (cos, sin))
-```
-
-Replace the illustrative booleans with the official layout in real code.
-Preserve partial rotary dimensions by splitting the rotated prefix and concatenating
-the pass-through suffix. If 3D cos/sin includes a batch axis, the current
-shared path assumes the values are shared across the batch; per-sample
-different positions require a proven representation/path.
-
-For packed `[tokens, heads, head_dim]` tensors with a non-interleaved
-`[cos | sin]` table, the current shared fused boundary is:
-
-```python
-from vllm_omni.diffusion.layers.fused_qk_norm_rope import fused_qk_norm_rope
-
-q, k = fused_qk_norm_rope(
-    q,
-    k,
-    self.norm_q.weight,
-    self.norm_k.weight,
-    rope_table,
-    self.norm_q.variance_epsilon,
-)
-```
-
-Its CUDA fast path currently specializes BF16 `head_dim=128` and
-`rotary_dim=96`; other valid inputs take the eager implementation. The separate
-shared RMSNorm and RotaryEmbedding still provide the general cross-platform
-baseline. Treat NPU/MUSA and other vendor fusions as independent qualification
-rows, and do not generalize a model-specific Triton kernel without compatible
-shape, layout, frequency partition, dtype, and platform evidence.
-
-### Fused QKV projection
-
-Use `QKVParallelLinear` as shown above. Qualification includes:
-
-- `num_heads` and `num_kv_heads` divisibility by TP;
-- local split sizes from the layer, never reusing global counts after TP;
-- Q/K/V checkpoint order conversion before the vLLM loader;
-- all Q, K, and V shards present;
-- quantization scale/loader mapping when FP8 or another method is enabled.
-
-### Gate-up projection + fused SwiGLU
-
-```python
-from vllm.model_executor.layers.activation import SiluAndMul
-from vllm.model_executor.layers.linear import (
-    MergedColumnParallelLinear,
-    RowParallelLinear,
-)
-
-self.gate_up_proj = MergedColumnParallelLinear(
-    input_size=hidden_size,
-    output_sizes=[intermediate_size, intermediate_size],
-    bias=False,
-    quant_config=quant_config,
-    prefix=f"{prefix}.gate_up_proj",
-)
-self.act_fn = SiluAndMul()
-self.down_proj = RowParallelLinear(
-    input_size=intermediate_size,
-    output_size=hidden_size,
-    bias=False,
-    input_is_parallel=True,
-    quant_config=quant_config,
-    prefix=f"{prefix}.down_proj",
-)
-
-gate_up, _ = self.gate_up_proj(x)
-x = self.act_fn(gate_up)
-x, _ = self.down_proj(x)
-```
-
-Map both checkpoint `gate_proj` and `up_proj` shards into the merged parameter
-and require both. A merged projection followed by `F.silu(gate) * up` is not
-the same claim as the `SiluAndMul` fused activation kernel; identify what the
-trace actually uses.
-
-### AdaLN
-
-Use a shared layer only when its mathematical/return contract matches the
-official model:
-
-```python
-from vllm_omni.diffusion.layers.adalayernorm import AdaLayerNorm
-
-self.adaln = AdaLayerNorm(
-    hidden_size,
-    elementwise_affine=False,
-    eps=1e-6,
-)
-x = self.adaln(x, scale[:, None, :], shift[:, None, :])
-```
-
-`AdaLayerNormZero`, `AdaLayerNormZeroSingle`, and continuous variants include
-different projections, broadcasts, and tuple outputs; they are not drop-in
-replacements. A shared AdaLN may have an NPU fast path while using native ops on
-CUDA/HIP/XPU. Do not call it a universal fused AdaLN kernel.
-
-For sigma schedules with a small finite set of timesteps, profile whether
-precomputing the timestep/AdaLN projection once per request is worthwhile.
-Cache keys must include the exact schedule/config/device/dtype and stay bounded.
-
-### Layout, residual, and other candidates
-
-After profiling, inspect:
-
-- bias/dropout/add or gated residual chains;
-- norm + linear or norm + modulation boundaries;
-- repeated transpose/reshape/contiguous around attention;
-- VAE normalization/activation/convolution boundaries;
-- MoE routing and fused experts for models that actually use MoE.
-
-For fused residual-plus-norm or modulation chains, preserve the unfused
-materialization boundary. If the residual was stored in BF16 before RMSNorm,
-round to BF16 and promote again inside the fused kernel; using the unrounded
-FP32 temporary changes the denoising trajectory even when the stored residual
-looks correct.
-
-Keep DLO materialize/release hooks, process-group collectives, request cache
-transitions, and cleanup outside compiled/fused regions unless lifecycle safety
-is explicitly tested. Compile only stable regions and retain eager fallback.
-
-## Remove redundant computation
-
-### Hoist request invariants
-
-Audit the denoise loop for work that depends only on the request rather than
-the current timestep:
-
-```python
-# Before the denoise loop.
-condition = self.condition_proj(text_embeddings)
-refined_condition = self.token_refiner(condition, packed_metadata)
-rope_tables = self.build_rope_tables(position_ids)
-attn_metadata = self.build_attention_metadata(layout)
-
-for step, sigma in enumerate(sigmas):
-    timestep_state = self.prepare_timestep_state(sigma)
-    noise = self.transformer(
-        latents,
-        refined_condition,
-        rope_tables=rope_tables,
-        attn_metadata=attn_metadata,
-        timestep_state=timestep_state,
-    )
-    latents = self.scheduler_step(noise, sigma, latents)
-```
-
-Candidates include prompt conditioning, token-refiner output, reference
-encoding/scans, position IDs/RoPE tables, packed masks/metadata, static CFG
-branches, and finite-schedule AdaLN projections. First prove they are invariant
-for all official tasks, LoRA/adapters, prompt-update features, shapes, and
-schedules. In continuous batching, store request-specific results on request
-state rather than the pipeline singleton.
-
-### Eliminate per-layer synchronization
-
-Avoid device-to-host scalar extraction such as `.item()` inside every layer:
-
-```python
-# Bad: forces a GPU/CPU synchronization in every attention layer.
-max_len = int(cu_seqlens[-1].item())
-
-# Better: compute validated host metadata once during request packing and pass it.
-packed = PackedLayout(
-    cu_seqlens=cu_seqlens,
-    max_seqlen=max(sample_lengths),
-)
-for block in self.blocks:
-    hidden = block(hidden, packed_layout=packed)
-```
-
-Trace to verify the synchronization disappears. Do not replace a correct
-dynamic value with a stale constant merely to remove `.item()`.
-
-### Remove unused masks and allocations
-
-If packed attention consumes `cu_seqlens` and ignores a dense mask, do not
-construct the dense mask. Reuse immutable request metadata and preallocated
-buffers when safe. Prefer direct indexed scatter/gather over constructing large
-temporary tensors only when duplicate-index semantics and gradients (normally
-disabled for inference) remain correct.
-
-Any cross-request cache needs a bounded lifetime and a complete key containing
-shape, layout, device, dtype, schedule, task, adapter/LoRA, and any value that
-changes the tensor. Otherwise keep it request-scoped.
-
-### VAE and reference work
-
-Profile duplicate VAE decode/postprocess, repeated reference decoding/scans,
-unnecessary host-device round trips, and full-frame materialization. Apply VAE
-tiling, slicing, patch parallelism, or on-demand staging only with seam,
-ordering, color/range, and memory parity tests.
-
-## USP communication optimization
-
-“USP works” is not evidence that USP communication is optimized. For Ulysses,
-Ring, hybrid, and AllGather-KV separately, profile:
-
-- collective count, bytes, duration, stream, and overlap with compute;
-- Q/K/V and output all-to-all boundaries;
-- pack/unpack, padding, metadata broadcast, transpose, and contiguous copies;
-- head divisibility or advanced uneven-head support;
-- scaling efficiency from 1 to N devices on the same workload/interconnect;
-- load balance for unequal packed samples.
-
-Declare `_sp_plan` split/gather boundaries around meaningful module outputs,
-not arbitrary Python statements. Keep packed metadata sharded/gathered in the
-same ownership model as tokens.
-
-When the target revision offers accelerated Ulysses transport such as
-`--ulysses-a2a-permute`, qualify it separately from regular Ulysses. Prove the
-strict scatter-heads/gather-sequence layout selected the fast path, record
-one-time JIT/readiness cost, prewarm the maximum workspace shape before CUDA
-graph capture, keep the grow-only workspace on one stream, and exercise cleanup
-before process-group destruction. Compare collective and E2E deltas after the
-dense backend and topology are frozen; keep unsupported layouts on an explicit
-regular-Ulysses fallback.
-
-For GQA, validate `Q_heads % KV_heads == 0`, pad KV heads first, and derive Q
-padding from the original ratio. Exercise both contiguous and row-strided QKV
-layouts; a fast copy path must preserve bitwise values and packed boundaries.
-
-Cross-attention sometimes uses replicated text K/V while only video/audio Q is
-sequence-sharded. In that specific case, `skip_sequence_parallel=True` can
-avoid incorrect or wasteful Q/K/V redistribution:
-
-```python
-self.cross_attention = Attention(
-    num_heads=local_heads,
-    num_kv_heads=local_kv_heads,
-    head_size=head_dim,
-    softmax_scale=head_dim**-0.5,
-    causal=False,
-    role="cross",
-    skip_sequence_parallel=True,
-    prefix=f"{prefix}.cross_attn",
-)
-```
-
-Use it only after proving K/V are replicated and Q/output ownership is correct.
-Otherwise it disables needed communication. Report whether the optimization
-reduces collective bytes/time and improves E2E scaling.
-
-## Text encoder and VAE disaggregation
-
-Disaggregation is a measured architecture decision. First collect:
-
-| Question | Required measurement |
-|---|---|
-| Is the stage large enough? | Encode/decode share at target concurrency and shape |
-| Is transfer economical? | Tensor/media bytes, serialization, connector latency and bandwidth |
-| Is there reuse? | Prompt/reference reuse rate, cacheability, fan-out |
-| Can it scale independently? | Queue saturation and resource utilization per stage |
-| Is the contract stable? | Shape, dtype, mask/tag/position metadata, ordering |
-
-### Text encoder candidate
-
-Define an explicit tensor contract containing hidden states, masks, tags or
-position IDs, dtype, shape, task/checkpoint revision, and request ID. Support
-disabling the local encoder so the DiT cannot accidentally recompute it. Test
-all tasks, prompt lengths, partial TP group membership, retries, backpressure,
-abort, connector cleanup, and independent replica scaling.
-
-### VAE candidate
-
-Separate VAE encode and decode contracts if both exist. Define latent shape,
-scale/shift, dtype/range, tiling/chunk order, output media ordering, and stage
-metadata. Measure latent/media transfer against local decode cost. Test tiling
-seams, RNG/determinism where relevant, chunk ordering, codec/postprocess,
-backpressure, abort, and worker failure.
-
-Do not label VAE tiling, patch parallelism, compile, or offload as VAE
-disaggregation. Likewise, a separate text-encoder TP group is not a separate
-stage. If no reusable generic stage/connector exists in the target revision,
-produce a feasibility result or RFC and keep the capability `not tested` rather
-than inventing a model-local production architecture.
-
-## Performance acceptance
-
-For each accepted optimization, require:
-
-- same fixed workload and correctness artifacts before/after;
-- actual fast-path execution proven by logs/trace/counters;
-- local metric plus E2E latency/throughput and memory;
-- at least three raw fixed-work repetitions after explicit warmup, or a
-  declared arrival-load sample set large enough for every tail percentile;
-- named hardware, backend, topology, dtype, and task scope;
-- fallback/rejection tests for unsupported shapes/platforms;
-- no regression in abort/error cleanup or long-run memory trend.
-
-Update the feature matrix narrowly. Do not convert a profile-driven direction
-into a generic “supported” feature merely because a model-specific PR exists.

.claude/skills/production-add-diffusion-model/references/production-validation.md

@@ -1,464 +0,0 @@
-# Production Serving and Validation
-
-## Contents
-
-1. [Execution-mode decision](#execution-mode-decision)
-2. [Request-level batching](#request-level-batching)
-3. [Step execution and continuous batching](#step-execution-and-continuous-batching)
-4. [Abort and cleanup](#abort-and-cleanup)
-5. [Mixed-RPS soak](#mixed-rps-soak)
-6. [Four CI tracks](#four-ci-tracks)
-7. [PR evidence checklist](#pr-evidence-checklist)
-
-## Execution-mode decision
-
-Request batching and step-wise execution solve different problems:
-
-| Mode | Pipeline contract | Scheduling/abort boundary |
-|---|---|---|
-| Serial request | `forward(DiffusionRequestBatch)` for one request | Whole request; in-denoise preemption is unavailable |
-| Request batching | `supports_request_batch = True`; one forward consumes compatible independent requests | Whole batch/request |
-| Single-request step | Full `SupportsStepExecution`; `max_num_seqs=1` | Between denoise steps |
-| Step continuous batching | Full step contract plus heterogeneous batched-step evidence | Between denoise waves/steps |
-
-Do not expose four stub methods as an opt-out. The current protocol is
-structural; if a native pipeline is not step-capable, do not define the four
-step methods. If it is capable, implement and test all four.
-
-Treat structural support, a generic request-batch bridge, and model benefit as
-three separate claims. Before recommending step execution, name a hypothesis
-that request mode cannot satisfy, define a success threshold, and run a matched
-request-mode control. If throughput or latency regresses, retain step execution
-as functional/limited and keep request mode as the production recommendation.
-
-Choose compatibility keys from every value that changes tensor shapes,
-semantics, hooks, or scheduler behavior: task, output shape/count, frames,
-steps/schedule, guidance/CFG, dtype/backend, LoRA/adapter, cache quality policy,
-and model-specific conditions. Reject or schedule separately rather than
-silently using the first request's settings.
-
-## Request-level batching
-
-A request-batched pipeline returns exactly one `DiffusionOutput` per logical
-request. This is model-owned pseudocode: `validate_batch_compatibility()` and
-`_generate_batch()` are placeholders to implement from the official contract,
-not shared APIs to import.
-
-```python
-import torch
-
-from vllm_omni.diffusion.data import DiffusionOutput
-from vllm_omni.diffusion.worker.request_batch import (
-    DiffusionRequestBatch,
-    split_diffusion_output_by_request,
-)
-
-class MyPipeline(torch.nn.Module):
-    supports_request_batch = True
-
-    def forward(self, req: DiffusionRequestBatch) -> list[DiffusionOutput]:
-        validate_batch_compatibility(req.sampling_params_list, req.prompts)
-        common = req.sampling_params_list[0]
-
-        text_prompts: list[str] | None = [
-            prompt
-            if isinstance(prompt, str)
-            else (prompt.get("prompt") or "")
-            for prompt in req.prompts
-        ]
-
-        prompt_fields = DiffusionRequestBatch.collate_prompt_field_map(
-            req.prompts,
-            {"prompt_embeds": None, "prompt_mask": None},
-        )
-        if prompt_fields["prompt_embeds"] is not None:
-            text_prompts = None
-        generators = req.collate_request_generators(
-            common.num_outputs_per_prompt,
-            default_generator=None,
-        )
-        output = self._generate_batch(
-            prompt=text_prompts,
-            prompt_fields=prompt_fields,
-            generators=generators,
-            sampling=common,
-        )
-        return split_diffusion_output_by_request(
-            DiffusionOutput(output=output),
-            req,
-            num_outputs_per_prompt=common.num_outputs_per_prompt,
-        )
-```
-
-The prompt-field collator only handles tensor fields; extract raw text prompts
-and model-specific negative/alias fields separately, as current in-tree batched
-pipelines do. Set a raw prompt field to `None` when the matching precomputed
-embedding is supplied. The exact aliases and generator device depend on the
-model. Use current collators rather than manual `torch.cat` when they cover the
-contract; they reject mixed present/missing tensors and incompatible
-shape/dtype/device.
-
-Before reading `sampling_params_list[0]`, validate every field that must be
-common. Tests must cover:
-
-- two and N requests, independent seeds/generators, one output each;
-- mixed prompt lengths and packed/padded boundaries;
-- multiple outputs per prompt and correct result slicing;
-- compatible versus incompatible shapes, steps, guidance, LoRA, and quality;
-- one request failure without cross-request output or state corruption;
-- output ordering and request IDs under different completion times.
-
-After the model advertises support, candidate serving is:
-
-```bash
-vllm serve '<model>' --omni \
-  --max-num-seqs 4 \
-  --request-batch-max-wait-ms 20
-```
-
-The wait window trades admission latency for batch formation. Benchmark at 0
-and candidate nonzero values under the target arrival process.
-
-## Step execution and continuous batching
-
-### Full protocol skeleton
-
-Use the current shared state types and make every mutable value request-local:
-
-The helpers beginning with `_` and `validate_step_batch()` below are
-model-owned placeholders; scheduler initialization is scheduler-specific.
-
-```python
-import copy
-from collections.abc import Sequence
-from typing import Any, ClassVar
-
-import torch
-
-from vllm_omni.diffusion.data import DiffusionOutput
-from vllm_omni.diffusion.models.interface import SupportsStepExecution
-from vllm_omni.diffusion.worker.input_batch import InputBatch
-from vllm_omni.diffusion.worker.utils import StepRequestState
-
-class MyPipeline(torch.nn.Module, SupportsStepExecution):
-    supports_step_execution: ClassVar[bool] = True
-
-    def prepare_encode(
-        self,
-        state: StepRequestState,
-        **kwargs: Any,
-    ) -> StepRequestState:
-        prepared = self._prepare_request(state.prompt, state.sampling)
-        state.prompt_embeds = prepared.prompt_embeds
-        state.latents = prepared.latents
-        state.timesteps = prepared.timesteps
-        state.scheduler = copy.deepcopy(self.scheduler)
-        state.scheduler.set_begin_index(0)
-        state.step_index = 0
-        state.extra["task_state"] = prepared.task_state
-        return state
-
-    def denoise_step(
-        self,
-        input_batch: InputBatch,
-        *,
-        states: Sequence[StepRequestState] | None = None,
-        **kwargs: Any,
-    ) -> torch.Tensor | None:
-        validate_step_batch(input_batch, states)
-        return self.transformer(**build_batched_denoise_inputs(input_batch))
-
-    def step_scheduler(
-        self,
-        state: StepRequestState,
-        noise_pred: torch.Tensor,
-        **kwargs: Any,
-    ) -> None:
-        timestep = state.current_timestep
-        if timestep is None:
-            raise RuntimeError("step_scheduler called without a current timestep")
-        state.latents = state.scheduler.step(
-            noise_pred,
-            timestep,
-            state.latents,
-            return_dict=False,
-        )[0].to(state.latents.dtype)
-        state.step_index += 1
-
-    def post_decode(
-        self,
-        state: StepRequestState,
-        **kwargs: Any,
-    ) -> DiffusionOutput:
-        return DiffusionOutput(output=self.decode(state.latents))
-```
-
-This is a shape guide, not drop-in code: verify the target revision's
-`StepRequestState.extra`, scheduler signature, CFG handling, state fields, and
-decode contract. Follow a current native pipeline such as Qwen-Image for the
-exact revision.
-
-Never use the shared `self.scheduler` to advance a request. Store scheduler,
-latents, embeddings, timesteps, masks, RNG/generator, task metadata, cache
-policy, adapter, and output assembly state on `StepRequestState` or its
-request-local extension. Pipeline fields may hold immutable weights/config and
-bounded immutable caches only.
-
-### Qualification sequence
-
-1. Compare request-mode and step-mode at `max_num_seqs=1` for every task using
-   identical seed/schedule/shape and intermediate latents.
-2. Abort queued and in-flight requests at several step boundaries.
-3. Increase to two compatible requests; verify independent scheduler/RNG/state
-   and output order.
-4. Increase to the target capacity and mix prompt lengths.
-5. Exercise mixed shape, frames, steps/schedule, task, guidance, and quality.
-   Batch only compatible rows; otherwise split/reject predictably.
-6. Inject a failure in encode, denoise, scheduler, and decode independently.
-7. Run the production soak and fault matrix.
-
-Candidate commands only after capability validation:
-
-```bash
-# Conservative step execution and step-boundary abort.
-vllm serve '<model>' --omni \
-  --step-execution \
-  --max-num-seqs 1
-
-# Experimental batched-step candidate after multi-request E2E.
-vllm serve '<model>' --omni \
-  --step-execution \
-  --max-num-seqs 4
-```
-
-If an attention backend is only correct for single-request step mode, fail
-early or keep the recipe at `max_num_seqs=1`. Do not describe generic runtime
-support as model support.
-
-## Abort and cleanup
-
-### Required semantics
-
-For queued and in-flight abort/disconnect:
-
-- emit exactly one terminal aborted/error result;
-- do not run `post_decode` after abort unless a documented partial-output mode
-  explicitly requires it;
-- do not increment `step_index` or apply a scheduler result after cancellation;
-- remove scheduler/state cache entries and request IDs;
-- release latents, prompt/reference tensors, cache hooks, DLO buffers, connector
-  messages, uploaded/temp files, and output buffers;
-- keep other requests alive;
-- make cleanup idempotent when disconnect, timeout, and worker error race;
-- drop late results/exceptions for already-resolved or cancelled futures
-  without terminating the shared result-pump thread;
-- complete the next valid request with its requested cache/offload policy.
-
-Test cancellation before engine admission, while queued, after encode, during
-early/middle/final denoise steps, before decode, and while returning/persisting
-output. Include async job cancellation and client socket disconnect.
-
-### Failure injection matrix
-
-| Location | Injection | Assertions |
-|---|---|---|
-| Validation/upload | bad MIME, count/bytes overflow, truncated media | 400/413, no job/engine submit, no temp leak |
-| Load/start | missing fused shard, unsupported topology | startup fails, actionable error, no silent init |
-| Encode | exception/OOM/timeout | one error, request state/temp released |
-| Denoise | exception/OOM/worker exit | peers handled, hooks/buffers released, health behavior explicit |
-| Scheduler | invalid state/shape | no latent/step partial commit |
-| Decode/post | exception/codec error | no corrupt successful result, partial output removed |
-| Result/IPC | late result after cancel, large split tensor, orphan SHM | pump remains alive, one terminal result, exact ownership cleanup |
-| Connector/disagg | timeout/retry/duplicate | idempotency, backpressure, no orphan payload |
-
-After each injection, check server health and issue a known-good request. If
-the process must restart, document and test the restart policy; do not claim
-transparent recovery.
-
-### Large-output transport
-
-Treat worker-to-engine IPC and engine-to-client transport as separate hops.
-For batched video, prove that each logical output does not accidentally retain
-or serialize the complete batch storage. Exercise the target revision's shared
-memory or artifact-handle path above and below its threshold, including
-non-contiguous views, multiple outputs, consumer failure, timeout, abort, and
-worker exit. Verify every shared segment/file is released exactly once.
-
-Freeze the media contract at each hop: dtype, value range, shape/layout,
-contiguity, request ownership, encoding/color model, payload bytes, and pending
-consumers. Device-side float-to-uint8 preparation, D2H/IPC, planar/interleaved
-conversion, CPU encode/mux, and delivery off the asyncio event loop are
-different optimizations. Validate their route selection and fallbacks
-independently. Byte-identical HTTP MP4 output does not prove that a raw offline
-tensor kept its dtype or range.
-
-For long-window output, validate segment ordering and audio/video continuity,
-then exercise payloads around serializer, shared-memory, HTTP, and codec size
-limits. Chunking must reassemble exactly, reject malformed/missing chunks, and
-release partial state after abort, timeout, or peer failure.
-
-Qualify chunk production, transport, and public streaming as separate layers.
-A VAE callback must define owned representation, frame offsets, final marker,
-rank ownership, source/config gate, and collective-safe failure, but it does not
-prove SHM/ZMQ delivery, bounded backpressure, cancellation, codec reuse, or
-client streaming. Measure time to first owned chunk, first encoded fragment,
-steady cadence, mux finalization, and complete client artifact independently.
-
-Measure remote MP4 encoding, serialization/copies, network transfer, and client
-materialization separately from denoise/decode. An inference-kernel speedup does
-not establish serving throughput if output transport dominates or leaks.
-
-## Mixed-RPS soak
-
-Use a realistic request mix and arrival distribution. Include small/large
-shapes, short/long prompts, all advertised tasks, cache tiers, and a small rate
-of abort/disconnect/error injections. Run phases below, near, and above measured
-saturation so queue/backpressure behavior is visible.
-
-Example serving benchmark phase:
-
-```bash
-python benchmarks/diffusion/diffusion_benchmark_serving.py \
-  --base-url http://127.0.0.1:8091 \
-  --endpoint /v1/videos \
-  --model '<model>' \
-  --dataset trace \
-  --dataset-path /path/to/versioned-production-trace.jsonl \
-  --task t2v \
-  --num-prompts 500 \
-  --request-rate 0.5 \
-  --max-concurrency 8 \
-  --warmup-requests 8 \
-  --warmup-concurrency 8 \
-  --output-file soak-near-saturation.json
-```
-
-Use a task-appropriate trace/media dataset. Run the same phase at rates selected
-from an initial saturation curve; a literal `0.5` is illustrative only.
-
-On supported GPU hosts, wrap a test or soak with the repository monitor:
-
-```bash
-bash tests/dfx/stability/scripts/resource_monitor.sh run --backend gpu -- \
-  pytest -s -v tests/dfx/stability/scripts/test_stability_<model>.py -m slow
-```
-
-Check the target revision's monitor backends. Do not claim CPU/NPU monitoring
-from a reserved CLI choice unless collection is implemented and verified.
-
-Report:
-
-- offered and achieved RPS, success/error/abort rates;
-- p50/p95/p99 and queue/service/stage times, with the successful sample count
-  and arrival process used for each percentile;
-- throughput per device and batch/wave utilization;
-- per-rank HBM and process-tree PSS trend/slope;
-- cache size/hit rate, temp-file/disk growth, open FDs/handles;
-- outstanding shared-memory/artifact handles and encoded/network bytes;
-- worker/engine health, restarts, OOM/timeouts, backpressure response;
-- total duration, request count/mix, raw logs/JSON/monitor artifact hashes.
-
-A short burst or repeated serial curl loop is not a stability test.
-
-## Four CI tracks
-
-Keep the tracks independent so a failure answers a specific question. These
-four readiness tracks are not the repository's numbered CI levels. Map them
-onto the current L1-L5 taxonomy:
-
-| Readiness track | Repository CI placement |
-|---|---|
-| Function | L1 unit/logic plus L2 basic online/offline E2E; extend real-model scenarios in L3/L4 |
-| Accuracy | L3 real-model gates for important cases; deeper and broader nightly coverage in L4 |
-| Performance | Important thresholds in L3 where they fit the time budget; full baselines/regression jobs in nightly L4 |
-| Reliability | Cheap rejection/recovery assertions can run earlier, but long stability, fault injection, and reliability suites belong to weekly L5 |
-
-Distributed topology is a test case within the appropriate level, not a
-replacement definition for a level.
-
-### 1. Function CI
-
-Cover:
-
-- strict unquantized and advertised quantized loading;
-- fused source-shard completeness and prefix routing;
-- official task positive boundaries and neighboring negative cases;
-- offline, sync, async normalization/parity;
-- single-device reference plus every advertised topology smoke;
-- output type/count/order and request ID isolation;
-- feature rejection paths that must fail early.
-
-### 2. Accuracy CI
-
-Pin official implementation/checkpoint/vLLM-Omni revisions, prompts and media
-hashes, seed, scheduler/sigma/steps, dimensions/frames/FPS, guidance, dtype,
-backend, and topology. Compare at suitable levels:
-
-- component outputs or selected blocks;
-- one or more intermediate denoise latents;
-- final image/video/audio artifact metrics;
-- temporal consistency and audio/video synchronization where relevant.
-
-Set tolerance per modality/hardware/backend with rationale. Keep failure
-artifacts and metadata. A single visual inspection does not define CI.
-
-For quantization trajectory tooling, inspect the target revision's current
-`vllm_omni/quantization/tools/compare_diffusion_trajectory_similarity.py` and
-reuse it when compatible instead of creating an ad hoc comparator.
-
-### 3. Performance CI
-
-Pin workload and best deployment. Include a resident recommended row and a
-small-HBM/DLO row when both are production targets. Declare:
-
-- one explicit warmup exclusion and at least three measured fixed-work
-  repetitions;
-- raw result JSON and environment identity;
-- median/mean plus range for small fixed-work A/Bs; p50/p95/p99 only for an
-  arrival-load run with enough samples, plus throughput/RPS, stage time,
-  HBM/PSS;
-- regression metric, threshold, baseline owner, and update procedure;
-- correctness/accuracy check linked to the same configuration.
-
-Use the current harness config style, for example:
-
-```bash
-pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py \
-  --test-config-file tests/dfx/perf/tests/test_<model>_vllm_omni.json
-```
-
-Verify parser options on the target revision. Do not copy a JSON schema from a
-different model without matching its endpoint, task, dataset, and deploy path.
-
-### 4. Reliability CI
-
-Run a bounded CI soak plus a longer scheduled soak. Cover:
-
-- below/near/above-saturation arrival phases;
-- queued/in-flight abort and disconnect;
-- OOM/timeout and worker/process failure appropriate to the platform;
-- DLO/cache/connector cleanup;
-- health, fast-fail/backpressure, restart/recovery policy;
-- memory/temp/FD trend and successful known-good request after each fault.
-
-Keep reliability thresholds independent from latency thresholds. A fast server
-that leaks memory or hangs after cancellation fails production readiness.
-
-## PR evidence checklist
-
-The PR description must include or link:
-
-- official source/checkpoint revisions and API/limit matrix;
-- scoped feature compatibility matrix with the four allowed states;
-- implementation notes for FP8, DLO, request cache, attention/fusion, batching;
-- exact environment, deploy/serve, every task curl, benchmark and soak commands;
-- fixed accuracy inputs and artifacts;
-- raw before/after performance runs and memory traces;
-- output-contract manifests and stage accounting from device preparation
-  through the complete client artifact;
-- abort/disconnect/fault results and cleanup evidence;
-- Function/Accuracy/Performance/Reliability CI files and ownership;
-- limitations/TODOs that remain `not tested` rather than implied support.
-
-Separate the Day-0 milestone from production readiness. If the PR only closes
-some gates, title and description should say which scoped gates it closes.

.claude/skills/readme.md

@@ -31,16 +31,9 @@ self-check.
 - [`diffusion-perf-opt`](diffusion-perf-opt/SKILL.md): guides diffusion model
   performance optimization, including profiling traces, parallel strategies,
   stage timing analysis, and benchmark-driven tuning
-- [`find-simplifications`](find-simplifications/SKILL.md): finds evidence-backed
-  opportunities to remove or merge dead, duplicated, speculative,
-  over-generalized, or unnecessarily defensive vLLM-Omni code
 - [`precheck-pr`](precheck-pr/SKILL.md): self-checks a branch before creating a
-  PR by validating title format, dead code, simplification opportunities,
+  PR by validating title format, dead code, accuracy and performance claims,
-  accuracy and performance claims, and merge readiness
+  and merge readiness
-- [`production-add-diffusion-model`](production-add-diffusion-model/SKILL.md):
-  takes a working Day-0 diffusion integration through API/limit parity,
-  feature-combination evidence, quantization/offload, performance, production
-  serving, hardware recipes, and CI readiness
 - [`quantization`](quantization/SKILL.md): guides quantization method selection,
   model integration, checkpoint loading, and quality/performance validation
   for vLLM-Omni

.claude/skills/review-pr/SKILL.md

@@ -21,10 +21,7 @@ Make every finding:
 - **Calibrated:** match severity to user and maintainer impact.
 
 Do not report unrelated backlog, style already enforced by pre-commit, or a
-missing test that would not protect changed behavior. Do report new allowlist
+missing test that would not protect changed behavior.
-or budget entries in `check_forbidden_imports.py` / `check_torch_cuda.py` /
-`check_tts_adapter.py` / `check_buildkite.py` unless the PR justifies them:
-those are policy changes, not lint noise.
 
 ## Select the input and depth
 
@@ -95,13 +92,11 @@ If docs and live code disagree, verify the code/tests and report the drift.
 
 | Reference | Read when |
 | --- | --- |
-| [model-addition-checklist.md](references/checks/model-addition-checklist.md) | A model, architecture, loader, processor, registry, pipeline config, or deploy config is added. |
+| [model-addition-checklist.md](references/checks/model-addition-checklist.md) | A model, architecture, loader, processor, registry, or stage config is added. |
 | [perf-verification.md](references/checks/perf-verification.md) | The PR makes a latency, throughput, memory, or quality claim. |
 | [test-quality-evaluation.md](references/checks/test-quality-evaluation.md) | Tests change, are absent for risky code, or may not exercise production behavior. |
 | [tests-docs-checklist.md](references/checks/tests-docs-checklist.md) | Coverage, CI markers, examples, user docs, or PR evidence need review. |
 | [verification.md](references/checks/verification.md) | Hardware, a server, or a runnable affected path is available for active verification. |
-| [examples-policy.md](../precheck-pr/references/examples-policy.md) | The PR adds, copies, or renames Python under `examples/`; apply the canonical policy shared with `precheck-pr`. |
-| [find-simplifications](../find-simplifications/SKILL.md) | Every review; run a diff-scoped subtraction and simplification pass after correctness blockers. |
 
 ### Delivery and reviewer coordination
 
@@ -157,13 +152,6 @@ contract they protect or use only the applicable evidence checks.
 Apply every category in [general-checks.md](references/process/general-checks.md) before
 lower-priority comments.
 
-If the diff census contains an added, copied, or renamed Python path under
-`examples/`, read and apply the canonical
-[examples policy](../precheck-pr/references/examples-policy.md). Treat a new
-model-specific Python example as blocking. Do not flag model-specific example
-debt that the PR only modifies or removes, and do not run the rest of the
-author-oriented `precheck-pr` workflow.
-
 For each changed value or behavior, trace:
 
 ```text
@@ -184,13 +172,9 @@ promotion gate. Candidate or draft rules are questions, not blockers, unless
 current code, tests, or policy enforce them. Inspect both sides of any config,
 registry, serialization, connector, cache, or stage boundary.
 
-Read and apply [find-simplifications](../find-simplifications/SKILL.md) on every
+When a diff adds or expands a helper, class, fallback, compatibility branch, or
-review. Constrain it to the diff and the adjacent ownership, callers, or
+public behavior, run a subtraction pass: remove out-of-scope behavior and check
-consumers needed to prove a candidate. Check whether added or expanded helpers,
+whether each new abstraction can be deleted, merged, moved, or inlined.
-classes, state, fallback and compatibility branches, data movement, or public
-behavior can be deleted, merged, moved, or inlined. Zero candidates is a valid
-result. Do not widen the review into repository backlog or report speculative
-style preferences as simplification findings.
 
 ### 6. Verify the changed path
 

.claude/skills/review-pr/references/checks/model-addition-checklist.md

@@ -1,7 +1,7 @@
 # Model Addition Checklist
 
 Read this reference when a PR adds or expands a model architecture, pipeline,
-loader, processor, registry entry, pipeline config, or deploy config.
+loader, processor, registry entry, or stage configuration.
 
 Official docs: [model contribution guides](https://docs.vllm.ai/projects/vllm-omni/en/latest/contributing/model/),
 [adding an omni model](https://docs.vllm.ai/projects/vllm-omni/en/latest/contributing/model/adding_omni_model/),

.claude/skills/review-pr/references/modules/configuration.md

@@ -3,7 +3,7 @@
 Primary design: [vLLM-Omni Configuration](https://docs.vllm.ai/projects/vllm-omni/en/latest/design/module/vllm_omni_config/).
 
 Use for config construction, deploy YAML, registries, schema, defaults,
-aliases, CLI projection, pipeline config, and topology. Inspect the reviewed
+aliases, CLI projection, stage config, and topology. Inspect the reviewed
 page's status: while its contract remains deferred or draft, current code and
 tests—not proposed precedence rules—are authoritative.
 

.claude/skills/review-pr/references/process/review-execution.md

@@ -111,17 +111,6 @@ Mark early findings as preliminary and continue. This is not a GitHub comment.
   but do not publish review events without separate authorization.
 - Record DCO, pre-commit, required CI, and mergeability. Pending or unknown gates
   do not block source review.
-- GitHub Actions `SKIP` omits SPDX, shellcheck, markdownlint, mypy-3.10, and
-  test-mark coverage. A green GHA pre-commit job does not prove those local
-  gates passed. New files still need the full Linting list: Omni SPDX
-  (`vLLM-Omni project`), no stdlib `re`/`base64` in `vllm_omni/`, no new
-  pickle / Hugging Face Hub API / `torch.cuda` call sites, test marks, TTS
-  adapter ratchet, Buildkite schema, and native shellcheck on macOS/Windows.
-  See [Linting](https://vllm-omni.readthedocs.io/en/latest/contributing/#linting).
-- Expanding `CHECK_IMPORTS[*].allowed_files`, `ALLOWED_FILES`,
-  `MAX_MODEL_TYPE_BRANCHES`, or Buildkite `SKIP_FILES` is a policy change.
-  Do not rubber-stamp allowlist/budget growth; require justification and
-  prefer fixing the call site.
 - Treat a failed gate as its own evidence. Do not restate its formatting/lint
   output as a new code-review finding.
 - Open CI logs only when the first failing step overlaps the frozen diff or

.claude/skills/review-pr/references/process/review-routing.md

@@ -48,7 +48,7 @@ compatibility changes.
 
 | Signal | Read | Optional repo-local skill |
 | --- | --- | --- |
-| New or expanded model, loader, processor, registry, pipeline config, or deploy config | [model-addition-checklist.md](../checks/model-addition-checklist.md) | [`add-tts-model`](../../../add-tts-model/SKILL.md) or [`add-diffusion-model`](../../../add-diffusion-model/SKILL.md) |
+| New or expanded model, loader, processor, registry, or stage config | [model-addition-checklist.md](../checks/model-addition-checklist.md) | [`add-tts-model`](../../../add-tts-model/SKILL.md) or [`add-diffusion-model`](../../../add-diffusion-model/SKILL.md) |
 | Latency, throughput, memory, scaling, precision, or quality claim | [perf-verification.md](../checks/perf-verification.md) | [`diffusion-perf-opt`](../../../diffusion-perf-opt/SKILL.md) for diffusion |
 | Tests changed, absent for risky behavior, or test-only | [test-quality-evaluation.md](../checks/test-quality-evaluation.md) | [`vllm-omni-test`](../../../vllm-omni-test/SKILL.md) |
 | CI, examples, docs, public behavior, or contributor evidence | [tests-docs-checklist.md](../checks/tests-docs-checklist.md) | None |

.claude/skills/vllm-omni-test/SKILL.md

@@ -52,8 +52,8 @@ Always attach markers deliberately:
   - `tts` — speech synthesis / TTS-only models (`/v1/audio/speech`, voice clone, etc.)
   - `diffusion` — generative diffusion models (image / audio / text / video from noise)
 - **Cross-cutting area** (when relevant): `parallel`, `cache`, `example`, `benchmark`
-- **Hardware**: `cpu`, `gpu`, `cuda`, `rocm`, `npu`, …; SKU (`H100`, `L4`, …) and `cards_{n}` via `@hardware_test(...)` / `hardware_marks(...)` only — never `pytest.mark.H100` (`check-mark` rejects it)
+- **Hardware**: `cpu`, `gpu`, `cuda`, `rocm`, `npu`, `L4`, `H100`, `distributed_cuda`, …
-- Optional: `slow`. Multi-card: `num_cards=N` on the helper (filter with `cards_N` / `not cards_1`)
+- Optional: `slow`, distributed markers when multi-card is required
 
 **Baseline smoke (L2 + L3):** The simplest e2e case per model — default deploy, minimal request — should usually carry **both** `@pytest.mark.core_model` and `@pytest.mark.advanced_model` on the **same** test function so `test-ready.yml` and `test-merge.yml` share one test. `send_*_request` picks validation depth from `--run-level`. References: `test_voxcpm2_tts.py::test_text_to_audio_001`, `test_qwen3_tts_customvoice.py::test_text_to_audio_001`. Heavier scenarios use **`advanced_model` only**; L4 expansion uses **`full_model`**.
 
@@ -94,8 +94,8 @@ Existing references: `tests/e2e/offline_inference/test_qwen2_5_omni.py` (L2-styl
 
 | Scenario | Typical location | Fixtures / runner pattern | Baseline markers & level |
 |----------|------------------|---------------------------|---------------------------|
-| **Offline inference e2e** | `tests/e2e/offline_inference/` | **Module (default):** `omni_runner` + `offline_client`. **Function (isolation only):** `omni_runner_function` + `offline_client_function`. Diffusion/TTS may use `Omni(...).generate` directly | L2: `core_model` + **one of** `omni` / `tts` / `diffusion`; `@hardware_test(...)` when GPU/NPU is required |
+| **Offline inference e2e** | `tests/e2e/offline_inference/` | **Module (default):** `omni_runner` + `omni_runner_handler`. **Function (isolation only):** `omni_runner_function` + `omni_runner_handler_function`. Diffusion/TTS may use `Omni(...).generate` directly | L2: `core_model` + **one of** `omni` / `tts` / `diffusion`; `@hardware_test(...)` when GPU/NPU is required |
-| **Online serving e2e** | `tests/e2e/online_serving/` | **Module (default):** `omni_server` + `online_client`. **Function (isolation only):** `omni_server_function` + `online_client_function`. Clients: `send_omni_request` (omni), `send_audio_speech_request` (tts), `send_diffusion_request` / `send_video_diffusion_request` / `send_images_generations_request` (diffusion) | Baseline smoke: **`core_model` + `advanced_model`**; heavier paths: `advanced_model` only; L4 expansion: `full_model` |
+| **Online serving e2e** | `tests/e2e/online_serving/` | **Module (default):** `omni_server` + `openai_client`. **Function (isolation only):** `omni_server_function` + `openai_client_function`. Clients: `send_omni_request` (omni), `send_audio_speech_request` (tts), `send_diffusion_request` / `send_video_diffusion_request` / `send_images_generations_request` (diffusion) | Baseline smoke: **`core_model` + `advanced_model`**; heavier paths: `advanced_model` only; L4 expansion: `full_model` |
 | **Documentation / runnable examples** | `tests/examples/offline_inference/`, `tests/examples/online_serving/` | **Offline docs (preferred):** extract Python/Bash blocks from the doc README (e.g. `ReadmeSnippet.extract_readme_snippets`), `pytest.mark.parametrize` each snippet, run via `example_runner.run` with a stable `output_subfolder`. **Online docs:** copy client/request scripts into dedicated tests and keep them in sync with the doc page. | Usually **L4**: `advanced_model`, often `example` plus hardware marks matching the nightly docs-example job (see `.buildkite/cuda/test-nightly.yml`). Full conventions: [docs/contributing/ci/test_examples/l4_doc_example_tests.inc.md](../../../docs/contributing/ci/test_examples/l4_doc_example_tests.inc.md) (introduced in [PR #1910](https://github.com/vllm-project/vllm-omni/pull/1910): naming, output directory layout, skip rules, avoid trimming `num_inference_steps` without a strong CI reason). |
 | **Performance / benchmark** | `tests/dfx/perf/tests/*.json` + `run_*_benchmark.py` | JSON or script-driven server + load config; assert explicit metrics / baselines | L4 Perf: JSON `mark` with `full_model` + `omni`/`tts`/`diffusion`; wire `test-nightly.yml` Perf steps |
 | **Invalid parameter / negative HTTP validation** | `tests/dfx/reliability/invalid_param_test/` | Live `omni_server` + low-level `send_*_http_request` with `err_code` / `err_message` | `pytest.mark.slow` + `omni` / `tts` / `diffusion` + `@hardware_marks` (`H100` or `L4`); CI in **`test-weekly.yml`** (not ready/merge/nightly) |
@@ -110,10 +110,10 @@ After choosing offline/online/docs/perf, classify the **product under test** and
 |-----------|------------------------------|----------------------------|----------------------------------------|
 | **What it is** | Multimodal LLM pipeline (thinker/talker/stages; text + vision + audio I/O) | Speech synthesis / voice models | Generative diffusion (noise → image, audio, text, or video) |
 | **Examples** | Qwen2.5-Omni, Qwen3-Omni | Qwen3-TTS, VoxCPM2, Higgs-Audio, Step-Audio2 | Qwen-Image, BAGEL, Wan2.2, HunyuanVideo |
-| **Offline runner** | `omni_runner` + `offline_client`, `generate_multimodal` | `Omni(...).generate` with deploy YAML + TTS params | `Omni(...).generate` + `OmniDiffusionSamplingParams` |
+| **Offline runner** | `omni_runner` + `omni_runner_handler`, `generate_multimodal` | `Omni(...).generate` or stage YAML + TTS params | `Omni(...).generate` + `OmniDiffusionSamplingParams` |
-| **Online client** | `online_client.send_omni_request` (chat completions, modalities) | `online_client.send_audio_speech_request` (`/v1/audio/speech`) | `send_diffusion_request` (chat/T2I), `send_video_diffusion_request` (`/v1/videos`, X2V), or `send_images_generations_http_request` / `send_images_edits_http_request` (DALL-E routes) |
+| **Online client** | `openai_client.send_omni_request` (chat completions, modalities) | `openai_client.send_audio_speech_request` (`/v1/audio/speech`) | `send_diffusion_request` (chat/T2I), `send_video_diffusion_request` (`/v1/videos`, X2V), or `send_images_generations_http_request` / `send_images_edits_http_request` (DALL-E routes) |
-| **Typical assertions** | Stage outputs, text/audio keywords via `OfflineOmniClient` / response handler | WAV bytes, stream chunks, speech endpoint contract | Image/video dimensions, `final_output_type`, `assert_diffusion_response` |
+| **Typical assertions** | Stage outputs, text/audio keywords via `OmniRunnerHandler` / response handler | WAV bytes, stream chunks, speech endpoint contract | Image/video dimensions, `final_output_type`, `assert_diffusion_response` |
-| **Deploy YAML** | Per-model deploy configs (`ci/qwen3_omni_moe.yaml`, …) | `qwen3_tts.yaml`, `voxcpm2.yaml`, … | Often default serve; parallel/offload YAML for heavy DiT |
+| **Stage / deploy YAML** | Per-model omni stage configs (`ci/qwen3_omni_moe.yaml`, …) | `qwen3_tts.yaml`, `voxcpm2.yaml`, … | Often default serve; parallel/offload YAML for heavy DiT |
 | **Nightly group** (`test-nightly.yml`) | **Omni Model Test** — `-m "full_model and omni"` | **TTS Model Test** — `-m "full_model and tts"` | **Diffusion X2I(&A&T)** *or* **Diffusion X2V** (see Step 3 table; same `diffusion` marker, different YAML group / file shard) |
 | **L4 pressure** | Expansion per modality/model as needed | Expansion + accuracy/perf in TTS group | X2I: merge feature combos per [#1832](https://github.com/vllm-project/vllm-omni/issues/1832); X2V: separate nightly group |
 
@@ -175,9 +175,9 @@ See **L1 unit test constraints (mocking)** below for the full do/don't list.
 @pytest.mark.omni
 @hardware_test(...)
 @pytest.mark.parametrize("omni_runner", test_params, indirect=True)
-def test_<scenario>(omni_runner, offline_client) -> None:
+def test_<scenario>(omni_runner, omni_runner_handler) -> None:
     request_config = {"prompts": ..., "modalities": [...]}  # optional: images, videos, audios
-    offline_client.send_omni_request(request_config)
+    omni_runner_handler.send_omni_request(request_config)
 ```
 
 *Offline generative e2e — **Diffusion** (representative):*
@@ -187,8 +187,8 @@ def test_<scenario>(omni_runner, offline_client) -> None:
 @pytest.mark.diffusion
 @hardware_test(...)
 @pytest.mark.parametrize("omni_runner", test_params, indirect=True)
-def test_text_to_image_001(offline_client) -> None:
+def test_text_to_image_001(omni_runner_handler) -> None:
-    offline_client.send_diffusion_request({"prompt": "...", "extra_body": {"num_inference_steps": 4, ...}})
+    omni_runner_handler.send_diffusion_request({"prompt": "...", "extra_body": {"num_inference_steps": 4, ...}})
 ```
 
 *Offline TTS e2e — **Qwen3-TTS** (two-stage; representative):*
@@ -198,8 +198,8 @@ def test_text_to_image_001(offline_client) -> None:
 @pytest.mark.tts
 @hardware_test(...)
 @pytest.mark.parametrize("omni_runner", tts_server_params, indirect=True)
-def test_text_to_audio_001(omni_runner, offline_client) -> None:
+def test_text_to_audio_001(omni_runner, omni_runner_handler) -> None:
-    offline_client.send_audio_speech_request({
+    omni_runner_handler.send_audio_speech_request({
         "input": "...",
         "task_type": "Base",
         "ref_audio": REF_AUDIO_URL,
@@ -214,8 +214,8 @@ def test_text_to_audio_001(omni_runner, offline_client) -> None:
 @pytest.mark.tts
 @hardware_test(...)
 @pytest.mark.parametrize("omni_runner", tts_server_params, indirect=True)
-def test_voice_clone_001(offline_client) -> None:
+def test_voice_clone_001(omni_runner_handler) -> None:
-    offline_client.send_single_stage_tts_request({
+    omni_runner_handler.send_single_stage_tts_request({
         "input": "...",
         "language": "en",
         "prompt_audio_path": REF_AUDIO_PATH,
@@ -231,9 +231,9 @@ def test_voice_clone_001(offline_client) -> None:
 @pytest.mark.omni
 @hardware_test(...)
 @pytest.mark.parametrize("omni_server", test_params, indirect=True)
-def test_text_to_text_001(omni_server, online_client) -> None:
+def test_text_to_text_001(omni_server, openai_client) -> None:
     request_config = {"model": omni_server.model, "messages": ..., "modalities": ["text"]}
-    online_client.send_omni_request(request_config)
+    openai_client.send_omni_request(request_config)
 ```
 
 *Online serving e2e — **TTS** (representative):*
@@ -244,9 +244,9 @@ def test_text_to_text_001(omni_server, online_client) -> None:
 @pytest.mark.tts
 @hardware_test(...)
 @pytest.mark.parametrize("omni_server", tts_server_params, indirect=True)
-def test_text_to_audio_001(omni_server, online_client) -> None:
+def test_text_to_audio_001(omni_server, openai_client) -> None:
     request_config = {"model": omni_server.model, "input": "...", "response_format": "wav", ...}
-    online_client.send_audio_speech_request(request_config)
+    openai_client.send_audio_speech_request(request_config)
 ```
 
 *Online serving e2e — **Diffusion** X2I (representative):*
@@ -255,8 +255,8 @@ def test_text_to_audio_001(omni_server, online_client) -> None:
 @pytest.mark.core_model
 @pytest.mark.diffusion
 @pytest.mark.parametrize("omni_server", _get_default_case(MODEL), indirect=True)
-def test_text_to_image_001(omni_server, online_client) -> None:
+def test_text_to_image_001(omni_server, openai_client) -> None:
-    online_client.send_diffusion_request({...})  # chat completions + extra_body
+    openai_client.send_diffusion_request({...})  # chat completions + extra_body
 ```
 
 *Online serving e2e — **Diffusion** X2V (representative):*
@@ -264,8 +264,8 @@ def test_text_to_image_001(omni_server, online_client) -> None:
 ```python
 @pytest.mark.core_model
 @pytest.mark.diffusion
-def test_text_to_video_001(omni_server, online_client) -> None:
+def test_text_to_video_001(omni_server, openai_client) -> None:
-    online_client.send_video_diffusion_request({"model": ..., "form_data": {...}})  # /v1/videos
+    openai_client.send_video_diffusion_request({"model": ..., "form_data": {...}})  # /v1/videos
 ```
 
 *Documentation example tests:* follow the **Preferred Test Strategy** in [l4_doc_example_tests.inc.md](../../../docs/contributing/ci/test_examples/l4_doc_example_tests.inc.md): dynamic extraction for offline READMEs; explicit copied client code for online pages until extraction is justified; use the documented **naming**, **output directory** (page folder + case id), and **skipping** rules (e.g. Gradio-only scripts).
@@ -274,7 +274,6 @@ def test_text_to_video_001(omni_server, online_client) -> None:
 
 **3. Cross-cutting rules**
 
-- New `.py` files need the Omni SPDX header (`Copyright contributors to the vLLM-Omni project`, not `vLLM project`). `tests/` may keep stdlib `re`/`base64`; pickle is still banned unless the file is already on the pickle allowlist. Every collected `tests/**/test_*.py` needs a CI **level** mark and a **hardware** platform mark/helper (`check-mark` runs locally; GHA skips it). Do not write `pytest.mark.H100` / `pytest.mark.L4`; use the helpers so `cards_{n}` is attached. Do not grow `allowed_files` to land a test. Policy: [docs/contributing/README.md](../../../docs/contributing/README.md#linting).
 - Reuse existing fixtures for the chosen scenario; do not mix “online client” assumptions into offline `OmniRunner` tests without a clear reason.
 - Avoid external network dependency in assertions unless the scenario is explicitly “online serving” or doc examples that require a model hub (then align with CI secrets/cache).
 - Keep **one test function = one intent** (one modality combo, one endpoint contract, or one acceleration combo).
@@ -324,21 +323,21 @@ vLLM-Omni e2e tests start a real **OmniServer** (online) or **OmniRunner** (offl
 
 | Scope | Online fixtures | Offline fixtures | When to use |
 |-------|-----------------|------------------|-------------|
-| **Module** (default) | `omni_server` → `online_client` | `omni_runner` → `offline_client` | **Default for L2/L3/L4 expansion** — amortize model/server init across `test_*` in the same module. Same `OmniServerParams` / runner config can be reused by multiple tests. |
+| **Module** (default) | `omni_server` → `openai_client` | `omni_runner` → `omni_runner_handler` | **Default for L2/L3/L4 expansion** — amortize model/server init across `test_*` in the same module. Same `OmniServerParams` / runner config can be reused by multiple tests. |
-| **Function** | `omni_server_function` → `online_client_function` | `omni_runner_function` → `offline_client_function` | **Only when required** — each `test_*` must get a **clean** server/runner (no shared engine/GPU state). Typical: `tests/dfx/reliability/`, sleep/wakeup, crash/restart, tests that mutate global server state. |
+| **Function** | `omni_server_function` → `openai_client_function` | `omni_runner_function` → `omni_runner_handler_function` | **Only when required** — each `test_*` must get a **clean** server/runner (no shared engine/GPU state). Typical: `tests/dfx/reliability/`, sleep/wakeup, crash/restart, tests that mutate global server state. |
 
 **Rules:**
 
 1. **Default to module scope** (`omni_server` / `omni_runner`) unless the scenario **explicitly** needs a fresh instance per test function.
-2. **Indirect parametrize name must match the fixture name**: `@pytest.mark.parametrize("omni_server", ...)` with `omni_server` + `online_client`; `@pytest.mark.parametrize("omni_server_function", ...)` with `omni_server_function` + `online_client_function`. Do not mix module fixture with function client (or vice versa).
+2. **Indirect parametrize name must match the fixture name**: `@pytest.mark.parametrize("omni_server", ...)` with `omni_server` + `openai_client`; `@pytest.mark.parametrize("omni_server_function", ...)` with `omni_server_function` + `openai_client_function`. Do not mix module fixture with function client (or vice versa).
 3. Different `OmniServerParams` per `test_*` (e.g. default vs `--enable-cpu-offload`) is still OK with **module** `omni_server` — pytest parametrizes per test node; only switch to `_function` when isolation between tests matters, not merely because `server_args` differ.
 4. One `test_*` with **many** server configs (expansion matrix) → single function + `@pytest.mark.parametrize("omni_server", [...], indirect=True)` + module `omni_server` (see in-tree `test_qwen_image_expansion.py`).
 
 ```python
 # Default — module-scoped server (L4 expansion)
 @pytest.mark.parametrize("omni_server", [pytest.param(OmniServerParams(model=MODEL), marks=H100)], indirect=True)
-def test_foo_images_generations_default_1024(omni_server, online_client) -> None:
+def test_foo_images_generations_default_1024(omni_server, openai_client) -> None:
-    online_client.send_images_generations_request({...})
+    openai_client.send_images_generations_request({...})
 
 # Function-scoped — reliability / per-test clean state only
 @pytest.mark.parametrize(
@@ -346,9 +345,9 @@ def test_foo_images_generations_default_1024(omni_server, online_client) -> None
     [pytest.param(OmniServerParams(model=MODEL), marks=H100)],
     indirect=True,
 )
-def test_foo_sleep_wakeup_cycle(omni_server_function, online_client_function) -> None:
+def test_foo_sleep_wakeup_cycle(omni_server_function, openai_client_function) -> None:
-    online_client_function.send_omni_sleep_http_request({...})
+    openai_client_function.send_omni_sleep_http_request({...})
-    online_client_function.send_omni_wakeup_http_request({...})
+    openai_client_function.send_omni_wakeup_http_request({...})
 ```
 
 #### Runtime send helpers — online **and** offline (`tests/helpers/runtime.py`)
@@ -357,19 +356,19 @@ def test_foo_sleep_wakeup_cycle(omni_server_function, online_client_function) ->
 
 | Principle | Action |
 |-----------|--------|
-| **Reuse first** | Grep `runtime.py` for an existing **`send_*_request`** on `OnlineOmniClient` (online) or `OfflineOmniClient` (offline) that matches the endpoint / pipeline shape. |
+| **Reuse first** | Grep `runtime.py` for an existing **`send_*_request`** on `OpenAIClientHandler` (online) or `OmniRunnerHandler` (offline) that matches the endpoint / pipeline shape. |
 | **Extend when close** | If an existing `send_*_request` almost fits (e.g. missing one optional field), extend it in `runtime.py` — do not fork logic in the test file. |
 | **Add when missing** | No suitable helper → add **`send_<feature>_request`** (high-level: call + general `assert_*`) or **`send_<route>_<verb>_http_request`** (low-level HTTP for negative/dfx) in **`runtime.py` first**, then call it from tests. |
 | **Test module owns payload only** | In `test_*.py`: `MODEL`, deploy path, vendored media, `get_prompt()`, and inline **`request_config` dicts** only. **No** `omni.generate(...)`, raw `requests.post`, OpenAI SDK calls, or `_collect_audio()` / `_process_output()` in e2e tests. |
 
-**Online** (`online_client` from `omni_server`): `OnlineOmniClient.send_*_request`.
+**Online** (`openai_client` from `omni_server`): `OpenAIClientHandler.send_*_request`.
 
-**Offline** (`offline_client` from `omni_runner`): `OfflineOmniClient.send_*_request`.
+**Offline** (`omni_runner_handler` from `omni_runner`): `OmniRunnerHandler.send_*_request`.
 
 | Do | Don't |
 |----|-------|
-| Online: `online_client.send_omni_request`, `send_diffusion_request`, `send_audio_speech_request`, `send_video_diffusion_request`, `send_images_generations_request`, … | `requests.post(f"{base_url}/v1/...", json=…)` or `client.chat.completions.create(...)` inside a test |
+| Online: `openai_client.send_omni_request`, `send_diffusion_request`, `send_audio_speech_request`, `send_video_diffusion_request`, `send_images_generations_request`, … | `requests.post(f"{base_url}/v1/...", json=…)` or `client.chat.completions.create(...)` inside a test |
-| Offline: `offline_client.send_omni_request`, `send_diffusion_request`, `send_audio_speech_request`, `send_single_stage_tts_request`, `send_single_stage_tts_batch_request`, … | `omni_runner.omni.generate(...)` + hand-rolled tensor/WAV extraction in `test_*.py` |
+| Offline: `omni_runner_handler.send_omni_request`, `send_diffusion_request`, `send_audio_speech_request`, `send_single_stage_tts_request`, `send_single_stage_tts_batch_request`, … | `omni_runner.omni.generate(...)` + hand-rolled tensor/WAV extraction in `test_*.py` |
 | Add missing `send_*` to **`runtime.py`** first; bundle general `assert_*` inside the send helper | A one-off `def _post_*` / `def _collect_audio` at the bottom of a test module |
 | Mirror naming/style of neighboring `send_*` (docstring, `request_config` dict, optional `run_level`, `err_code` / `err_message` for negative cases) | Different parameter shapes per test file for the same endpoint |
 
@@ -392,7 +391,7 @@ def test_foo_sleep_wakeup_cycle(omni_server_function, online_client_function) ->
 
 Changing `request_config` fields (`size`, `n`, `seed`, server flags) is **not** special validation — the general `assert_*` should read those from `request_config` inside `send_*`.
 
-**`send_*` ↔ `assert_*` pairing (online `OnlineOmniClient`):**
+**`send_*` ↔ `assert_*` pairing (online `OpenAIClientHandler`):**
 
 | High-level `send_*` (prefer in L2+ e2e) | Assert already invoked inside `runtime.py` | Low-level HTTP-only sibling |
 |----------------------------------------|--------------------------------------------|-----------------------------|
@@ -405,7 +404,7 @@ Changing `request_config` fields (`size`, `n`, `seed`, server flags) is **not**
 
 **Rule:** Tests call high-level `send_*_request` for the general contract. **Never** call the same bundled `assert_*` again. Call an extra `assert_*` in the test **only** for special, case-specific validation.
 
-**Common `OnlineOmniClient` entry points** (non-exhaustive — grep `runtime.py` before adding):
+**Common `OpenAIClientHandler` entry points** (non-exhaustive — grep `runtime.py` before adding):
 
 | Area | High-level (SDK + assert) | Low-level HTTP (`*_http_request`) |
 |------|---------------------------|-----------------------------------|
@@ -416,7 +415,7 @@ Changing `request_config` fields (`size`, `n`, `seed`, server flags) is **not**
 | TTS (online) | `send_audio_speech_request` | `send_audio_speech_http_request`, `send_audio_generate_http_request`, … |
 | Ops / meta | — | `send_health_http_request`, `send_models_http_request`, `send_omni_sleep_http_request`, … |
 
-**Common `OfflineOmniClient` entry points** (offline — grep `runtime.py` before adding):
+**Common `OmniRunnerHandler` entry points** (offline — grep `runtime.py` before adding):
 
 | Area | High-level (`send_*_request` + assert) | Notes |
 |------|----------------------------------------|-------|
@@ -426,7 +425,7 @@ Changing `request_config` fields (`size`, `n`, `seed`, server flags) is **not**
 | Single-stage TTS (Coqui XTTS, MOSS-TTS-Nano, …) | `send_single_stage_tts_request`, `send_single_stage_tts_batch_request` | `prompt` + `additional_information` + `omni.generate` — **do not** duplicate in tests |
 | New model family | **Add** `send_<family>_request` here first | Then call from `tests/e2e/offline_inference/test_*.py` |
 
-L1 tests under `tests/entrypoints/` may use **FastAPI `TestClient`** or direct handler calls with mocks; they do **not** need `OnlineOmniClient` / `OfflineOmniClient`. **L2+ online and offline e2e** must use `runtime.py` `send_*` helpers.
+L1 tests under `tests/entrypoints/` may use **FastAPI `TestClient`** or direct handler calls with mocks; they do **not** need `OpenAIClientHandler` / `OmniRunnerHandler`. **L2+ online and offline e2e** must use `runtime.py` `send_*` helpers.
 
 #### Invalid parameter validation (`tests/dfx/reliability/invalid_param_test/`)
 
@@ -452,7 +451,7 @@ When the user asks for **invalid parameter validation**, **invalid request bodie
 | **Module markers** | `pytestmark = [pytest.mark.slow, pytest.mark.<omni or tts or diffusion>]` |
 | **Server fixture** | `_PARAMS` / `_QWEN3_TTS_SPEECH`-style list of `pytest.param(OmniServerParams(...), id="...", marks=hardware_marks(...))`; `@pytest.mark.parametrize("omni_server", _PARAMS, indirect=True)` |
 | **Hardware** | `hardware_marks(res={"cuda": "H100"})` for heavy diffusion/omni/video; `hardware_marks(res={"cuda": "L4"})` for smaller TTS models (must match weekly `-m "slow and L4"` step) |
-| **HTTP client** | **Low-level** `online_client.send_*_http_request({..., "err_code": 400, "err_message": (...)} )` — **not** `send_*_request` (success path) |
+| **HTTP client** | **Low-level** `openai_client.send_*_http_request({..., "err_code": 400, "err_message": (...)} )` — **not** `send_*_request` (success path) |
 | **Case shape** | Prefer **one `test_*` per route family** + `@pytest.mark.parametrize("body_spec, err_message", [...])` with stable `id=` per case; or dedicated `test_<route>_malformed_json` when not parametrized |
 | **Body helpers** | `_minimal_<endpoint>_json(omni_server)` / `_minimal_*_form_data()`; merge overrides with `body.update(body_spec)` |
 | **Known gaps** | `pytest.mark.skip(reason="…#3649")` as `_SKIP_ISSUE_3649` when server validation is not yet strict (mirror neighboring cases) |
@@ -485,13 +484,13 @@ When the user asks for **invalid parameter validation**, **invalid request bodie
 @pytest.mark.parametrize("omni_server", _PARAMS, indirect=True)
 def test_images_generations_invalid_requests(
     omni_server: OmniServer,
-    online_client: OnlineOmniClient,
+    openai_client: OpenAIClientHandler,
     body_spec: dict[str, object],
     err_message: str | tuple[str, ...],
 ) -> None:
     body = _minimal_images_gen_json(omni_server)
     body.update(body_spec)
-    online_client.send_images_generations_http_request(
+    openai_client.send_images_generations_http_request(
         {"json": body, "timeout": 300, "err_code": 400, "err_message": err_message}
     )
 ```
@@ -542,18 +541,18 @@ def send_images_generations_request(self, request_config: dict[str, Any], ...) -
 
 # tests/e2e/.../test_foo_expansion.py — one case per test_* (no case_id branching)
 @pytest.mark.parametrize("omni_server", [pytest.param(OmniServerParams(model=MODEL), marks=SINGLE_L4)], indirect=True)
-def test_foo_images_generations_default_1024(omni_server, online_client) -> None:
+def test_foo_images_generations_default_1024(omni_server, openai_client) -> None:
-    online_client.send_images_generations_request({"json": body, "timeout": 300})
+    openai_client.send_images_generations_request({"json": body, "timeout": 300})
 
 @pytest.mark.parametrize("omni_server", [pytest.param(OmniServerParams(model=MODEL), marks=SINGLE_L4)], indirect=True)
-def test_foo_images_generations_sizes_256_512_1024(omni_server, online_client) -> None:
+def test_foo_images_generations_sizes_256_512_1024(omni_server, openai_client) -> None:
     for size in ("256x256", "512x512", "1024x1024"):
-        online_client.send_images_generations_request({"json": {**body, "size": size}, "timeout": 300})
+        openai_client.send_images_generations_request({"json": {**body, "size": size}, "timeout": 300})
 
 # chat diffusion — separate test; send_diffusion_request bundles assert_diffusion_response
 @pytest.mark.parametrize("omni_server", [pytest.param(OmniServerParams(model=MODEL), marks=SINGLE_L4)], indirect=True)
-def test_foo_chat_completions_t2i_fallback(omni_server, online_client) -> None:
+def test_foo_chat_completions_t2i_fallback(omni_server, openai_client) -> None:
-    online_client.send_diffusion_request({...})
+    openai_client.send_diffusion_request({...})
 ```
 
 L1 tests under `tests/entrypoints/` may keep **minimal** asserts next to the handler under test when validating a single branch; still prefer `assertions.py` when the same JSON/media check appears in more than one file or level (e.g. L1 protocol test + L4 e2e).

.claude/skills/vllm-omni-test/references/test-routing.md

@@ -79,7 +79,7 @@ E2E (L2+) should not rely on mocks unless documenting a rare exception; prefer r
    - `full_model` — L4 nightly (`test-nightly.yml`); some expansion tests still carry both `advanced_model` and `full_model` during migration
 2. **Model type** (pick one): `omni`, `tts`, or `diffusion`
 3. **Cross-cutting** when relevant: `parallel`, `cache`, `example`, `benchmark`
-4. **Hardware**: `cpu`, `cuda`, `rocm`, `npu`, `L4`, `H100`, `cards_1`, …
+4. **Hardware**: `cpu`, `cuda`, `rocm`, `npu`, `L4`, `H100`, `distributed_cuda`, …
 5. Multi-card: `@hardware_test(...)` in `tests/helpers/mark.py`
 
 ## Command Templates
@@ -244,7 +244,7 @@ pytest -s -v dfx/reliability/invalid_param_test/test_invalid_image_generation.py
 
 ```bash
 cd tests
-pytest -s -v -m "core_model and L4 and not cards_1" --run-level=core_model
+pytest -s -v -m "core_model and distributed_cuda and L4" --run-level=core_model
 ```
 
 ### Concrete e2e paths (common in-tree)
@@ -279,7 +279,7 @@ When adding or modifying tests, do not stop at “where the file lives” — al
 4. **API client + assert placement**: **General** validation → implement in `assertions.py`, call **inside** `send_*_request` in `runtime.py`; tests call **`send_*_request` only**. **Special** case validation → `assert_*` in `assertions.py`, called **in the test** after `send_*_request`. Low-level `send_*_http_request` is for negative/dfx tests (`err_code`), not ordinary L2+ success e2e.
 5. **Shared assertions**: logic in **`tests/helpers/assertions.py`**; general checks inside `send_*_request`; special checks only in the test. No `_assert_*` in `test_*.py`.
 6. **One case → one `test_*`**: function name reflects what is validated; no `if case_id == ...` mega-test merging multiple scenarios.
-7. **Fixture scope**: default **`omni_server` + `online_client`** / **`omni_runner` + `offline_client`** (module). Use **`omni_server_function` + `online_client_function`** / **`omni_runner_function` + `offline_client_function`** only when each `test_*` must spawn a fresh instance. Parametrize name must match fixture (`omni_server` vs `omni_server_function`).
+7. **Fixture scope**: default **`omni_server` + `openai_client`** / **`omni_runner` + `omni_runner_handler`** (module). Use **`omni_server_function` + `openai_client_function`** / **`omni_runner_function` + `omni_runner_handler_function`** only when each `test_*` must spawn a fresh instance. Parametrize name must match fixture (`omni_server` vs `omni_server_function`).
 8. **Type marker**: `omni`, `tts`, or `diffusion` on every model e2e module.
 9. **Diffusion L4 Function**: wire `*_expansion.py` into **X2I(&A&T)** or **X2V** **Function Test** in **`test-nightly.yml` only** — do not add `test-merge.yml` unless the user also requested L3.
 10. **Diffusion L4 Perf** (only when requested): add `tests/dfx/perf/tests/test_<slug>_vllm_omni.json` + **Perf Test · &lt;Model&gt;** step (artifact upload); not part of “L4 functional cases” by default.

.github/CODEOWNERS

@@ -79,6 +79,8 @@
 /vllm_omni/model_executor/models/ @tzhouam @gcanlin @Sy0307 @amy-why-3459 @linyueqian
 /vllm_omni/model_extras/ @tzhouam @gcanlin @princepride
 /vllm_omni/plugins/ @gcanlin @tjtanaa @xuechendi
+# Config override within model_executor — docs/design/module/vllm_omni_config.md
+/vllm_omni/model_executor/stage_configs/ @lishunyang12 @alex-jw-brooks
 
 # Execution platforms (hardware backends) — docs/design/module/execution_platforms.md
 /vllm_omni/platforms/ @gcanlin @tjtanaa @xuechendi

.github/ISSUE_TEMPLATE/400-bug-report.yml

@@ -83,7 +83,7 @@ body:
 
       omni = Omni(
           model="Qwen/Qwen-Image",
-          deploy_config="/path/to/deploy_config.yaml",
+          stage_configs_path="/path/to/stage_configs.yaml",
       )
 
       prompts = [{"prompt": "A scenic watercolor painting of a lighthouse at sunset"}]

.github/mergify.yml

@@ -45,7 +45,7 @@ pull_request_rules:
         > # For mypy (substitute "3.10" with the failing version if needed)
         > pre-commit run --hook-stage manual mypy-3.10
         > # For markdownlint
-        > pre-commit run markdownlint-cli2
+        > pre-commit run --hook-stage manual markdownlint
         > ```
         > </details>
 

.github/workflows/pre-commit.yml

@@ -24,7 +24,9 @@ jobs:
           python-version: "3.12"
           cache: 'pip'
       - uses: pre-commit/action@v3.0.1
-        # Don't run historical-debt hooks against the whole tree in CI yet.
+        # Don't run the pytest mark check against everything in the CI
-        # They still run locally on changed files.
+        # for now since a lot of files are still missing them. This check
+        # is used to help contributors remember to add the marks when they
+        # commit.
         env:
-          SKIP: check-mark,markdownlint-cli2,shellcheck,check-spdx-header,mypy-3.10
+          SKIP: check-test-ci-coverage

.gitignore

@@ -46,8 +46,6 @@ htmlcov/
 .cache
 nosetests.xml
 coverage.xml
-coverage-*.xml
-coverage-*.xml.gz
 *.cover
 *.py,cover
 .hypothesis/
@@ -242,9 +240,9 @@ data/
 datasets/
 *.csv
 *.json
-!recipes/cosmos3/negative_prompt.json
 !tests/dfx/perf/tests/*.json
 !apps/ComfyUI-vLLM-Omni/example_workflows/*.json
+!tests/assets/soulxsinger/*.json
 *.jsonl
 *.parquet
 
@@ -254,7 +252,6 @@ outputs/
 results/
 generated/
 output_*/
-/tests/e2e/accuracy/artifacts/
 
 # Configuration overrides
 configs/local.yaml
@@ -278,6 +275,3 @@ tests/.ci_generated/
 
 # Local Lance precision-matching dump scripts (not part of the repo)
 lance_compare/
-
-# Linting binaries downloaded by local pre-commit hooks
-shellcheck*/

.markdownlint.yaml

@@ -1,17 +0,0 @@
-# Keep these rules aligned with vLLM so MkDocs Material syntax is tolerated.
-MD007:
-  indent: 4
-MD013: false
-# GitHub READMEs start with a centered logo, not an H1.
-MD041: false
-MD024:
-  siblings_only: true
-MD025:
-  # Allow front matter title to be different from the first heading in the document.
-  front_matter_title: ""
-MD031:
-  list_items: false
-MD033: false
-MD046: false
-MD052: false
-MD059: false

.pre-commit-config.yaml

@@ -24,7 +24,7 @@ repos:
     - id: ruff-format
 
   - repo: https://github.com/crate-ci/typos
-    rev: v1.49.0
+    rev: typos-dict-v0.13.13
     hooks:
       - id: typos
         # only for staged files
@@ -38,19 +38,7 @@ repos:
       - id: actionlint
         files: ^\.github/workflows/.*\.ya?ml$
 
-  - repo: https://github.com/DavidAnson/markdownlint-cli2
-    rev: v0.23.2
-    hooks:
-      - id: markdownlint-cli2
-        language_version: lts
-        args: [--fix]
-        files: (^docs/|^recipes/|(^|/)README\.md$|^CONTRIBUTING\.md$)
-        # Agent skill docs and CLAUDE.md are not user-facing docs/READMEs.
-        exclude: (^|/)CLAUDE\.md$|^\.claude/|^\.cursor/
 
-  # Keep `local` last so `suggestion` stays the final hook. Autofix repos
-  # (ruff, typos, markdownlint, trailing-whitespace, etc.) must run first;
-  # otherwise their "files were modified" output displaces the SKIP tip.
   - repo: local
     hooks:
     - id: signoff-commit
@@ -66,47 +54,8 @@ repos:
       verbose: true
       stages: [commit-msg]
 
-    # TODO: Use https://github.com/pre-commit/mirrors-mypy when mypy setup
+    - id: check-test-ci-coverage
-    # is less awkward. Local wrapper groups files so follow_imports=skip on
+      name: Ensure test files have pytestmarks
-    # tests does not leak into library code.
-    # mypy-3.10 runs on git commit (changed files). CI SKIPs it because
-    # --all-files is not clean yet. Extra versions are manual-stage only
-    # (not in default `pre-commit run --all-files`):
-    #   pre-commit run --hook-stage manual mypy-3.12
-    - id: mypy-3.10
-      name: Run mypy for Python 3.10
-      entry: python tools/pre_commit/mypy.py "3.10"
-      <<: &mypy_common
-        language: python
-        types_or: [python, pyi]
-        require_serial: true
-        additional_dependencies:
-          - mypy==1.11.1
-          - regex
-          - types-cachetools
-          - types-setuptools
-          - types-PyYAML
-          - types-requests
-          - types-torch
-          - pydantic
-    - id: mypy-3.11
-      name: Run mypy for Python 3.11
-      entry: python tools/pre_commit/mypy.py "3.11"
-      <<: *mypy_common
-      stages: [manual]
-    - id: mypy-3.12
-      name: Run mypy for Python 3.12
-      entry: python tools/pre_commit/mypy.py "3.12"
-      <<: *mypy_common
-      stages: [manual]
-    - id: mypy-3.13
-      name: Run mypy for Python 3.13
-      entry: python tools/pre_commit/mypy.py "3.13"
-      <<: *mypy_common
-      stages: [manual]
-
-    - id: check-mark
-      name: Ensure test files have CI marks (no direct SKU)
       entry: python tools/pre_commit/check_test_marks.py
       language: python
       types: [python]
@@ -117,50 +66,12 @@ repos:
       entry: python tools/pre_commit/check_tts_adapter.py
       language: python
       pass_filenames: false
-      files: ^(vllm_omni/entrypoints/openai/serving_speech\.py|tools/pre_commit/check_tts_adapter\.py)$
+      files: ^(vllm_omni/entrypoints/openai/(serving_speech\.py|tts_adapters/__init__\.py)|tools/pre_commit/check_tts_adapter\.py)$
       # The budget-is-now-slack reminder is printed on a PASSING run, and
       # pre-commit hides output from passing hooks unless the hook is verbose.
       verbose: true
 
-    - id: shellcheck
+    # Keep `suggestion` last
-      name: Lint shell scripts
-      # language: script breaks on Windows (shebang path loses slashes).
-      # Invoke the vLLM-style wrapper with bash so Linux CI and Win/Git Bash work.
-      entry: bash
-      args: [tools/pre_commit/shellcheck.sh]
-      language: system
-      types: [shell]
-
-    - id: check-spdx-header
-      name: Check SPDX headers
-      entry: python tools/pre_commit/check_spdx_header.py
-      language: python
-      types_or: [python, pyi, rust, proto, shell]
-      files: \.(py|pyi|rs|proto|sh)$
-
-    - id: check-forbidden-imports
-      name: Check forbidden imports
-      entry: python tools/pre_commit/check_forbidden_imports.py
-      language: python
-      types: [python]
-      additional_dependencies: [regex]
-
-    - id: check-torch-cuda-call
-      name: Prevent new torch.cuda API calls
-      entry: python tools/pre_commit/check_torch_cuda.py
-      language: python
-      types: [python]
-      additional_dependencies: [regex]
-
-    - id: check-buildkite
-      name: Validate Buildkite Pipelines
-      entry: python tools/pre_commit/check_buildkite.py
-      language: python
-      files: ^\.buildkite/.*\.ya?ml$
-      additional_dependencies: [pyyaml, check-jsonschema==0.38.0]
-
-    # Keep `suggestion` as the final hook. Run after autofix hooks so the
-    # bypass/SKIP tip is not displaced.
     - id: suggestion
       name: Suggestion
       entry: bash -c 'echo "To bypass all the pre-commit hooks, add --no-verify to git commit. To skip a specific hook, prefix the commit command with SKIP=<hook-id>."'
@@ -168,3 +79,10 @@ repos:
       verbose: true
       pass_filenames: false
     # Insert new entries above the `suggestion` entry
+
+    - id: check-pickle-imports
+      name: Prevent new pickle/cloudpickle imports
+      entry: python tools/pre_commit/check_pickle_imports.py
+      language: python
+      types: [python]
+      additional_dependencies: [regex]

.readthedocs.yml

@@ -11,15 +11,10 @@ build:
     post_checkout:
       - git fetch --unshallow || true
     pre_install:
-      # `pre_install` runs before RTD upgrades pip, and the seeded pip 23.1
-      # rejects the `typing_extensions` wheel torch needs (fixed in pip 23.2).
-      - python -m pip install --upgrade pip
       # Pre-install CPU torch so the subsequent `pip install .[docs]` does not
       # pull ~2 GB of CUDA 13 libs (cuBLAS/cuDNN/NCCL/...) that exceed RTD's
-      # 1 GB memory cap and time out the mkdocs build. Wheel-only: `--index-url`
+      # 1 GB memory cap and time out the mkdocs build.
-      # replaces PyPI, so an sdist fallback cannot reach its build deps. The
+      - python -m pip install --index-url https://download.pytorch.org/whl/cpu torch
-      # quotes keep YAML from reading `:all:` as a mapping key.
-      - "python -m pip install --only-binary=:all: --index-url https://download.pytorch.org/whl/cpu torch"
 
 mkdocs:
   configuration: mkdocs.yml

CONTRIBUTING.md

@@ -1,11 +1,5 @@
 # Contributing to vLLM-Omni
 
-You may find information about contributing to vLLM-Omni on [Contributing](https://vllm-omni.readthedocs.io/en/latest/contributing/).
+You may find information about contributing to vLLM-Omni on [Contributing](https://vllm-omni.readthedocs.io/en/latest/contributing/)
-
-Local `pre-commit` gates (SPDX, forbidden imports, `torch.cuda`, shellcheck,
-mypy, test marks, TTS adapter ratchet, markdownlint, Buildkite schema, and
-how to extend allowlists) are documented in the
-[Linting](https://vllm-omni.readthedocs.io/en/latest/contributing/#linting)
-section of that guide.
 
 Before submitting a PR, run the [precheck-pr skill](https://github.com/vllm-project/vllm-omni/blob/main/.claude/skills/precheck-pr/SKILL.md) with the code agent for a self-check against project conventions.

README.md

@@ -12,12 +12,10 @@ Easy, fast, and cheap omni-modality model serving for everyone
 | <a href="https://vllm-omni.readthedocs.io/en/latest/"><b>Documentation</b></a> | <a href="https://deepwiki.com/vllm-project/vllm-omni"><b>DeepWiki</b></a> | <a href="https://discuss.vllm.ai"><b>User Forum</b></a> | <a href="https://slack.vllm.ai"><b>Developer Slack</b></a> | <a href="docs/assets/WeChat.jpg"><b>WeChat</b></a> | <a href="https://arxiv.org/abs/2602.02204"><b>Paper</b></a> | <a href="https://docs.google.com/presentation/d/1aPj0OGl_-ZVoib-Qne5dGDAlrRFB-PdHl6E-EE99g8E/edit?usp=sharing"><b>Slides</b></a> |
 </p>
 
+
 ---
 
 *Latest News* 🔥
-
-- [2026/08] We released [0.28.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.28.0), featuring production-ready [MiniMax H3](recipes/MiniMaxAI/MiniMax-H3.md) serving on GPU and NPU, a unified AR/DiT paged KV cache runtime, and enhanced realtime full-duplex serving for the [MiniCPM-o series](recipes/OpenBMB/MiniCPM-o-4_5.md).
-- [2026/08] [VeRL-Omni](https://github.com/verl-project/verl-omni) `v0.2.0` is released: faster diffusion RL powered by vLLM-Omni (request-level/step-wise batching with FA3), rebuilt Qwen3-Omni multimodal training (DPO & GSPO), plus LTX-2.3, Qwen-Image-Edit support and more. See the [release notes](https://github.com/verl-project/verl-omni/releases/tag/v0.2.0).
 - [2026/08] We released [0.26.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.26.0) - aligned with the vLLM 0.26 release line, featuring [MiniMax H3](recipes/MiniMaxAI/MiniMax-H3.md) joint video/audio generation, an experimental full-duplex realtime runtime for [MiniCPM-o 4.5](recipes/OpenBMB/MiniCPM-o-4_5.md), distributed layerwise diffusion offload, and broader model, hardware, streaming, TTS, and quantization support.
 - [2026/07] We released [0.24.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.24.0) - aligned with the vLLM 0.24 release line, expanding production-ready coverage across TTS, speech, diffusion, image/video generation, and robot-policy serving, with major Omni stage runtime refactoring, diffusion request-level batching, async output materialization, quantization/cache/memory improvements, and broad CUDA/ROCm/XPU/NPU support.
 - [2026/06] Starting with [0.14.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.14.0), vLLM-Omni publishes a stable release aligned with every even-numbered upstream vLLM minor version. [0.16.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.16.0), [0.18.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.18.0), [0.20.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.20.0), and [0.22.0](https://github.com/vllm-project/vllm-omni/releases/tag/v0.22.0) continued this cadence, expanding omni and world-model support with [NVIDIA Cosmos3](recipes/cosmos3/Cosmos3-Nano.md) and DreamZero, adding models such as MiniCPM-o 4.5, MOSS-TTS, and Lance, and advancing TTS, diffusion, distributed execution, quantization, RL integration through [VeRL-Omni](https://github.com/verl-project/verl-omni), and CUDA/ROCm/MUSA/NPU/XPU coverage.
@@ -58,9 +56,9 @@ vLLM-Omni is flexible and easy to use with:
 vLLM-Omni seamlessly supports most popular open-source models on HuggingFace, including:
 
 - **Omni-modality models** (e.g. Qwen3-Omni, MiniCPM-o 4.5, Cosmos3, HunyuanImage, BAGEL)
-- **TTS models** (e.g. Qwen3-TTS, IndexTTS 2.5, dots.tts, CosyVoice3)
+- **TTS models** (e.g. Qwen3-TTS, VoxCPM2, Ming-Omni-TTS, CosyVoice3)
-- **Diffusion models** — image, video, and audio generation (e.g. MiniMax H3, LTX-2.5, SANA-Video, Wan2.2)
+- **Diffusion models** — image, video, and audio generation (e.g. MiniMax H3, Qwen-Image, Wan2.2, FLUX)
-- **Robot-policy and action models** (e.g. π0, GR00T-N1.7, DreamZero-DROID, InternVLA-A1)
+- **Robot-policy and action models** (e.g. GR00T-N1.7, DreamZero-DROID, InternVLA-A1, Cosmos3 action policy)
 
 ## Getting Started
 
@@ -68,7 +66,6 @@ Visit our [documentation](https://vllm-omni.readthedocs.io/en/latest/) to learn
 
 - [Installation](https://vllm-omni.readthedocs.io/en/latest/getting_started/installation/)
 - [Quickstart](https://vllm-omni.readthedocs.io/en/latest/getting_started/quickstart/)
-- [Nightly CUDA Docker images](https://hub.docker.com/r/vllm/vllm-omni/tags?name=nightly) built from the latest `main`
 - [List of Supported Models](https://vllm-omni.readthedocs.io/en/latest/models/supported_models/)
 - [Deployment Recipes](https://recipes.vllm.ai) for vLLM-Omni model serving
 
@@ -91,18 +88,11 @@ If you use vLLM-Omni for your research, please cite our [paper](https://arxiv.or
 ```
 
 ## Join the Community
-
 Feel free to ask questions, provide feedbacks and discuss with fellow users of vLLM-Omni in `#sig-omni` slack channel at [slack.vllm.ai](https://slack.vllm.ai) or vLLM user forum at [discuss.vllm.ai](https://discuss.vllm.ai).
 
 ## Star History
 
-<a href="https://www.star-history.com/?repos=vllm-project%2Fvllm-omni&type=date&legend=top-left">
+[![Star History Chart](https://api.star-history.com/svg?repos=vllm-project/vllm-omni&type=date&legend=top-left)](https://www.star-history.com/#vllm-project/vllm-omni&type=date&legend=top-left)
-  <picture>
-    <source media="(prefers-color-scheme: dark)" srcset="https://api.star-history.com/chart?repos=vllm-project/vllm-omni&type=date&theme=dark&legend=top-left&sealed_token=ExgLDZJoQEg27Zfhhut2LqN0GYO6Fw2PWLwPE6JYBUp2BgM3hmsYlwaIVopnUEfbRXidQ4nisumrTdKYydiKhy1SZXipw47qY2_tiUDhCpsPXeXtPuEVKVzBwKs3pw0tiHsJgtSfwXx5yjHXck0Y2SblzFWeJYCkTe1WLGTbUAOIETjXXQJjyCGZvKz5" />
-    <source media="(prefers-color-scheme: light)" srcset="https://api.star-history.com/chart?repos=vllm-project/vllm-omni&type=date&legend=top-left&sealed_token=ExgLDZJoQEg27Zfhhut2LqN0GYO6Fw2PWLwPE6JYBUp2BgM3hmsYlwaIVopnUEfbRXidQ4nisumrTdKYydiKhy1SZXipw47qY2_tiUDhCpsPXeXtPuEVKVzBwKs3pw0tiHsJgtSfwXx5yjHXck0Y2SblzFWeJYCkTe1WLGTbUAOIETjXXQJjyCGZvKz5" />
-    <img alt="Star History Chart" src="https://api.star-history.com/chart?repos=vllm-project/vllm-omni&type=date&legend=top-left&sealed_token=ExgLDZJoQEg27Zfhhut2LqN0GYO6Fw2PWLwPE6JYBUp2BgM3hmsYlwaIVopnUEfbRXidQ4nisumrTdKYydiKhy1SZXipw47qY2_tiUDhCpsPXeXtPuEVKVzBwKs3pw0tiHsJgtSfwXx5yjHXck0Y2SblzFWeJYCkTe1WLGTbUAOIETjXXQJjyCGZvKz5" />
-  </picture>
-</a>
 
 ## License
 

benchmarks/diffusion/quantization_quality.py

@@ -158,6 +158,7 @@ def _build_omni_kwargs(args, quantization=None):
 def _generate_image(omni, args, prompt, seed):
     """Generate a single image and return (PIL.Image, time_seconds, memory_gib)."""
     from vllm_omni.inputs.data import OmniDiffusionSamplingParams
+    from vllm_omni.outputs import OmniRequestOutput
     from vllm_omni.platforms import current_omni_platform
 
     generator = torch.Generator(device=current_omni_platform.device_type).manual_seed(seed)
@@ -175,7 +176,7 @@ def _generate_image(omni, args, prompt, seed):
     elapsed = time.perf_counter() - start
     peak_mem = torch.accelerator.max_memory_allocated() / (1024**3)
 
-    req_out = outputs[0] if isinstance(outputs, list) else outputs
+    req_out = OmniRequestOutput.unwrap_result(outputs)
     if not req_out.images:
         raise ValueError("Could not extract image output from result.")
     img = req_out.images[0]
@@ -205,15 +206,15 @@ def _generate_video(omni, args, prompt, seed):
     elapsed = time.perf_counter() - start
     peak_mem = torch.accelerator.max_memory_allocated() / (1024**3)
 
-    if isinstance(outputs, list) and isinstance(outputs[0], OmniRequestOutput):
+    first = outputs[0]
-        first = outputs[0]
+    if hasattr(first, "request_output") and isinstance(first.request_output, list):
-    elif isinstance(outputs, OmniRequestOutput):
+        inner = first.request_output[0]
-        first = outputs
+        if isinstance(inner, OmniRequestOutput) and hasattr(inner, "images"):
-    else:
+            frames = inner.images[0] if inner.images else None
-        raise ValueError("Could not extract video frames from output.")
+        else:
-
+            frames = inner
-    if hasattr(first, "images"):
+    elif hasattr(first, "images") and first.images:
-        frames = first.images[0] if first.images else None
+        frames = first.images[0]
     else:
         raise ValueError("Could not extract video frames from output.")
 

benchmarks/glm_image/vllm-omni/inference.py

@@ -300,7 +300,7 @@ def benchmark(args: argparse.Namespace) -> None:
         output_idx = 0
         for stage_outputs in omni.generate(all_prompts, sampling_params_list, py_generator=True):
             if stage_outputs.final_output_type == "image":
-                request_output = stage_outputs
+                request_output = stage_outputs.request_output
                 request_id = getattr(request_output, "request_id", "")
 
                 images = getattr(request_output, "images", [])

benchmarks/kernels/benchmark_diffusion_lora_expand.py

@@ -1,156 +0,0 @@
-# SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
-"""Benchmark diffusion LoRA expand-and-accumulate implementations.
-
-This compares the previous functional expand path with direct accumulation into
-the selected output slice. It is a synthetic kernel microbenchmark and does not
-load a diffusion model.
-
-Example:
-    python benchmarks/kernels/benchmark_diffusion_lora_expand.py \
-        --m 1,16,128,512,4096,16384 \
-        --in-dim 4096 --out-dim 4096 --rank 64 \
-        --dtype bfloat16 --packed --warmup 30 \
-        --small-iterations 500 --large-iterations 100
-"""
-
-import argparse
-import math
-import statistics
-from collections.abc import Callable
-
-import torch
-
-MiB = 1024**2
-
-
-def parse_args() -> argparse.Namespace:
-    parser = argparse.ArgumentParser(description=__doc__)
-    parser.add_argument("--m", default="1,16,128,512,4096,16384")
-    parser.add_argument("--in-dim", type=int, default=4096)
-    parser.add_argument("--out-dim", type=int, default=4096)
-    parser.add_argument("--rank", type=int, default=64)
-    parser.add_argument("--dtype", choices=("bfloat16", "float16"), default="bfloat16")
-    parser.add_argument(
-        "--packed",
-        action="store_true",
-        help="Accumulate into the second half of a packed output tensor.",
-    )
-    parser.add_argument("--warmup", type=int, default=30)
-    parser.add_argument("--small-iterations", type=int, default=500)
-    parser.add_argument("--large-iterations", type=int, default=100)
-    parser.add_argument("--large-m-threshold", type=int, default=1024)
-    return parser.parse_args()
-
-
-def percentile(samples: list[float], probability: float) -> float:
-    ordered = sorted(samples)
-    position = probability * (len(ordered) - 1)
-    lower = math.floor(position)
-    upper = math.ceil(position)
-    if lower == upper:
-        return ordered[lower]
-    weight = position - lower
-    return ordered[lower] * (1 - weight) + ordered[upper] * weight
-
-
-def measure_latency_ms(operation: Callable[[], None], warmup: int, iterations: int) -> list[float]:
-    for _ in range(warmup):
-        operation()
-    torch.accelerator.synchronize()
-
-    starts = [torch.cuda.Event(enable_timing=True) for _ in range(iterations)]
-    ends = [torch.cuda.Event(enable_timing=True) for _ in range(iterations)]
-    for start, end in zip(starts, ends, strict=True):
-        start.record()
-        operation()
-        end.record()
-    torch.accelerator.synchronize()
-    return [start.elapsed_time(end) for start, end in zip(starts, ends, strict=True)]
-
-
-def measure_extra_peak_mib(operation: Callable[[], None]) -> float:
-    torch.accelerator.synchronize()
-    torch.accelerator.reset_peak_memory_stats()
-    allocated = torch.accelerator.memory_allocated()
-    operation()
-    torch.accelerator.synchronize()
-    peak = torch.accelerator.max_memory_allocated()
-    return (peak - allocated) / MiB
-
-
-def print_header(args: argparse.Namespace, dtype: torch.dtype) -> None:
-    device_name = torch.cuda.get_device_name()
-    print(f"device={device_name} torch={torch.__version__} dtype={dtype}")
-    print(f"in_dim={args.in_dim} out_dim={args.out_dim} rank={args.rank} packed={args.packed} warmup={args.warmup}")
-    print("M       iters  old_p10  old_p50  old_p90  new_p10  new_p50  new_p90  speedup  old_MiB  new_MiB  max_abs")
-
-
-def benchmark_shape(m: int, args: argparse.Namespace, dtype: torch.dtype) -> None:
-    device = torch.device("cuda")
-    output_width = args.out_dim * (2 if args.packed else 1)
-    offset = args.out_dim if args.packed else 0
-    output_slice = slice(offset, offset + args.out_dim)
-    scale = 1 / math.sqrt(args.in_dim)
-
-    x = torch.randn(m, args.in_dim, device=device, dtype=dtype)
-    lora_a = torch.randn(args.rank, args.in_dim, device=device, dtype=dtype) * scale
-    lora_b = torch.randn(args.out_dim, args.rank, device=device, dtype=dtype) / math.sqrt(args.rank)
-    initial_output = torch.randn(m, output_width, device=device, dtype=dtype)
-
-    def baseline(output: torch.Tensor) -> None:
-        buffer = x @ lora_a.t()
-        delta = buffer @ lora_b.t()
-        selected = output[:, output_slice]
-        selected[:] = selected + delta
-
-    def optimized(output: torch.Tensor) -> None:
-        buffer = x @ lora_a.t()
-        selected = output[:, output_slice]
-        torch.addmm(selected, buffer, lora_b.t(), out=selected)
-
-    reference = initial_output.clone()
-    candidate = initial_output.clone()
-    baseline(reference)
-    optimized(candidate)
-    torch.testing.assert_close(candidate, reference, rtol=1e-2, atol=1e-2)
-    max_abs = (candidate - reference).abs().max().item()
-
-    old_output = initial_output.clone()
-    new_output = initial_output.clone()
-    iterations = args.large_iterations if m >= args.large_m_threshold else args.small_iterations
-
-    old_peak = measure_extra_peak_mib(lambda: baseline(old_output))
-    new_peak = measure_extra_peak_mib(lambda: optimized(new_output))
-    old_samples = measure_latency_ms(lambda: baseline(old_output), args.warmup, iterations)
-    new_samples = measure_latency_ms(lambda: optimized(new_output), args.warmup, iterations)
-
-    old_p50 = statistics.median(old_samples)
-    new_p50 = statistics.median(new_samples)
-    print(
-        f"{m:<7} {iterations:<6} "
-        f"{percentile(old_samples, 0.1):>8.5f} {old_p50:>8.5f} {percentile(old_samples, 0.9):>8.5f} "
-        f"{percentile(new_samples, 0.1):>8.5f} {new_p50:>8.5f} {percentile(new_samples, 0.9):>8.5f} "
-        f"{old_p50 / new_p50:>8.3f} {old_peak:>8.2f} {new_peak:>8.2f} {max_abs:>8.5f}"
-    )
-
-
-def main() -> None:
-    args = parse_args()
-    if not torch.cuda.is_available():
-        raise RuntimeError("This benchmark requires a CUDA GPU")
-    if args.warmup < 0 or args.small_iterations <= 0 or args.large_iterations <= 0:
-        raise ValueError("warmup must be non-negative and iteration counts must be positive")
-
-    dtype = getattr(torch, args.dtype)
-    torch.manual_seed(0)
-    print_header(args, dtype)
-    with torch.inference_mode():
-        for m in (int(value) for value in args.m.split(",")):
-            if m <= 0:
-                raise ValueError(f"M must be positive, got {m}")
-            benchmark_shape(m, args, dtype)
-
-
-if __name__ == "__main__":
-    main()

benchmarks/lingbot_video/dense_pipeline_parity.py

@@ -167,11 +167,9 @@ def main() -> None:
     ]
     native_cmd = [
         sys.executable,
-        "examples/offline_inference/text_to_video/text_to_video.py",
+        "examples/offline_inference/text_to_video/text_to_video_lingbot.py",
         "--model",
         args.model,
-        "--model-class-name",
-        "LingBotVideoPipeline",
         "--prompt",
         args.prompt,
         "--output",

benchmarks/tts/README.md

@@ -18,8 +18,8 @@ vllm serve Qwen/Qwen3-TTS-12Hz-1.7B-Base --omni --port 8000
 ```
 
 The server auto-loads its Deploy YAML from `vllm_omni/deploy/qwen3_tts.yaml`
-(Pipeline + Deploy schema introduced in #2383). No explicit `--deploy-config`
+(Pipeline + Deploy schema introduced in #2383). No `--stage-configs-path` or
-flag is needed for a registered model.
+`--deploy-config` flag is needed for any registered model.
 
 ### 2. Run the benchmark (`vllm bench serve --omni`)
 

benchmarks/tts/benchmark_cosyvoice3_trt_streams.py

@@ -1,151 +0,0 @@
-"""Benchmark CosyVoice3-style CUDA stream handoff strategies."""
-
-import argparse
-import json
-import statistics
-import time
-
-import torch
-
-from vllm_omni.platforms import current_omni_platform
-
-
-def run_case(
-    *,
-    device: torch.device,
-    steps: int,
-    warmup_steps: int,
-    producer_cycles: int,
-    estimator_cycles: int,
-    use_stream_dependencies: bool,
-) -> dict[str, float | int | str]:
-    caller_stream = torch.cuda.Stream(device=device)
-    estimator_stream = torch.cuda.Stream(device=device)
-    input_tensor = torch.zeros(1024, device=device)
-    output_tensor = torch.empty_like(input_tensor)
-    submission_ms = []
-
-    def step(measure: bool) -> None:
-        with torch.cuda.stream(caller_stream):
-            torch.cuda._sleep(producer_cycles)
-            input_tensor.add_(1)
-
-        start = time.perf_counter_ns()
-        if use_stream_dependencies:
-            estimator_stream.wait_stream(caller_stream)
-        else:
-            caller_stream.synchronize()
-
-        with torch.cuda.stream(estimator_stream):
-            torch.cuda._sleep(estimator_cycles)
-            output_tensor.copy_(input_tensor)
-
-        if use_stream_dependencies:
-            caller_stream.wait_stream(estimator_stream)
-        else:
-            estimator_stream.synchronize()
-
-        if measure:
-            submission_ms.append((time.perf_counter_ns() - start) / 1e6)
-
-    for _ in range(warmup_steps):
-        step(measure=False)
-    caller_stream.synchronize()
-    estimator_stream.synchronize()
-    input_tensor.zero_()
-    torch.accelerator.reset_peak_memory_stats(device)
-    initial_allocated = torch.accelerator.memory_allocated(device)
-
-    start_event = torch.cuda.Event(enable_timing=True)
-    end_event = torch.cuda.Event(enable_timing=True)
-    wall_start = time.perf_counter()
-    with torch.cuda.stream(caller_stream):
-        start_event.record()
-    for _ in range(steps):
-        step(measure=True)
-    with torch.cuda.stream(caller_stream):
-        end_event.record()
-    end_event.synchronize()
-    wall_ms = (time.perf_counter() - wall_start) * 1e3
-    peak_delta_bytes = torch.accelerator.max_memory_allocated(device) - initial_allocated
-
-    expected = torch.full_like(output_tensor, steps)
-    torch.testing.assert_close(output_tensor, expected, rtol=0, atol=0)
-    return {
-        "mode": "wait_stream" if use_stream_dependencies else "synchronize",
-        "steps": steps,
-        "submission_median_ms": statistics.median(submission_ms),
-        "submission_p95_ms": statistics.quantiles(submission_ms, n=20)[18],
-        "submission_mean_ms": statistics.mean(submission_ms),
-        "gpu_elapsed_ms": start_event.elapsed_time(end_event),
-        "wall_elapsed_ms": wall_ms,
-        "peak_allocation_delta_bytes": peak_delta_bytes,
-        "final_value": int(output_tensor[0].item()),
-    }
-
-
-def parse_args() -> argparse.Namespace:
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--device", default="cuda:0")
-    parser.add_argument("--steps", type=int, default=100)
-    parser.add_argument("--warmup-steps", type=int, default=10)
-    parser.add_argument("--repeats", type=int, default=7)
-    parser.add_argument("--producer-cycles", type=int, default=1_000_000)
-    parser.add_argument("--estimator-cycles", type=int, default=2_000_000)
-    return parser.parse_args()
-
-
-def main() -> None:
-    args = parse_args()
-    device = torch.device(args.device)
-    current_omni_platform.set_device(device)
-    torch.manual_seed(0)
-
-    old_results = []
-    new_results = []
-    for repeat in range(args.repeats):
-        common_args = {
-            "device": device,
-            "steps": args.steps,
-            "warmup_steps": args.warmup_steps,
-            "producer_cycles": args.producer_cycles,
-            "estimator_cycles": args.estimator_cycles,
-        }
-        if repeat % 2 == 0:
-            old_results.append(run_case(**common_args, use_stream_dependencies=False))
-            new_results.append(run_case(**common_args, use_stream_dependencies=True))
-        else:
-            new_results.append(run_case(**common_args, use_stream_dependencies=True))
-            old_results.append(run_case(**common_args, use_stream_dependencies=False))
-
-    old_submission_ms = statistics.median(result["submission_median_ms"] for result in old_results)
-    new_submission_ms = statistics.median(result["submission_median_ms"] for result in new_results)
-    old_gpu_ms = statistics.median(result["gpu_elapsed_ms"] for result in old_results)
-    new_gpu_ms = statistics.median(result["gpu_elapsed_ms"] for result in new_results)
-    print(
-        json.dumps(
-            {
-                "device": torch.cuda.get_device_name(device),
-                "torch": torch.__version__,
-                "cuda": torch.version.cuda,
-                "repeats": args.repeats,
-                "warmup_steps": args.warmup_steps,
-                "producer_cycles": args.producer_cycles,
-                "estimator_cycles": args.estimator_cycles,
-                "old_submission_median_ms": old_submission_ms,
-                "new_submission_median_ms": new_submission_ms,
-                "submission_median_speedup": old_submission_ms / new_submission_ms,
-                "old_gpu_elapsed_median_ms": old_gpu_ms,
-                "new_gpu_elapsed_median_ms": new_gpu_ms,
-                "gpu_elapsed_change_percent": (new_gpu_ms - old_gpu_ms) / old_gpu_ms * 100,
-                "parity": all(result["final_value"] == args.steps for result in old_results + new_results),
-                "old_runs": old_results,
-                "new_runs": new_results,
-            },
-            indent=2,
-        )
-    )
-
-
-if __name__ == "__main__":
-    main()

benchmarks/tts/model_configs.yaml

@@ -132,14 +132,3 @@ models:
     endpoint: /v1/audio/speech
     task_extra_body:
       voice_design: {}
-
-  # OmniVoice serves plain text-to-speech over /v1/audio/speech; voice cloning
-  # and voice design are offline-only for this model, so default_voice is the
-  # only task the server can answer.
-  k2-fsa/OmniVoice:
-    supported_tasks: [default_voice]
-    backend: openai-audio-speech
-    endpoint: /v1/audio/speech
-    trust_remote_code: true
-    task_extra_body:
-      default_voice: {}

collect_env.py

@@ -1,5 +1,5 @@
 # SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
+# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
 
 # ruff: noqa
 # code borrowed from https://github.com/pytorch/pytorch/blob/main/torch/utils/collect_env.py
@@ -494,16 +494,6 @@ def is_xnnpack_available():
 
 
 def get_env_vars():
-    try:
-        from vllm_omni.config.environment_variable_inventory import ENVIRONMENT_VARIABLE_INVENTORY
-    except Exception:
-        # Keep environment collection usable when an incomplete/broken Omni
-        # installation is the reason the user is running this diagnostic.
-        # Importing an Omni submodule executes package initialization, which
-        # may fail with RuntimeError, AssertionError, or OSError as well as
-        # ImportError when vLLM or the accelerator stack is broken.
-        ENVIRONMENT_VARIABLE_INVENTORY = {}
-
     env_vars = ""
     secret_terms = ("secret", "token", "api", "access", "password")
     report_prefix = (
@@ -517,17 +507,12 @@ def get_env_vars():
         "MKL_",
         "NVIDIA",
     )
-    for k, v in sorted(os.environ.items()):
+    for k, v in os.environ.items():
-        omni_classification = ENVIRONMENT_VARIABLE_INVENTORY.get(k)
+        if any(term in k.lower() for term in secret_terms):
-        if (omni_classification is not None and omni_classification.redact_value) or any(
-            term in k.lower() for term in secret_terms
-        ):
             continue
-        if (
+        if k in environment_variables:
-            k in environment_variables
+            env_vars = env_vars + "{}={}".format(k, v) + "\n"
-            or (omni_classification is not None and omni_classification.is_public_omni)
+        if k.startswith(report_prefix):
-            or k.startswith(report_prefix)
-        ):
             env_vars = env_vars + "{}={}".format(k, v) + "\n"
 
     return env_vars

docker/Dockerfile.ci

@@ -1,4 +1,4 @@
-# Target vllm version: v0.28.0 (tag v0.28.0 @ 2cf0a6915ce5).
+# Target vllm version: v0.27.0 (releases/v0.27.0 branch).
 # Keep VLLM_BASE_TAG aligned with the target vLLM release on every rebase.
 #
 # Release-boundary form: there is no precompiled-wheel pin here, because the
@@ -14,14 +14,11 @@
 # core init), registering libnvJitLink.so.13 with ldconfig, and re-pinning numpy
 # under numba's ceiling. All of it was compensation for a base image and a vLLM
 # that disagreed. At a real release boundary they agree, so none of it is
-# needed. The 0.28 rebase repeated the cycle: the v0.28.0rc / final-tag wheels
+# needed. If you pin an untagged commit again, restore the reinstall block —
-# rode on the v0.27.0 base image with the full reinstall block until the
+# agent/tasks/20_update_wheel_url.sh fails the run rather than letting the
-# official v0.28.0 image shipped, at which point this file dropped back to the
+# Dockerfile silently ship the wrong vLLM.
-# form you are reading. If you pin an untagged commit again, restore the
-# reinstall block — agent/tasks/20_update_wheel_url.sh fails the run rather
-# than letting the Dockerfile silently ship the wrong vLLM.
 ARG VLLM_BASE_IMAGE=vllm/vllm-openai
-ARG VLLM_BASE_TAG=v0.28.0
+ARG VLLM_BASE_TAG=v0.27.0
 
 # We use registry-based layer caching for the dependencies,
 # where the tag is determined based on the hash of the contents

docker/Dockerfile.cuda

@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=vllm/vllm-openai:v0.28.0
+ARG BASE_IMAGE=vllm/vllm-openai:v0.26.0
 FROM ${BASE_IMAGE}
 
 ARG COMMON_WORKDIR=/app

docker/Dockerfile.npu

@@ -1,5 +1,5 @@
 ARG VLLM_ASCEND_IMAGE=quay.io/atlas-ci/vllm-ascend
-ARG VLLM_ASCEND_TAG=v0.28.0
+ARG VLLM_ASCEND_TAG=v0.26.0
 FROM ${VLLM_ASCEND_IMAGE}:${VLLM_ASCEND_TAG}
 
 # WORKDIR /vllm-workspace/vllm
@@ -30,7 +30,6 @@ RUN python3 -m pip install -v --no-deps \
       "stepaudio2-minicpmo>=0.1.1" \
       "step-audio2>=1.0.0"
 
-ENV VLLM_TARGET_DEVICE=npu
 ENV VLLM_WORKER_MULTIPROC_METHOD=spawn
 
 ENTRYPOINT []

docker/Dockerfile.npu.a3

@@ -1,5 +1,5 @@
 ARG VLLM_ASCEND_IMAGE=quay.io/atlas-ci/vllm-ascend
-ARG VLLM_ASCEND_TAG=v0.28.0-a3
+ARG VLLM_ASCEND_TAG=v0.26.0-a3
 FROM ${VLLM_ASCEND_IMAGE}:${VLLM_ASCEND_TAG}
 
 # WORKDIR /vllm-workspace/vllm
@@ -30,7 +30,6 @@ RUN python3 -m pip install -v --no-deps \
       "stepaudio2-minicpmo>=0.1.1" \
       "step-audio2>=1.0.0"
 
-ENV VLLM_TARGET_DEVICE=npu
 ENV VLLM_WORKER_MULTIPROC_METHOD=spawn
 
 ENTRYPOINT []

docker/Dockerfile.npu.ci

@@ -1,5 +1,5 @@
 ARG VLLM_ASCEND_IMAGE=quay.io/atlas-ci/vllm-ascend
-ARG VLLM_ASCEND_TAG=v0.28.0
+ARG VLLM_ASCEND_TAG=v0.26.0
 FROM ${VLLM_ASCEND_IMAGE}:${VLLM_ASCEND_TAG}
 
 ARG APP_DIR=/vllm-workspace/vllm-omni
@@ -65,7 +65,6 @@ RUN python3 -m pip install -v --no-deps \
       --trusted-host cache-service.nginx-pypi-cache.svc.cluster.local \
       --default-timeout=60
 
-ENV VLLM_TARGET_DEVICE=npu
 ENV VLLM_WORKER_MULTIPROC_METHOD=spawn
 
 ENTRYPOINT []

docker/Dockerfile.npu.ci.a3

@@ -1,5 +1,5 @@
 ARG VLLM_ASCEND_IMAGE=quay.io/atlas-ci/vllm-ascend
-ARG VLLM_ASCEND_TAG=v0.28.0-a3
+ARG VLLM_ASCEND_TAG=v0.26.0-a3
 FROM ${VLLM_ASCEND_IMAGE}:${VLLM_ASCEND_TAG}
 
 ARG APP_DIR=/vllm-workspace/vllm-omni
@@ -66,7 +66,6 @@ RUN python3 -m pip install -v --no-deps \
       --trusted-host cache-service.nginx-pypi-cache.svc.cluster.local \
       --default-timeout=60
 
-ENV VLLM_TARGET_DEVICE=npu
 ENV VLLM_WORKER_MULTIPROC_METHOD=spawn
 
 ENTRYPOINT []

docker/Dockerfile.rocm

@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=vllm/vllm-openai-rocm:v0.28.0
+ARG BASE_IMAGE=vllm/vllm-openai-rocm:v0.26.0
 FROM ${BASE_IMAGE} AS base
 
 # Declare a variable to know if we want to use the nightly build or the stable build.
@@ -10,7 +10,7 @@ FROM ${BASE_IMAGE} AS base
 #    we should swap over to use stable release ASAP.
 #    We should avoid relying on custom commits.
 ARG USE_NIGHTLY_BUILD=0
-ARG VLLM_VERSION_OR_COMMIT_HASH=v0.28.0
+ARG VLLM_VERSION_OR_COMMIT_HASH=v0.26.0
 ARG ARG_PYTORCH_ROCM_ARCH
 ENV PYTORCH_ROCM_ARCH=${ARG_PYTORCH_ROCM_ARCH:-${PYTORCH_ROCM_ARCH}}
 
@@ -79,16 +79,9 @@ RUN printf '%s\n' \
     '    os._exit(_exit_code)' \
     > ${COMMON_WORKDIR}/vllm-omni/conftest.py
 
-# Some optional dependencies pull in the CPU ONNX Runtime distribution. The
-# CPU and ROCm wheels own the same import path and cannot coexist, so remove
-# the CPU wheel and reinstall the ROCm wheel after dependency resolution.
 RUN cd ${COMMON_WORKDIR}/vllm-omni && \
     uv pip install --python "$(python3 -c 'import sys; print(sys.executable)')" \
         --no-cache-dir ".[dev]" --no-build-isolation && \
-    python3 -m pip uninstall -y onnxruntime && \
-    python3 -m pip install --no-cache-dir --force-reinstall --no-deps \
-        "onnxruntime-rocm>=1.22.2" && \
-    python3 -c "import importlib.metadata as m; import onnxruntime as ort; assert 'ROCMExecutionProvider' in ort.get_available_providers(), ort.get_available_providers(); assert m.version('onnxruntime-rocm'); assert not any(d.metadata['Name'] == 'onnxruntime' for d in m.distributions())" && \
     python3 -c "from torchcodec.decoders import VideoDecoder"
 
 RUN ln -sf /usr/bin/python3 /usr/bin/python

docker/Dockerfile.xpu

@@ -1,33 +1,140 @@
-# Target vLLM version: v0.28.0. Keep VLLM_VERSION aligned with the target vLLM
+# Argument to configure vllm base image if pre-built
-# release on every rebase.
+ARG VLLM_BASE=vllm-base
-#
+
-# This file layers vLLM-Omni on top of the published upstream XPU image, which
+FROM intel/deep-learning-essentials:2025.3.2-0-devel-ubuntu24.04 AS vllm-base
-# already ships a self-consistent XPU runtime (torch-xpu, oneCCL/UCX+NIXL, the
+
-# pinned UMD/compute-runtime stack, and vLLM itself) built from
+WORKDIR /workspace/
-# https://github.com/vllm-project/vllm/blob/main/docker/Dockerfile.xpu. Docker
+
-# pulls that base automatically, so there is nothing to build here beyond the
+ARG PYTHON_VERSION=3.12
-# Omni layers. Mirrors how docker/Dockerfile.ci consumes vllm/vllm-openai.
+ARG PIP_EXTRA_INDEX_URL="https://download.pytorch.org/whl/xpu"
-#
+
-# If the tag does not exist yet (targeting an unreleased vLLM commit), build the
+RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | gpg --dearmor | tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null && \
-# base from upstream's own Dockerfile and point VLLM_BASE at the result:
+    echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | tee /etc/apt/sources.list.d/oneAPI.list
-#
+
-#   git clone --depth 1 --branch $VLLM_VERSION https://github.com/vllm-project/vllm /tmp/vllm
+RUN apt clean && apt-get update -y && \
-#   docker build -t vllm-openai-xpu:local --target vllm-openai \
+    apt-get install -y --no-install-recommends --fix-missing \
-#       -f /tmp/vllm/docker/Dockerfile.xpu /tmp/vllm
+    curl \
-#   docker build -f docker/Dockerfile.xpu --build-arg VLLM_BASE=vllm-openai-xpu:local .
+    ffmpeg \
-#
+    git \
-# .buildkite/intel/scripts/run-xpu-test.sh automates that pull-then-build fallback.
+    libsndfile1 \
-ARG VLLM_BASE_IMAGE=vllm/vllm-openai-xpu
+    libsm6 \
-ARG VLLM_VERSION=v0.28.0
+    libxext6 \
-ARG VLLM_BASE=${VLLM_BASE_IMAGE}:${VLLM_VERSION}
+    libgl1 \
+    lsb-release \
+    libaio-dev \
+    numactl \
+    wget \
+    vim \
+    python3.12 \
+    python3.12-dev \
+    python3-pip
+
+# Add oneAPI repo, pin oneAPI to 2025.3, then install pinned packages in one layer.
+RUN wget -O- https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB | gpg --dearmor | tee /usr/share/keyrings/oneapi-archive-keyring.gpg > /dev/null && \
+    echo "deb [signed-by=/usr/share/keyrings/oneapi-archive-keyring.gpg] https://apt.repos.intel.com/oneapi all main" | tee /etc/apt/sources.list.d/oneAPI.list && \
+    printf '%s\n' \
+    'Package: intel-oneapi-* intel-deep-learning-essentials* intel-pti*' \
+    'Pin: version 2025.3*' \
+    'Pin-Priority: 1001' \
+    > /etc/apt/preferences.d/oneapi-2025.3.pref && \
+    apt-get update -y && \
+    apt-get install -y --no-install-recommends \
+    intel-oneapi-compiler-dpcpp-cpp-2025.3 \
+    intel-oneapi-mkl-devel-2025.3 \
+    intel-oneapi-dnnl-devel-2025.3 && \
+    rm -rf /var/lib/apt/lists/*
+
+# Install UMD with fixed version
+RUN mkdir neo && \
+    cd neo && \
+    wget https://github.com/intel/intel-graphics-compiler/releases/download/v2.24.8/intel-igc-core-2_2.24.8+20344_amd64.deb && \
+    wget https://github.com/intel/intel-graphics-compiler/releases/download/v2.24.8/intel-igc-opencl-2_2.24.8+20344_amd64.deb && \
+    wget https://github.com/intel/compute-runtime/releases/download/25.48.36300.8/intel-ocloc_25.48.36300.8-0_amd64.deb && \
+    wget https://github.com/intel/compute-runtime/releases/download/25.48.36300.8/intel-opencl-icd_25.48.36300.8-0_amd64.deb && \
+    wget https://github.com/intel/compute-runtime/releases/download/25.48.36300.8/libigdgmm12_22.8.2_amd64.deb && \
+    wget https://github.com/intel/compute-runtime/releases/download/25.48.36300.8/libze-intel-gpu1_25.48.36300.8-0_amd64.deb && \
+    wget https://github.com/oneapi-src/level-zero/releases/download/v1.26.0/level-zero_1.26.0+u24.04_amd64.deb && \
+    dpkg -i *.deb && \
+    cd .. && \
+    rm -rf neo
+
+ENV PATH="/root/.local/bin:$PATH"
+ENV VIRTUAL_ENV="/opt/venv"
+ENV UV_PYTHON_INSTALL_DIR=/opt/uv/python
+RUN curl -LsSf https://astral.sh/uv/install.sh | sh
+RUN uv venv --python ${PYTHON_VERSION} --seed ${VIRTUAL_ENV}
+ENV PATH="$VIRTUAL_ENV/bin:$PATH"
+
+# This oneccl contains the BMG support which is not the case for default version of oneapi 2025.3.
+ARG ONECCL_INSTALLER="intel-oneccl-2021.15.9.14_offline.sh"
+RUN wget "https://github.com/uxlfoundation/oneCCL/releases/download/2021.15.9/${ONECCL_INSTALLER}" && \
+    bash "${ONECCL_INSTALLER}" -a --silent --eula accept && \
+    rm "${ONECCL_INSTALLER}" && \
+    echo "source /opt/intel/oneapi/setvars.sh --force" >> /root/.bashrc && \
+    echo "source /opt/intel/oneapi/ccl/2021.15/env/vars.sh --force" >> /root/.bashrc
+RUN rm -f /opt/intel/oneapi/ccl/latest && \
+    ln -s /opt/intel/oneapi/ccl/2021.15 /opt/intel/oneapi/ccl/latest
+
+SHELL ["bash", "-c"]
+CMD ["bash", "-c", "source /root/.bashrc && exec bash"]
+
+WORKDIR /workspace/
+
+ENV UV_HTTP_TIMEOUT=500
+
+# Configure package index for XPU
+ENV PIP_EXTRA_INDEX_URL=${PIP_EXTRA_INDEX_URL}
+ENV UV_EXTRA_INDEX_URL=${PIP_EXTRA_INDEX_URL}
+ENV UV_INDEX_STRATEGY="unsafe-best-match"
+ENV UV_LINK_MODE="copy"
+
+ARG VLLM_VERSION=v0.27.0
+RUN git clone -b ${VLLM_VERSION} https://github.com/vllm-project/vllm
+WORKDIR /workspace/vllm
+
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv pip install --upgrade pip && \
+    uv pip install -r requirements/xpu.txt && \
+    uv pip install setuptools_scm &&\
+    uv pip install grpcio-tools protobuf nanobind && \
+    source /opt/intel/oneapi/setvars.sh --force && \
+    source /opt/intel/oneapi/ccl/2021.15/env/vars.sh --force && \
+    export CMAKE_PREFIX_PATH="$(python3 -c 'import site; print(site.getsitepackages()[0])'):${CMAKE_PREFIX_PATH}"
+
+ENV LD_LIBRARY_PATH="/opt/intel/oneapi/ccl/2021.15/lib:$LD_LIBRARY_PATH:/usr/local/lib/"
+
+ENV VLLM_TARGET_DEVICE=xpu
+ENV VLLM_WORKER_MULTIPROC_METHOD=spawn
+
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv pip uninstall triton triton-xpu && \
+    uv pip install triton-xpu==3.7.2 && \
+    uv pip install --no-build-isolation --no-deps .
+
+CMD ["/bin/bash"]
+
+FROM vllm-base AS vllm-openai
+
+# install nixl from source code
+RUN uv pip install nixl
+
+# install additional dependencies for openai api server
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv pip install accelerate hf_transfer pytest pytest_asyncio lm_eval[api] modelscope
+
+# install development dependencies (for testing)
+RUN uv pip install -e tests/vllm_test_utils
+
+# ensure vllm is properly installed
+RUN python -c "import vllm, inspect; print(vllm.__file__)"
+RUN uv pip show vllm
+
+CMD ["/bin/bash"]
+
+ENTRYPOINT []
 
 FROM ${VLLM_BASE} AS vllm-omni
 
-# The base image ships ENTRYPOINT ["vllm", "serve"]; clear it so this stage stays
-# usable as a plain shell/test container regardless of which VLLM_BASE was used.
-ENTRYPOINT []
-CMD ["/bin/bash"]
-
 RUN apt-get update && \
     apt-get install -y espeak-ng ffmpeg git jq && \
     apt-get clean && \

docs/.nav.yml

@@ -5,22 +5,14 @@ nav:
     - getting_started/quickstart.md
     - getting_started/installation/*
   - Serving:
-    - API Server:
+    - OpenAI-Compatible API:
-      - Overview: serving/README.md
+      - Chat Completions API: serving/chat_completions_api.md
-      - Core APIs:
+      - Image Generation: serving/image_generation_api.md
-        - Chat Completions: serving/chat_completions_api.md
+      - Image Edit: serving/image_edit_api.md
-        - Text to Speech: serving/speech_api.md
+      - Text to Speech: serving/speech_api.md
-        - Audio Generation: serving/audio_generate_api.md
+      - Videos: serving/videos_api.md
-        - Image Generation: serving/image_generation_api.md
+      - Audio Generation: serving/audio_generate_api.md
-        - Image Edit: serving/image_edit_api.md
+      - Streaming Video Input: serving/video_stream_api.md
-        - Videos: serving/videos_api.md
-      - Streaming and Realtime:
-        - Realtime Audio: serving/realtime_api.md
-        - Full Duplex: serving/full_duplex_api.md
-        - Streaming Video Input: serving/video_stream_api.md
-        - Streaming Video Output: serving/streaming_video_output_api.md
-        - OpenPI Robot Policy: serving/openpi_api.md
-      - Standalone Experimental Servers: serving/standalone_servers.md
   - Examples:
     - examples/README.md
     - Offline Inference:
@@ -32,6 +24,7 @@ nav:
       - Text-To-Speech: user_guide/examples/offline_inference/text_to_speech.md
       - Text-To-Video: user_guide/examples/offline_inference/text_to_video.md
       - X-To-Text: user_guide/examples/offline_inference/x_to_text.md
+      - X-To-Video-Audio: user_guide/examples/offline_inference/x_to_video_audio.md
     - Online Serving:
       - Image-To-Image: user_guide/examples/online_serving/image_to_image.md
       - Image-To-Video: user_guide/examples/online_serving/image_to_video.md
@@ -50,7 +43,6 @@ nav:
     - Overview: features/README.md
     - Runtime and Stage Execution:
       - Execution Modes and Streaming: user_guide/diffusion/execution_modes.md
-      - Prefill-Decode Disaggregation (experimental): features/pd_disaggregation.md
       - Sleep Mode: features/sleep_mode.md
     - Quantization:
       - Overview: user_guide/quantization/overview.md
@@ -61,7 +53,6 @@ nav:
       - BitsAndBytes W4: user_guide/quantization/bitsandbytes.md
       - MXFP8 W8A8: user_guide/quantization/mxfp8.md
       - MXFP4 W4A4: user_guide/quantization/mxfp4.md
-      - SVDQuant W4A4: user_guide/quantization/svdquant.md
       - ModelOpt: user_guide/quantization/modelopt.md
       - AutoRound: user_guide/quantization/autoround.md
       - msModelSlim: user_guide/quantization/msmodelslim.md
@@ -69,11 +60,7 @@ nav:
       - Overview and Compatibility:
         - Overview: user_guide/diffusion_features.md
         - Feature Compatibility: user_guide/feature_compatibility.md
-      - CPU Offloading:
+      - CPU Offloading: user_guide/diffusion/cpu_offload.md
-        - Overview: user_guide/diffusion/cpu_offload.md
-        - Model-Level: user_guide/diffusion/offloader/cpu_offload.md
-        - Layerwise: user_guide/diffusion/offloader/layerwise_offload.md
-        - Distributed Layerwise: user_guide/diffusion/offloader/distributed_layerwise_offload.md
       - Cache Acceleration:
         - TeaCache: user_guide/diffusion/cache_acceleration/teacache.md
         - Cache-DiT: user_guide/diffusion/cache_acceleration/cache_dit.md
@@ -85,13 +72,7 @@ nav:
         - Sequence Parallel: user_guide/diffusion/parallelism/sequence_parallel.md
         - Tensor Parallel: user_guide/diffusion/parallelism/tensor_parallel.md
         - VAE Parallelism: user_guide/diffusion/parallelism/vae_parallelism.md
-      - Attention Backends:
+      - Attention Backends: user_guide/diffusion/attention_backends.md
-        - Overview: user_guide/diffusion/attention_backends.md
-        - Dense Backends: user_guide/diffusion/attention_backends/dense_backends.md
-        - TRTLLM: user_guide/diffusion/attention_backends/trtllm.md
-        - SageAttention: user_guide/diffusion/attention_backends/sage.md
-        - Hugging Face Hub: user_guide/diffusion/attention_backends/huggingface_hub.md
-        - RainFusion: user_guide/diffusion/attention_backends/rainfusion.md
       - Regional Compilation: user_guide/diffusion/regional_compilation.md
       - Frame Interpolation: user_guide/diffusion/frame_interpolation.md
       - Startup and Loading: user_guide/diffusion/startup_and_loading.md
@@ -121,12 +102,10 @@ nav:
     - Feature Design:
       - Runtime and Stage Execution:
         - Disaggregated Inference: design/feature/disaggregated_inference.md
-        - Host Weight Runtime: design/feature/host_weight_runtime.md
         - Async Chunk: design/feature/async_chunk.md
         - Async Diffusion Output: design/feature/async_diffusion_output.md
         - Async Omni Output Materialization: design/feature/omni_async_output_materialization.md
         - Automatic Prefix Caching: design/feature/prefix_caching.md
-        - Realtime AR-Diffusion: design/feature/realtime_ar_diffusion.md
       - Communication:
         - OmniConnector Implementations:
           - Mooncake Store Connector: design/feature/omni_connectors/mooncake_store_connector.md
@@ -146,16 +125,11 @@ nav:
           - Tensor Parallel: design/feature/tensor_parallel.md
           - VAE Patch Parallelism: design/feature/vae_parallel.md
         - Attention Optimization:
-          - Backend Selection: design/feature/attention_backend_selection.md
           - Skip-Softmax: design/feature/skip_softmax.md
-        - CPU Offloading:
-          - Overview: design/feature/offloader/README.md
-          - Model-Level: design/feature/offloader/cpu_offload.md
-          - Layerwise: design/feature/offloader/layerwise_offload.md
-          - Distributed Layerwise: design/feature/offloader/distributed_layerwise_offload.md
         - Cache-DiT: design/feature/cache_dit.md
         - TeaCache: design/feature/teacache.md
         - Diffusion Continuous Batching: design/feature/diffusion_continuous_batching.md
+        - Distributed Layerwise Offload: design/feature/distributed_layerwise_offload.md
     - Infrastructure and Performance:
       - Prometheus Metrics: design/metrics.md
       - Speech Generation Performance Optimizations: design/qwen3_omni_tts_performance_optimization.md
@@ -178,7 +152,6 @@ nav:
         - Offloader: design/module/diffusion/offloader.md
       - Execution Platforms: design/module/execution_platforms.md
       - Cache Management: design/module/cache_management.md
-      - Host Weight Runtime: design/module/host_weight_runtime.md
       - Quantization: design/module/quantization.md
       - Observability: design/module/observability.md
       - Profiling: design/module/profiling.md
@@ -190,8 +163,7 @@ nav:
 - CLI Reference: cli
 - Community:
   - Governance: community/governance.md
-  - Contact Us: community/contact_us.md
+  - community/*
-  - Community Meetings: community/meetings.md
   - Slack: https://slack.vllm.ai
   - Blog: https://blog.vllm.ai
   - Forum: https://discuss.vllm.ai

docs/README.md

@@ -57,9 +57,9 @@ vLLM-Omni is flexible and easy to use with:
 vLLM-Omni seamlessly supports most popular open-source models on HuggingFace, including:
 
 - Omni-modality models (e.g. Qwen3-Omni, MiniCPM-o 4.5, Cosmos3, HunyuanImage, BAGEL)
-- TTS models (e.g. Qwen3-TTS, IndexTTS 2.5, dots.tts, CosyVoice3)
+- TTS models (e.g. Qwen3-TTS, VoxCPM2, Ming-Omni-TTS, CosyVoice3)
-- Diffusion models — image, video, and audio generation (e.g. MiniMax H3, LTX-2.5, SANA-Video, Wan2.2)
+- Diffusion models — image, video, and audio generation (e.g. MiniMax H3, Qwen-Image, Wan2.2, FLUX)
-- Robot-policy and action models (e.g. π0, GR00T-N1.7, DreamZero-DROID, InternVLA-A1)
+- Robot-policy and action models (e.g. GR00T-N1.7, DreamZero-DROID, InternVLA-A1, Cosmos3 action policy)
 
 For more information, checkout the following:
 

docs/cli/README.md

@@ -2,11 +2,6 @@
 
 The CLI for vLLM-Omni inherits from vllm with some additional arguments.
 
-Environment variables are process configuration rather than CLI-only options.
-See the [Environment Variables](../configuration/environment_variables.md)
-reference for ownership, precedence, evaluation time, and per-stage `env`
-scoping.
-
 ## serve
 
 Starts the vLLM-Omni OpenAI Compatible API server.
@@ -44,5 +39,4 @@ vllm bench serve --omni \
     --num-prompts  5
 ```
 
-See [vllm bench serve](./bench/serve.md) for serving benchmark arguments and
+See [vllm bench serve](./bench/serve.md) for the full reference of all available arguments.
-dataset-specific examples, including OmniInteract native-duplex sessions.

docs/cli/bench/serve.md

@@ -2,7 +2,7 @@
 The vllm bench command launches the vLLM-Omni benchmark to evaluate the performance of multimodal models.
 
 ## Notes
-vLLM-Omni registers the `openai-chat-omni`, `openai-audio-speech`, `openai-image-edits-omni`, `daily-omni`, and `openai-realtime-duplex` serving benchmark backends. It also adds the `omniinteract` dataset.
+vLLM-Omni registers the `openai-chat-omni`, `openai-audio-speech`, `openai-image-edits-omni`, and `daily-omni` serving benchmark backends.
 
 ## Basic Parameter Description
 You can use `vllm bench serve --omni --help=all` to get descriptions of all parameters. The commonly used parameters are described below:
@@ -10,7 +10,7 @@ You can use `vllm bench serve --omni --help=all` to get descriptions of all para
   Enable Omni (multimodal) mode, supporting multimodal inputs and outputs such as images, videos, and audio.
 
 - `--backend`
-  Specify the backend adapter. vLLM-Omni adds `openai-chat-omni`, `openai-audio-speech`, `openai-image-edits-omni`, `daily-omni`, and `openai-realtime-duplex` to the upstream vLLM backend choices.
+  Specify the backend adapter. vLLM-Omni adds `openai-chat-omni`, `openai-audio-speech`, `openai-image-edits-omni`, and `daily-omni` to the upstream vLLM backend choices.
 
 - `--model`
   The model identifier to load, filled according to the models supported by vLLM-Omni.
@@ -19,7 +19,7 @@ You can use `vllm bench serve --omni --help=all` to get descriptions of all para
   The API endpoint exposed externally, to which clients send their requests.
 
 - `--dataset-name`
-  The name of the dataset used; random-mm indicates generating random multimodal inputs (images, videos, audio), while `omniinteract` replays official OmniInteract videos.
+  The name of the dataset used; random-mm indicates generating random multimodal inputs (images, videos, audio).
 
 - `--num-prompts`
   The total number of requests to send, an integer.
@@ -269,44 +269,6 @@ We use audio generation time / audio duration to calculate RTF.
 
 </details>
 
-### OmniInteract Realtime Benchmark
-
-OmniInteract runs each video as one native-duplex WebSocket sample in the standard serving-benchmark lifecycle:
-
-```bash
-vllm bench serve --omni \
-  --backend openai-realtime-duplex \
-  --dataset-name omniinteract \
-  --dataset-path /path/to/OmniInteract \
-  --model openbmb/MiniCPM-o-4_5 \
-  --base-url http://127.0.0.1:8000 \
-  --endpoint /v1/realtime \
-  --omniinteract-ref-audio /path/to/reference.wav \
-  --omniinteract-output-dir ./omniinteract-artifacts \
-  --num-prompts 3 \
-  --num-warmups 0
-```
-
-`--dataset-path` accepts an extracted directory, `data.tar[.gz]`, or a Hugging Face dataset ID; omitting it uses
-`lucky-lance/OmniInteract`. `--num-prompts` is the total across subsets and defaults to 3 for OmniInteract; explicit `0`
-selects all and oversize values use all available cases. Reference audio is required, and OmniInteract uses the
-`/v1/realtime` endpoint.
-
-Audio is replayed as 16 kHz PCM16 in 200 ms chunks and video at 1 FPS with real-time pacing. All selected media is decoded
-before timing and remains in client memory for the run, so `--max-concurrency` does not limit media preparation memory; use
-explicit `--num-prompts 0` only when the client has enough RAM for the full dataset. Media commands are bounded by
-`--omniinteract-media-timeout-s`, and concurrency defaults to 1. Standard request-rate, warmup, result-saving, and summary
-options apply. Use `--omniinteract-require-response` only for functional E2E cases; LISTEN is a valid benchmark result.
-
-Each completed case writes `output.wav`, `wav_transcript.json`, `events.json`, `result.json`, and a final `.done` marker under
-`--omniinteract-output-dir`. The root also contains `batch_summary.json` and `official_eval_manifest.jsonl`; failed cases write
-`.failed.json`. Runs sharing one output root are serialized. Completion validates transport, response lifecycle, and artifacts,
-not answer accuracy. Transcript timestamps are serialized playback-queue times. Clipped or cancelled outputs are ineligible and
-omitted from the official manifest; `audio_clipped_bytes` records output beyond the rounded video horizon.
-
-TTFT, TTFP, and RTF start at client receipt of `response.created`. TPOT/ITL use engine stage-0 timing; ITL is emitted only when
-every token interval is present.
-
 ### Multi-Modal Benchmark
 
 <details class="admonition abstract" markdown="1">

docs/community/contact_us.md

@@ -1,9 +1,5 @@
 # Contact Us
 
-- For bugs and feature requests, search or open a GitHub [issue](https://github.com/vllm-project/vllm-omni/issues).
+- For technical questions and feature requests, please use GitHub [Issues](https://github.com/vllm-project/vllm-omni/issues)
-- For technical and usage questions, join the `sig-omni` channel in [vLLM Slack](https://slack.vllm.ai/) or use the [vLLM Forum](https://discuss.vllm.ai/).
+- For coordinating contributions and development and discussing with other users and developers, please join `sig-omni` channel in our [Slack](https://slack.vllm.ai/) or use the [vLLM Forum](https://discuss.vllm.ai/)
-- For coordinating contributions and development, use `sig-omni`, the forum, or join a [community meeting](meetings.md).
+- For security disclosures, please use GitHub's [Security Advisories](https://github.com/vllm-project/vllm-omni/security/advisories) feature
-
-## Security Disclosures
-
-Do not share vulnerability details in public GitHub issues, pull requests, discussions, or Slack channels. To arrange a private report, send a direct message to one of the [lead maintainers](governance.md#lead-maintainers) in [vLLM Slack](https://slack.vllm.ai/). If you cannot send a direct message, you may post a detail-free request in `sig-omni` asking a lead maintainer to contact you privately; do not include any vulnerability details in that request.

docs/community/governance.md

@@ -52,11 +52,31 @@ Committers have write access and merge rights. They typically have deep expertis
 - [@ZJY0516](https://github.com/ZJY0516): Diffusion attention backend, kernel fusion, and CustomOp
 - [@yuanheng-zhao](https://github.com/yuanheng-zhao): Diffusion cache, offload, and Omni Support
 
-## Reviewer Routing
+## Diffusion Workload Division
 
-For path-based reviewer routing, consult [`.github/CODEOWNERS`](https://github.com/vllm-project/vllm-omni/blob/main/.github/CODEOWNERS). Detailed subsystem boundaries and safe-change guidance live in the [module design documents](../design/index.md#module-design-documents).
+This section breaks down diffusion-related responsibilities. If you have PRs touching these areas, please ping the listed owners for review.
 
-`CODEOWNERS` is operational routing metadata, not a governance or approval policy. Being listed does not grant committer status or merge rights, and it does not mean every listed reviewer must approve a change. The maintainer roster above is authoritative for project roles and merge rights.
+### Runtime-related
+
+- Diffusion models: @RuixiangMa, @wtomin, @xuechendi
+
+### Optimization
+
+- Parallel: @RuixiangMa, @wtomin
+- Attention backend: @ZJY0516
+- Cache: @yuanheng-zhao, @RuixiangMa, @SamitHuang
+- Offload: @yuanheng-zhao
+- Intel GPU support: @xuechendi
+- Quantization: @lishunyang12, @david6666666
+- Kernel fusion / communication-computation: @ZJY0516
+
+### Docs & Test
+
+- Diffusion docs: @RuixiangMa, @wtomin
+
+## Meetings
+
+Committers hold **bi-weekly meetings** to discuss future directions and collaborations of the project.
 
 ## Committer Nomination Process
 

docs/community/meetings.md

@@ -1,8 +0,0 @@
-# Community Meetings
-
-vLLM-Omni hosts developer-facing meetings for Chinese- and English-language audiences. Schedules and access details can change, so use these maintained documents as the source of truth:
-
-- [Chinese-language meeting document](https://docs.google.com/document/d/1pdUBiS_7mdOUNDtdwy-9OUf7jsMWN324BIbps5olDME/edit?tab=t.0)
-- [English-language meeting document](https://docs.google.com/document/d/19AlD08T8uJ7wqfs8VtYCi0iGMTwlTYVVlAxFZ3S7OIg/edit?tab=t.0)
-
-Each document contains the current schedule, access details, agendas, and past meeting notes. For broader vLLM community events, see the [vLLM events calendar](https://vllm.ai/events/).

docs/community/meetups.md

@@ -0,0 +1 @@
+# Meetups

docs/community/volunteers.md

@@ -0,0 +1,12 @@
+# Volunteers for Bugfix and CI
+
+We encourage you to check current docs and [issues](https://github.com/vllm-project/vllm-omni/issues) to find possible solutions for your questions. If non of these can solve it, please propose an issue to describe your questions about bug or CI problems for developing.
+
+If you have urgent need for locating and solving bugfix or CI problems, please find community volunteers below.
+
+| Dec 4-Dec 12 | Dec 15-Dec 19 | Dec 22-Dec 26 | Dec 29- Jan 2, 2026| Jan 5-Jan 9 | Jan 12-Jan 16 |
+|----------|----------|----------|----------|----------|----------|
+| <a href="https://github.com/congw729">Conw729</a> | <a href="https://github.com/yinpeiqi">yinpeiqi</a> | <a href="https://github.com/tzhouam">tzhouam</a> | <a href="https://github.com/SamitHuang">SamitHuang</a> | <a href="https://github.com/gcanlin">gcanlin</a> | <a href="https://github.com/natureofnature">natureofnature</a> |
+| <a href="https://github.com/david6666666">david6666666</a> | <a href="https://github.com/R2-Y">R2-Y</a> | <a href="https://github.com/hsliuustc0106">hsliuustc0106</a> | <a href="https://github.com/Gaohan123">Gaohan123</a> | <a href="https://github.com/ZJY0516">ZJY0516</a> | <a href="https://github.com/qibaoyuan">qibaoyuan</a> |
+
+We kindly welcome more contributors to fix bugs and contribute new features!

docs/configuration/README.md

@@ -2,22 +2,22 @@
 
 This section lists the most common options for running vLLM-Omni.
 
-For options within a vLLM Engine, please refer to the [vLLM 0.28 configuration guide](https://docs.vllm.ai/en/v0.28.0/configuration/index.html).
+For options within a vLLM Engine, please refer to the [vLLM 0.26 configuration guide](https://docs.vllm.ai/en/v0.26.0/configuration/index.html).
 
-Each model defines fixed topology in a registered `PipelineConfig` and runtime
+Currently, the main options are maintained by stage configs for each model.
-overrides in a deploy YAML.
-
-For process-level settings shared by the CLI, deploy YAML, workers, and model
-integrations, see [Environment Variables](./environment_variables.md).
 
 For a specific example, see the [Qwen2.5-Omni deploy config](gh-file:vllm_omni/deploy/qwen2_5_omni.yaml). The matching frozen pipeline topology lives at [vllm_omni/model_executor/models/qwen2_5_omni/pipeline.py](gh-file:vllm_omni/model_executor/models/qwen2_5_omni/pipeline.py).
 
-For an introduction, see [Pipeline and deploy configurations](./stage_configs.md).
+For introduction, please check [Introduction for stage config](./stage_configs.md)
 
 ## Memory Configuration
 
 - **[GPU Memory Calculation and Configuration](./gpu_memory_utilization.md)** - Guide on how to calculate memory requirements and set up `gpu_memory_utilization` for optimal performance
 
+## Multi-Stage Recipes
+
+- **[Prefill-Decode Disaggregation](./pd_disaggregation.md)** - How to derive a PD-aware Qwen3-Omni stage config from the default config without introducing another bundled YAML
+
 ## Optimization Features
 
 - **[Diffusion Features Overview](../user_guide/diffusion_features.md)** - Complete overview of all diffusion model features and supported models

docs/configuration/composable_parallel.md

@@ -1,6 +1,6 @@
 # Composable parallel strategies
 
-In vLLM-Omni, the `composable_parallel` system layers a declarative *parallel strategy* on top of the registered pipeline and deploy config. A strategy YAML maps each pipeline stage (by its `model_stage` name) to a stack of axis declarations — one per parallelism dimension — that the engine translates into per-stage parallel sizing (`tensor_parallel_size`, `data_parallel_size`, `pipeline_parallel_size`, expert-parallel toggle, replica count) and, when relevant, an orchestrator-wide load-balancer policy.
+In vLLM-Omni, the `composable_parallel` system layers a declarative *parallel strategy* on top of the existing stage configs. A strategy YAML maps each pipeline stage (by its `model_stage` name) to a stack of axis declarations — one per parallelism dimension — that the engine translates into per-stage parallel sizing (`tensor_parallel_size`, `data_parallel_size`, `pipeline_parallel_size`, expert-parallel toggle, replica count) and, when relevant, an orchestrator-wide load-balancer policy.
 
 Strategies are intentionally narrow: each one only writes the axes it declares, leaving every other engine arg untouched. This makes a strategy file a small, reusable overlay you can swap independently of the deploy YAML (see `configuration/stage_configs.md`).
 
@@ -195,7 +195,7 @@ Because `omni_lb_policy` is read once at `AsyncOmniEngine` construction — not
 
 Default: no default — required for every entry under `strategies:`.
 
-Strategy keys MUST be `model_stage` *names* (strings), the same labels defined by [`StagePipelineConfig.model_stage`](stage_configs.md#pipeline-configuration). Integer `stage_id` values are not accepted; the resolver matches on `model_stage` exactly (`_resolve_stage` in `vllm_omni/config/composable_parallel/apply.py`).
+Strategy keys MUST be `model_stage` *names* (strings), the same labels stage configs use for `engine_args.model_stage` (see [`model_stage` in `stage_configs.md`](stage_configs.md#engine_argsmodel_stage)). Integer `stage_id` values are not accepted; the resolver matches on `model_stage` exactly (`_resolve_stage` in `vllm_omni/config/composable_parallel/apply.py`).
 
 ## Common pitfalls
 

docs/configuration/environment_variables.md

@@ -1,238 +0,0 @@
-# Environment Variables
-
-Environment variables are process-level configuration shared by the CLI,
-deploy YAML, Python APIs, worker processes, and model integrations. They are
-therefore documented under **User Guide → Configuration**, rather than under
-CLI Reference.
-
-This page describes the public vLLM-Omni environment-variable contract on the
-`latest` documentation branch. Use the documentation for your installed
-vLLM-Omni release when behavior differs.
-
-!!! important "Public contract"
-    Only names in [Public vLLM-Omni variables](#public-vllm-omni-variables) are
-    public Omni configuration. A name found in source code is not automatically
-    supported: model-specific, benchmark, platform, and internal variables have
-    separate ownership and lifecycle rules.
-
-## Naming and implementation boundary
-
-New public variables owned by vLLM-Omni must use the `VLLM_OMNI_` prefix.
-Existing public names with older prefixes remain supported for compatibility,
-but they do not establish naming precedent for new settings. Inherited vLLM
-and third-party names retain the prefix chosen by their owning project.
-
-The source [environment-variable inventory](gh-file:vllm_omni/config/environment_variable_inventory.py)
-records ownership, migration disposition, and redaction metadata. Unlike
-`vllm.envs`, it is not an executable value resolver: existing consumers still
-own parsing, defaults, precedence, and evaluation time. The public tables below
-describe those live consumer contracts. Stable model or stage behavior should
-move to typed configuration, and request-varying behavior should move to a
-request schema instead of adding another environment variable.
-
-## Precedence and evaluation time
-
-There is no repository-wide rule that makes environment variables override (or
-lose to) every CLI, YAML, or request option. The tables below state precedence
-for each public setting. In general:
-
-- request fields should control behavior that can vary per request;
-- typed deploy or stage configuration should control stable model and stage
-  behavior;
-- environment variables are appropriate for process launch, platform
-  integration, emergency diagnostics, and compatibility fallbacks.
-
-An environment variable can be read while a module is imported, when server or
-model configuration is constructed, when a worker is spawned, or whenever a
-property is accessed. Set import- and startup-time variables before invoking
-`vllm serve` or importing vLLM-Omni. Changing the parent shell after workers
-start does not update existing worker environments.
-
-The lifecycle labels used below are:
-
-- **Stable** — reviewed public configuration; changes require compatibility
-  consideration.
-- **Experimental** — public for evaluation, but its contract may still evolve.
-- **Diagnostic** — supported for investigation or temporary mitigation, not as
-  a normal tuning interface.
-- **Deprecated** — compatibility alias; migrate to the named replacement.
-
-## Public vLLM-Omni variables
-
-### Diffusion backend and cache selection
-
-| Name | Type and default | Applies to and read time | Precedence and invalid values | Lifecycle |
-| --- | --- | --- | --- | --- |
-| `DIFFUSION_ATTENTION_BACKEND` | Backend name or `auto`; default `auto` (platform selection) | Diffusion stages; read when `OmniDiffusionConfig` is constructed | `diffusion_attention_config.default` wins. An unknown backend fails during backend resolution. | Stable fallback |
-| `DIFFUSION_CACHE_BACKEND` | `none`, `cache_dit`, `tea_cache`, `mag_cache`, `step_cache`, `stepcache`, or `step_cache_dit`; default `none` | Diffusion runner startup | Explicit `cache_backend` in config wins. Otherwise this name wins over the deprecated alias. Unsupported values raise `ValueError` during runner setup. | Stable fallback |
-| `DIFFUSION_CACHE_ADAPTER` | Same values as `DIFFUSION_CACHE_BACKEND`; default `none` | Diffusion runner startup | Used only when neither explicit `cache_backend` nor `DIFFUSION_CACHE_BACKEND` is set. Unsupported values raise `ValueError`. | Deprecated; use `DIFFUSION_CACHE_BACKEND` |
-| `OMNI_DIFFUSION_PROMPT_EMBED_CACHE` | Boolean: `1`, `true`, `yes`, `on`, `0`, `false`, `no`, or `off`; default disabled | Each diffusion runner; resolved during model setup | A recognized environment value overrides the explicit enable setting. An unrecognized value is ignored. | Experimental |
-| `OMNI_DIFFUSION_PROMPT_EMBED_CACHE_SIZE` | Positive integer; default `32` entries | Each diffusion runner; resolved during model setup | A valid environment value overrides the explicit cache size. A non-integer logs a warning; a non-positive value is ignored. | Experimental |
-| `OMNI_DIFFUSION_SESSION_STATE_MANAGER` | Boolean with the same accepted spellings as the prompt cache; default disabled | Experimental diffusion session manager; model setup | A recognized environment value overrides the explicit enable setting. An unrecognized value is ignored. | Experimental |
-| `OMNI_DIFFUSION_SESSION_STATE_MANAGER_MAX_SESSIONS` | Positive integer; default `64` | Experimental diffusion session manager; model setup | A valid environment value overrides the explicit maximum. A non-integer logs a warning; a non-positive value is ignored. | Experimental |
-
-Backend names for `DIFFUSION_ATTENTION_BACKEND` are the members of
-`DiffusionAttentionBackendEnum`, such as `FLASH_ATTN`, `TORCH_SDPA`,
-`SAGE_ATTN`, `FLASHINFER_ATTN`, and `TRTLLM_ATTN`. Platform support still
-depends on the installed kernels and model path.
-
-### Serving and runtime
-
-| Name | Type and default | Applies to and read time | Precedence and invalid values | Lifecycle |
-| --- | --- | --- | --- | --- |
-| `SPEAKER_SAMPLES_DIR` | Filesystem path; default `~/.cache/vllm-omni/speakers` | Speech server; read when speaker storage initializes | Environment-only setting. The directory is created; filesystem errors propagate. | Stable |
-| `SPEAKER_MAX_UPLOADED` | Integer; default `1000` | Speech server; read when speaker storage initializes | Environment-only setting. A non-integer logs a warning and uses `1000`; range is not otherwise validated. | Stable |
-| `VLLM_OMNI_ASYNC_OUTPUT_TIMEOUT` | Float seconds; default `600` | Diffusion engine async-output wait in `step_streaming`; resolved per call on the request path, not at import | Environment-only setting. A non-float or `<=0` value warns once and uses the default. | Experimental |
-| `VLLM_OMNI_EVENT_DRIVEN_ORCH` | `1`, `true`, `yes` or `on` enables; default `0` (off) | Orchestration loop and the serving-side final-output drain; read once when the `Orchestrator` is constructed | Environment-only setting. Values are stripped and case-normalized; any unrecognized value leaves the legacy poll loop selected. | Experimental |
-| `VLLM_OMNI_INPUT_WAIT_TIMEOUT_S` | Float seconds; default `600`; `<=0` disables | Full-payload input coordinator, not async-chunk transfer; read when the scheduler module imports in each worker | Environment-only setting. A non-float logs a warning and uses `600`. | Stable operational control |
-| `VLLM_OMNI_ORCH_MONITOR_PATH` | Filesystem path; default `<current-working-directory>/vllm_omni_orch_monitor_<timestamp>.json` | Orchestrator monitor enabled by `--enable-orch-monitor`; read when the monitor is created | Environment-only path override. Parent directories are created; write errors are logged. | Diagnostic |
-| `VLLM_OMNI_VIDEO_SYNC_TIMEOUT` | Float seconds; default `600` | Synchronous Videos API; read when the API server module imports | Environment-only setting. A non-float raises `ValueError` during import. | Experimental |
-| `VLLM_VIDEO_ASYNC_CHUNK` | `on` or `off`; default `on` | Streaming video output; read on attribute access | Environment-only setting. Values are trimmed and case-normalized; an invalid value warns once and uses `on`. | Experimental |
-| `VLLM_VIDEO_AUDIO_DELTA_MODE` | `fast` or `slow`; default `fast` | Streaming video audio deltas; read on attribute access | Environment-only setting. Values are trimmed and case-normalized; an invalid value warns once and uses `fast`. | Experimental |
-
-### Server storage
-
-Storage names use Pydantic's nested-settings delimiter (`__`). They are read
-when `ServerSettings` is initialized, normally during server import/startup.
-Invalid integer values raise a Pydantic validation error.
-
-| Name | Type and default | Meaning | Lifecycle |
-| --- | --- | --- | --- |
-| `VLLM_OMNI_SERVER_STORAGE__PATH` | Path; default `/tmp/storage` | Directory for completed server files. | Stable |
-| `VLLM_OMNI_SERVER_STORAGE__FILE_CONCURRENCY` | Integer; default `4` | Maximum concurrent file operations. No additional range validation is currently applied. | Stable |
-| `VLLM_OMNI_SERVER_STORAGE__FILE_TTL` | Integer seconds or unset; default unset | Optional lifetime for locally stored files. | Stable |
-| `VLLM_OMNI_SERVER_STORAGE__TTL_SWEEP_INTERVAL` | Integer seconds or unset | Sweep frequency. If file TTL is set and this is unset, the effective default is `300`. | Stable |
-| `VLLM_OMNI_STORAGE_PATH` | Path | Deprecated alias for `VLLM_OMNI_SERVER_STORAGE__PATH`. The new name wins when both are set. | Deprecated |
-| `VLLM_OMNI_STORAGE_MAX_CONCURRENCY` | Integer | Deprecated alias for `VLLM_OMNI_SERVER_STORAGE__FILE_CONCURRENCY`. The new name wins when both are set. | Deprecated |
-
-The deprecated aliases emit `DeprecationWarning`. They do not affect file TTL
-settings.
-
-### Quantization diagnostics and performance
-
-| Name | Type and default | Applies to and read time | Precedence and invalid values | Lifecycle |
-| --- | --- | --- | --- | --- |
-| `VLLM_OMNI_SKIP_NVFP4_NAN_CLAMP` | Boolean truthy spellings: `1`, `true`, `yes`, `on`; default false | ModelOpt NVFP4 compatibility patch; read when `vllm_omni.patch` imports | Environment-only escape hatch. Any other value means false. Set only to diagnose the upstream NaN-scale issue. | Diagnostic and temporary |
-| `VLLM_OMNI_USE_QUACK_FP8` | Boolean truthy spellings: `1`, `true`, `yes`, `on`; unset means hardware auto-detection | FP8 scaled matrix multiplication; evaluated when quack capability is selected | A set value overrides auto-detection. Any non-truthy value forces quack off. If quack cannot load, vLLM-Omni warns and falls back to FlashInfer. | Experimental performance control |
-
-`QUACK_CACHE_DIR` is owned by the external quack library and is not an
-Omni-owned variable, even though vLLM-Omni supplies a persistent default when
-it is unset.
-
-## Per-stage environment
-
-Deploy configurations can set arbitrary environment keys for one stage:
-
-```yaml
-stages:
-  - stage_id: 0
-    env:
-      HCCL_IF_BASE_PORT: 23000
-      MY_MODEL_CACHE: /mnt/model-cache
-```
-
-The legacy nested form under `runtime.env` is also accepted. vLLM-Omni converts
-keys and values to strings, applies them while the stage engine or child process
-is launched, logs keys but not values, and restores the parent process's prior
-environment afterward. A child process inherits the scoped values. This keeps
-sibling stages from permanently changing one another's parent environment.
-
-This scoping has two consequences:
-
-1. A setting read inside stage/model initialization can see the stage value.
-2. A setting already cached by an import that occurred before stage launch
-   cannot be changed retroactively with stage `env`.
-
-Stage `env` is an escape hatch for process and third-party settings, not a
-replacement for typed stage configuration. Do not put request-varying behavior
-there. Secret values may reach child processes; logs and diagnostics must emit
-their keys only.
-
-## Inherited vLLM variables
-
-vLLM-Omni also reads variables through its aligned vLLM dependency. Refer to
-the [vLLM 0.28 environment-variable reference](https://docs.vllm.ai/en/v0.28.0/configuration/env_vars.html)
-for their definitions. This includes vLLM launch, cache, logging, plugin, ROCm,
-XPU, ModelScope, and FlashInfer workspace settings.
-
-An inherited name does not become Omni-owned because Omni reads or forwards it.
-Where behavior differs, the Omni-specific page should document only that
-difference. For example, device-visibility values can be further narrowed by a
-stage's `devices` setting.
-
-## Platform, external, and secret variables
-
-Hardware runtimes, distributed launchers, and third-party libraries retain
-ownership of their variables. Common groups include:
-
-- device visibility (`ASCEND_RT_VISIBLE_DEVICES`, `HIP_VISIBLE_DEVICES`, and
-  `MUSA_VISIBLE_DEVICES`);
-- distributed launch (`MASTER_ADDR`, `MASTER_PORT`, `RANK`, and `WORLD_SIZE`);
-- Hugging Face cache and authentication (`HF_HOME`, `HF_TOKEN`, and related
-  names);
-- accelerator libraries such as NCCL, FlashInfer, MindIE, MORI, and quack.
-
-Consult the owning project's documentation for accepted values and support.
-`HF_TOKEN`, `HUGGINGFACE_HUB_TOKEN`, and `OPENAI_API_KEY` are explicitly marked
-for redaction in the reviewed inventory. Never print their values in environment
-collection, examples, bug reports, or logs.
-
-## Model-specific variables
-
-The audit found 58 variables read by a single model or pipeline family. They are
-not listed as public usage options here because doing so would turn implementation
-escape hatches into an accidental compatibility contract.
-
-Every audited model-specific name has a migration disposition in the
-[environment-variable inventory](gh-file:vllm_omni/config/environment_variable_inventory.py):
-
-| Disposition | Count | Required outcome |
-| --- | ---: | --- |
-| Promote | 35 | Move a stable setting into typed stage or model configuration. |
-| Request scope | 6 | Move request-varying behavior into a declared request-option schema. |
-| External | 0 | Retain only when a supported third-party library owns the contract. |
-| Internalize | 11 | Keep a debug or diagnostic switch out of public documentation and configuration. |
-| Deprecate/remove | 6 | Remove a compatibility escape hatch that has no continuing contract. |
-
-The disposition is a migration target, not a statement that the existing
-environment switch is stable. Promote or request-scope work should land in
-model-owner-reviewed follow-up changes with validation and model-page updates.
-The family-sized migration packages and their ownership gate are tracked in
-[#6232](https://github.com/vllm-project/vllm-omni/issues/6232).
-
-## Environment collection and troubleshooting
-
-Run the repository collector when preparing a bug report:
-
-```bash
-python collect_env.py
-```
-
-The environment section includes the safe public Omni allowlist, registered
-vLLM variables, and selected platform prefixes. It omits names marked for
-redaction and names containing common secret terms. Review the output before
-sharing it. A key supplied through stage `env` is not collected merely because
-it appears in deploy configuration; the same allowlist and prefix rules apply.
-
-If a setting appears ineffective, first check its read time in the tables
-above. Restart processes after changing import- or startup-time values, verify
-that the variable reaches the intended worker, and check whether an explicit
-typed configuration field takes precedence.
-
-## Benchmark and internal variables
-
-The reviewed inventory marks 20 benchmark variables as transitional while
-benchmark configuration moves to CLI/request schemas in
-[issue #5376](https://github.com/vllm-project/vllm-omni/issues/5376). They are not
-part of the serving/runtime contract.
-
-Internal process-coordination and debug variables are deliberately excluded
-from this page. The CI scanner resolves string literals and module-level string
-constants used by direct `os` access, local wrapper helpers, aliased `os`
-imports, and environment membership checks. Server-storage names generated by
-Pydantic settings are checked separately against their model fields.
-
-Names assembled from cross-module metadata and arbitrary keys supplied through
-stage `env` cannot be inferred reliably. Contributors must classify those names
-explicitly. CI fails when a statically resolved name has no classification or
-when a reviewed public name is absent from this page.

docs/configuration/gpu_memory_utilization.md

@@ -80,15 +80,19 @@ When multiple stages share the same GPU, you must ensure the sum of their `gpu_m
 
 **Example: Two stages on GPU 0**
 ```yaml
-stages:
+stage_args:
   - stage_id: 0
-    devices: "0"
+    runtime:
-    gpu_memory_utilization: 0.6  # Uses 60% of GPU 0
+      devices: "0"
+    engine_args:
+      gpu_memory_utilization: 0.6  # Uses 60% of GPU 0
 
   - stage_id: 1
-    devices: "0"
+    runtime:
-    gpu_memory_utilization: 0.3  # Uses 30% of GPU 0
+      devices: "0"
-    # Total: 90% of GPU 0 (safe, leaves 10% buffer)
+    engine_args:
+      gpu_memory_utilization: 0.3  # Uses 30% of GPU 0
+      # Total: 90% of GPU 0 (safe, leaves 10% buffer)
 ```
 
 **Important:** If stages run on different GPUs, each can use up to 1.0 independently.
@@ -99,12 +103,14 @@ When using `tensor_parallel_size > 1`, the model is split across multiple GPUs,
 
 **Example: 2-way tensor parallelism**
 ```yaml
-stages:
+stage_args:
   - stage_id: 0
-    devices: "0,1"  # Uses both GPUs
+    runtime:
-    tensor_parallel_size: 2
+      devices: "0,1"  # Uses both GPUs
-    gpu_memory_utilization: 0.6  # 60% per GPU
+    engine_args:
-    # Model is split, so each GPU uses ~30% of model memory
+      tensor_parallel_size: 2
+      gpu_memory_utilization: 0.6  # 60% per GPU
+      # Model is split, so each GPU uses ~30% of model memory
 ```
 
 ## Examples
@@ -112,25 +118,27 @@ stages:
 ### Qwen3-Omni-MoE on 2x H100-80GB
 
 ```yaml
-base_config: /path/to/vllm_omni/deploy/qwen3_omni_moe.yaml
+stage_args:
-
-stages:
   - stage_id: 0  # Thinker stage with TP=2
-    devices: "0,1"
+    runtime:
-    tensor_parallel_size: 2
+      devices: "0,1"
-    gpu_memory_utilization: 0.6  # 48GB per GPU
+    engine_args:
+      tensor_parallel_size: 2
+      gpu_memory_utilization: 0.6  # 48GB per GPU
 
   - stage_id: 1  # Talker stage
-    devices: "1"
+    runtime:
-    gpu_memory_utilization: 0.3  # 24GB on GPU 1
+      devices: "1"
+    engine_args:
+      gpu_memory_utilization: 0.3  # 24GB on GPU 1
 
   - stage_id: 2  # Code2Wav stage
-    devices: "0"
+    runtime:
-    gpu_memory_utilization: 0.1  # 8GB on GPU 0
+      devices: "0"
+    engine_args:
+      gpu_memory_utilization: 0.1  # 8GB on GPU 0
 ```
-**Note:** Stage 0 uses GPUs 0 and 1, so the combined utilization is `0.7` on
+**Note:** In this configuration, stages 0 and 2 share GPU 0, but they run at different times in the pipeline, so their memory usage doesn't overlap.
-GPU 0 and `0.9` on GPU 1. Keep the sum of all resident stages below `1.0` on
-each device.
 
 ## Troubleshooting
 

docs/configuration/pd_disaggregation.md

@@ -0,0 +1,171 @@
+# Prefill-Decode (PD) Disaggregation
+
+PD disaggregation splits the Qwen3-Omni thinker into separate prefill and decode
+stages so prompt processing and token generation can run on different workers.
+
+This is documented as a stage-config recipe instead of a bundled YAML because the
+deployment-specific values usually change per environment:
+
+- GPU placement
+- `tensor_parallel_size`
+- connector backend and connector ports
+- connector IPs or bootstrap addresses
+
+Start from the [default Qwen3-Omni stage config](gh-file:vllm_omni/deploy/qwen3_omni_moe.yaml)
+and copy it to your own file, for example `qwen3_omni_pd.yaml`. Then apply the
+changes below.
+
+## Requirements
+
+- 3+ GPUs for a basic layout: prefill, decode, and talker+code2wav
+- A KV connector supported by vLLM, such as `MooncakeConnector`
+- Matching `tensor_parallel_size` on the prefill and decode thinker stages
+
+## 1. Split the thinker into prefill and decode stages
+
+Replace the original thinker stage with two stages:
+
+```yaml
+stage_args:
+  - stage_id: 0
+    stage_type: llm
+    is_prefill_only: true
+    runtime:
+      devices: "0"
+    engine_args:
+      max_num_seqs: 16
+      model_stage: thinker
+      model_arch: Qwen3OmniMoeForConditionalGeneration
+      worker_type: ar
+      scheduler_cls: vllm_omni.core.sched.omni_ar_scheduler.OmniARScheduler
+      gpu_memory_utilization: 0.9
+      enforce_eager: true
+      trust_remote_code: true
+      engine_output_type: latent
+      distributed_executor_backend: "mp"
+      enable_prefix_caching: false
+      max_num_batched_tokens: 32768
+      hf_config_name: thinker_config
+      tensor_parallel_size: 1
+      kv_transfer_config:
+        kv_connector: "MooncakeConnector"
+        kv_role: "kv_producer"
+        kv_rank: 0
+        kv_parallel_size: 2
+        kv_connector_extra_config:
+          mooncake_bootstrap_port: 25201
+    final_output: false
+    is_comprehension: true
+    default_sampling_params:
+      temperature: 0.4
+      top_p: 0.9
+      top_k: 1
+      max_tokens: 2048
+      seed: 42
+      detokenize: True
+      repetition_penalty: 1.05
+
+  - stage_id: 1
+    stage_type: llm
+    is_decode_only: true
+    runtime:
+      devices: "1"
+    engine_args:
+      max_num_seqs: 64
+      model_stage: thinker
+      model_arch: Qwen3OmniMoeForConditionalGeneration
+      worker_type: ar
+      scheduler_cls: vllm_omni.core.sched.omni_ar_scheduler.OmniARScheduler
+      gpu_memory_utilization: 0.9
+      enforce_eager: true
+      trust_remote_code: true
+      engine_output_type: latent
+      distributed_executor_backend: "mp"
+      enable_prefix_caching: false
+      max_num_batched_tokens: 32768
+      hf_config_name: thinker_config
+      tensor_parallel_size: 1
+      kv_transfer_config:
+        kv_connector: "MooncakeConnector"
+        kv_role: "kv_consumer"
+        kv_rank: 1
+        kv_parallel_size: 2
+        kv_connector_extra_config:
+          mooncake_bootstrap_port: 25202
+    engine_input_source: [0]
+    final_output: true
+    final_output_type: text
+    is_comprehension: true
+    default_sampling_params:
+      temperature: 0.4
+      top_p: 0.9
+      top_k: 1
+      max_tokens: 2048
+      seed: 42
+      detokenize: True
+      repetition_penalty: 1.05
+```
+
+Notes:
+
+- `is_prefill_only: true` marks the thinker stage that only saves KV.
+- `is_decode_only: true` marks the thinker stage that resumes from remote KV.
+- `kv_transfer_config` is required on both stages.
+- The orchestrator forces the prefill stage to run with `max_tokens=1`, so the
+  prefill side only processes the prompt and exports KV.
+
+## 2. Shift the downstream stages by one index
+
+After inserting the extra thinker stage, renumber the remaining stages:
+
+```yaml
+  - stage_id: 2
+    runtime:
+      devices: "2"
+    engine_input_source: [1]
+    sync_process_input_func: vllm_omni.model_executor.stage_input_processors.qwen3_omni.thinker2talker_token_only
+
+  - stage_id: 3
+    runtime:
+      devices: "2"
+    engine_args:
+      max_num_seqs: 1
+    engine_input_source: [2]
+    sync_process_input_func: vllm_omni.model_executor.stage_input_processors.qwen3_omni.talker2code2wav_token_only
+```
+
+Compared with the default Qwen3-Omni config:
+
+- the talker becomes stage `2` instead of stage `1`
+- the code2wav stage becomes stage `3` instead of stage `2`
+- the talker now reads from decode stage `1`
+
+## 3. Add runtime edges for the four-stage pipeline
+
+```yaml
+runtime:
+  enabled: true
+  edges:
+    - from: 0
+      to: 1
+    - from: 1
+      to: 2
+    - from: 2
+      to: 3
+```
+
+## 4. Launch with your custom config
+
+```bash
+vllm serve Qwen/Qwen3-Omni-30B-A3B-Instruct --omni --port 8091 \
+    --deploy-config /path/to/qwen3_omni_pd.yaml
+```
+
+## Operational Notes
+
+- `MooncakeConnector` does not support heterogeneous TP sizes across the PD
+  pair. Keep prefill and decode at the same `tensor_parallel_size`.
+- If the thinker requires TP=2, both thinker stages must use TP=2 and be given
+  separate GPU sets, for example `"0,1"` for prefill and `"2,3"` for decode.
+- Choose connector ports and addresses that match your deployment. The values
+  shown above are examples only.

docs/configuration/stage_configs.md

@@ -3,104 +3,52 @@
 In vLLM-Omni, a model's `PipelineConfig` defines its fixed stage topology, while a deploy configuration controls how those stages run.
 
 !!! note
-    Default deploy config YAMLs (for example, `vllm_omni/deploy/qwen2_5_omni.yaml`, `vllm_omni/deploy/qwen3_omni_moe.yaml`, and `vllm_omni/deploy/qwen3_tts.yaml`) are bundled and loaded automatically when `--deploy-config` is omitted. The resolved pipeline selects its default through `default_deploy_config_name`.
+    Default deploy config YAMLs (for example, `vllm_omni/deploy/qwen2_5_omni.yaml`, `vllm_omni/deploy/qwen3_omni_moe.yaml`, and `vllm_omni/deploy/qwen3_tts.yaml`) are bundled and loaded automatically when `--deploy-config` is omitted. The model registry resolves the right pipeline and deploy YAML by `model_type`.
-
-## Pipeline configuration
-
-`PipelineConfig` and its `StagePipelineConfig` entries are Python definitions
-owned by the model implementation and registered in `OMNI_PIPELINES`. They
-describe fixed model topology and are not accepted in deploy YAMLs.
-
-Common `PipelineConfig` fields include:
-
-| Field | Description |
-| ------- | ------------- |
-| `model_type` | Pipeline identifier used during model and config resolution. |
-| `default_deploy_config_name` | Bundled deploy YAML loaded when the user does not pass `deploy_config`. |
-| `model_arch` | Default Hugging Face architecture for the pipeline. |
-| `hf_architectures` | Architecture names used to identify checkpoints whose `model_type` is shared. |
-| `hf_config_predicate` | Optional predicate used to select between pipelines with otherwise identical HF metadata. |
-| `diffusers_class_name` | Diffusers `_class_name` used to identify Diffusers-style repositories. |
-| `stages` | Ordered tuple of fixed `StagePipelineConfig` definitions. |
-
-Common `StagePipelineConfig` fields include:
-
-| Field | Description |
-| ------- | ------------- |
-| `stage_id` | Stable stage identifier. |
-| `model_stage` | Logical stage name used by runtime and strategy resolution. |
-| `execution_type` | `LLM_AR`, `LLM_GENERATION`, or `DIFFUSION`. |
-| `input_sources` | Upstream stage IDs that provide this stage's inputs. |
-| `final_output` / `final_output_type` | Whether the stage produces a user-visible output and its modality. |
-| `owns_tokenizer` | Whether this stage owns the pipeline tokenizer. |
-| `model_arch`, `hf_config_name` | Stage-specific model architecture and nested HF config selector. |
-| `engine_output_type` | Runtime output representation such as `text`, `latent`, or `audio`. |
-| `custom_process_input_func` | Processor applied to this stage's incoming payload. |
-| `custom_process_next_stage_input_func` | Processor used for full-payload handoff to the next stage. |
-| `async_chunk_process_next_stage_input_func` | Processor used for async chunk handoff. |
-| `sampling_constraints` | Model-owned sampling constraints that deploy defaults cannot override. |
-
-To add or change topology, define and register a new pipeline variant. Use
-deploy YAML only for runtime placement, resource sizing, connectors, and other
-deployment overrides.
 
 ## Deploy configuration schema
 
 The new deploy schema lives under `vllm_omni/deploy/` and is paired with a frozen `PipelineConfig` registered by the model's `pipeline.py`. Each deploy YAML has these top-level fields:
 
 | Field | Type | Required | Default | Description |
-| ------- | ------ | ---------- | --------- | ------------- |
+|-------|------|----------|---------|-------------|
 | `base_config` | str (path) | optional | — | Overlay parent (relative or absolute). `stages:` / `platforms:` deep-merged by stage_id; other scalars overlay-wins. Intended for user-authored overlays; prod yamls stay flat. |
 | `async_chunk` | bool | optional | `true` | Enable chunked streaming between stages. Pin to `false` if the pipeline runs end-to-end. |
-| `session_mode` | str | optional | `"turn"` | Session behavior. MiniCPM-o 4.5 deploy YAMLs set `"duplex"` so `/v1/realtime` and chat share the same profile. |
-| `active_stream_window` | int | optional | `0` | Number of active downstream stream slots; `0` preserves all-stream cycling. |
-| `duplex_session` | dict | optional | runtime defaults | Full-duplex session lifecycle, buffering, replay, and capacity limits. |
 | `connectors` | dict | optional | `null` | Named connector specs (`{name, extra}`). Referenced by each stage's `input_connectors` / `output_connectors`. See [Connector schema](#connector-schema). |
 | `edges` | list | optional | `null` | Explicit edge list for the KV transfer graph. Auto-derived from stage inputs if omitted. |
-| `stages` | list | optional | `[]` | Per-stage runtime overrides matched by `stage_id`. Pipeline stages are still created from `PipelineConfig` when this list is empty. |
+| `stages` | list | required | — | Per-stage engine args + wiring (see [Stage fields](#stage-fields)). |
 | `platforms` | dict | optional | `null` | Keyed by `npu` / `rocm` / `xpu`, each contains a `stages:` list with per-platform overrides applied on top of the CUDA defaults. |
-| `pipeline` | str | optional | `null` | Override the auto-detected pipeline registry key (used for structural variants like `qwen2_5_omni_thinker_only` / `qwen3_omni_moe_thinker_only`). |
+| `pipeline` | str | optional | `null` | Override the auto-detected pipeline registry key (used for structural variants like `qwen2_5_omni_thinker_only`). |
-| `trust_remote_code` | bool \| null | optional | `null` | **Pipeline-wide.** Trust HF remote code on model load; applies to every stage when specified. |
+| `trust_remote_code` | bool | optional | `true` | **Pipeline-wide.** Trust HF remote code on model load; applies to every stage. |
 | `distributed_executor_backend` | str \| null | optional | `null` | **Pipeline-wide.** Distributed executor backend forwarded to vLLM (`"mp"`, `"ray"`, `"external_launcher"`). If omitted, vLLM auto-selects backend from runtime topology. |
 | `dtype` | str \| null | optional | `null` | **Pipeline-wide.** Model dtype for every stage. |
 | `quantization` | str \| null | optional | `null` | **Pipeline-wide.** Quantization method for every stage. |
-| `enable_prefix_caching` | bool \| null | optional | `null` | **Pipeline-wide.** Prefix cache toggle applied to every stage when specified. |
+| `enable_prefix_caching` | bool | optional | `false` | **Pipeline-wide.** Prefix cache toggle applied to every stage. |
 | `enable_chunked_prefill` | bool \| null | optional | `null` | **Pipeline-wide.** Chunked prefill toggle applied to every stage. |
-| `data_parallel_size` | int \| null | optional | `null` | **Pipeline-wide.** DP degree for every stage. |
+| `data_parallel_size` | int | optional | `1` | **Pipeline-wide.** DP degree for every stage. |
-| `pipeline_parallel_size` | int \| null | optional | `null` | **Pipeline-wide.** PP degree for every stage. |
+| `pipeline_parallel_size` | int | optional | `1` | **Pipeline-wide.** PP degree for every stage. |
-| `custom_voice_dir` | str \| null | optional | `null` | **Pipeline-wide.** Directory containing custom voice profiles for supported TTS models. |
 
-For fields whose deploy default is `null`, the deploy layer contributes no
+Note: for diffusion path, `distributed_executor_backend` currently defaults to
-override. The effective value may still come from a platform section, an
+`mp`, and `ray` / `external_launcher` are not fully supported yet.
-explicit CLI or stage override, or the downstream vLLM engine default.
-
-Note: for the diffusion path, an omitted `distributed_executor_backend` selects
-`uni` on a single GPU (in-process worker, no MessageQueue / `/dev/shm` output
-segments) and `mp` when `num_gpus > 1`. Set `mp` explicitly to keep a worker
-subprocess on one GPU. `ray` / `external_launcher` are not fully supported yet.
 
 ### Stage fields
 
-Each entry under `stages:` accepts any `StageDeployConfig` field directly (no nested `engine_args:`). Only fields whose value legitimately varies across stages live here; pipeline-wide settings (trust_remote_code, distributed_executor_backend, dtype, quantization, prefix/chunked prefill, DP/PP sizes) are declared at the top level and applied to every stage. Unknown keys fall through to `engine_extras:` and are forwarded to the engine. Frequently used fields are listed below; the source-of-truth schema is `StageDeployConfig` in `vllm_omni/config/stage_config.py`.
+Each entry under `stages:` accepts any `StageDeployConfig` field directly (no nested `engine_args:`). Only fields whose value legitimately varies across stages live here; pipeline-wide settings (trust_remote_code, distributed_executor_backend, dtype, quantization, prefix/chunked prefill, DP/PP sizes) are declared at the top level and applied to every stage. Unknown keys fall through to `engine_extras:` and are forwarded to the engine.
 
 | Field | Type | Required | Default | Description |
-| ------- | ------ | ---------- | --------- | ------------- |
+|-------|------|----------|---------|-------------|
 | `stage_id` | int | required | — | Stage identity; matched against `PipelineConfig.stages[*].stage_id`. |
-| `max_num_seqs` | int \| null | optional | `null` | Max concurrent sequences per stage. |
+| `max_num_seqs` | int | optional | `64` | Max concurrent sequences per stage. |
-| `gpu_memory_utilization` | float \| null | optional | `null` | Per-stage total memory target; used for automatic KV-cache sizing. |
+| `gpu_memory_utilization` | float | optional | `0.9` | Per-stage memory budget. |
-| `kv_cache_memory_bytes` | int \| null | optional | `null` | Explicit per-rank KV-cache budget in bytes; overrides automatic sizing. |
+| `tensor_parallel_size` | int | optional | `1` | TP degree for this stage. |
-| `tensor_parallel_size` | int \| null | optional | `null` | TP degree for this stage. |
+| `enforce_eager` | bool | optional | `false` | Disable CUDA graphs. |
-| `enforce_eager` | bool \| null | optional | `null` | Disable CUDA graphs. |
+| `max_num_batched_tokens` | int | optional | `32768` | Prefill budget. |
-| `max_num_batched_tokens` | int \| null | optional | `null` | Per-stage prefill/token budget; also contributes to the native maximum in-flight token limit. |
+| `max_model_len` | int \| null | optional | `null` | Per-stage context length (auto-sets `VLLM_ALLOW_LONG_MAX_MODEL_LEN=1` when larger than HF default). |
-| `max_model_len` | int \| null | optional | `null` | Per-sequence context or KV length; `-1` enables native cache-capacity auto-fitting, while values above the HF default auto-set `VLLM_ALLOW_LONG_MAX_MODEL_LEN=1`. |
 | `async_scheduling` | bool \| null | optional | `null` | Per-stage async scheduling toggle. |
-| `devices` | str \| null | optional | `null` | Device list assigned to this stage. The number of device ids must equal this stage's local world size (`tensor_parallel_size` × local data-parallel size × `pipeline_parallel_size`, or `num_replicas` × that product for a replica pool); a mismatch fails early — see the note below. |
+| `devices` | str | optional | `"0"` | `CUDA_VISIBLE_DEVICES`-style device list. |
 | `output_connectors` | dict \| null | optional | `null` | Keyed by `to_stage_<n>`; values are names registered under top-level `connectors:`. |
 | `input_connectors` | dict \| null | optional | `null` | Keyed by `from_stage_<n>`; values are names registered under top-level `connectors:`. |
 | `default_sampling_params` | dict \| null | optional | `null` | Baseline sampling params. Deep-merged with pipeline `sampling_constraints` (pipeline wins). |
-| `engine_extras` | dict | optional | `{}` | Catch-all for engine fields not listed above; deep-merged across overlays and forwarded to the stage engine. |
+| `engine_extras` | dict | optional | `{}` | Catch-all for keys not listed above; deep-merged across overlays. Also carries per-stage overrides of pipeline-wide settings (e.g. stage-specific `dtype`). |
-
-**Note:** a stage's `devices` count must equal its local world size (`tensor_parallel_size` × `data_parallel_size_local` × `pipeline_parallel_size`, falling back to global `data_parallel_size` when the local size is unset), or `num_replicas` × that product for a replica pool. A mismatch fails early and names the offending stage. A top-level `--tensor-parallel-size` is broadcast to every stage, so it can make a single-GPU stage violate this contract ([issue #5003](gh-issue:5003)); fix that case with `--stage-overrides` (set `tensor_parallel_size` and `devices` together per stage) or set TP only on the multi-GPU stage.
 
 ### Connector schema
 
@@ -112,12 +60,12 @@ connectors:
     name: <ConnectorClassName>     # required — class registered in vllm_omni.distributed
     extra:                         # optional — forwarded to the connector's __init__
       <key>: <value>
-      # Additional connector-specific options
+      ...
 ```
 
-| Connector class          | Use case                                                              | `extra` keys                                                                                                      |
+| Connector class | Use case | `extra` keys |
-|--------------------------|-----------------------------------------------------------------------|-------------------------------------------------------------------------------------------------------------------|
+|-----------------|----------|--------------|
-| `SharedMemoryConnector`  | Same-host KV transfer between stages (default for bundled YAMLs).     | None. All payloads use shared memory.                                                                             |
+| `SharedMemoryConnector` | Same-host KV transfer between stages (default for bundled YAMLs). | None. All payloads use shared memory. |
 | `MooncakeStoreConnector` | Cross-host KV transfer over TCP. Required for multi-node deployments. | `host`, `metadata_server`, `master`, `segment` (int bytes), `localbuf` (int bytes), `proto` (`"tcp"` / `"rdma"`). |
 
 A stage references a connector by name in its `input_connectors` / `output_connectors`:
@@ -137,7 +85,7 @@ stages:
 ### CLI flags
 
 | Flag | Description |
-| ------ | ------------- |
+|------|-------------|
 | `--deploy-config PATH` | Load a deploy YAML. **Optional** — when omitted, the bundled `vllm_omni/deploy/<model_type>.yaml` is auto-loaded by the model registry. |
 | `--stage-overrides JSON` | Per-stage JSON overrides, e.g. `'{"0":{"gpu_memory_utilization":0.5}}'`. Per-stage values always win over global flags. |
 | `--async-chunk` / `--no-async-chunk` | Flip the deploy YAML's `async_chunk:` bool. Unset (default) leaves the YAML value in force. |
@@ -251,7 +199,7 @@ CUDA_VISIBLE_DEVICES=0 vllm serve Qwen/Qwen3-Omni-30B-A3B-Instruct --omni \
 Effective config per stage after the merge:
 
 | Stage | Field | Final value | Source |
-| ------- | ------- | ------------- | -------- |
+|-------|-------|-------------|--------|
 | 0 | `gpu_memory_utilization` | `0.9` | base YAML (overlay didn't touch stage 0) |
 | 0 | `max_num_seqs` | `8` | per-stage CLI (`--stage-overrides`) — wins over base `32` |
 | 0 | `max_model_len` | `16384` | global CLI |
@@ -275,17 +223,300 @@ when loading a deploy configuration.
 Examples:
 
 For offline inference (assuming the necessary dependencies have been imported):
-
 ```python
 model_name = "Qwen/Qwen2.5-Omni-7B"
 omni = Omni(model=model_name, deploy_config="/path/to/deploy_config.yaml")
 ```
 
 For online serving:
-
 ```bash
 vllm serve Qwen/Qwen2.5-Omni-7B --omni --port 8091 --deploy-config /path/to/deploy_config.yaml
 ```
 
 !!! important
     We are actively iterating on the definition of deployment configurations, and we welcome feedback from users and developers.
+
+## Legacy stage configuration reference
+
+The remainder of this page documents the deprecated `stage_args` schema for models that have not migrated to deploy configurations. The following historical Qwen2.5-Omni configuration illustrates that schema; current Qwen2.5-Omni deployments should use `vllm_omni/deploy/qwen2_5_omni.yaml` instead.
+
+```python
+# Historical stage configuration for Qwen2.5-Omni.
+stage_args:
+  - stage_id: 0 # mark the unique id for each stage
+    runtime: # The disaggregated configuration
+      process: true  # Run this stage in a separate process
+      devices: "0" # Logical device index for this stage (mapped through CUDA_VISIBLE_DEVICES / ASCEND_RT_VISIBLE_DEVICES if set)
+    engine_args: # Engine arguments for a certain engine
+      model_stage: thinker
+      max_num_seqs: 1
+      model_arch: Qwen2_5OmniForConditionalGeneration # The model implementation registered in model_executor/models/registry.py
+      worker_type: ar # The specific worker used
+      scheduler_cls: vllm_omni.core.sched.omni_ar_scheduler.OmniARScheduler # The specific scehduler used
+      gpu_memory_utilization: 0.8 # The gpu memory allocation for the stage within a single chip
+      enforce_eager: true  # Now we only support eager mode
+      trust_remote_code: true # Needed by huggingface config parsing
+      engine_output_type: latent  # It claims that the stage will input latent hiddenstates besides token ids
+      enable_prefix_caching: false # For request with hiddenstates output, the prefix caching is not supported now
+    is_comprehension: true # If the stage is a text or multimodal comprehension module. If it is, the AsyncOmni will use its tokenizer as default
+    final_output: true # If the stage has output as part of final outputs. If it is false, which means that the stage only works as a intermediate role.
+    final_output_type: text # What is the final output type. It can be text and audio now.
+    default_sampling_params: # sampling parameters for the stage. Their meaning aligns with vLLM.
+      temperature: 0.0
+      top_p: 1.0
+      top_k: -1
+      max_tokens: 2048
+      seed: 42
+      detokenize: True
+      repetition_penalty: 1.1
+  - stage_id: 1
+    runtime:
+      process: true
+      devices: "1"
+    engine_args:
+      model_stage: talker
+      max_num_seqs: 3
+      model_arch: Qwen2_5OmniForConditionalGeneration
+      worker_type: ar
+      scheduler_cls: vllm_omni.core.sched.omni_ar_scheduler.OmniARScheduler
+      gpu_memory_utilization: 0.8
+      enforce_eager: true
+      trust_remote_code: true
+      enable_prefix_caching: false
+      engine_output_type: latent
+    engine_input_source: [0]
+    default_sampling_params:
+      temperature: 0.9
+      top_p: 0.8
+      top_k: 40
+      max_tokens: 2048
+      seed: 42
+      detokenize: True
+      repetition_penalty: 1.05
+      stop_token_ids: [8294]
+  - stage_id: 2
+    runtime:
+      process: true
+      devices: "0"            # Example: use a different GPU than the previous stage; use "0" if single GPU
+    engine_args:
+      model_stage: code2wav
+      max_num_seqs: 1
+      model_arch: Qwen2_5OmniForConditionalGeneration
+      worker_type: generation
+      scheduler_cls: vllm_omni.core.sched.omni_generation_scheduler.OmniGenerationScheduler
+      gpu_memory_utilization: 0.15
+      enforce_eager: true
+      trust_remote_code: true
+      enable_prefix_caching: false
+      engine_output_type: audio
+    engine_input_source: [1]
+    final_output: true
+    final_output_type: audio
+    default_sampling_params:
+      temperature: 0.0
+      top_p: 1.0
+      top_k: -1
+      max_tokens: 2048
+      seed: 42
+      detokenize: True
+      repetition_penalty: 1.1
+
+# Top-level runtime config (concise): default windows and stage edges
+runtime:
+  enabled: true
+
+  edges:
+    - from: 0                   # thinker → talker: trigger only after receiving full input (-1)
+      to: 1
+    - from: 1                   # talker → code2wav: trigger only after receiving full input (-1)
+      to: 2
+
+```
+
+## Legacy stage configuration arguments
+
+Each stage in the `stage_args` list contains the following configuration options:
+
+### `stage_id`
+
+A unique identifier for each stage in the multi-stage pipeline. Stages are numbered sequentially starting from 0, and this ID is used to reference stages in inter-stage dependencies (e.g., `engine_input_source`).
+
+### `prompt_expand_func` (Optional)
+
+A custom Python function hook for the LLM stage (Stage 0) that expands a single incoming prompt object into multiple prompts. This is primarily used for multi-modal Classifier-Free Guidance (CFG), where it generates the necessary companion requests (like a negative text prompt) and tags them with internal roles (e.g., `cfg_text`). This ensures the upstream LLM generates the needed contextual hidden states for both the conditional and unconditional generations simultaneously.
+
+### `cfg_kv_collect_func` (Optional)
+
+A custom Python function hook for downstream diffusion stages (Stage 1+) to collect, map, and process the KV caches transferred from the companion requests fired by `prompt_expand_func`. It aggregates the hidden condition states cleanly (e.g., binding them as `cfg_text_past_key_values` and `cfg_text_kv_metadata`), allowing the diffusion runtime to perform CFG smoothly without redundantly evaluating text paths on the DiT workers.
+
+### `runtime`
+
+Configuration for disaggregated execution of the stage, controlling how the stage is deployed and executed.
+
+#### `runtime.process`
+
+Whether to run this stage in a separate process. When set to `true`, the stage will be executed in an isolated process, enabling better resource isolation and parallel execution across different stages. This is essential for multi-GPU deployments where different stages run on different devices.
+
+Default: `true`
+
+#### `runtime.devices`
+
+Logical device indices for this stage, specified as a string. Values are **logical indices** (`0`, `1`, `2`, ...) — not physical GPU IDs — and are mapped through the platform's visibility env var (`CUDA_VISIBLE_DEVICES` on CUDA, `ASCEND_RT_VISIBLE_DEVICES` on NPU) before being applied via `torch.cuda.set_device()` (or the equivalent).
+
+Example: if `CUDA_VISIBLE_DEVICES=0,2,4` is set in the environment, then `devices: "0"` selects physical GPU 0 (the first visible), `devices: "1"` selects physical GPU 2, and `devices: "0,1"` makes physical GPUs 0 and 2 available to the stage. If no visibility env var is set, logical and physical IDs coincide.
+
+Default: `"0"`
+
+#### `engine_args.max_num_seqs`
+
+The maximum number of sequences for concurrent processing in this stage. For
+LLM stages, this controls the vLLM scheduler's maximum concurrent sequences.
+For diffusion stages, this controls scheduler wave capacity in both request
+batch mode and step batch mode. `max_num_seqs: 1` is the serial/conservative
+path; values above `1` allow compatible requests to batch when the selected
+diffusion pipeline supports that batching mode.
+
+Default: `1`
+
+#### `engine_args.request_batch_max_wait_ms`
+
+The maximum time, in milliseconds, that a diffusion request-mode stage may wait
+before the first `schedule()` of a new scheduler wave so compatible requests can
+accumulate for request-level batching. This only applies to diffusion pipelines
+that support request-level batching with `step_execution` disabled.
+
+Use this together with `max_num_seqs > 1` for bursty serving traffic. `0`
+disables admission waiting and preserves the lowest first-request latency.
+For diffusion execution and batching tuning, see
+[Diffusion Execution Modes](../user_guide/diffusion/execution_modes.md).
+
+Default: `0.0`
+
+### `engine_args`
+
+Engine arguments for configuring the LLM engine, diffusion engine, or other engine types used by this stage.
+
+#### `engine_args.model_stage`
+
+The name identifier for this model stage within the multi-stage architecture. This is used internally to distinguish different stages of the same model (e.g., "thinker", "talker", "code2wav" in Qwen2.5-Omni).
+
+#### `engine_args.model_arch`
+
+The model architecture class name that is registered in `model_executor/models/registry.py`. This specifies which model implementation to use for this stage. The class must be registered in the model registry for vLLM-Omni to locate and instantiate it.
+
+#### `engine_args.worker_cls`
+
+The specific worker class to use for this stage. This determines how the model computations are executed. Examples include `vllm_omni.worker.gpu_ar_worker.GPUARWorker` for autoregressive stages and `vllm_omni.worker.gpu_generation_worker.GPUGenerationWorker` for diffusion-based stages.
+
+#### `engine_args.scheduler_cls`
+
+The scheduler class to use for this stage. The scheduler manages request queuing, batching, and execution order. Examples include `vllm_omni.core.sched.omni_ar_scheduler.OmniARScheduler` for standard stages and `vllm_omni.core.sched.omni_generation_scheduler.OmniGenerationScheduler` for diffusion stages.
+
+#### `engine_args.gpu_memory_utilization`
+
+The fraction of GPU memory to allocate for this stage within a single GPU chip. This is a value between 0.0 and 1.0, where 0.8 means 80% of the GPU memory will be used by this stage. This allows fine-grained control over memory allocation when multiple stages share the same GPU or when reserving memory for other operations.
+
+Default: `0.8`
+
+!!! tip "Memory Configuration Guide"
+    For detailed information on how to calculate memory requirements and properly configure `gpu_memory_utilization`, see the [GPU Memory Calculation and Configuration Guide](./gpu_memory_utilization.md).
+
+#### `engine_args.enforce_eager`
+
+Whether to enforce eager execution mode. When set to `true`, the engine will run in eager mode without using CUDA graphs or other compilation optimizations. Currently, vLLM-Omni only supports eager mode.
+
+Default: `true`
+
+#### `engine_args.trust_remote_code`
+
+Whether to trust remote code when loading models from Hugging Face. This is required for models that use custom code in their configuration files. Set to `true` when loading models that require custom model implementations.
+
+Default: `true`
+
+#### `engine_args.engine_output_type`
+
+Specifies the type of output produced by this stage's engine. This determines what kind of data flows to downstream stages. Possible values include `latent` (hidden states), `text` (tokenized text), and `audio` (audio waveforms). When set to `latent`, the stage outputs latent hidden states in addition to token IDs, which are consumed by downstream stages.
+
+Default: `latent`
+
+#### `engine_args.enable_prefix_caching`
+
+Whether to enable prefix caching for this stage. Prefix caching can improve performance by caching KV cache for common prompt prefixes. However, for requests that output hidden states (when `engine_output_type` is `latent`), prefix caching is not currently supported and should be set to `false`.
+
+Default: `false`
+
+### `is_comprehension`
+
+Whether this stage is a text or multimodal comprehension module. When set to `true`, the stage acts as a comprehension module that processes input text or multimodal content. If this is the first comprehension stage, `AsyncOmni` will use its tokenizer as the default tokenizer for the entire pipeline.
+
+Default: `true`
+
+### `final_output`
+
+Whether this stage produces output that is part of the final outputs returned to the user. When set to `false`, the stage only works as an intermediate stage, processing data that flows to downstream stages but not contributing directly to the final response.
+
+Default: `true`
+
+### `final_output_type`
+
+The type of final output produced by this stage. This specifies what format the output will be in when returned to the user. Currently supported values are `text` (for text generation) and `audio` (for audio generation).
+
+Default: `text`
+
+### `default_sampling_params`
+
+Default sampling parameters for this stage. These parameters control the generation behavior and align with vLLM's sampling parameter semantics. These defaults are used when no explicit sampling parameters are provided in the request.
+
+#### `default_sampling_params.temperature`
+
+Sampling temperature for controlling randomness. Lower values (e.g., 0.0) make the output more deterministic and focused, while higher values increase randomness.
+
+Default: `0.0`
+
+#### `default_sampling_params.top_p`
+
+Nucleus sampling parameter. Only tokens with cumulative probability mass up to `top_p` are considered. This helps filter out low-probability tokens while maintaining diversity.
+
+Default: `1.0`
+
+#### `default_sampling_params.top_k`
+
+Top-k sampling parameter. Only the top `k` most likely tokens are considered. Set to `-1` to disable top-k filtering and consider all tokens.
+
+Default: `-1`
+
+#### `default_sampling_params.max_tokens`
+
+Maximum number of tokens to generate in this stage. This limits the length of the output sequence.
+
+Default: `2048`
+
+#### `default_sampling_params.seed`
+
+Random seed for reproducible generation. When set, the random number generator will be initialized with this seed to ensure consistent outputs across runs.
+
+Default: `42`
+
+#### `default_sampling_params.detokenize`
+
+Whether to detokenize the output tokens into text. When set to `true`, token IDs are converted back to readable text strings.
+
+Default: `True`
+
+#### `default_sampling_params.repetition_penalty`
+
+Penalty applied to tokens that have already appeared in the generated sequence. Values greater than 1.0 discourage repetition, while values less than 1.0 encourage it. A value of 1.0 applies no penalty.
+
+Default: `1.1`
+
+### `tts_args` (TTS stages only)
+
+Configuration for Text-to-Speech specific parameters. This section is only applicable to TTS model stages (e.g., `qwen3_tts`).
+
+#### `tts_args.max_instructions_length`
+
+Maximum character length for voice style/emotion instructions. Instructions exceeding this limit will be rejected with a validation error.
+
+Default: `500`
+
+This value can be overridden at runtime using the `--tts-max-instructions-length` CLI parameter when starting the server.

docs/contributing/DOCS_GUIDE.md

@@ -46,7 +46,7 @@ class Omni:
 
     Args:
         model: Model name or path
-        deploy_config: Optional path to a deploy configuration
+        stage_configs: Optional stage configurations
         **kwargs: Additional arguments passed to the engine
 
     Example:

docs/contributing/README.md

@@ -3,7 +3,7 @@
 Thank you for your interest in contributing to vLLM-Omni! This document provides guidelines and instructions for contributing.
 
 !!! note
-    vLLM-Omni hosts developer-facing meetings for Chinese- and English-language audiences. See [Community Meetings](../community/meetings.md) for current schedules, access details, agendas, and past notes.
+    We host weekly developer-facing online meetings to discuss milestones and updates **every Tuesday at 19:30 PDT**. Meeting link as well as the past meeting notes can be found [here](https://tinyurl.com/vllm-omni-meeting).
 
 ## Getting Started
 
@@ -44,151 +44,6 @@ vLLM-Omni's `pre-commit` hooks will now run automatically every time you commit.
     pre-commit run --show-diff-on-failure --color=always --all-files  # runs on all files (short for --all-files)
     ```
 
-!!! warning
-    GitHub Actions `pre-commit` **skips** several local gates so a whole-tree
-    `--all-files` run stays green while historical debt is cleaned up. The
-    current `SKIP` list in
-    [`.github/workflows/pre-commit.yml`](https://github.com/vllm-project/vllm-omni/blob/main/.github/workflows/pre-commit.yml)
-    is `check-mark`, `markdownlint-cli2`, `shellcheck`,
-    `check-spdx-header`, and `mypy-3.10`. Those hooks still run on **your
-    commit** for changed files. A passing GitHub pre-commit check does not mean
-    they passed locally. Hooks **not** on that list (forbidden imports,
-    `torch.cuda`, TTS adapter ratchet, Buildkite schema, Ruff, typos, …) run in
-    both places.
-
-The hooks below are new relative to the previous Omni config (Ruff, typos,
-actionlint, YAML/whitespace, DCO sign-off, and the pickle-only checker were
-already there). `check-pickle-imports` is gone: pickle is now one rule inside
-`check-forbidden-imports`.
-
-| Hook id | Enforces | GitHub Actions |
-| ------- | -------- | -------------- |
-| `markdownlint-cli2` | Markdown in `docs/`, `recipes/`, `README.md`, `CONTRIBUTING.md` (not `.claude/` / `.cursor/` / `CLAUDE.md`) | skipped |
-| `mypy-3.10` | Type-check changed `vllm_omni/` files (model trees excluded). `tests/` uses `--follow-imports skip` | skipped |
-| `mypy-3.11` / `3.12` / `3.13` | Same checker, extra Python versions | not installed; `pre-commit run --hook-stage manual mypy-3.12` |
-| `check-mark` | Every `tests/**/test_*.py` has a CI level mark and a hardware platform mark or helper; no direct `pytest.mark.H100` / SKU | skipped |
-| `check-tts-adapter-migration` | `self._tts_model_type` branches in `serving_speech.py` must not increase | runs |
-| `shellcheck` | `*.sh` quoting / undefined vars | skipped |
-| `check-spdx-header` | Omni SPDX header on `.py` / `.pyi` / `.sh` / `.rs` / `.proto` | skipped |
-| `check-forbidden-imports` | pickle, stdlib `re`/`base64`, Triton, TileLang, Hugging Face Hub download/API | runs |
-| `check-torch-cuda-call` | Direct `torch.cuda.*` helpers outside platform adapters | runs |
-| `check-buildkite` | `.buildkite/*.yml` against the Buildkite schema | runs |
-
-Related config (not a new hook id): Ruff `TID251` also bans `librosa` and several
-`torch.cuda.*` names; typos is pinned and ignores git-describe hashes (`+g` +
-hex) and NumPy `writeable`. Keep the `suggestion` hook last in
-`.pre-commit-config.yaml` so autofix output does not bury the SKIP tip.
-
-#### SPDX headers
-
-Source files must start with:
-
-```text
-# SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
-```
-
-Rust and proto files use `//` instead of `#`. Empty files are exempt. A stale
-`Copyright contributors to the vLLM project` line is **rewritten in place** to
-the Omni copyright; the hook then exits non-zero so you restage the rewrite.
-
-#### Forbidden imports
-
-Under `vllm_omni/`, do not use:
-
-- stdlib `re` → `import regex as re`
-- stdlib `base64` → `import pybase64` (or `import pybase64 as base64`)
-- `pickle` / `cloudpickle` (also banned in most tests)
-- `from huggingface_hub import snapshot_download` / `hf_hub_download` / `HfApi`
-  / similar → `vllm.transformers_utils.repo_utils`
-- direct `triton` / `tilelang` → `vllm.triton_utils` / `vllm.tilelang_utils`
-
-`examples/`, `tests/`, `benchmarks/`, `apps/`, `docs/`, `tools/`, `scripts/`,
-`.buildkite/`, and `.github/` may keep stdlib `re` and `base64`
-(`_NON_LIBRARY_DIRS`). Pickle is stricter: tests are not exempt unless the
-file is on the pickle allowlist.
-
-#### `torch.cuda` call sites
-
-Prefer `current_omni_platform` / `OmniPlatform`. Platform adapters under
-`vllm_omni/platforms/` are allowed; other files must not add new
-`torch.cuda.*` helpers (`device_count`, `synchronize`, `empty_cache`,
-`device()`, `manual_seed`, and similar). `import torch` is fine.
-
-#### shellcheck
-
-The hook does **not** download a binary. Install `shellcheck` with a signed
-package manager, then re-run:
-
-- Debian/Ubuntu/WSL: `sudo apt-get install shellcheck`
-- Fedora: `sudo dnf install ShellCheck`
-- macOS: `brew install shellcheck`
-- Git Bash (MINGW): `scoop install shellcheck` so `shellcheck.exe` is on PATH.
-  WSL is Linux: use `apt-get`, not a Windows `.exe`.
-
-If no native binary is found, the wrapper prints those commands (and
-[the upstream install page](https://github.com/koalaman/shellcheck?tab=readme-ov-file#installing))
-and exits 1. AMD and other vendor scripts are **not** excluded.
-
-#### mypy
-
-`mypy-3.10` runs on commit for changed files. `[tool.mypy]` uses Python 3.12,
-`ignore_missing_imports = true`, and `follow_imports = "silent"`. Wrapper
-`tools/pre_commit/mypy.py` skips `vllm_omni/model_executor/models/` and
-`vllm_omni/diffusion/models/`, and type-checks `tests/` with
-`--follow-imports skip`. Extra versions:
-
-```bash
-pre-commit run --hook-stage manual mypy-3.12
-```
-
-#### Test CI marks
-
-`check-mark` requires each collected test module under `tests/` to
-have at least one CI **level** mark (`core_model`, `advanced_model`,
-`full_model`, `local_model`, or `slow`) and a **hardware** platform mark
-(`cpu`, `cuda`, …) or helper (`hardware_test(` / `hardware_marks(`). SKU marks
-(`H100`, `L4`, …) must come from those helpers so `cards_{n}` is attached;
-direct `pytest.mark.H100` is rejected. GitHub Actions skips this hook; local
-commit does not. See the
-[test writing guide](./ci/test_writing_guide.md).
-
-#### TTS adapter ratchet
-
-`check-tts-adapter-migration` counts `self._tts_model_type` comparisons in
-`vllm_omni/entrypoints/openai/serving_speech.py`. The count must stay at or
-below `MAX_MODEL_TYPE_BRANCHES` in
-[`tools/pre_commit/check_tts_adapter.py`](https://github.com/vllm-project/vllm-omni/blob/main/tools/pre_commit/check_tts_adapter.py).
-New per-model behavior belongs in
-`vllm_omni/entrypoints/openai/tts_adapters/`. Removing branches **must** lower
-the budget in the same change; raising it is a reviewable policy edit.
-
-#### Markdownlint and Buildkite
-
-`markdownlint-cli2` auto-fixes `docs/`, `recipes/`, and the root README /
-CONTRIBUTING (rules in `.markdownlint.yaml`). Skill files under `.claude/` and
-`.cursor/` are excluded.
-
-`check-buildkite` validates `.buildkite/*.yml` against the official schema
-after the same expansion `upload_pipeline.py` uses. Omni-only keys such as
-`mirror_hardwares` are stripped or rendered first. A few files are skipped in
-`SKIP_FILES` inside `tools/pre_commit/check_buildkite.py`.
-
-#### Extending allowlists and budgets
-
-Prefer fixing the call site. Growing an allowlist or raising a budget is a
-policy change and needs review — do not edit these just to make the hook pass.
-
-| Gate | File | What to edit |
-| ---- | ---- | ------------ |
-| pickle, `re`, `base64`, Hugging Face Hub, Triton/TileLang | [`tools/pre_commit/check_forbidden_imports.py`](https://github.com/vllm-project/vllm-omni/blob/main/tools/pre_commit/check_forbidden_imports.py) | `CHECK_IMPORTS[<rule>].allowed_files`. Use `allowed_dirs` only when a whole tree should stay exempt. |
-| `torch.cuda` | [`tools/pre_commit/check_torch_cuda.py`](https://github.com/vllm-project/vllm-omni/blob/main/tools/pre_commit/check_torch_cuda.py) | `ALLOWED_FILES`, or `ALLOWED_PREFIXES` for a platform adapter tree. |
-| TTS `_tts_model_type` branches | [`tools/pre_commit/check_tts_adapter.py`](https://github.com/vllm-project/vllm-omni/blob/main/tools/pre_commit/check_tts_adapter.py) | `MAX_MODEL_TYPE_BRANCHES` (down only, unless the PR justifies a raise). |
-| Buildkite skip list | [`tools/pre_commit/check_buildkite.py`](https://github.com/vllm-project/vllm-omni/blob/main/tools/pre_commit/check_buildkite.py) | `SKIP_FILES` for pipelines that cannot be expanded. |
-
-To skip one hook for a single commit (discouraged): `SKIP=<hook-id> git commit`.
-To bypass every hook: `git commit --no-verify` (also discouraged).
-
 ### Documentation
 
 MkDocs is a fast, simple and downright gorgeous static site generator that's geared towards building project documentation. Documentation source files are written in Markdown, and configured with a single YAML configuration file, `mkdocs.yml`.
@@ -267,7 +122,7 @@ Repository skills guide the agent, but they do not replace contributor judgment,
 If you encounter a bug or have a feature request, please search existing issues first to see if it has already been reported. If not, please file a new issue, providing as much relevant information as possible.
 
 !!! important
-    Do not report suspected security vulnerabilities through a public issue, pull request, discussion, or Slack channel. Follow the [security disclosure instructions](../community/contact_us.md#security-disclosures) to arrange a private report.
+    If you discover a security vulnerability, please report it by creating a GitHub issue with the `security` label.
 
 ## Pull Requests & Code Reviews
 
@@ -307,14 +162,12 @@ Only specific types of PRs will be reviewed. The PR title is prefixed appropriat
 Before submitting a PR, run the [precheck-pr skill](https://github.com/vllm-project/vllm-omni/blob/main/.claude/skills/precheck-pr/SKILL.md) with the code agent for a self-review against project conventions:
 
 The skill offers two modes:
-
 - **Quick (~3 min):** catches showstoppers — PR title format, missing benchmark claims, rebase status
 - **Full (~10 min):** thorough maintainer-grade review — dead code scan, copy-paste detection, import hygiene
 
 The precheck covers five PR types: Bug Fix, Performance, New Model, Diffusion Model, and General. Each type has a tailored checklist that validates evidence quality (repro steps, A/B benchmarks, registry entries, etc.). See the [precheck-pr skill](https://github.com/vllm-project/vllm-omni/blob/main/.claude/skills/precheck-pr/SKILL.md) for the full checklist.
 
 ### Local Test
-
 Please run the L1 and L2 test cases locally first and attach the results before contacting us to add the "ready" label. Please refer to the [test instructions](./ci/test_execution_guide.md) for running the test cases.
 
 ### Automatic skip-ci (docs and pytest skip marks)
@@ -336,7 +189,7 @@ These PR shapes all trigger skip-ci:
 Skip-ci does **not** apply when the diff also touches product code (for example `vllm_omni/`), or when test files change assertions, imports, fixtures, or other non-skip logic. If the diff cannot be resolved (non-PR branches outside `main`), CI runs as usual.
 
 !!! note
-    Skipping L2/L3 does **not** disable the Docker image build step. Nightly (L4) upload can still run when the PR has a `nightly-test` label or on scheduled `main` builds with `NIGHTLY=1`. Weekly (L5) upload can still run with `weekly-test`, `WEEKLY=1`, or `NON_CRITICAL=1`. Bootstrap child steps live in `bootstrap-upload-steps.yml`; the hook entry step runs `upload_pipeline.py --upload <platform>/bootstrap-upload-steps.yml`, which injects `if` by step `key` from skip-ci before upload. See [CI Settings — Diff-aware CI](./ci/ci_settings.md#diff-aware-ci) and [Test System Overview](./ci/test_system_overview.md).
+    Skipping L2/L3 does **not** disable the Docker image build step. Nightly (L4) upload can still run when the PR has a `nightly-test` label or on scheduled `main` builds with `NIGHTLY=1`. Bootstrap child steps live in `bootstrap-upload-steps.yml`; the hook entry step runs `upload_pipeline.py --upload <platform>/bootstrap-upload-steps.yml`, which injects `if` by step `key` from skip-ci before upload. See [CI Settings — Diff-aware CI](./ci/ci_settings.md#diff-aware-ci) and [Test System Overview](./ci/test_system_overview.md).
 
 ### Code Quality
 

docs/contributing/ci/ci_settings.md

@@ -8,7 +8,7 @@ For CI levels (L1–L5) and triggers, see [Test System Overview](./test_system_o
 
 Canonical layout (prefer these paths for new changes):
 
-```text
+```
 .buildkite/
 ├── common/                          # Shared across platforms
 │   ├── scripts/
@@ -48,7 +48,7 @@ Canonical layout (prefer these paths for new changes):
     └── scripts/
 ```
 
-### Placement rules
+**Placement rules**
 
 | Rule | Detail |
 | ---- | ------ |
@@ -57,11 +57,13 @@ Canonical layout (prefer these paths for new changes):
 | Bootstrap vs test YAML | **Bootstrap** (`pipeline*.yml`) builds images and uploads **child** test pipelines. **Test** YAML (`test-*.yml`) lists pytest steps only. |
 | Register CI YAML in `skip_ci.py` | If you add a new whitelisted test pipeline file, update `L2_YAML_FILES`, `L3_YAML_FILES`, or `L45_YAML_FILES` in `.buildkite/common/scripts/skip_ci.py` so skip-ci paths stay correct. |
 
+There are still **legacy copies** at `.buildkite/*.yaml` (without the `cuda/` prefix). Treat `.buildkite/cuda/*` as source of truth.
+
 ## Platform comparison
 
 | Platform | Bootstrap entry | Test job files | Upload mechanism | Job hardware in YAML |
 | -------- | ---------------- | -------------- | ---------------- | -------------------- |
-| **CUDA** | `cuda/pipeline.yml` | `test-ready.yml`, `test-merge.yml`, `test-nightly.yml`, `test-weekly.yml` | `upload_pipeline.py --upload` (expands uploader-only keys) | omit `mirror_hardwares` (from `-m`) / preset string + `MIRROR_HW` |
+| **CUDA** | `cuda/pipeline.yml` | `test-ready.yml`, `test-merge.yml`, `test-nightly.yml`, `test-weekly.yml` | `upload_pipeline.py --upload` (expands uploader-only keys) | `mirror_hardwares: <preset>` (string) |
 | **NPU** | `npu/pipeline-npu.yml` | `test-npu-ready.yml`, `test-npu-nightly.yml` | `upload_pipeline.py --upload` | `mirror_hardwares: a2b3_npu_1` / `a2b3_npu_4` / `a3_npu_2` |
 | **AMD** | `amd/scripts/bootstrap-amd-omni.sh` | `test-amd-ready.yml`, `test-amd-merge.yml` | Jinja (`test-template-amd-omni.j2`) → `pipeline upload` | `agent_pool` + `mirror_hardwares: [amdproduction]` (array, template filter) |
 | **Intel** | `intel/scripts/bootstrap-intel-omni.sh` | `intel/pipeline-intel.yml` (steps inline) | Direct `pipeline upload` | Inline `agents.queue` on each step |
@@ -79,34 +81,11 @@ Canonical layout (prefer these paths for new changes):
     | L2 | `cuda/test-ready.yml` | `ready` label |
     | L3 | `cuda/test-merge.yml` | `merge-test` label / main merge |
     | L4 | `cuda/test-nightly.yml` | `nightly-test` label or `NIGHTLY=1` |
-    | L5 | `cuda/test-weekly.yml` | `weekly-test` label, `WEEKLY=1`, or `NON_CRITICAL=1` (see step `if`s below) |
+    | L5 | `cuda/test-weekly.yml` | `weekly-test` label or `WEEKLY=1` |
-
-    On scheduled `main` builds, `test-weekly.yml` is uploaded when **`WEEKLY=1` or `NON_CRITICAL=1`**. Inside that file:
-
-    - **Reliability** / **Perf Test** / **Simple · CPU Coverage Test** → `WEEKLY=1` (or PR label `weekly-test` for Reliability/Perf). Pipeline upload for scheduled env vars still requires `main`.
-    - **E2E Tests** group (slow Omni/TTS/Diffusion sweeps) → `NON_CRITICAL=1`
-
-    **`--e2e`:** when uploading ready/merge on `main` with **`WEEKLY=1`**, bootstrap passes `--e2e` so only the **E2E Test** group is kept (same flag `run_cov_split.sh` uses to enable per-model coverage).
 
     **Upload:** `upload_pipeline.py --upload` expands uploader-only keys before Buildkite upload.
 
-    **Hardware in YAML:** omit `mirror_hardwares` to compose `{chip}_{n}` from pytest `-m` (SKU + `cards_n`). Several positive `cards_*` take the **max** n (`cards_2 and cards_3` → `{chip}_3`). `not cards_1` with no positive `cards_*` uses that chip's **highest existing** preset in `ci_mirror_hardwares.yml` (L4 → `l4_4`, H100 → `h100_4`, B200 → `b200_4`). Set a **preset string** to force a pool (marks are ignored). Do **not** set `agents` / `plugins` on the same step as `mirror_hardwares`. `MIRROR_HW=b200` remaps inferred jobs whose `-m` includes `B200`, and omits CUDA `h100_*` / `l4_*` strings. Jobs whose `-m` does not match (unset: no H100/L4; set: no B200) are **skipped**.
+    **Hardware in YAML:** `mirror_hardwares: <preset>` (string)—preset names in [`common/ci_mirror_hardwares.yml`](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/common/ci_mirror_hardwares.yml). Do **not** set `agents` / `plugins` on the same step.
-
-    ```yaml
-    # inferred: H100+B200 + cards_2 → h100_2, or b200_2 when MIRROR_HW=b200
-    commands:
-      - pytest -sv tests/e2e -m "full_model and H100 and B200 and omni and cards_2"
-    # no mirror_hardwares
-
-    # inferred: not cards_1 (no positive cards_*) → that chip's highest pool (l4_4 / h100_4 / b200_4)
-    commands:
-      - pytest -sv tests/e2e -m "full_model and diffusion and L4 and not cards_1"
-
-    # forced preset (ignores -m SKU/cards for pool selection)
-    mirror_hardwares: h100_1
-    ```
-
-    Inferred form: unset/`empty` `MIRROR_HW` matches `H100` or `L4` in `-m` (both present → H100); no match skips the step. `MIRROR_HW=b200` must appear in `-m` or the step is skipped. The uploader does **not** rewrite pytest `-m`; B200 collection comes from the YAML expression and from tests that declare `B200` in `hardware_test` / `hardware_marks` (for example `res={"cuda": ["H100", "B200"]}`). `MIRROR_HW` must be empty or `b200` (case-insensitive); unknown values (for example `b20o`) **fail the upload**. A CUDA preset string such as `mirror_hardwares: h100_4` is skipped when `MIRROR_HW=b200`. Unset `MIRROR_HW` keeps string presets. NPU presets ignore `MIRROR_HW`. A `mirror_hardwares` name that is not a key in `ci_mirror_hardwares.yml` **fails the upload**.
 
     **Conventions**
 
@@ -117,9 +96,9 @@ Canonical layout (prefer these paths for new changes):
 
     **Adding a job**
 
-    1. Pick the level file (`test-ready.yml` / `test-merge.yml` / `test-nightly.yml` / `test-weekly.yml`).
+    1. Pick the level file (ready / merge / nightly / weekly).
     2. Add a step under the right **group** (usually **E2E Test** for model pytest).
-    3. Set `label`, `commands`, `mirror_hardwares`, `depends_on: upload-ready-pipeline` (or `upload-merge-pipeline` / `upload-nightly-pipeline` / `upload-weekly-pipeline`).
+    3. Set `label`, `commands`, `mirror_hardwares`, `depends_on: upload-<level>-pipeline`.
     4. For L2/L3 E2E, add `source_file_dependencies` (pytest + model + deploy YAML prefixes).
     5. Dry-run:
 
@@ -133,8 +112,6 @@ Canonical layout (prefer these paths for new changes):
 
     **Test YAML:** `npu/test-npu-ready.yml` (L2), `test-npu-nightly.yml` (L4).
 
-    **L2 trigger:** PR label `ready` (CUDA/AMD L2). PR labels `ready` + `npu-test` trigger NPU CI only.
-
     **Upload:** same as CUDA—`upload_pipeline.py --upload`.
 
     **Hardware in YAML:** `mirror_hardwares` preset (string), expanded to `agents`, top-level `image`, and `plugins`. Presets: `a2b3_npu_1`, `a2b3_npu_4`, `a3_npu_2` in `common/ci_mirror_hardwares.yml`.
@@ -227,15 +204,15 @@ Label triggers (`ready`, `merge-test`) are unchanged—diff-aware logic only red
 | Diff shape | Path | Default L2/L3 |
 | --- | --- | --- |
 | Product code / non-whitelisted paths | normal CI | all on |
-| Docs and/or qualifying skip-mark only | `skip_all` | all off (scheduled `main` NIGHTLY/WEEKLY/`NON_CRITICAL` exceptions below) |
+| Docs and/or qualifying skip-mark only | `skip_all` | all off (scheduled `main` NIGHTLY/WEEKLY exceptions below) |
 | Whitelisted CI YAML only | yaml-gated (`skip_l2_l3`) | per platform/level matrix |
 | Docs/skip-mark **+** whitelisted CI YAML | yaml-gated (`skip_l2_l3`) | same as CI-YAML-only (does **not** widen to normal CI) |
 | Non-qualifying skip-mark and no CI YAML | normal CI | all on |
 | Diff unavailable | normal CI | all on |
 
-**`skip_all` exceptions (CUDA/NPU bootstrap only):** PR labels (`nightly-test`, `merge-test`, `weekly-test`, `ready`, …) do **not** revive jobs under `skip_all`. On `main`, scheduled `NIGHTLY=1` still builds the image and uploads L4; `WEEKLY=1` or `NON_CRITICAL=1` still builds the image and uploads L5 (CUDA). `WEEKLY=1` on `main` also uploads L2/L3 with `--e2e`.
+**`skip_all` exceptions (CUDA/NPU bootstrap only):** PR labels (`nightly-test`, `merge-test`, `npu-test`, `weekly-test`, …) do **not** revive jobs. On `main`, scheduled `NIGHTLY=1` still builds the image and uploads L4 plus L2/L3 (with `--e2e`); `WEEKLY=1` still builds the image and uploads L5 (CUDA).
 
-**Yaml-gated nightly/weekly:** L4/L5 upload steps keep their normal label / `NIGHTLY` / `WEEKLY` / `NON_CRITICAL` conditions (for example PR `nightly-test` still uploads nightly). Only L2/L3 upload steps are matrix-gated.
+**Yaml-gated nightly/weekly:** L4/L5 upload steps keep their normal label / `NIGHTLY` / `WEEKLY` conditions (for example PR `nightly-test` still uploads nightly). Only L2/L3 upload steps are matrix-gated.
 
 ##### Whitelisted CI YAML → platform
 
@@ -308,12 +285,12 @@ CUDA **L2** (`.buildkite/cuda/test-ready.yml`) and **L3** (`.buildkite/cuda/test
 | `source_file_dependencies` | List of path **prefixes**. If any changed file equals a prefix or starts with `prefix/`, keep the step (or group); otherwise omit it. |
 | `mirror_hardwares` | Expand to `agents` + `plugins` (+ optional `image`) from `ci_mirror_hardwares.yml`. |
 
-### Policy
+**Policy**
 
 - **Always uploaded** (no key): groups outside **E2E Test**—Simple Test, Diffusion unit tests, Engine/Model Executor, Distributed, Custom Pipeline, Entrypoints (L2), LoRA / Entrypoints (L3).
 - **Diff-gated**: every **E2E Test** leaf job. List the smallest prefix set per step—pytest file(s), model code under `vllm_omni/model_executor/models/` or `vllm_omni/diffusion/models/`, plus `stage_input_processors/` and `vllm_omni/deploy/*.yaml` when applicable. A **group** may define the key instead; the whole group drops if no prefix matches.
 
-### YAML examples
+**YAML examples**
 
 ```yaml
       - label: "Diffusion · Qwen Image Test"
@@ -335,7 +312,7 @@ CUDA **L2** (`.buildkite/cuda/test-ready.yml`) and **L3** (`.buildkite/cuda/test
         mirror_hardwares: l4_4
 ```
 
-### Local dry-run
+**Local dry-run**
 
 ```bash
 python3 .buildkite/common/scripts/upload_pipeline.py .buildkite/cuda/test-ready.yml
@@ -347,111 +324,15 @@ On PR builds, `upload_pipeline.py` logs `skip '…' (no changes under …)` for
 
 ### Labels and grouping
 
-- Use **groups** for dashboard readability; keep **E2E Test** as the group name CUDA diff filtering expects for `--e2e` weekly runs on main (`WEEKLY=1`).
+- Use **groups** for dashboard readability; keep **E2E Test** as the group name CUDA diff filtering expects for `--e2e` nightly runs on main.
 - Prefix labels by model domain: **Omni ·**, **TTS ·**, **Diffusion ·**, **Simple ·**, etc., matching existing steps.
 
-### Per-model coverage
-
-Pilot (v1): one Merge-tier (L3) job in `.buildkite/cuda/test-merge.yml` uploads
-per-model, per-entry-mode coverage as Buildkite artifacts — "TTS · Qwen3-TTS Base
-Test". Coverage + artifact upload run only when bootstrap uploaded that job with
-`--e2e` (`WEEKLY=1` and `BUILDKITE_BRANCH=main`, via `run_cov_split.sh`); otherwise
-the same job runs plain pytest without `--cov`. It runs on a single GPU so the
-pilot is cheap to reproduce.
-
-**L1 package coverage (CPU):** ready/merge Simple Test jobs run without `--cov`. Scheduled weekly on `main` with `WEEKLY=1` runs **Simple · CPU Coverage Test** in `.buildkite/cuda/test-weekly.yml` (`pytest -m 'core_model and cpu'` + Cobertura XML artifact).
-
-**Naming convention**: `coverage-<model_id>-<mode>-<step_id>.xml.gz`, where
-`<model_id>` is the model's directory name under
-`vllm_omni/diffusion/models/<model_id>/` or
-`vllm_omni/model_executor/models/<model_id>/` (e.g. `z_image`, `qwen3_omni`),
-`<mode>` is `online` or `offline`, and `<step_id>` is `BUILDKITE_STEP_ID`
-(`local` outside Buildkite). `gunzip` before feeding a report to a Cobertura
-consumer.
-
-Reports are gzipped because `--cov=vllm_omni` sets coverage's `source`, which
-makes it walk the package and emit one `<line>` element per statement for every
-file — including the ones a single model's tests never import. That inventory,
-not the coverage, is the size: a Bagel report is 7.1 MB, of which 96.5% is
-`<line>` elements and 77% belongs to the 840 of 1169 files that run never
-imported. A run covering zero lines produces the same 7 MB. gzip takes it to
-roughly 410 KB and changes nothing about the data.
-
-`<step_id>` is what keeps two steps that cover the same model apart. Artifacts are
-scoped to the build, and a weekly build can carry ready/merge (with `--e2e`) and
-weekly tiers at once, so several same-model steps land in one namespace — `qwen3_tts` already has
-both a Base and a CustomVoice job, and `minicpmo_4_5` a base and a duplex one. Two
-uploads on one path do not fail loudly: an exact-name download can report the path
-as ambiguous, and a glob download fetches every match concurrently and renames each
-onto the same destination, so whichever finishes last wins. The step id remains the
-logical step identifier across retries, while Buildkite's default artifact lookup
-selects only the latest attempt — `BUILDKITE_JOB_ID` would instead give every
-attempt its own filename, leaving a failed attempt's partial report to be picked up
-alongside the real one. A step using `parallelism`/matrix would need
-`BUILDKITE_PARALLEL_JOB` added, since its jobs share a step id; neither pilot is
-one.
-
-**Opting in a new model**: replace the job's combined `pytest` command with
-[`run_cov_split.sh`](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/common/scripts/run_cov_split.sh),
-which runs each mode, names the reports, uploads them, and fails the step if
-either half or the upload failed:
-
-```yaml
-commands:
-  - |
-    .buildkite/common/scripts/run_cov_split.sh \
-      --model-id <model_id> \
-      --offline <offline test path> \
-      --online <online test path> \
-      -- <the job's existing pytest flags, e.g. -m '...' --run-level '...'>
-```
-
-Everything after `--` is forwarded verbatim to both runs with quoting preserved,
-so an expression like `-m 'advanced_model and cuda'` survives intact and a job can
-pass anything else it needs. `--offline`/`--online` are repeatable and only one is
-required, so a job with tests in a single mode runs just that half. Runtime is
-bounded solely by the step's `timeout_in_minutes` — raise it, since the split
-loads the model once per mode.
-
-Before splitting, confirm each half actually collects at least one test under the job's forwarded filters — pytest exits with code 5 ("no tests collected") if a half is empty, which the script reports as a failure, red-ing a job that used to pass as one combined invocation.
-
-Also check the tests' `num_cards` against the job's `mirror_hardwares` preset.
-`_cuda_marks` turns `num_cards > 1` into `skipif(device_count() < num_cards)`, so a
-test asking for more GPUs than the preset provides is silently skipped and its
-coverage artifact comes back empty. Splitting such a job produces a file that
-looks fine and measures nothing — "Diffusion · Z Image Test" is the current
-example (online tests want 4 cards, the job runs on `l4_1`), which is why it is
-not a coverage pilot.
-
-Online-mode coverage depends on the `[tool.coverage.run]` `patch`/`sigterm`
-settings in `pyproject.toml`: online tests launch the server with
-`subprocess.Popen` and stop it with SIGTERM, and without those settings the XML
-reflects only the pytest parent process, not the server's code paths.
-
-The upload depends on `buildkite-agent` being callable inside the container.
-CUDA (`l4_*`, `h100_*`) and NPU (`*_npu_*`) presets all use the `kubernetes`
-plugin, which provides the agent. `l4_*` jobs run on the EKS `l4-k8s` queue
-(agent-stack-k8s); they no longer use the docker plugin or
-`mount-buildkite-agent`. A new **docker** preset that runs a coverage job
-still needs `mount-buildkite-agent: true`.
-
-List both `run_cov_split.sh` and `pyproject.toml` in every opted-in job's
-`source_file_dependencies` — both change what the job measures, so without them a
-change there is filtered out of normal PR builds and only surfaces in a later
-nightly. `tests/buildkite/test_upload_pipeline.py` covers the filter behavior with
-a synthetic job (it does not pin real merge labels).
-Editing only the surrounding CI YAML still does not schedule them, so a PR that
-touches just the wiring needs a full E2E run (or the commands run on a GPU host)
-to produce artifacts. When checking a new model's
-artifacts, compare `lines-covered` between the online and offline XML rather than
-just confirming both files exist.
-
 ### Validation checklist
 
 | Check | Command / location |
 | ----- | ------------------ |
 | CUDA render | `python3 .buildkite/common/scripts/upload_pipeline.py .buildkite/cuda/test-<level>.yml` |
-| No leaked uploader keys | Pipe the render into `grep -E 'mirror_hardwares\|source_file_dependencies'` — expect empty |
+| No leaked uploader keys | `… \| grep -E 'mirror_hardwares|source_file_dependencies'` → empty |
 | Skip-ci / upload unit tests | `pytest tests/buildkite/` |
 | Local job replay (CUDA L2+) | `tools/run_ready_jobs.sh`, `tools/run_merge_jobs.sh`, `tools/nightly/run_nightly_jobs.sh` (read YAML from `cuda/`) |
 

docs/contributing/ci/failures.md

@@ -6,7 +6,7 @@ When you open a PR against vLLM-Omni, several CI checks run automatically:
 
 | Check | Platform | What it does |
 | ----- | -------- | ------------ |
-| **pre-commit** | GitHub Actions | Runs Ruff, formatting, typos, YAML validation, forbidden imports, `torch.cuda`, TTS adapter ratchet, and Buildkite schema. Local-only gates (SPDX, shellcheck, markdownlint, mypy-3.10, test marks) are in the workflow `SKIP` env — see [Linting](../README.md#linting). |
+| **pre-commit** | GitHub Actions | Runs linting (Ruff), formatting, spell-checking (typos), and YAML validation. |
 | **Build Wheel** | GitHub Actions | Builds Python wheels for Python 3.11 and 3.12 on Ubuntu. Skipped for docs-only or Markdown-only changes (controlled by `paths-ignore` in the workflow). |
 | **DCO** | GitHub | Verifies every commit has a `Signed-off-by` line. |
 | **docs/readthedocs.org:vllm-omni** | Read the Docs | Builds the MkDocs documentation site. |
@@ -29,13 +29,6 @@ pre-commit run --all-files
 
 Then commit the fixes and push.
 
-A green GitHub pre-commit job can still miss SPDX, shellcheck, mypy-3.10,
-markdownlint, and test-mark coverage because CI skips those IDs. If your
-**local** commit hook fails, follow [Linting](../README.md#linting) for the
-full hook list (forbidden imports, `torch.cuda`, TTS adapter ratchet, and
-Buildkite schema still run in GHA). Do not grow allowlists or raise
-`MAX_MODEL_TYPE_BRANCHES` without review.
-
 ### DCO failures
 
 Every commit must include a `Signed-off-by` line. If you forgot, amend your commits:

docs/contributing/ci/test_examples/l4_functionality_tests.inc.md

@@ -1,8 +1,8 @@
-#### Scope
+**Scope**
 
 For diffusion models, the L4 functionality test covers all or common *diffusion features* that are supported by this model, including several [parallelism acceleration methods](https://docs.vllm.ai/projects/vllm-omni/en/latest/user_guide/diffusion/parallelism_acceleration/), [CPU offloading](https://docs.vllm.ai/projects/vllm-omni/en/latest/user_guide/diffusion/cpu_offload/), [TeaCache](https://docs.vllm.ai/projects/vllm-omni/en/latest/user_guide/diffusion/teacache/) and [Cache-DiT](https://docs.vllm.ai/projects/vllm-omni/en/latest/user_guide/diffusion/cache_dit_acceleration/) cache backends, [quantization methods](https://docs.vllm.ai/projects/vllm-omni/en/latest/user_guide/quantization/overview/).
 
-#### Test Case Design
+**Test Case Design**
 
 For a *high priority* model (currently listed in [issue #1832](https://github.com/vllm-project/vllm-omni/issues/1832)), we use several test cases, each with multiple features turned on, so that each supported feature is tested in at least one test case. This is to relieve the GPU workload on the CI machine. The suggested test case combination is as follows:
 
@@ -34,12 +34,12 @@ For a *normal priority* model, further reduce the number of test cases.
 
 Currently all the features are available in online serving mode. Hence, only need to add `tests/e2e/online_serving/test_{model}_expansion.py`.
 
-#### Code Style
+**Code Style**
 
-- Validation: test that the multimodal output files of your model have the correct shapes. `OnlineOmniClient.send_diffusion_request` should have taken care of this.
+- Validation: test that the multimodal output files of your model have the correct shapes. `OpenAIClientHandler.send_diffusion_request` should have taken care of this.
 - Test marks: always add `full_model` and `diffusion` for L4 nightly `test_*_expansion.py` cases. Add GPU-related marks if needed. Ref: [Markers for Tests](https://docs.vllm.ai/projects/vllm-omni/en/latest/contributing/ci/test_writing_guide/#markers-for-tests).
 - To maximize code reuse, you may refer to
-    - `tests/conftest.py` for `omni_server` (running server in subprocess) and `online_client` fixtures (sending requests and validating output), `generate_synthetic_image` and `assert_XXX_valid` helper.
+    - `tests/conftest.py` for `omni_server` (running server in subprocess) and `openai_client` fixtures (sending requests and validating output), `generate_synthetic_image` and `assert_XXX_valid` helper.
     - `tests/helpers/mark.py` for `@hardware_test(...)` and `hardware_marks`.
     - [Parametrizing tests (pytest doc)](https://docs.pytest.org/en/stable/example/parametrize.html) to reuse test function implementation for different cases.
 - Doc: add a concise docstring for each test function.

docs/contributing/ci/test_examples/l4_performance_tests.inc.md

@@ -1,15 +1,14 @@
 When you want to add L4-level ***performance test*** cases, add entries to JSON files under `tests/dfx/perf/tests/` and run them via `tests/dfx/perf/scripts/run_benchmark.py` (omni / TTS) or `run_diffusion_benchmark.py` (diffusion).
 
-#### Config file layout (in-tree examples)
+**Config file layout (in-tree examples)**
 
 | Model type | Runner | Example JSON files |
 | ---------- | ------ | ------------------ |
-| Omni (nightly) | `run_benchmark.py` | `test_qwen3_omni_no_async_chunk.json`, `test_qwen3_omni_async_chunk.json` (`full_model` without `slow` in `mark`) |
+| Omni | `run_benchmark.py` | `test_qwen3_omni_no_async_chunk.json`, `test_qwen3_omni_async_chunk.json`, `test_qwen3_omni_vllm_text.json`, `test_qwen3_omni_multi_replicas.json` |
-| Omni (weekly) | `run_benchmark.py` | `test_qwen3_omni_async_chunk.json` (CUDA only), `test_qwen3_omni_vllm_text.json`, `test_qwen3_omni_multi_replicas.json` (`slow` in `mark`; **Perf Test** in `test-weekly.yml`) |
 | TTS | `run_benchmark.py` | `test_tts.json`, `test_voxcpm2.json`, `test_higgs_audio_v3.json` |
 | Diffusion | `run_diffusion_benchmark.py` | `test_qwen_image_vllm_omni.json`, `test_bagel_vllm_omni.json`, `test_wan22_i2v_vllm_omni.json`, `test_cosmos3_vllm_omni.json`, … |
 
-#### How runners pick cases
+**How runners pick cases**
 
 Without **`--test-config-file`**, each runner scans all `*.json` under `tests/dfx/perf/tests/` but only keeps its own model type:
 
@@ -18,9 +17,9 @@ Without **`--test-config-file`**, each runner scans all `*.json` under `tests/df
 
 Diffusion cases are detected when the JSON has `server_type` (typically `"vllm-omni"`) or `"diffusion"` in the `mark` array. Omni / TTS JSON has neither.
 
-#### Running perf cases
+**Running perf cases**
 
-Pass **`--test-config-file`** to load one JSON file, or omit it for the bulk scan above. In either mode, pytest **`-m`** filters each case using its JSON `mark`. A single file can contain nightly, weekly, and platform-specific case objects. For example, the nightly object in `test_qwen3_omni_async_chunk.json` declares both H100 and A3, while its `slow` weekly object is CUDA-only; there is no NPU weekly counterpart. Use `-m "H100 and full_model and not slow"`, `-m "H100 and full_model and slow"`, or `-m "npu"` to select the intended schedule and platform.
+Pass **`--test-config-file`** to run one JSON file as-is, or omit it for the bulk scan above and filter with pytest **`-m`** on each case's JSON `mark` (for example `pytest … run_benchmark.py -m "full_model and tts and H100"`).
 
 ```JSON
 {
@@ -32,7 +31,7 @@ Pass **`--test-config-file`** to load one JSON file, or omit it for the bulk sca
     ],
     "server_params": {
         "model": "Qwen/Qwen3-Omni-30B-A3B-Instruct",
-        "stage_config_name": "qwen3_omni_moe.yaml"
+        "stage_config_name": "qwen3_omni.yaml"
     },
     "benchmark_params": [
         {
@@ -56,33 +55,31 @@ Pass **`--test-config-file`** to load one JSON file, or omit it for the bulk sca
 }
 ```
 
-#### Parameter Explanation
+**Parameter Explanation**
 
-##### Overview
+*Overview*
 
-| Field              | Required       | Description                                  |
+| Field            | Required | Description                                                     |
-| ------------------ | -------------- | -------------------------------------------- |
+| ---------------- | -------- | --------------------------------------------------------------- |
-| test_name          | Yes            | Unique identifier for the test case          |
+| test_name        | Yes      | Unique identifier for the test case                             |
-| mark               | No             | Pytest marks; see **`mark` field** below     |
+| mark             | No       | Pytest marks for this case (see **`mark` field** below). Omit only for configs not meant to be filtered by `-m`. |
-| server_params      | Yes            | Server-side configuration parameters         |
+| server_params    | Yes      | Server-side configuration parameters                            |
-| benchmark_params   | Yes            | Benchmark running parameters                 |
+| benchmark_params | Yes      | Benchmark running parameters (supports multiple configurations) |
-| server_type        | Diffusion only | Routes case to run_diffusion_benchmark.py    |
+| server_type      | Diffusion only | When set (e.g. `"vllm-omni"`), routes the case to `run_diffusion_benchmark.py`. |
-| benchmark_endpoint | Diffusion only | Benchmark API path                           |
+| benchmark_endpoint | Diffusion only | API path for the benchmark client (e.g. `/v1/videos`, `/v1/images/generations`). |
 
-Omit `mark` only for configs not meant to be filtered by `-m`. `server_type` is typically `"vllm-omni"`. `benchmark_endpoint` examples: `/v1/videos`, `/v1/images/generations`.
+**`mark` field**
-
-#### `mark` field
 
 Optional top-level field on each perf JSON **case object** (one per `test_name`). `run_benchmark.py` and `run_diffusion_benchmark.py` read it via `tests.dfx.conftest.resolve_pytest_marks` and attach the marks to the corresponding `pytest.param`, so you can filter locally with `-m`.
 
-When `mark` is present, it must be an **array** with exactly one ``hardware_marks`` object (same shape as `@hardware_test` / `hardware_marks()` in `tests/helpers/mark.py`), followed by registered pytest marker name strings such as `full_model`, `slow`, `omni`, `tts`, `diffusion`, or `local_model`.
+When `mark` is present, it must be an **array** with exactly one ``hardware_marks`` object (same shape as `@hardware_test` / `hardware_marks()` in `tests/helpers/mark.py`), followed by registered pytest marker name strings such as `full_model`, `omni`, `tts`, `diffusion`, or `local_model`.
 
 Supported form:
 
 | Form | Example | Effect |
 | ---- | ------- | ------ |
-| Array (single platform) | `"mark": [{"hardware_marks": {"res": {"cuda": "H100"}, "num_cards": 2}}, "full_model", "diffusion"]` | Hardware marks from `hardware_marks(...)`; `num_cards` adds `cards_{n}` (including `cards_1` when omitted). String entries become extra pytest marks. |
+| Array (single platform) | `"mark": [{"hardware_marks": {"res": {"cuda": "H100"}, "num_cards": 2}}, "full_model", "diffusion"]` | Hardware marks from `hardware_marks(...)`; `num_cards` > 1 adds `distributed_*` (+ CUDA `skipif_cuda` when applicable). String entries become extra pytest marks. |
-| Array (multi-platform) | `"mark": [{"hardware_marks": {"res": {"cuda": "H100", "rocm": "MI325"}, "num_cards": 2}}, "full_model", "omni"]` | Same as above, but declares **multiple platforms** on one item. All platforms must share the same `num_cards` (int or identical dict values). Different counts need `@hardware_test` or one `hardware_marks()` per `pytest.param`. Filter examples: `-m "full_model and H100 and cards_2 and cuda"`, `-m "full_model and MI325 and cards_2 and rocm"`. |
+| Array (multi-platform) | `"mark": [{"hardware_marks": {"res": {"cuda": "H100", "rocm": "MI325", "npu": "A2"}, "num_cards": {"cuda": 2, "rocm": 2, "npu": 1}}}, "full_model", "omni"]` | Same as above, but declares **multiple platforms** in one case. Filter examples: `-m "full_model and H100 and cuda"`, `-m "full_model and MI325 and rocm"`. |
 
 Recommended for L4 perf cases:
 
@@ -122,11 +119,11 @@ Multi-GPU diffusion (example: Cosmos3 with `cfg-parallel-size=2`):
 - Diffusion perf: include `"diffusion"` in the `mark` array
 - HunyuanImage local-weight cases: add `"local_model"` to the `mark` array
 
-#### Parametrization IDs
+**Parametrization IDs**
 
 Each `(server, benchmark index)` pair becomes one `pytest.param` with id `{test_name}-{suffix}`. The suffix comes from `benchmark_params[].name` when set (for example `test_tts-p0`, `test_omni-p1`); otherwise it is derived from fields like `task` / `eval_phase`.
 
-#### Benchmark result filenames
+**Benchmark result filenames**
 
 Result files use the **runtime** hardware label from `get_runtime_resource_label()` (detected GPU/NPU name on the machine that ran the job), **not** `mark.hardware_marks.res`. On the default H100 CI pool, `H100` is **omitted** from filenames (`resource_label_for_filename`).
 
@@ -135,40 +132,32 @@ Examples:
 - Omni/TTS: `result_{test_name}_{optional_hw}_{dataset}_....json` under `BENCHMARK_DIR`
 - Diffusion: one aggregate `diffusion_result_{config_stem}_{optional_hw}_{timestamp}.json` per source JSON file (array of all runs from that file)
 
-#### Local commands
+**Local commands**
 
 ```bash
 # Bulk load + filter by JSON mark
 pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py -m "full_model and H100 and diffusion"
 pytest -s -v tests/dfx/perf/scripts/run_benchmark.py -m "full_model and omni and H100"
 
-# Single file (same selectors as the CI Perf steps)
+# Single file (same as nightly CI Perf steps)
 pytest -s -v tests/dfx/perf/scripts/run_diffusion_benchmark.py \
   --test-config-file tests/dfx/perf/tests/test_bagel_vllm_omni.json
 pytest -s -v tests/dfx/perf/scripts/run_benchmark.py \
-  --test-config-file tests/dfx/perf/tests/test_qwen3_omni_async_chunk.json \
+  --test-config-file tests/dfx/perf/tests/test_qwen3_omni_async_chunk.json
-  -m "H100 and full_model and not slow"
-pytest -s -v tests/dfx/perf/scripts/run_benchmark.py \
-  --test-config-file tests/dfx/perf/tests/test_qwen3_omni_async_chunk.json \
-  -m "H100 and full_model and slow"
-pytest -s -v tests/dfx/perf/scripts/run_benchmark.py \
-  --test-config-file tests/dfx/perf/tests/test_qwen3_omni_async_chunk.json \
-  -m "npu"
 ```
 
-<!-- markdownlint-disable-next-line MD051 -->
+See also [Markers for Tests](#markers-for-tests) for registered hardware markers (`H100`, `L4`, `cuda`, `distributed_cuda`, …).
-See also [Markers for Tests](#markers-for-tests) for registered hardware markers (`H100`, `L4`, `cuda`, `cards_1`, …).
 
-#### `server_params` Configuration
+**`server_params` Configuration**
 
-##### Basic Parameters
+*Basic Parameters*
 
-| Parameter         | Required | Example                   | Description                    |
+| Parameter         | Required | Example                            | Description                   |
-| ----------------- | -------- | ------------------------- | ------------------------------ |
+| ----------------- | -------- | ---------------------------------- | ----------------------------- |
-| model             | Yes      | Qwen/Qwen3-Omni-30B-A3B…  | Model name or path             |
+| model             | Yes      | "Qwen/Qwen3-Omni-30B-A3B-Instruct" | Model name or path            |
-| stage_config_name | Yes      | qwen3_omni_moe.yaml       | Deploy configuration file name |
+| stage_config_name | Yes      | "qwen3_omni.yaml"                  | Stage configuration file name |
 
-##### Dynamic Configuration (update/delete)
+*Dynamic Configuration (update/delete)*
 
 Supports incremental modifications based on the basic configuration:
 
@@ -177,40 +166,38 @@ Supports incremental modifications based on the basic configuration:
 | update    | Update or add configuration items    |
 | delete    | Delete specified configuration items |
 
-##### Example
+**Example**:
 
-```json
+```
 "update": {
     "async_chunk": true,  // Enable asynchronous chunk processing
-    "stages": {
+    "stage_args": {
         "0": {
-            "max_num_seqs": 64
+            "engine_args.custom_process_next_stage_input_func": "vllm_omni.model_executor.stage_input_processors.qwen3_omni.thinker2talker_async_chunk"
         }
     }
 },
 "delete": {
-    "stages": {
+    "stage_args": {
-        "2": ["max_num_batched_tokens"]  // Delete this deploy override for stage 2
+        "2": ["custom_process_input_func"]  // Delete this configuration for stage 2
     }
 }
 ```
 
-#### `benchmark_params` Configuration
+**`benchmark_params` Configuration**
 
 You can add any benchmark running parameters you need here. For all optional parameters, refer to the [benchmark documentation](https://github.com/vllm-project/vllm-omni/blob/main/docs/cli/bench/serve.md). General modifications are as follows:
 
-1. Change the --xxx-xx-xx running parameters to xxx_xx_xx format and fill them as keys in the JSON file.
+1.  Change the --xxx-xx-xx running parameters to xxx_xx_xx format and fill them as keys in the JSON file.
-2. For boolean variables in the running parameters, modify them to forms such as ignore_eos: true/false and fill them into the JSON file.
+2.  For boolean variables in the running parameters, modify them to forms such as ignore_eos: true/false and fill them into the JSON file.
-3. Optionally add a `baseline` object (see **Baseline thresholds** below). If you omit `baseline` or leave it empty, the performance test still runs but does not assert metric thresholds from this field.
+3.  Optionally add a `baseline` object (see **Baseline thresholds** below). If you omit `baseline` or leave it empty, the performance test still runs but does not assert metric thresholds from this field.
-4. Set `"name"` on each `benchmark_params` entry for stable pytest ids and readable result keys.
+4.  Set `"name"` on each `benchmark_params` entry for stable pytest ids and readable result keys.
-5. The qps and concurrency modes are recommended to be mutually exclusive. For detailed explanations, see the table below:
+5.  The qps and concurrency modes are recommended to be mutually exclusive. For detailed explanations, see the table below:
 
-| Parameter       | Type          | Required | Example/Values       | Description                          |
+| Parameter       | Type        | Required | Example/Values  | Description                                                                                                                                                                                                                                                          |
-| --------------- | ------------- | -------- | -------------------- | ------------------------------------ |
+| --------------- | ----------- | -------- | --------------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| name            | string        | No       | `"1024x1024_steps4"` | Stable suffix for pytest param id    |
+| name            | string      | No       | `"1024x1024_steps4"` | Stable suffix for pytest param id and logs. |
-| num_prompts     | int / array   | Yes      | 10, [10, 20, 30]     | Number of requests (scalar or sweep) |
+| num_prompts     | int / array | Yes      | 10,[10, 20, 30] | Number of requests. Supports single values or arrays. If a single value is used, it will be automatically expanded to match the number of qps or max_concurrency, e.g., [10,10,10]. If an array is used, its length must match the number of qps or max_concurrency. |
-| request_rate    | float / array | No       | 0.5, [0.5, 1, inf]   | Queries per second (QPS mode)        |
+| request_rate    | float / array | No  | 0.5, [0.5, 1, inf] | Queries per second. Supports single values or arrays. If a single value is used, it will be automatically expanded to match the number of num_prompts, e.g., [1,1,1]. If an array is used, its length must match the number of num_prompts.                          |
-| max_concurrency | int / array   | No       | 1, [1, 2, 3]         | Max in-flight requests               |
+| max_concurrency | int / array | No       | 1, [1, 2, 3]    | Maximum concurrent in-flight requests. Same array / expansion rules as `request_rate` (mutually exclusive with QPS mode).                                                                                                                                                                                                             |
-| baseline        | object        | No       | see below            | Optional hardware-nested thresholds  |
+| baseline        | object      | No       | see below       | Optional thresholds. **Hardware-nested only**: top-level keys must be `[hardware-resource]` pytest markers from `pyproject.toml` (`get_hardware_mark_list()`). Under each hardware key, metric names are free-form; values are scalars or sweep-aligned lists. Example: `{"H100": {"throughput_qps": […], "custom_stage_ms": 1.0}, "A3": {…}}`. Runtime filename aliases (`_RUNTIME_DEVICE_ALIASES`) are separate and do not expand this allowlist. |
-
-`num_prompts`, `request_rate`, and `max_concurrency` accept scalars or arrays; single values expand to match sweep length. QPS and concurrency modes are mutually exclusive. `baseline` top-level keys must be hardware pytest markers from `pyproject.toml`; metric names and values are free-form (scalars or sweep-aligned lists). Omit `baseline` to run without threshold assertions.

docs/contributing/ci/test_execution_guide.md

@@ -1,20 +1,13 @@
 # Test Execution Guide
-
 ## Setting Up the Test Environment
-
 ### Creating a Container
-
 vLLM-Omni provides an official Docker image for deployment. These images are built upon vLLM Docker images and are available on [Docker Hub](https://hub.docker.com/r/vllm/vllm-omni/tags). The version of vLLM-Omni indicates which vLLM release it is based on.
 For a local test environment, you can follow the steps below to create a container:
-
 ## Installing Dependencies
-
 ### vLLM & vLLM-Omni
-
 vLLM-Omni is built based on vLLM. You can follow [install guide](../../getting_started/installation/README.md) to build your local environment.
 
 ### Test Case Dependencies
-
 When running test cases, you may need to install the following dependencies:
 
 ```bash
@@ -23,7 +16,6 @@ apt-get install -y espeak-ng jq
 ```
 
 ## Running Tests
-
 Our test scripts use the pytest framework. First, please use `git clone https://github.com/vllm-project/vllm-omni.git` to download the vllm-omni source code. Then, in the root directory of vllm-omni, you can run the following commands in your local test environment to execute the corresponding test cases.
 
 ### CI job runners (L2–L4): logs, timing, and timeouts
@@ -33,7 +25,7 @@ Our test scripts use the pytest framework. First, please use `git clone https://
 **Log layout** (default under `logs/ready_jobs`, `logs/merge_jobs`, or `logs/nightly_jobs`; override with `--log-dir`):
 
 | Path | Description |
-| ------ | ------------- |
+|------|-------------|
 | `jobs/*.sh` | Generated per-step bash wrappers |
 | `jobs/.job_timeouts` | Job key → `timeout_in_minutes` from Buildkite YAML (when set) |
 | `<job_key>.log` | Tee output for each job |
@@ -41,7 +33,7 @@ Our test scripts use the pytest framework. First, please use `git clone https://
 
 **Timing:** while each job runs, stderr prints `finished in …` or `failed after …`. When all jobs finish, a summary is printed and written to `timing_summary.log`, for example:
 
-```text
+```
 === Job timing summary ===
   TTS_VoxCPM2_Test  25m 40s  OK
   Omni_Qwen3-Omni_Test  8m 05s  TIMED OUT
@@ -88,7 +80,7 @@ Failed jobs: 1/2
     ```bash
     cd tests
     pytest -s -v test_xxxx.py --run-level=core_model
-    pytest -s -v -m "core_model and L4 and not cards_1" --run-level=core_model
+    pytest -s -v -m "core_model and distributed_cuda and L4" --run-level=core_model
     ```
 
 === "L3 level"
@@ -119,9 +111,10 @@ Failed jobs: 1/2
 
     ```bash
     pytest -s -v test_xxxx.py --run-level=advanced_model
-    pytest -s -v -m "advanced_model and L4 and not cards_1" --run-level=advanced_model
+    pytest -s -v -m "advanced_model and distributed_cuda and L4" --run-level=advanced_model
     ```
 
+
 === "L4 level"
 
     **Recommended:** run CI-aligned nightly jobs from the repo root via [`tools/nightly/run_nightly_jobs.sh`](https://github.com/vllm-project/vllm-omni/blob/main/tools/nightly/run_nightly_jobs.sh). The script reads [`.buildkite/cuda/test-nightly.yml`](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/cuda/test-nightly.yml), generates per-step bash wrappers, runs pytest (perf jobs first, then aggregates perf JSON into Excel), and tees logs under `logs/nightly_jobs/` (see [CI job runners](#ci-job-runners-l2l4-logs-timing-and-timeouts); `generate_nightly_perf_excel` is included in the timing summary). Requires `bash`, `python3`, and PyYAML (`pip install pyyaml`).
@@ -153,12 +146,12 @@ Failed jobs: 1/2
     ```bash
     cd tests
     pytest -s -v test_xxxx.py --run-level=full_model
-    pytest -s -v -m "full_model and L4 and not cards_1" --run-level=full_model
+    pytest -s -v -m "full_model and distributed_cuda and L4" --run-level=full_model
     pytest -s -v -m "full_model and (omni or tts) and H100" --run-level=full_model
     ```
     If you only want to run specific test cases on a particular platform, you can use:
     ```bash
-    pytest -s -v -m "full_model and L4 and not cards_1"  --run-level=full_model
+    pytest -s -v -m "full_model and distributed_cuda and L4"  --run-level=full_model
     ```
     Note: ``run_benchmark.py`` and ``run_diffusion_benchmark.py`` accept an optional ``--test-config-file``. If omitted, each loads every ``*.json`` under ``tests/dfx/perf/tests/`` (omni/tts vs diffusion split by ``is_diffusion_perf_config``) and pytest ``-m`` filters by each case's JSON ``mark``:
     ```bash
@@ -167,7 +160,7 @@ Failed jobs: 1/2
     pytest -sv tests/dfx/perf/scripts/run_benchmark.py --test-config-file tests/dfx/perf/tests/test_tts.json
     pytest -sv tests/dfx/perf/scripts/run_diffusion_benchmark.py --test-config-file tests/dfx/perf/tests/test_cosmos3_vllm_omni.json
     ```
-    Nightly **Perf Test** jobs in [``test-nightly.yml``](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/cuda/test-nightly.yml) use ``--test-config-file`` only (no ``-m``). Weekly **Perf Test** in [``test-weekly.yml``](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/cuda/test-weekly.yml) runs ``test_qwen3_omni_vllm_text.json`` and ``test_qwen3_omni_multi_replicas.json`` (JSON ``mark`` includes ``slow``). E2e L4 function tests use ``full_model`` + ``--run-level full_model``. Example:
+    Nightly **Perf Test** jobs in ``test-nightly.yml`` use ``--test-config-file`` only (no ``-m``). E2e L4 function tests still use ``full_model`` + ``--run-level full_model`` (see [test-nightly.yml](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/test-nightly.yml)). Example:
 
 === "L5 level"
 
@@ -191,15 +184,14 @@ Failed jobs: 1/2
     # Reliability: HunyuanImage DiT (H100 × 4)
     pytest -s -v dfx/reliability/test_reliability_hunyuan_image.py -m slow
 
+    # Reliability: VoxCPM2 (L4)
+    pytest -s -v dfx/reliability/test_reliability_voxcpm2.py -m slow
+
     ```
 
-    The latest L5 CI jobs are in [test-weekly.yml](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/cuda/test-weekly.yml):
+    The latest L5 CI jobs (reliability + invalid-parameter weekly steps) are in [test-weekly.yml](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/cuda/test-weekly.yml).
-
-    - **Reliability** (+ invalid-parameter), **Perf Test** (Omni vLLM Text / Multi-Replica), and **Simple · CPU Coverage Test** → `WEEKLY=1` (or PR label `weekly-test` for Reliability/Perf). Scheduled upload of those env-gated steps still requires `main`.
-    - **E2E Tests** (slow Omni / TTS / Diffusion sweeps) → `NON_CRITICAL=1` (pipeline upload also accepts `WEEKLY=1` / `weekly-test`; the group `if` still needs `NON_CRITICAL`)
 
 You can find more information about markers in [Test Writing Guide — Markers for Tests](./test_writing_guide.md#markers-for-tests).
 
 ## Adding New Test Cases
-
 Please refer to the [Test System Overview](./test_system_overview.md) and [Test Writing Guide](./test_writing_guide.md).

docs/contributing/ci/test_system_overview.md

@@ -12,7 +12,8 @@ Through five levels (L1-L5) and common (Common) specifications, the system clari
       <th>Level</th>
       <th>Scope & Focus</th>
       <th>Model Coverage Strategy</th>
-      <th>Module Strategy</th>
+      <th>Feature Coverage Strategy</th>
+      <th>Interface Coverage Strategy</th>
       <th>Tags</th>
       <th>Time Cost</th>
       <th>Test Dir</th>
@@ -30,6 +31,7 @@ Through five levels (L1-L5) and common (Common) specifications, the system clari
       <td>/</td>
       <td>/</td>
       <td>/</td>
+      <td>/</td>
       <td><a href="https://github.com/vllm-project/vllm-omni/blob/main/.github/PULL_REQUEST_TEMPLATE.md">PR Checklist</a></td>
       <td>/</td>
       <td>/</td>
@@ -41,6 +43,7 @@ Through five levels (L1-L5) and common (Common) specifications, the system clari
       <td>/</td>
       <td>/</td>
       <td>/</td>
+      <td>/</td>
       <td><a href="./failures.md">CI Failures</a></td>
       <td>/</td>
       <td>/</td>
@@ -50,6 +53,7 @@ Through five levels (L1-L5) and common (Common) specifications, the system clari
       <td>Unit tests for components like entrypoints, models</td>
       <td>/</td>
       <td>/</td>
+      <td>/</td>
       <td><code>core_model and cpu</code></td>
       <td rowspan="2">&lt;15min</td>
       <td>
@@ -67,8 +71,9 @@ Through five levels (L1-L5) and common (Common) specifications, the system clari
     <tr>
       <td><strong>L2</strong><br>(E2E across models & GPU-required UT)</td>
       <td>Online (basic deployment scenarios):<br>dummy, normal inference function (output format, stream), some instance startup UT</td>
-      <td>Key models + online basic scenarios; request success, non-empty output, format match (no Whisper/accuracy)</td>
+      <td>High-priority models + online basic scenarios; request success, non-empty output, format match (no Whisper/accuracy)</td>
-      <td>Key features and modules that require launching instances (prefer random weights)</td>
+      <td>High-priority features and component tests that require launching instances</td>
+      <td>High-priority interfaces (using random lightweight models)</td>
       <td><code>core_model and hardware_test(H100, L4, etc.) and omni/tts/diffusion</code></td>
       <td>
         <strong>Model E2E:</strong><br>
@@ -90,8 +95,9 @@ Through five levels (L1-L5) and common (Common) specifications, the system clari
     <tr>
       <td><strong>L3</strong><br>(Important Perf & Integration & Accuracy)</td>
       <td>Online & Offline (multiple deployment scenarios):<br>real model, normal inference function, normal accuracy</td>
-      <td>Key models + key online/offline scenarios; real weights, Whisper/similarity, preset voice gender, basic accuracy</td>
+      <td>High/medium-priority models + key online/offline scenarios; real weights, Whisper/similarity, preset voice gender, basic accuracy</td>
-      <td>Key features and modules that require launching instances (using real weights)</td>
+      <td>Medium-priority features and component tests that require launching instances</td>
+      <td>Medium-priority interfaces (using random lightweight models)</td>
       <td><code>advanced_model and hardware_test(H100, L4, etc.) and omni/tts/diffusion</code></td>
       <td>&lt;30min</td>
       <td>
@@ -114,8 +120,9 @@ Through five levels (L1-L5) and common (Common) specifications, the system clari
     <tr>
       <td><strong>L4</strong><br>(Perf & Integration & Accuracy)</td>
       <td>Online: full functional scenarios + performance test + doc test + accuracy test</td>
-      <td>Key models: function, performance, accuracy, and doc testing</td>
+      <td>High-priority models: function, performance, accuracy, and doc testing<br>Medium-priority models: function and doc testing</td>
-      <td>Other features and modules that require launching instances (using real weights)</td>
+      <td>Low-priority features and component tests that require launching instances</td>
+      <td>Low-priority interfaces (using real weights)</td>
       <td><code>full_model and hardware_test(H100, L4, etc.) and omni/tts/diffusion</code></td>
       <td>&lt;3 hour</td>
       <td>
@@ -136,14 +143,15 @@ Through five levels (L1-L5) and common (Common) specifications, the system clari
         <a href="test_writing_guide.md#l4-level-testing-full-functionality-performance-and-documentation-testing">L4</a><br>
         L4: Purpose, Test Content, Directory Location, Example
       </td>
-      <td>Nightly / Days before Release<br>(additional GPU SKUs)</td>
+      <td>Nightly</td>
       <td>GPU</td>
     </tr>
     <tr>
-      <td><strong>L5</strong><br>(Stability & Reliability & selected Perf)</td>
+      <td><strong>L5</strong><br>(Stability & Reliability)</td>
-      <td>Online: long-term stability + reliability; coverage test</td>
+      <td>Online: long-term stability test + reliability test</td>
-      <td>Long-term stability and reliability testing for key models<br>Non-critical scenario performance and accuracy testing for key models<br>Other models: function and doc testing</td>
+      <td>Long-term stability and reliability testing for high-priority models<br>Low-priority models: function and doc testing</td>
       <td>/</td>
+      <td>Invalid-parameter validation for high-priority interfaces</td>
       <td><code>slow and hardware_test(H100, L4, etc.) and omni/tts/diffusion</code></td>
       <td> Depends on reality </td>
       <td>
@@ -170,20 +178,11 @@ For per-level test authoring (markers, examples), see [Test Writing Guide](./tes
 
 Before entering specific testing levels, the project establishes two common specifications aimed at standardizing the development process and quickly locating issues.
 
-1. ***PR Checklist ([`.github/PULL_REQUEST_TEMPLATE.md`](https://github.com/vllm-project/vllm-omni/blob/main/.github/PULL_REQUEST_TEMPLATE.md))***: This template defines the self-check items that must be completed before submitting a code review (Pull Request). It ensures that each code change meets basic requirements such as code style, dependency updates, and documentation synchronization before entering the automated testing pipeline, serving as the first manual line of defense for quality assurance.
+1.  ***PR Checklist ([`.github/PULL_REQUEST_TEMPLATE.md`](https://github.com/vllm-project/vllm-omni/blob/main/.github/PULL_REQUEST_TEMPLATE.md))***: This template defines the self-check items that must be completed before submitting a code review (Pull Request). It ensures that each code change meets basic requirements such as code style, dependency updates, and documentation synchronization before entering the automated testing pipeline, serving as the first manual line of defense for quality assurance.
-2. ***CI Failure Explanation ([CI Failures](./failures.md))***: This document archives and explains common failure patterns in the Continuous Integration (CI) pipeline, error log interpretation, and preliminary troubleshooting steps. It helps developers and testers quickly diagnose the causes of automated test failures, improving problem-solving efficiency.
+2.  ***CI Failure Explanation ([CI Failures](./failures.md))***: This document archives and explains common failure patterns in the Continuous Integration (CI) pipeline, error log interpretation, and preliminary troubleshooting steps. It helps developers and testers quickly diagnose the causes of automated test failures, improving problem-solving efficiency.
 
 ## Notes
 
-### L4 cadence (nightly and pre-release)
-
-CUDA **L4** (`full_model`) jobs are defined once in [`.buildkite/cuda/test-nightly.yml`](https://github.com/vllm-project/vllm-omni/blob/main/.buildkite/cuda/test-nightly.yml). That file is reused at two cadences:
-
-- **Nightly:** scheduled `main` builds with `NIGHTLY=1` (and PR labels such as `nightly-test` / `omni-test`). Default fleet is typically H100.
-- **Pre-release:** the same L4 suite runs again on extra GPU SKUs before a release (for example B200 via `MIRROR_HW=b200`), so `full_model` coverage is not limited to the nightly machine type.
-
-How count-form `mirror_hardwares` and `MIRROR_HW` pick a SKU: [CI Settings](./ci_settings.md).
-
 ### L2 / L3 diff-aware CI (CUDA)
 
 On CUDA **L2** and **L3**, **E2E Test** Buildkite jobs may be omitted at pipeline upload when the PR diff does not touch their path prefixes; other groups still always upload. Full two-layer mechanics and YAML examples: [CI Settings — Diff-aware CI](./ci_settings.md#diff-aware-ci) ([step filtering](./ci_settings.md#step-filtering)).

docs/contributing/ci/test_writing_guide.md

@@ -17,43 +17,47 @@ Summary:
 - **Reliability:** `tests/dfx/reliability/`
 - Do **not** add new top-level `tests/` directories unrelated to a component (or to `e2e` / `dfx` / `helpers` / `examples` / `buildkite`)
 
-Deploy YAMLs: `vllm_omni/deploy/` via `tests.helpers.stage_config.get_deploy_config_path` (see [pipeline and deploy configs](../../configuration/stage_configs.md)).
+Deploy YAMLs: `vllm_omni/deploy/` via `tests.helpers.stage_config.get_deploy_config_path` (see [stage configs](../../configuration/stage_configs.md)).
 
 ## Markers for Tests
 
 By adding markers before test functions, tests can later be executed uniformly by simply declaring the corresponding marker type.
 
 ### Current Markers
-
 Defined in `pyproject.toml`:
 
-| Marker             | Description                                                                                             |
+| Marker             | Description                                               |
-| ------------------ | ------------------------------------------------------------------------------------------------------- |
+| ------------------ | --------------------------------------------------------- |
-| `core_model`       | L1&L2 tests (run in each PR)                                                                            |
+| `core_model`       | L1&L2 tests (run in each PR)                              |
-| `advanced_model`   | L3 tests (run on each merge to main)                                                                    |
+| `advanced_model`   | L3 tests (run on each merge to main)                 |
-| `full_model`       | L4 tests (run nightly)                                                                                  |
+| `full_model`       | L4 tests (run nightly) |
-| `diffusion`        | Diffusion model tests                                                                                   |
+| `diffusion`        | Diffusion model tests                                     |
-| `omni`             | Omni multimodal model tests                                                                             |
+| `omni`             | Omni multimodal model tests |
-| `tts`              | TTS model tests                                                                                         |
+| `tts`              | TTS model tests     |
-| `cache`            | Cache backend tests                                                                                     |
+| `cache`            | Cache backend tests                                       |
-| `parallel`         | Parallelism/distributed tests                                                                           |
+| `parallel`         | Parallelism/distributed tests                             |
-| `cpu`              | Tests that run on CPU                                                                                   |
+| `cpu`              | Tests that run on CPU                                     |
-| `gpu`              | Tests that run on GPU *                                                                                 |
+| `gpu`              | Tests that run on GPU *                                   |
-| `cuda`             | Tests that run on CUDA *                                                                                |
+| `cuda`             | Tests that run on CUDA *                                  |
-| `rocm`             | Tests that run on AMD/ROCm *                                                                            |
+| `rocm`             | Tests that run on AMD/ROCm *                              |
-| `xpu`              | Tests that run on Intel XPU *                                                                           |
+| `xpu`              | Tests that run on Intel XPU *                             |
-| `npu`              | Tests that run on NPU/Ascend *                                                                          |
+| `npu`              | Tests that run on NPU/Ascend *                            |
-| `H100`             | Tests that require H100 GPU  *                                                                          |
+| `H100`             | Tests that require H100 GPU  *                            |
-| `L4`               | Tests that require L4 GPU *                                                                             |
+| `L4`               | Tests that require L4 GPU *                               |
-| `MI325`            | Tests that require MI325 GPU (AMD/ROCm) *                                                               |
+| `MI325`            | Tests that require MI325 GPU (AMD/ROCm) *                 |
-| `A2`               | Tests that require A2 NPU *                                                                             |
+| `A2`               | Tests that require A2 NPU *                               |
-| `A3`               | Tests that require A3 NPU *                                                                             |
+| `A3`               | Tests that require A3 NPU *                               |
-| `cards_{n}`        | Tests that require *n* accelerator cards (`cards_1` … `cards_8`); auto-added from `num_cards` *         |
+| `distributed_cuda` | Tests that require multi cards on CUDA platform *         |
-| `slow`             | Slow tests (may skip in quick CI)                                                                       |
+| `distributed_rocm` | Tests that require multi cards on ROCm platform  *        |
+| `distributed_npu`  | Tests that require multi cards on NPU platform  *         |
+| `skipif_cuda`      | Skip if the num of CUDA cards is less than the required * |
+| `skipif_rocm`      | Skip if the num of ROCm cards is less than the required * |
+| `skipif_npu`       | Skip if the num of NPU cards is less than the required *  |
+| `slow`             | Slow tests (may skip in quick CI)                         |
 | `benchmark`        | Benchmark tests (decorator on runner test functions; perf JSON uses `full_model` + type marker instead) |
-| `local_model`      | Tests requiring local / non-HF-hub model weights                                                        |
+| `local_model`      | Tests requiring local / non-HF-hub model weights          |
 
-\* Means those markers are auto-added by `@hardware_test` (parametrization decorator) or `hardware_marks` (only returning the list of marks for flexibility). Do **not** write `pytest.mark.H100` / `pytest.mark.L4` (or other SKU names) by hand: `check-mark` rejects that so `cards_{n}` cannot be skipped.
+\* Means those markers are auto-added by `@hardware_test` (parametrization decorator) or `hardware_marks` (only returning the list of marks for flexibility).
 
 #### Example usage for markers
 
@@ -76,26 +80,28 @@ def test_video_to_audio()
 This decorator is intended to make hardware-aware, cross-platform test authoring easier and more robust for CI/CD environments. The `hardware_test` decorator in `vllm-omni/tests/helpers/mark.py` performs the following actions:
 
 1. **Applies platform and resource markers**  
-   Adds the appropriate pytest markers for each specified hardware platform (e.g., `cuda`, `rocm`, `xpu`, `npu`) and resource type (e.g., `L4`, `H100`, `H200`, `B200`, `MI325`, `B60`, `A2`, `A3`).
+   Adds the appropriate pytest markers for each specified hardware platform (e.g., `cuda`, `rocm`, `xpu`, `npu`) and resource type (e.g., `L4`, `H100`, `MI325`, `B60`, `A2`, `A3`).
-   ```python
+   ```
    @pytest.mark.cuda
    @pytest.mark.L4
    ```
-2. **Handles card-count filtering**  
+2. **Handles multi-card (distributed) scenarios**  
-   Adds a filterable marker `cards_{n}` (including `cards_1` when `num_cards` is omitted / 1). Use this with the SKU and platform, e.g. `-m "H100 and cards_4 and cuda"` vs `-m "H100 and cards_2 and cuda"`. Any multi-card CUDA test is `-m "cuda and not cards_1"`.
+   For tests requiring multiple cards, it automatically adds distributed markers such as `distributed_cuda`, `distributed_rocm`, or `distributed_npu`.
-
+   ```
-   When `res` lists more than one platform **and** `num_cards` differs by platform, `@hardware_test` collects **one variant per platform** so those `cards_{n}` marks do not stack. `-m "H100 and cards_4 and cuda"` then selects only the CUDA/H100 4-card variant, not a ROCm 2-card sibling. The same count on every platform stays a single item.
+   @pytest.mark.distributed_cuda(num_cards=num_cards)
+   ```
 3. **Supports flexible card requirements**  
    Accepts `num_cards` as either a single integer for all platforms or as a dictionary with per-platform values. If not specified, defaults to 1 card per platform.
 
 4. **Integrates resource validation**  
-   On CUDA and XPU, when `num_cards > 1`, attaches pytest's built-in `skipif` if the machine has fewer devices than requested. This is not a named `skipif_*` marker and is not used for `-m` filtering.
+   On CUDA, adds a skip marker (`skipif_cuda`) if the system does not have the required number of devices.
+   Support for `skipif_rocm` and `skipif_npu` will be implemented later.
+
 
 5. **Works with pytest filtering**  
-   Allows tests to be filtered and selected at runtime using standard pytest marker expressions (e.g., `-m "H100 and cards_2"`, `-m "L4 and not cards_1"`).
+   Allows tests to be filtered and selected at runtime using standard pytest marker expressions (e.g., `-m "distributed_cuda and L4"`).
 
 ##### Example usage for decorator
-
 - Single call for multiple platforms:
     ```python
     @hardware_test(
@@ -110,14 +116,13 @@ This decorator is intended to make hardware-aware, cross-platform test authoring
         num_cards=2,
     )
     ```
-- `res` must be a dict. SKU values come from `pyproject.toml` markers tagged `[hardware-resource]` plus the platform key (`[cuda]`, `[rocm]`, `[xpu]`, `[npu]`, `[musa]`).
+- `res` must be a dict; supported resources: CUDA (L4/H100), ROCm (MI325), XPU (B60), MUSA (S5000), NPU (A2/A3)
 - `num_cards` can be int (all platforms) or dict (per platform); defaults to 1 when missing
+- Distributed markers (`distributed_cuda`, `distributed_rocm`, `distributed_npu`) are auto-added for multi-card cases
 - Filtering examples:
-    - CUDA H100 4-card: `pytest -m "H100 and cards_4 and cuda"`
+    - CUDA only: `pytest -m "distributed_cuda and L4"`
-    - CUDA H100 2-card: `pytest -m "H100 and cards_2 and cuda"`
+    - ROCm only: `pytest -m "distributed_rocm and MI325"`
-    - CUDA multi-card (any count): `pytest -m "L4 and not cards_1"`
+    - NPU only: `pytest -m "distributed_npu"`
-    - ROCm multi-card: `pytest -m "MI325 and not cards_1"`
-    - NPU multi-card: `pytest -m "npu and not cards_1"`
 
 #### Function: `hardware_marks`
 
@@ -173,30 +178,37 @@ If you want to add support for a new platform (e.g., "tpu" for a new accelerator
    [tool.pytest.ini_options]
    markers = [
        # ... existing markers ...
-       "tpu: [hardware-platform] Tests that require TPU device",
+       "tpu: Tests that require TPU device",
-       # Add [gpu] next to [hardware-platform] if this platform should also get pytest.mark.gpu.
+       "TPU_V3: Tests that require TPU v3 hardware",
-       "TPU_V3: [hardware-resource] [tpu] Tests that require TPU v3 hardware",
+       "distributed_tpu: Tests that require multiple TPU devices",
-       "cards_1: [hardware-cards] Tests that require 1 accelerator card",
-       "cards_2: [hardware-cards] Tests that require 2 accelerator cards",
    ]
    ```
-   `check-mark` treats `[hardware-platform]` names (including `cpu`/`gpu`) as allowed `pytest.mark.<name>` values. `hardware_test(res=...)` still rejects `cpu`/`gpu` as `res` keys.
 2. **Implement a marker construction function for your platform** in `vllm-omni/tests/helpers/mark.py`:
    ```python
    # In vllm-omni/tests/helpers/mark.py
 
-   def _tpu_marks(*, res: str, num_cards: int):
+   def tpu_marks(*, res: str, num_cards: int):
-       _require_sku("tpu", res)
+       test_platform = pytest.mark.tpu
-       return [pytest.mark.tpu, getattr(pytest.mark, res), _cards_mark(num_cards)]
+       if res == "TPU_V3":
+           test_resource = pytest.mark.TPU_V3
+       else:
+           raise ValueError(
+               f"Invalid TPU resource type: {res}. Supported: TPU_V3")
+
+       if num_cards == 1:
+           return [test_platform, test_resource]
+       else:
+           test_distributed = pytest.mark.distributed_tpu(num_cards=num_cards)
+           # Optionally: add skipif_tpu when implemented
+           return [test_platform, test_resource, test_distributed]
    ```
-3. **Register the builder** in `_marks_for_platform`:
+3. **Update `hardware_marks` to recognize your new platform**:
+    In the relevant place (see the `hardware_marks` implementation), add:
     ```python
-    builders = {
+    if platform == "tpu":
-        # ... existing platforms ...
+        marks = tpu_marks(res=resource, num_cards=cards)
-        "tpu": _tpu_marks,
-    }
     ```
-    (`hardware_test` / `hardware_marks` both go through `_marks_for_platform`.)
+    (`hardware_test` calls `hardware_marks` internally, so both will pick up the change.)
 4. **(Recommended) Add a test using your new markers**:
    ```python
    @hardware_test(
@@ -208,13 +220,12 @@ If you want to add support for a new platform (e.g., "tpu" for a new accelerator
    ```
 
 **Summary**:  
-
 - Add pytest markers for your new platform/resources  
-- Implement a private marker function (`_xxx_marks`)
+- Implement a marker function (`xxx_marks`)  
-- Register it in `_marks_for_platform`
+- Plug into `hardware_marks`  
-- You're done: tests using `@hardware_test` or `hardware_marks` with your platform now automatically get the correct markers and `cards_{n}` filtering.
+- You're done: tests using `@hardware_test` or `hardware_marks` with your platform now automatically get the correct markers, distribution, and isolation!
 
-See code in `vllm-omni/tests/helpers/mark.py` for existing examples (`_cuda_marks`, `_rocm_marks`, `_npu_marks`).
+See code in `vllm-omni/tests/helpers/mark.py` for existing examples (`cuda_marks`, `rocm_marks`, `npu_marks`).
 
 ## Test case style
 
@@ -226,46 +237,47 @@ L1 and L2 level testing form the foundation of the quality assurance system. L1
 
 L2 level testing builds upon L1 by introducing GPU resources and verifying that the end-to-end (E2E) process of the model in basic deployment scenarios is smooth. For example, it uses dummy models to confirm that core interfaces like the inference pipeline, output format, and streaming response work properly. The common goal of these two levels is to provide developers with rapid feedback, discovering and fixing issues early in the development cycle.
 
+
+
 #### 1.2 Testing Content and Scope
 
-- ***L1 (Unit & Logic Testing)***:
+-   ***L1 (Unit & Logic Testing)***:
-    - ***Scope***: Tests internal functions and methods of core components such as `entrypoints`, `models`.
+-   -   ***Scope***: Tests internal functions and methods of core components such as `entrypoints`, `models`.
-        - ***Focus***: Branch coverage, exception handling, algorithm logic correctness. Does not involve external dependencies or the complete service stack.
+    -   ***Focus***: Branch coverage, exception handling, algorithm logic correctness. Does not involve external dependencies or the complete service stack.
-        - ***Time Cost***: Execution time is controlled within ***15 minutes*** to ensure fast feedback.
+    -   ***Time Cost***: Execution time is controlled within ***15 minutes*** to ensure fast feedback.
-- ***L2 (Basic End-to-End Testing)***:
+-   ***L2 (Basic End-to-End Testing)***:
-    - ***Scope***: Covers two basic deployment scenarios: `online` (serving) and `offline` (inference).
+-   -   ***Scope***: Covers two basic deployment scenarios: `online` (serving) and `offline` (inference).
-        - ***Focus***: Uses `dummy` weights (via deploy YAML patching at `core_model`) or lightweight real models to verify that the entire chain from request input to result output works normally, including output data structure, streaming (stream) support, and **cheap payload checks** at `--run-level core_model` (see below). Also includes some unit tests that require launching independent service instances.
+    -   ***Focus***: Uses `dummy` weights (via deploy YAML patching at `core_model`) or lightweight real models to verify that the entire chain from request input to result output works normally, including output data structure, streaming (stream) support, and **cheap payload checks** at `--run-level core_model` (see below). Also includes some unit tests that require launching independent service instances.
-        - ***L2 response validation (`core_model`)***: Implemented in `tests/helpers/assertions.py` and invoked by `OpenAIClientHandler` based on `--run-level`. At L2 we require **request success** plus minimal output sanity—not full accuracy:
+    -   ***L2 response validation (`core_model`)***: Implemented in `tests/helpers/assertions.py` and invoked by `OpenAIClientHandler` based on `--run-level`. At L2 we require **request success** plus minimal output sanity—not full accuracy:
-            - **Speech / TTS** (`assert_audio_speech_response`): decoded audio must be present and non-empty (or exceed `min_audio_bytes` when set in `request_config`); `response_format` must match the returned content-type (e.g. `wav`, `pcm`). Whisper transcript similarity, PCM HNR, and preset-voice gender checks run only at L3+.
+        -   **Speech / TTS** (`assert_audio_speech_response`): decoded audio must be present and non-empty (or exceed `min_audio_bytes` when set in `request_config`); `response_format` must match the returned content-type (e.g. `wav`, `pcm`). Whisper transcript similarity, PCM HNR, and preset-voice gender checks run only at L3+.
-            - **Diffusion** (`assert_diffusion_response`): at least one non-empty image, video, or audio artifact. Resolution/frame-count parity and other deep checks run only at L3+.
+        -   **Diffusion** (`assert_diffusion_response`): at least one non-empty image, video, or audio artifact. Resolution/frame-count parity and other deep checks run only at L3+.
-            - **Omni multimodal** (`assert_omni_response`): L2 asserts successful completion; keyword, transcript, and cross-modal similarity checks run only at L3+.
+        -   **Omni multimodal** (`assert_omni_response`): L2 asserts successful completion; keyword, transcript, and cross-modal similarity checks run only at L3+.
-        - ***Characteristic***: Requires ***GPU*** resources to perform model computations.
+    -   ***Characteristic***: Requires ***GPU*** resources to perform model computations.
 
 #### 1.3 Test Directory and Execution Files
 
 A clear directory structure is key to managing test cases efficiently. See the **Test Dir** column in [Test System Overview](./test_system_overview.md) for the canonical placement rules.
 
-- ***L1 (component / unit)***: `tests/{component_name}/…` mirroring `vllm_omni/{component_name}/` (e.g. `tests/diffusion/`, `tests/entrypoints/`). Do **not** invent new top-level `tests/` folders unrelated to a component.
+-   ***L1 (component / unit)***: `tests/{component_name}/…` mirroring `vllm_omni/{component_name}/` (e.g. `tests/diffusion/`, `tests/entrypoints/`). Do **not** invent new top-level `tests/` folders unrelated to a component.
-- ***L2 (basic model E2E)***:
+-   ***L2 (basic model E2E)***:
-    - Online serving: `tests/e2e/online_serving/test_{model_name}.py`
+    -   Online serving: `tests/e2e/online_serving/test_{model_name}.py`
-    - Offline inference: `tests/e2e/offline_inference/test_{model_name}.py`
+    -   Offline inference: `tests/e2e/offline_inference/test_{model_name}.py`
-- ***Feature-level integration*** (cross-cutting, not a single-model smoke): `tests/e2e/features/<feature>/` (e.g. `fullduplex/`, `custom_pipeline/`, `rlhf_test/`).
+-   ***Feature-level integration*** (cross-cutting, not a single-model smoke): `tests/e2e/features/<feature>/` (e.g. `fullduplex/`, `custom_pipeline/`, `rlhf_test/`).
 
 #### 1.4 Execution Method and Example
 
-- ***Trigger Timing***: **`PR with ready label`**. That is, when a developer adds a "ready for review" or similar label to a PR on platforms like GitHub, L1 and L2 tests are automatically triggered.
+-   ***Trigger Timing***: **`PR with ready label`**. That is, when a developer adds a "ready for review" or similar label to a PR on platforms like GitHub, L1 and L2 tests are automatically triggered.
-- ***Diff-aware step skipping***: On L2, **E2E Test** jobs may be omitted at pipeline upload when the PR diff does not touch their [`source_file_dependencies`](ci_settings.md#step-filtering) prefixes; non-E2E groups still always upload. See [CI Settings — Diff-aware CI](ci_settings.md#diff-aware-ci).
+-   ***Diff-aware step skipping***: On L2, **E2E Test** jobs may be omitted at pipeline upload when the PR diff does not touch their [`source_file_dependencies`](ci_settings.md#step-filtering) prefixes; non-E2E groups still always upload. See [CI Settings — Diff-aware CI](ci_settings.md#diff-aware-ci).
-- ***Execution Environment***: L1 uses ***CPU*** environment; L2 requires ***GPU*** environment.
+-   ***Execution Environment***: L1 uses ***CPU*** environment; L2 requires ***GPU*** environment.
-- ***Script Example***:
+-   ***Script Example***:
 
 <details>
 <summary> L1 Test Examples</summary>
 
 Examples from `tests/model_executor/models/qwen2_5_omni/test_audio_length.py`
-
 ```python
 # SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
+# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
 
 import pytest
 
@@ -319,7 +331,6 @@ def test_cap_and_align_mel_length_no_mismatch(repeats, code_len, max_mel_frames)
     if max_mel_frames is not None and int(max_mel_frames) > 0 and int(max_mel_frames) >= repeats:
         assert target_mel_len <= int(max_mel_frames)
 ```
-
 </details>
 
 <details>
@@ -327,7 +338,7 @@ def test_cap_and_align_mel_length_no_mismatch(repeats, code_len, max_mel_frames)
 You can refer to Test Examples in the L3 section to see example test cases that incorporate both L2 and L3 testing logic.
 </details>
 
-- ***Run Command***:
+-   -   ***Run Command***:
 
     `pytest -s -v /tests/e2e/online_serving/test_{model_name}.py`
     `pytest -s -v -m 'core_model and cpu' --run-level=core_model`
@@ -340,34 +351,36 @@ L3 level testing executes after code is merged into the main branch. Its core pu
 
 . It acts as the "quality gatekeeper" for the main branch, ensuring that no merge breaks the core capabilities of the model service. Testing needs to provide clear conclusions within a relatively short time (<30min), balancing test depth with feedback speed.
 
+
+
 #### 2.2 Testing Content and Scope
 
-- ***Deployment Scenarios***: Covers richer `online` and `offline` deployment configurations, which may include different hardware configurations, batch sizes, concurrency levels, etc.
+-   ***Deployment Scenarios***: Covers richer `online` and `offline` deployment configurations, which may include different hardware configurations, batch sizes, concurrency levels, etc.
-- ***Core Verification***:
+-   ***Core Verification***:
-    1. ***Inference Functionality***: Ensures real models can perform forward computation normally and return results.
+-   1.  ***Inference Functionality***: Ensures real models can perform forward computation normally and return results.
-    2. ***Accuracy Compliance***: Verifies that the model's evaluation metrics (e.g., accuracy) meet the expected baseline, preventing code changes from introducing accuracy issues.
+    2.  ***Accuracy Compliance***: Verifies that the model's evaluation metrics (e.g., accuracy) meet the expected baseline, preventing code changes from introducing accuracy issues.
-    3. ***Important Performance***: Verifies whether performance (e.g., P99 latency, throughput) in core scenarios meets preset thresholds.
+    3.  ***Important Performance***: Verifies whether performance (e.g., P99 latency, throughput) in core scenarios meets preset thresholds.
-- ***L3 response validation (`advanced_model`)***: At `--run-level advanced_model`, `tests/helpers/assertions.py` adds semantic checks on top of L2 payload gates (see L1 & L2 above):
+-   ***L3 response validation (`advanced_model`)***: At `--run-level advanced_model`, `tests/helpers/assertions.py` adds semantic checks on top of L2 payload gates (see L1 & L2 above):
-    - **Speech / TTS** (`assert_audio_speech_response`): Whisper transcript of returned audio vs `request_config["input"]` (cosine similarity &gt; 0.9 when input text is set); optional `min_audio_bytes` floor; PCM harmonic-to-noise ratio when `response_format` is `pcm`; **preset voice gender** via `_assert_preset_voice_gender_from_audio` when `voice` matches a known preset in `_PRESET_VOICE_GENDER_MAP` (pitch/F0-based classifier; skipped for unknown voices or `pcm` output).
+    -   **Speech / TTS** (`assert_audio_speech_response`): Whisper transcript of returned audio vs `request_config["input"]` (cosine similarity &gt; 0.9 when input text is set); optional `min_audio_bytes` floor; PCM harmonic-to-noise ratio when `response_format` is `pcm`; **preset voice gender** via `_assert_preset_voice_gender_from_audio` when `voice` matches a known preset in `_PRESET_VOICE_GENDER_MAP` (pitch/F0-based classifier; skipped for unknown voices or `pcm` output).
-    - **Omni multimodal** (`assert_omni_response`): non-empty text/audio outputs; `key_words` in transcript or text; text–audio similarity / containment; preset **`speaker`** gender check (same helper as TTS).
+    -   **Omni multimodal** (`assert_omni_response`): non-empty text/audio outputs; `key_words` in transcript or text; text–audio similarity / containment; preset **`speaker`** gender check (same helper as TTS).
-    - **Diffusion** (`assert_*_diffusion_response`): image/video dimension and frame-count parity with request parameters where configured.
+    -   **Diffusion** (`assert_*_diffusion_response`): image/video dimension and frame-count parity with request parameters where configured.
-    - **Accuracy suites**: pixel/video similarity and metric baselines under `/tests/e2e/accuracy/` (separate from inline helper assertions).
+    -   **Accuracy suites**: pixel/video similarity and metric baselines under `/tests/e2e/accuracy/` (separate from inline helper assertions).
 
 #### 2.3 Test Directory and Execution Files
 
-- ***Functional / model E2E***:
+-   ***Functional / model E2E***:
-    - Online serving: `tests/e2e/online_serving/test_{model_name}_expansion.py`
+    -   Online serving: `tests/e2e/online_serving/test_{model_name}_expansion.py`
-    - Offline inference: `tests/e2e/offline_inference/test_{model_name}_expansion.py`
+    -   Offline inference: `tests/e2e/offline_inference/test_{model_name}_expansion.py`
-    - Accuracy / similarity: `tests/e2e/accuracy/`
+    -   Accuracy / similarity: `tests/e2e/accuracy/`
-    - (`_expansion.py` holds broader scenario coverage than the L2 smoke file.)
+    -   (`_expansion.py` holds broader scenario coverage than the L2 smoke file.)
-- ***Feature-level integration***: `tests/e2e/features/<feature>/` (not under a new top-level `tests/` directory).
+-   ***Feature-level integration***: `tests/e2e/features/<feature>/` (not under a new top-level `tests/` directory).
 
 #### 2.4 Execution Method and Example
 
-- ***Trigger Timing***: **`PR Merged`**. Automatically triggered after code review is approved and merged into the main branch (typically via `merge-test` label on the PR before merge).
+-   ***Trigger Timing***: **`PR Merged`**. Automatically triggered after code review is approved and merged into the main branch (typically via `merge-test` label on the PR before merge).
-- ***Diff-aware step skipping***: Same [`source_file_dependencies`](ci_settings.md#step-filtering) mechanism as L2—**all E2E Test** leaf steps are diff-gated; other L3 groups always upload. See [CI Settings — Diff-aware CI](ci_settings.md#diff-aware-ci).
+-   ***Diff-aware step skipping***: Same [`source_file_dependencies`](ci_settings.md#step-filtering) mechanism as L2—**all E2E Test** leaf steps are diff-gated; other L3 groups always upload. See [CI Settings — Diff-aware CI](ci_settings.md#diff-aware-ci).
-- ***Execution Environment***: ***GPU*** servers.
+-   ***Execution Environment***: ***GPU*** servers.
-- ***Script Example***:
+-   ***Script Example***:
 
 ???+ example "Test Examples"
 
@@ -483,9 +496,9 @@ L3 level testing executes after code is merged into the main branch. Its core pu
 
     **Audio output debugging**: Deep validation may transcribe returned audio via `convert_audio_bytes_to_text` (Whisper). If an audio keyword or text–audio similarity assertion fails, set `VLLM_OMNI_KEEP_REQUEST_MEDIA=1` before running pytest to keep the intermediate WAV files for inspection (see [Test helper environment variables](test_system_overview.md#test-helper-environment-variables)).
 
-- ***Run Command (L3 merge)***: `pytest -s -v /tests/e2e/online_serving/test_{model_name}.py -m advanced_model --run-level=advanced_model`
+-   ***Run Command (L3 merge)***: `pytest -s -v /tests/e2e/online_serving/test_{model_name}.py -m advanced_model --run-level=advanced_model`
 
-- ***Run Command (L4 nightly expansion)***: `pytest -s -v /tests/e2e/online_serving/test_{model_name}_expansion.py -m full_model --run-level=full_model`
+-   ***Run Command (L4 nightly expansion)***: `pytest -s -v /tests/e2e/online_serving/test_{model_name}_expansion.py -m full_model --run-level=full_model`
 
 ### L4 Level Testing - Full Functionality, Performance, and Documentation Testing
 
@@ -493,25 +506,27 @@ L3 level testing executes after code is merged into the main branch. Its core pu
 
 L4 level testing is a comprehensive quality audit before a version release. It expands upon L3, executing ***full*** functional scenarios, conducting systematic ***performance stress tests***, and simultaneously verifying the correctness of accompanying ***example documentation***. Its purpose is to perform deep validation of the system during off-peak nighttime hours, providing quality trend reports for daytime development and data support for release decisions.
 
+
+
 #### 3.2 Testing Content and Scope
 
-- ***Full Functionality Testing***: Executes all test cases defined in `test_{model_name}_expansion.py`, covering all implemented features, positive flows, boundary conditions, and exception handling.
+-   ***Full Functionality Testing***: Executes all test cases defined in `test_{model_name}_expansion.py`, covering all implemented features, positive flows, boundary conditions, and exception handling.
-- ***Performance Testing***: JSON configs under `tests/dfx/perf/tests/` drive Omni, TTS, and diffusion serving benchmarks via `run_benchmark.py` or `run_diffusion_benchmark.py`, measuring throughput, latency, and related metrics under configured workloads.
+-   ***Performance Testing***: Uses `tests/dfx/perf/tests/test_qwen3_omni_*.json` (Omni), `test_tts.json` / `test_voxcpm2.json` / `test_higgs_audio_v3.json` (TTS), and diffusion configs `tests/dfx/perf/tests/test_*_vllm_omni.json` (passed to `run_benchmark.py` or `run_diffusion_benchmark.py` via `--test-config-file` in nightly **Perf Test** steps) to drive throughput, latency, and memory benchmarks. Each JSON **case** may declare an optional top-level **`mark`** array: exactly one ``hardware_marks`` object plus pytest marker name strings (`full_model`, `omni` / `tts` / `diffusion`, …). Runners attach those marks to each parametrized `(server, benchmark index)` pair so **local** bulk runs can filter with `-m` (for example `-m "full_model and H100 and diffusion"`). Nightly CI perf jobs select workloads by **`--test-config-file`**, not `-m`. Details are in the Performance Tests example below.
-- ***Documentation Testing***: Verifies whether the example code provided to users is runnable and its results match the description.
+-   ***Documentation Testing***: Verifies whether the example code provided to users is runnable and its results match the description.
 
 #### 3.3 Test Directory and Execution Files
 
-- ***Functional Testing***: Same directories as L3.
+-   ***Functional Testing***: Same directories as L3.
-- ***Performance Test Configuration***: `tests/dfx/perf/tests/test_qwen3_omni_*.json`, `test_tts.json`, `test_voxcpm2.json`, `test_higgs_audio_v3.json`, and diffusion configs `tests/dfx/perf/tests/test_*_vllm_omni.json` (e.g. `test_qwen_image_vllm_omni.json`, `test_cosmos3_vllm_omni.json`). Optional per-case `mark` for local `-m` filtering is documented in Section 3.4 Performance Tests.
+-   ***Performance Test Configuration***: `tests/dfx/perf/tests/test_qwen3_omni_*.json`, `test_tts.json`, `test_voxcpm2.json`, `test_higgs_audio_v3.json`, and diffusion configs `tests/dfx/perf/tests/test_*_vllm_omni.json` (e.g. `test_qwen_image_vllm_omni.json`, `test_cosmos3_vllm_omni.json`). Optional per-case `mark` for local `-m` filtering is documented in Section 3.4 Performance Tests.
-- ***Documentation Example Tests***:
+-   ***Documentation Example Tests***:
-    - `tests/examples/online_serving/test_{model_name}.py`
+    -   `tests/examples/online_serving/test_{model_name}.py`
-    - `tests/examples/offline_inference/test_{model_name}.py`
+    -   `tests/examples/offline_inference/test_{model_name}.py`
 
 #### 3.4 Execution Method and Example
 
-- ***Trigger Timing***: **`Nightly`**, automatically executed every night.
+-   ***Trigger Timing***: **`Nightly`**, automatically executed every night.
-- ***Execution Environment***: ***GPU*** server clusters to meet the resource demands of performance testing.
+-   ***Execution Environment***: ***GPU*** server clusters to meet the resource demands of performance testing.
-- ***Script Example***:
+-   ***Script Example***:
 
 ??? example "Test Examples: Documentation Example Tests"
 
@@ -525,7 +540,7 @@ L4 level testing is a comprehensive quality audit before a version release. It e
 
     --8<-- "docs/contributing/ci/test_examples/l4_functionality_tests.inc.md"
 
-- ***Run Command***: (Specific commands would depend on the performance testing tool and configuration defined in `nightly.json`).
+-   ***Run Command***: (Specific commands would depend on the performance testing tool and configuration defined in `nightly.json`).
 
 ### L5 Level Testing - Stability and Reliability Testing
 
@@ -533,43 +548,41 @@ L4 level testing is a comprehensive quality audit before a version release. It e
 
 L5 level testing focuses on the performance of model services under ***long-running*** and ***abnormal fault*** scenarios. It aims to uncover deep-seated issues that only manifest under sustained pressure or extreme conditions, such as memory leaks, resource contention, gradual performance degradation, and lack of fault tolerance mechanisms. This is the final, yet crucial, line of defense for ensuring service high availability and production environment robustness.
 
+
+
 #### 4.2 Testing Content and Scope
 
-- ***Long-term Stability (Stability) Testing***: Uses JSON under `tests/dfx/stability/tests/` (for example `test_qwen3_omni.json` and `test_wan22.json`) to run the service under moderate load for an extended period (e.g., over 12 hours), monitoring whether metrics like memory/VRAM usage, response time, and throughput degrade over time, and whether the service process remains stable.
+-   ***Long-term Stability (Stability) Testing***: Uses JSON under `tests/dfx/stability/tests/` (for example `test_qwen3_omni.json` and `test_wan22.json`) to run the service under moderate load for an extended period (e.g., over 12 hours), monitoring whether metrics like memory/VRAM usage, response time, and throughput degrade over time, and whether the service process remains stable.
-- ***Reliability Testing***: Uses pytest suites under `tests/dfx/reliability/` to inject controlled faults against a **live** `vllm_omni serve` instance (same **`omni_server` / `omni_server_function`** fixture style as E2E). Current suites emphasize **GPU memory pressure** (CUDA sidecar “memory hog”), **worker / runtime process kill** (`SIGKILL` on `VLLM::Worker` for Qwen3-Omni, `multiprocessing.spawn` for Wan2.2 video workers, or `vLLM-Omni::` for HunyuanImage DiT workers), **large multimodal chat**, **`/v1/videos`**, **`/v1/images/generations`**, or **`/v1/audio/speech`** jobs under OOM, **`/health` → 503** and **fast-fail / non-hanging concurrent** requests after kill, and **OpenAI-style 5xx error contracts** (e.g. text vs text+audio under OOM). **Post-fault recovery** checks exist where enabled (some cases may be `skip` while issues are tracked). See Section 4.3 for file-level responsibilities and Section 4.4 Test Examples (Reliability) for CI markers (`slow`, `hardware_test`, POSIX-only kill).
+-   ***Reliability Testing***: Uses pytest suites under `tests/dfx/reliability/` to inject controlled faults against a **live** `vllm_omni serve` instance (same **`omni_server` / `omni_server_function`** fixture style as E2E). Current suites emphasize **GPU memory pressure** (CUDA sidecar “memory hog”), **worker / runtime process kill** (`SIGKILL` on `VLLM::Worker` for Qwen3-Omni, `multiprocessing.spawn` for Wan2.2 video workers, or `vLLM-Omni::` for HunyuanImage DiT workers), **large multimodal chat**, **`/v1/videos`**, **`/v1/images/generations`**, or **`/v1/audio/speech`** jobs under OOM, **`/health` → 503** and **fast-fail / non-hanging concurrent** requests after kill, and **OpenAI-style 5xx error contracts** (e.g. text vs text+audio under OOM). **Post-fault recovery** checks exist where enabled (some cases may be `skip` while issues are tracked). See the Reliability `<details>` block in Section 4.4 for file-level responsibilities and CI markers (`slow`, `hardware_test`, POSIX-only kill).
 
 #### 4.3 Test Directory and Execution Files
 
-- ***Stability Test Configuration***: `tests/dfx/stability/tests/test_qwen3_omni.json`, `tests/dfx/stability/tests/test_wan22.json` (one JSON per model / runner family)
+-   ***Stability Test Configuration***: `tests/dfx/stability/tests/test_qwen3_omni.json`, `tests/dfx/stability/tests/test_wan22.json` (one JSON per model / runner family)
-- ***Reliability Test Suite*** (`tests/dfx/reliability/`):
+-   ***Reliability Test Suite*** (`tests/dfx/reliability/`):
-    - `test_reliability_qwen3_omni.py` — Qwen3-Omni chat / multimodal reliability (GPU OOM, process kill, recovery, error contract under `--async-chunk` vs default).
+    -   `test_reliability_qwen3_omni.py` — Qwen3-Omni chat / multimodal reliability (GPU OOM, process kill, recovery, error contract under `--async-chunk` vs default).
-    - `test_reliability_wan22.py` — Wan2.2 I2V video API reliability (`/v1/videos` under OOM and process kill, recovery).
+    -   `test_reliability_wan22.py` — Wan2.2 I2V video API reliability (`/v1/videos` under OOM and process kill, recovery).
-    - `test_reliability_hunyuan_image.py` — HunyuanImage-3.0-Instruct DiT-only reliability (`/v1/images/generations` under OOM and process kill; deploy `hunyuan_image3_dit.yaml`, H100 × 4).
+    -   `test_reliability_hunyuan_image.py` — HunyuanImage-3.0-Instruct DiT-only reliability (`/v1/images/generations` under OOM and process kill; deploy `hunyuan_image3_dit.yaml`, H100 × 4).
-    - `helpers.py` — Shared primitives used by current suites: raw HTTP probes for `/v1/chat/completions` and `/health`, OpenAI-style error parsing, GPU OOM sidecar (`inject_gpu_oom` / `stop_gpu_oom_hogs`), and `pgrep`-based process-kill injector construction (`make_process_kill_fault_injector`).
+    -   `test_reliability_voxcpm2.py` — VoxCPM2 TTS reliability (`/v1/audio/speech` under OOM and process kill; deploy `voxcpm2.yaml`, L4 × 1).
-    - `conftest.py` — `fault_injector` and `omni_server_after_fault` / `omni_server_after_fault_function` fixtures to run a callable **after** the server is ready.
+    -   `helpers.py` — Shared primitives used by current suites: raw HTTP probes for `/v1/chat/completions` and `/health`, OpenAI-style error parsing, GPU OOM sidecar (`inject_gpu_oom` / `stop_gpu_oom_hogs`), and `pgrep`-based process-kill injector construction (`make_process_kill_fault_injector`).
-    - `README.md` — Short local run commands for this directory.
+    -   `conftest.py` — `fault_injector` and `omni_server_after_fault` / `omni_server_after_fault_function` fixtures to run a callable **after** the server is ready.
+    -   `README.md` — Short local run commands for this directory.
 
 #### 4.4 Execution Method and Example
 
-- ***Trigger Timing***: **`Weekly`** (weekly) or **`Days before Release`** (several days before a major release). Due to long execution times, the frequency is lower.
+-   ***Trigger Timing***: **`Weekly`** (weekly) or **`Days before Release`** (several days before a major release). Due to long execution times, the frequency is lower.
-- ***Run Command***:
+-   ***Execution Environment***: ***GPU*** servers, requiring a stable and exclusive testing environment.
-    - ***Stability***: `pytest -s -v tests/dfx/stability/scripts/test_stability_qwen3_omni.py` or `pytest -s -v tests/dfx/stability/scripts/test_stability_wan22.py` (or add `test_stability_<model>.py` alongside a matching JSON config)
+-   ***Script Example***:
-    - ***Reliability***: `pytest -s -v tests/dfx/reliability/test_reliability_<model>.py -m slow` (current suites: `qwen3_omni`, `wan22`, `hunyuan_image`).
-- ***Script Example***:
-
 <details>
 <summary> Test Examples</summary>
 
-##### Stability
+When you want to add L5-level stability test cases, add or extend the appropriate JSON file under `tests/dfx/stability/tests/` (for example `test_qwen3_omni.json` for Omni bench traffic, or `test_wan22.json` for diffusion `/v1/videos` workloads). The following illustrates the Qwen3-Omni shape:
-
-When you want to add L5-level stability test cases, add or extend the appropriate JSON file under `tests/dfx/stability/tests/` (for example `test_qwen3_omni.json` for Omni bench traffic, or `test_wan22.json` for diffusion `/v1/videos` workloads). Pair the JSON with `tests/dfx/stability/scripts/test_stability_<model>.py`. The following illustrates the Qwen3-Omni shape:
 
 ```json
 {
     "test_name": "test_qwen3_omni_stability",
     "server_params": {
         "model": "Qwen/Qwen3-Omni-30B-A3B-Instruct",
-        "stage_config_name": "qwen3_omni_moe.yaml"
+        "stage_config_name": "qwen3_omni.yaml"
     },
     "benchmark_params": [
         {
@@ -588,61 +601,112 @@ When you want to add L5-level stability test cases, add or extend the appropriat
 }
 ```
 
-###### Parameter Explanation
+##### Parameter Explanation
 
 ***Overview***
 
 | Field            | Required | Description                                                                 |
 | ---------------- | -------- | --------------------------------------------------------------------------- |
 | test_name        | Yes      | Unique identifier for the stability test case                               |
-| server_params    | Yes      | Server-side configuration parameters (model, deploy configuration, etc.)    |
+| server_params    | Yes      | Server-side configuration parameters (model, stage configuration, etc.)     |
 | benchmark_params | Yes      | Stability benchmark running parameters (supports multiple configurations)   |
 
-###### server_params Configuration
+##### server_params Configuration
 
-Basic parameters:
+###### Basic Parameters
 
 | Parameter         | Required | Example                            | Description                         |
 | ----------------- | -------- | ---------------------------------- | ----------------------------------- |
 | model             | Yes      | "Qwen/Qwen3-Omni-30B-A3B-Instruct" | Model name or path                  |
-| stage_config_name | Yes      | "qwen3_omni_moe.yaml"              | Deploy configuration file name      |
+| stage_config_name | Yes      | "qwen3_omni.yaml"                  | Stage configuration file name       |
 
-Dynamic configuration (`update` / `delete`) supports incremental modifications based on the basic configuration:
+###### Dynamic Configuration (update/delete)
+
+Supports incremental modifications based on the basic configuration:
 
 | Operation | Description                          |
 | --------- | ------------------------------------ |
 | update    | Update or add configuration items    |
 | delete    | Delete specified configuration items |
 
-***Example***: You can refer to Test Examples in L4 §3.4.
+***Example***:
+You can refer to Test Examples in L4 §3.4
 
-###### benchmark_params Configuration
+##### benchmark_params Configuration
 
 For stability testing, the key parameters are:
 
-- **duration_sec**: Total duration (in seconds) during which benchmark traffic is sent. The stability benchmark will keep sending batches until this duration is reached.
+-   **duration_sec**: Total duration (in seconds) during which benchmark traffic is sent. The stability benchmark will keep sending batches until this duration is reached.
-- **request_rate** / **max_concurrency**: Exactly one of them must be specified. They control how the traffic is generated for each batch:
+-   **request_rate** / **max_concurrency**: Exactly one of them must be specified. They control how the traffic is generated for each batch:
-    - `request_rate`: Number of requests per second. The benchmark will send `num_prompts_per_batch` requests at the given rate.
+    -   `request_rate`: Number of requests per second. The benchmark will send `num_prompts_per_batch` requests at the given rate.
-    - `max_concurrency`: Maximum number of concurrent requests. When this is used, `request_rate` is set to `inf` internally.
+    -   `max_concurrency`: Maximum number of concurrent requests. When this is used, `request_rate` is set to `inf` internally.
-- **num_prompts_per_batch**: Number of prompts sent in each batch. Multiple batches will be executed sequentially within `duration_sec`.
+-   **num_prompts_per_batch**: Number of prompts sent in each batch. Multiple batches will be executed sequentially within `duration_sec`.
 
 All other optional parameters follow the same rules as in L4 §3.4.
 
-##### Reliability
+</details>
 
-Add pytest modules under `tests/dfx/reliability/` (for example `test_reliability_qwen3_omni.py`, `test_reliability_wan22.py`, `test_reliability_hunyuan_image.py`). Reuse `helpers.py` and `conftest.py` rather than duplicating OOM / process-kill / raw HTTP plumbing.
+<details>
+<summary> Reliability test suite (<code>tests/dfx/reliability</code>)</summary>
 
-###### Parametrization and markers
+##### Purpose and relationship to stability
+
+Reliability tests are **short fault-injection** integration runs (L5 **(b)** in `tests/dfx/reliability/README.md`). They complement **stability** JSON-driven long runs: instead of hours of steady traffic, they **perturb** the server (GPU OOM sidecar, fatal signals on selected processes) and check **failure mode** and **latency bounds** (e.g. chat or `/v1/videos` must not hang under concurrent fault-time load).
+
+##### Directory layout
+
+| Path | Responsibility |
+| ---- | -------------- |
+| `helpers.py` | Shared helpers used by current reliability suites: raw `POST`/`GET` probes (`/v1/chat/completions`, `/health`), OpenAI error parsing (`extract_openai_error_contract_from_bytes`), GPU OOM sidecar lifecycle (`inject_gpu_oom`, `stop_gpu_oom_hogs`), and process-kill injector builder (`make_process_kill_fault_injector`). |
+| `conftest.py` | Pytest fixtures: indirect `fault_injector`, `omni_server_after_fault` / `omni_server_after_fault_function` (run injector after server is ready, then yield server). |
+| `test_reliability_qwen3_omni.py` | Qwen3-Omni: OOM vs **text vs text+audio** error contract, large multimodal chat under OOM, concurrent pressure, **SIGKILL** on `VLLM::Worker`, `/health` → 503 + fast-fail + concurrent chat; optional OOM recovery scenario (may be skipped while tracked in issues). |
+| `test_reliability_wan22.py` | Wan2.2 I2V: large `/v1/videos` under OOM, **SIGKILL** on `multiprocessing.spawn` chain, health / fast-fail / concurrent video requests; optional recovery test (may be skipped). |
+| `test_reliability_hunyuan_image.py` | HunyuanImage DiT-only: large `/v1/images/generations` under OOM, **SIGKILL** on `vLLM-Omni::` workers and serve/tree targets, health / fast-fail / concurrent image requests; some OOM/recovery cases may be skipped while tracked in issues. |
+| `test_reliability_voxcpm2.py` | VoxCPM2: `/v1/audio/speech` under OOM (error contract), **SIGKILL** on `VLLM::` workers and serve/tree targets, health / fast-fail / concurrent speech requests; some OOM cases may be skipped while tracked in issues. |
+| `README.md` | Minimal run / collect examples. |
+
+##### Parametrization and markers
 
 - Each test module defines a **`RELIABILITY_SCENARIOS`** list (`test_name`, `server_params`: model, `stage_config_name` or diffusion `server_args`, etc.). **`create_reliability_omni_server_params()`** in `tests/dfx/conftest.py` resolves stage paths (including XPU substitutions where applicable) and builds **`OmniServerParams`** lists consumed by **`@pytest.mark.parametrize(..., indirect=True)`** on `omni_server` or `omni_server_function`.
-- Cases are tagged **`@pytest.mark.slow`** for weekly / selective CI. GPU-heavy suites use **`@hardware_test(res={"cuda": "H100"}, num_cards=...)`** (Qwen3-Omni **2**× H100; Wan2.2 **1**× H100; HunyuanImage DiT **4**× H100).
+- Cases are tagged **`@pytest.mark.slow`** for weekly / selective CI. GPU-heavy suites use **`@hardware_test(res={"cuda": "H100"}, num_cards=...)`** or **`@hardware_test(res={"cuda": "L4"}, num_cards=1)`** (Qwen3-Omni **2**× H100; Wan2.2 **1**× H100; HunyuanImage DiT **4**× H100; VoxCPM2 **1**× L4).
 - **Process-kill** tests use **`@pytest.mark.skipif(os.name == "nt", ...)`** because injection uses POSIX **`pgrep` / `kill`**.
 
-###### Adding a new model suite
+##### CI trigger
+
+Weekly Buildkite (`.buildkite/cuda/test-weekly.yml`) runs one step per model suite (trigger: `WEEKLY=1` or PR label `weekly-test`), for example:
+
+| Buildkite step | Test file | CI hardware |
+| -------------- | --------- | ----------- |
+| Reliability Test - qwen3-omni | `test_reliability_qwen3_omni.py` | H100 × 2 (`mithril-h100-pool`) |
+| Reliability Test - wan22 | `test_reliability_wan22.py` | H100 × 2 (`mithril-h100-pool`) |
+| Reliability Test - hunyuan-image | `test_reliability_hunyuan_image.py` | H100 × 4 (`mithril-h100-pool`) |
+| Reliability Test - voxcpm2 | `test_reliability_voxcpm2.py` | L4 × 1 (`gpu_1_queue`) |
+
+```bash
+pytest -s -v tests/dfx/reliability/test_reliability_qwen3_omni.py -m "slow"
+pytest -s -v tests/dfx/reliability/test_reliability_wan22.py -m "slow"
+pytest -s -v tests/dfx/reliability/test_reliability_hunyuan_image.py -m "slow"
+pytest -s -v tests/dfx/reliability/test_reliability_voxcpm2.py -m "slow"
+```
+
+##### Local commands
+
+```bash
+pytest --collect-only tests/dfx/reliability
+pytest -s -v tests/dfx/reliability/test_reliability_qwen3_omni.py -m slow
+pytest -s -v tests/dfx/reliability/test_reliability_wan22.py -m slow
+pytest -s -v tests/dfx/reliability/test_reliability_hunyuan_image.py -m slow
+pytest -s -v tests/dfx/reliability/test_reliability_voxcpm2.py -m slow
+```
+
+##### Adding a new model suite
 
 1. Add `test_reliability_<model>.py` under `tests/dfx/reliability/`.
-2. Define **`RELIABILITY_SCENARIOS`** and pass them through **`create_reliability_omni_server_params()`** with the correct deploy or e2e deploy-config directory (same pattern as existing files).
+2. Define **`RELIABILITY_SCENARIOS`** and pass them through **`create_reliability_omni_server_params()`** with the correct deploy or e2e stage-config directory (same pattern as existing files).
-3. Reuse **`helpers`** for OOM / kill / raw HTTP; prefer **`assert_fault_exception()`** and **`resolve_oom_device_spec()`** from `tests/dfx/conftest.py` for consistent device selection vs deploy YAML.
+3. Reuse **`helpers`** for OOM / kill / raw HTTP; prefer **`assert_fault_exception()`** and **`resolve_oom_device_spec()`** from `tests/dfx/conftest.py` for consistent device selection vs stage YAML.
 4. Register **`slow`** (and **`hardware_test`** if needed); extend **`.buildkite/cuda/test-weekly.yml`** when the suite should run in weekly L5.
 
 </details>
+
+-   -   ***Stability***: `pytest -s -v tests/dfx/stability/scripts/test_stability_qwen3_omni.py` or `pytest -s -v tests/dfx/stability/scripts/test_stability_wan22.py` (or add `test_stability_<model>.py` alongside a matching JSON config)
+    -   ***Reliability***: `pytest -s -v tests/dfx/reliability/test_reliability_<model>.py -m slow` (current suites: `qwen3_omni`, `wan22`, `hunyuan_image`, `voxcpm2`; add `test_reliability_<suite>.py` and a matching step in `.buildkite/cuda/test-weekly.yml` for new models)

docs/contributing/model/adding_omni_model.md

@@ -9,7 +9,7 @@ This guide walks through the process of adding a new multi-stage model to vLLM-O
 3. [Step-by-Step Implementation](#step-by-step-implementation)
 4. [Key Components](#key-components)
 5. [Model Registration](#model-registration)
-6. [Pipeline and Deploy Configuration](#pipeline-and-deploy-configuration)
+6. [Stage Configuration](#stage-configuration)
 7. [Stage Input Processors](#stage-input-processors)
 8. [Testing](#testing)
 9. [Adding a Model Recipe](#adding-a-model-recipe)
@@ -316,28 +316,18 @@ _OMNI_MODELS = {
 
 The registry uses lazy loading, so the model class is imported only when needed.
 
-## Pipeline and Deploy Configuration
+## Stage Configuration
 
-Define fixed topology in `vllm_omni/model_executor/models/<model>/pipeline.py`
+Create a YAML configuration file in `vllm_omni/deploy/`. For a complete example, see the [Qwen3-Omni configuration file](gh-file:vllm_omni/deploy/qwen3_omni_moe.yaml).
-with `PipelineConfig` and `StagePipelineConfig`, then register it in
-`vllm_omni/config/pipeline_registry.py`. For a complete example, see the
-[Qwen3-Omni pipeline](gh-file:vllm_omni/model_executor/models/qwen3_omni/pipeline.py).
 
-### Pipeline topology fields
+### Key Configuration Fields
 
-- **`model_stage`**: Logical stage name ("thinker", "talker", "code2wav", etc.)
+- **`model_stage`**: Which stage to run ("thinker", "talker", "code2wav", etc.)
-- **`execution_type`**: Runtime family (`LLM_AR`, `LLM_GENERATION`, or `DIFFUSION`)
+- **`model_arch`**: The model architecture name (must match registry)
-- **`model_arch`**: Stage architecture name, when different from the pipeline default
+- **`engine_input_source`**: List of stage IDs that provide input to this stage
-- **`input_sources`**: Upstream stage IDs that provide input to this stage
+- **`custom_process_input_func`**: Function to process inputs from previous stages
-- **`custom_process_input_func`**: Function that processes inputs from previous stages
+- **`final_output`**: Whether this stage produces the final output (True/False)
-- **`final_output`**: Whether this stage produces user-facing output
+- **`final_output_type`**: Type of final output ("text", "audio", "image", etc.)
-- **`final_output_type`**: Output modality ("text", "audio", "image", etc.)
-- **`owns_tokenizer`**: Whether this stage owns the pipeline tokenizer
-
-Create the matching runtime defaults in `vllm_omni/deploy/<model>.yaml` and set
-`PipelineConfig.default_deploy_config_name` to that file. The deploy YAML owns
-GPU placement, memory sizing, parallelism, connectors, and sampling defaults;
-it must not redefine topology fields.
 
 ## Stage Input Processors
 
@@ -384,7 +374,7 @@ The stage transition process follows these steps:
 
 2. **Transition Detection**: The orchestrator checks if there's a next stage and calls `process_engine_inputs()` on it
 
-3. **Input Processing**: The stage input processor registered by the pipeline (implemented under `vllm_omni/model_executor/stage_input_processors/`) handles the transition:
+3. **Input Processing**: The stage input processor configured in stage YAML (under `vllm_omni/model_executor/stage_input_processors/`) handles the transition:
    ```python
    def process_engine_inputs(
        self, stage_list: list[Any], prompt: OmniTokensPrompt | TextPrompt = None
@@ -587,7 +577,7 @@ Adding a new model to vLLM-Omni involves:
 1. **Create model directory structure** with stage implementations
 2. **Implement unified model class** that orchestrates stages
 3. **Register model** in `registry.py`
-4. **Define pipeline topology and deploy defaults** in `pipeline.py` and `vllm_omni/deploy/`
+4. **Create stage configuration** YAML file
 5. **Implement stage input processors** for stage transitions
 6. **Write tests** to verify functionality
 7. **Add model recipe** to the [vllm-project/recipes](https://github.com/vllm-project/recipes) repository (see [Adding a Model Recipe](#adding-a-model-recipe) section)
@@ -600,8 +590,7 @@ For a complete reference implementation, see:
 - **Thinker**: `vllm_omni/model_executor/models/qwen3_omni/qwen3_omni_moe_thinker.py`
 - **Talker**: `vllm_omni/model_executor/models/qwen3_omni/qwen3_omni_moe_talker.py`
 - **Code2Wav**: `vllm_omni/model_executor/models/qwen3_omni/qwen3_omni_code2wav.py`
-- **Pipeline topology**: `vllm_omni/model_executor/models/qwen3_omni/pipeline.py`
+- **Stage config**: `vllm_omni/deploy/qwen3_omni_moe.yaml`
-- **Deploy config**: `vllm_omni/deploy/qwen3_omni_moe.yaml`
 - **Input processors**: `vllm_omni/model_executor/stage_input_processors/qwen3_omni.py`
 - **Registry**: `vllm_omni/model_executor/models/registry.py`
 - **Testing**: `vllm_omni/tests/e2e/offline_inference/test_qwen3_omni.py`
@@ -609,4 +598,4 @@ For a complete reference implementation, see:
 For more information, see:
 - [Architecture Overview](../../design/architecture_overview.md)
 - [Supported Models](../../models/supported_models.md)
-- [Pipeline and Deploy Configuration Guide](../../configuration/stage_configs.md)
+- [Stage Configuration Guide](../../configuration/stage_configs.md)

docs/contributing/model/adding_tts_model.md

@@ -20,7 +20,7 @@ pattern, refer to MOSS-TTS-Nano.
 4. [Step-by-Step Implementation](#step-by-step-implementation)
 5. [Key Components](#key-components)
 6. [Model Registration](#model-registration)
-7. [Pipeline and Deploy Configuration](#pipeline-and-deploy-configuration)
+7. [Stage Configuration](#stage-configuration)
 8. [Stage Input Processors](#stage-input-processors)
 9. [Online Serving Integration](#online-serving-integration)
 10. [Single-Stage Models](#single-stage-models)
@@ -78,7 +78,7 @@ Profile before optimizing.
 new TTS model must pass all three levels:
 
 | Layer | Catches | Tool |
-| ------- | --------- | ------ |
+|-------|---------|------|
 | Offline RTF / duration | Throughput regressions, missing audio, wrong sample rate | `end2end.py`, pytest e2e |
 | Browser streaming playback | Delta-vs-cumulative bugs, chunk boundary glitches, TTFP regressions | Gradio demo over `/v1/audio/speech?stream=true` |
 | Concurrent requests | Per-request state leaks, codec window round-robin gaps | `max_num_seqs>1` smoke with 4+ parallel prompts |
@@ -94,10 +94,10 @@ the symptom is crosstalk or truncation under load, nothing in single-request tes
 vLLM-Omni supports TTS models as multi-stage pipelines where each stage runs independently
 and can be placed on different devices. Qwen3-TTS has two stages:
 
-| Stage | Name                | Input           | Output                   |
+| Stage | Name | Input | Output |
-| ----- | ------------------- | --------------- | ------------------------ |
+|-------|------|-------|--------|
-| 0     | Code Predictor (AR) | Text tokens     | Discrete RVQ codec codes |
+| 0 | Code Predictor (AR) | Text tokens | Discrete RVQ codec codes |
-| 1     | Code2Wav (Decoder)  | RVQ codec codes | Audio waveform           |
+| 1 | Code2Wav (Decoder) | RVQ codec codes | Audio waveform |
 
 Each stage is a separate model class configured independently via YAML. The two stages
 are connected by the `async_chunk` framework, which enables inter-stage streaming for
@@ -170,7 +170,7 @@ and Qwen3-Omni).
 
 When adding a new TTS model, create the following structure:
 
-```text
+```
 vllm_omni/model_executor/models/
   your_model_name/
     __init__.py
@@ -201,7 +201,7 @@ it would not be picked up — the hub README is the documented surface.
 **Qwen3-TTS reference files:**
 
 | File | Purpose |
-| ------ | --------- |
+|------|---------|
 | `models/qwen3_tts/qwen3_tts.py` | Unified model class |
 | `models/qwen3_tts/qwen3_tts_code_predictor_vllm.py` | Stage 0 - optimized AR |
 | `models/qwen3_tts/qwen3_tts_code2wav.py` | Stage 1 - decoder |
@@ -427,92 +427,111 @@ _OMNI_MODELS = {
 
 The registry uses lazy loading - model classes are only imported when needed.
 
-## Pipeline and Deploy Configuration
+## Stage Configuration
 
-Define fixed topology in `models/<model>/pipeline.py`. The execution type
+Each stage has a `worker_type` that determines how it is scheduled:
-selects the scheduler and worker family:
 
-- `StageExecutionType.LLM_AR` uses the autoregressive runtime with PagedAttention.
+- `worker_type: ar` - autoregressive stage, uses `OmniARScheduler` with PagedAttention
-- `StageExecutionType.LLM_GENERATION` uses the generation runtime for stages such as audio decoders.
+- `worker_type: generation` - non-AR stage (e.g. decoder), uses `OmniGenerationScheduler`
 
-Key topology fields belong to `StagePipelineConfig`:
+Key configuration fields:
 
 | Field | Description |
-| ------- | ------------- |
+|-------|-------------|
-| `model_stage` | Logical stage name (`ar_stage`, `decoder`, etc.). |
+| `model_stage` | Which stage to initialize (`ar_stage`, `decoder`, etc.) |
-| `execution_type` | Runtime family for this stage. |
+| `model_arch` | Architecture name, must match `registry.py` |
-| `input_sources` | Upstream stage IDs that provide input. |
+| `engine_input_source` | List of upstream stage IDs that provide input (e.g. `[0]`) |
-| `model_arch` | Stage-specific architecture name, when it differs from the pipeline default. |
+| `engine_output_type` | Output type: `latent` for intermediate, `audio` for final |
-| `engine_output_type` | Output representation such as `latent` or `audio`. |
+| `custom_process_next_stage_input_func` | Async chunk processor function path (streaming only) |
-| `custom_process_next_stage_input_func` | Processor for full-payload stage handoff. |
+| `final_output` | Whether this stage produces the final user-facing output |
-| `async_chunk_process_next_stage_input_func` | Processor for streaming chunk handoff. |
+| `final_output_type` | Type of final output (`audio`, `text`, etc.) |
-| `final_output` / `final_output_type` | Whether this stage produces user-facing output and its modality. |
-| `owns_tokenizer` | Whether this stage supplies the pipeline tokenizer. |
 
-For example, a two-stage TTS topology can be defined as:
+!!! note
+    New in-tree models should define frozen topology in `models/<model>/pipeline.py`,
+    register it in `vllm_omni/config/pipeline_registry.py`, and put deployment
+    knobs in `vllm_omni/deploy/<model>.yaml`. Use `--deploy-config` to load a
+    custom deployment YAML.
 
-```python
+### Batch mode
-from vllm_omni.config.stage_config import (
-    PipelineConfig,
-    StageExecutionType,
-    StagePipelineConfig,
-)
-
-_PROC = "vllm_omni.model_executor.stage_input_processors.your_model_name"
-
-YOUR_TTS_PIPELINE = PipelineConfig(
-    model_type="your_model_name",
-    default_deploy_config_name="your_model_name.yaml",
-    model_arch="YourTTSModelForConditionalGeneration",
-    stages=(
-        StagePipelineConfig(
-            stage_id=0,
-            model_stage="ar_stage",
-            execution_type=StageExecutionType.LLM_AR,
-            input_sources=(),
-            owns_tokenizer=True,
-            engine_output_type="latent",
-            custom_process_next_stage_input_func=f"{_PROC}.ar2decoder",
-            async_chunk_process_next_stage_input_func=f"{_PROC}.ar2decoder_async_chunk",
-        ),
-        StagePipelineConfig(
-            stage_id=1,
-            model_stage="decoder",
-            execution_type=StageExecutionType.LLM_GENERATION,
-            input_sources=(0,),
-            model_arch="YourTTSDecoder",
-            engine_output_type="audio",
-            final_output=True,
-            final_output_type="audio",
-        ),
-    ),
-)
-```
-
-Register the pipeline in `vllm_omni/config/pipeline_registry.py`. Runtime
-placement and sizing belong in `vllm_omni/deploy/your_model_name.yaml`:
 
 ```yaml
-async_chunk: true
+# stage_configs/your_model_name.yaml
 
-stages:
+stage_args:
   - stage_id: 0
-    devices: "0"
+    stage_type: llm
-    max_num_seqs: 64
+    runtime:
+      devices: "0"
+    engine_args:
+      model_stage: ar_stage
+      max_num_seqs: 64
+      model_arch: YourTTSModelForConditionalGeneration
+      worker_type: ar
+      scheduler_cls: vllm_omni.core.sched.omni_ar_scheduler.OmniARScheduler
+      engine_output_type: latent
     default_sampling_params:
       temperature: 0.9
       top_k: 50
       max_tokens: 2048
 
   - stage_id: 1
-    devices: "0"
+    stage_type: llm
-    max_num_seqs: 1
+    runtime:
+      devices: "0"
+    engine_args:
+      model_stage: decoder
+      model_arch: YourTTSModelForConditionalGeneration
+      worker_type: generation
+      scheduler_cls: vllm_omni.core.sched.omni_generation_scheduler.OmniGenerationScheduler
+      engine_output_type: audio
+    engine_input_source: [0]
+    final_output: true
+    final_output_type: audio
 ```
 
-Set `async_chunk: false` for end-to-end handoff. The registered pipeline
+### Streaming mode (async_chunk)
-selects the full-payload or async chunk processor; processor paths and topology
+
-must not be moved into the deploy YAML. Use `--deploy-config` to load a custom
+Add `async_chunk: true` at the top level and specify `custom_process_next_stage_input_func`
-deployment overlay.
+on Stage 0 to define how intermediate outputs are chunked and forwarded:
+
+```yaml
+# stage_configs/your_model_name_async_chunk.yaml
+
+async_chunk: true
+
+stage_args:
+  - stage_id: 0
+    stage_type: llm
+    runtime:
+      devices: "0"
+    engine_args:
+      model_stage: ar_stage
+      max_num_seqs: 64
+      model_arch: YourTTSModelForConditionalGeneration
+      worker_type: ar
+      scheduler_cls: vllm_omni.core.sched.omni_ar_scheduler.OmniARScheduler
+      engine_output_type: latent
+      custom_process_next_stage_input_func: >
+        vllm_omni.model_executor.stage_input_processors.your_model_name.ar2decoder_async_chunk
+    default_sampling_params:
+      temperature: 0.9
+      top_k: 50
+      max_tokens: 2048
+
+  - stage_id: 1
+    stage_type: llm
+    runtime:
+      devices: "0"
+    engine_args:
+      model_stage: decoder
+      model_arch: YourTTSModelForConditionalGeneration
+      worker_type: generation
+      scheduler_cls: vllm_omni.core.sched.omni_generation_scheduler.OmniGenerationScheduler
+      engine_output_type: audio
+    engine_input_source: [0]
+    final_output: true
+    final_output_type: audio
+```
 
 ## Stage Input Processors
 
@@ -528,9 +547,9 @@ Understanding what's available in stage outputs:
 - `stage_list[source_id].engine_outputs` - list of `EngineCoreOutput` objects
 - Each `EngineCoreOutput` has `outputs` - list of `RequestOutput` objects
 - Each `RequestOutput` has:
-    - `token_ids` - generated token IDs
+  - `token_ids` - generated token IDs
-    - `multimodal_output` - dict with keys matching your model's `OmniOutput.multimodal_outputs`
+  - `multimodal_output` - dict with keys matching your model's `OmniOutput.multimodal_outputs`
-    - `prompt_token_ids` - original prompt token IDs
+  - `prompt_token_ids` - original prompt token IDs
 
 ### Batch mode (non-streaming)
 
@@ -610,7 +629,6 @@ def ar2decoder_async_chunk(
 ```
 
 Key points:
-
 - `transfer_manager` is the `OmniChunkTransferAdapter` that owns the chunk lifecycle
 - Each call appends one AR decode step's output; a chunk is emitted every `chunk_size` steps
 - The final (possibly partial) chunk is flushed when `is_finished` is true
@@ -665,7 +683,6 @@ Create `vllm_omni/entrypoints/openai/tts_adapters/your_model.py`:
 
 ```python
 # SPDX-License-Identifier: Apache-2.0
-# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
 """YourModel serving adapter."""
 
 from typing import TYPE_CHECKING
@@ -804,7 +821,7 @@ directly from the model's own generator.
 
 ### Directory structure
 
-```text
+```
 vllm_omni/model_executor/models/your_model_name/
     __init__.py
     modeling_your_model_name.py    # unified class: load_weights + forward + streaming
@@ -815,40 +832,32 @@ vllm_omni/deploy/your_model_name.yaml
 
 No stage input processor is needed.
 
-### Pipeline and deploy config
+### Stage config
 
-Use a single `LLM_AR` stage that owns the tokenizer and produces final audio:
+Use a single stage with `worker_type: ar`. The `is_comprehension: true` field and the
-
+top-level `async_chunk: false` are required — omitting them causes silent
-```python
+misclassification in the serving layer. Set `max_num_seqs` to at least 4 for
-YOUR_TTS_PIPELINE = PipelineConfig(
+concurrent production use.
-    model_type="your_model_name",
-    default_deploy_config_name="your_model_name.yaml",
-    model_arch="YourModelForCausalLM",
-    stages=(
-        StagePipelineConfig(
-            stage_id=0,
-            model_stage="your_model_stage_key",
-            execution_type=StageExecutionType.LLM_AR,
-            input_sources=(),
-            owns_tokenizer=True,
-            engine_output_type="audio",
-            final_output=True,
-            final_output_type="audio",
-        ),
-    ),
-)
-```
-
-The deploy YAML controls runtime behavior. Set `max_num_seqs` to at least 4
-for concurrent production use:
 
 ```yaml
+# stage_configs/your_model_name.yaml
 async_chunk: false
 
-stages:
+stage_args:
   - stage_id: 0
-    devices: "0"
+    stage_type: llm
-    max_num_seqs: 4
+    is_comprehension: true          # required for serving_speech.py dispatch
+    runtime:
+      devices: "0"
+    engine_args:
+      model_stage: your_model_stage_key
+      model_arch: YourModelForCausalLM
+      worker_type: ar
+      scheduler_cls: vllm_omni.core.sched.omni_ar_scheduler.OmniARScheduler
+      engine_output_type: audio
+      max_num_seqs: 4               # min 4 for concurrent requests; default 1 causes gaps
+    final_output: true
+    final_output_type: audio
 ```
 
 ### Generator-based streaming pattern
@@ -952,17 +961,10 @@ is correct. Stage the modified files and commit again — do not revert the chan
 Common failures and fixes:
 
 | Check | Cause | Fix |
-| ------- | ------- | ----- |
+|-------|-------|-----|
 | `ruff F841` | Local variable assigned but never used | Remove the extraction or forward it to the model call |
 | `ruff E402` | Module-level import not at top of file | Move import to the top-level import block |
 | `ruff format` | Line length, spacing, or quote style | Accept the auto-fix, stage, and re-commit |
-| `check-spdx-header` | Missing header, or copyright still says `vLLM project` | Use `Copyright contributors to the vLLM-Omni project`; restage the rewrite |
-| `check-forbidden-imports` | Stdlib `re`/`base64`, pickle, Hugging Face Hub API, or direct Triton/TileLang | `import regex as re` and `pybase64`; do not grow the allowlist without review |
-| `check-torch-cuda-call` | New `torch.cuda.*` call site | Use `current_omni_platform`; do not grow `ALLOWED_FILES` without review |
-| `check-tts-adapter-migration` | New `_tts_model_type` branch in `serving_speech.py` | Put logic in `tts_adapters/`; lower `MAX_MODEL_TYPE_BRANCHES` when removing branches |
-| `check-mark` | Test file missing level/hardware mark, or direct `pytest.mark.H100` | Add `core_model`/… plus `cpu`/`cuda`/`hardware_test(`; SKU marks only via helpers |
-| `shellcheck` | Native `shellcheck` missing, or script warning | Install via apt/dnf/brew (or `shellcheck.exe` on Windows). See [Linting](../README.md#linting) |
-| `mypy-3.10` / `markdownlint-cli2` / `check-buildkite` | Types, docs markdown, or Buildkite YAML | See [Linting](../README.md#linting) |
 
 ### DCO sign-off
 
@@ -1017,7 +1019,7 @@ Adding a TTS model to vLLM-Omni involves:
 ### Qwen3-TTS Reference Files
 
 | File | Purpose |
-| ------ | --------- |
+|------|---------|
 | `models/qwen3_tts/qwen3_tts.py` | Unified model class |
 | `models/qwen3_tts/qwen3_tts_code_predictor_vllm.py` | AR stage with vLLM fused ops |
 | `models/qwen3_tts/qwen3_tts_code2wav.py` | Decoder stage with `chunked_decode_streaming()` |
@@ -1029,4 +1031,4 @@ For more information, see:
 
 - [Architecture Overview](../../design/architecture_overview.md)
 - [Async Chunk Design](../../design/feature/async_chunk.md)
-- [Pipeline and Deploy Configuration Guide](../../configuration/stage_configs.md)
+- [Stage Configuration Guide](../../configuration/stage_configs.md)

docs/contributing/profiling.md

@@ -322,15 +322,6 @@ Each 1-second window records:
 
 On shutdown the server also logs a short summary (`loop_active_pct`, per-replica queue averages/maxima).
 
-`loop_idle` and `loop_active` count loop iterations, so their absolute scale is
-tied to which orchestration loop is running. Under the default poll loop an idle
-orchestrator records roughly one iteration per millisecond. Under the
-event-driven loop (`VLLM_OMNI_EVENT_DRIVEN_ORCH=1`, see
-[Speech API](../serving/speech_api.md#orchestration-loop-experimental)) an idle
-orchestrator wakes only once per 0.5 s reconcile timeout, while a busy one still
-records one iteration per routed output. Window counts and the `loop_active_pct`
-summary are therefore not comparable across the two modes.
-
 ### Relationship to other diagnostics
 
 This monitor is intentionally separate from the existing profiling tools:

docs/design/ar_diffusion_pipeline_capability.md

@@ -4,9 +4,6 @@
 > `experimental/ar_diffusion` contract. It follows the ownership and lifecycle
 > direction in [#5137](https://github.com/vllm-project/vllm-omni/issues/5137),
 > but it is not yet part of the stable module-design hierarchy.
->
-> The end-to-end request, identity, failure, and memory contracts are documented
-> in [Realtime AR-Diffusion sessions](feature/realtime_ar_diffusion.md).
 
 `ARDiffusionEngine` is an opt-in diffusion backend for pipelines that keep
 autoregressive attention KV across requests. The engine only selects
@@ -54,13 +51,14 @@ time, and its inherited batch/step entry points also fail explicitly. Supporting
 those paths requires a batch-aware state-binding contract rather than silently
 bypassing `bind_ar_diffusion_state()`.
 
-The implementation may retain multiple sessions per model instance. It selects
+The current implementation retains one session per model instance. A request
-the largest feasible capacity up to the pipeline's `session_capacity`, based on
+for a different `session_id` closes the resident session through the same LRU
-self-attention pages, cross-attention allocations, scratch, and declared
+lifecycle used by explicit close and failure cleanup before allocating the new
-model-owned state. One session is admitted whenever it fits actual free device
+one. Self-attention and cross-attention memory are therefore sized for one
-memory; `gpu_memory_fraction` controls additional residency. Requests beyond
+resident session even if the pipeline declares a larger `session_capacity`.
-that capacity evict the least-recently-used session through the same lifecycle
+This prioritizes rollout throughput and prevents block-pool exhaustion before a
-hook used by explicit close and failure cleanup.
+count-based eviction can run. Multi-session residency can be added later with a
+memory-backed capacity calculation or allocation-pressure eviction.
 
 Scratch storage is also capability-driven. `frames_per_block` reserves space for
 an uncommitted current video block, while `max_scratch_tokens_per_branch`

docs/design/architecture_overview.md

@@ -335,8 +335,8 @@ The single resolution boundary is implemented by
 `StageConfigFactory.create_from_model()` and
 `VllmOmniConfig.from_pipeline_config()` in
 [`vllm_omni/config`](https://github.com/vllm-project/vllm-omni/tree/main/vllm_omni/config).
-The legacy `stage_args` YAML path has been removed. Model topology now resolves
+The legacy `stage_args` YAML path remains only for models that have not yet
-through `PipelineConfig`, with runtime overrides supplied by `DeployConfig`.
+migrated to `PipelineConfig` and `DeployConfig`.
 
 In the typed path, each stage configuration derives from
 `BaseVllmOmniStageConfig` and is specialized as
@@ -406,7 +406,7 @@ the source for configuration, compatibility, and implementation details.
   stage contract.
 * **Quantization and memory efficiency:** [Quantization](feature/quantization.md)
   resolves per-pipeline or per-component quantization configurations, while
-  [Distributed Layerwise Offload](feature/offloader/distributed_layerwise_offload.md)
+  [Distributed Layerwise Offload](feature/distributed_layerwise_offload.md)
   streams diffusion blocks from host memory within the existing parallel
   topology.