95a9311696
Signed-off-by: wangyu <410167048@qq.com> Signed-off-by: lishunyang12 <lishunyang12@163.com> Signed-off-by: [Your Name] <your.email@example.com> Signed-off-by: wangyu <53896905+yenuo26@users.noreply.github.com> Signed-off-by: Bo Li <22713281+bobboli@users.noreply.github.com> Co-authored-by: lishunyang12 <lishunyang12@163.com> Co-authored-by: Cursor <cursoragent@cursor.com> Co-authored-by: Bo Li <22713281+bobboli@users.noreply.github.com>
50 lines
2.4 KiB
Bash
50 lines
2.4 KiB
Bash
#!/usr/bin/env bash
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
#
|
|
# Extract steps from .buildkite/cuda/test-ready.yml that contain pytest, synthesize
|
|
# small bash wrappers (exports + pytest), run them, and tee output to logs named
|
|
# after each step's Buildkite "key" when present (otherwise a slug of the label).
|
|
#
|
|
# Model area (--model-type / MODEL_TYPE), multiple allowed (OR semantics):
|
|
# e.g. --model-type omni,tts
|
|
# omni — label contains "Omni ·"
|
|
# tts — label contains "TTS ·"
|
|
# diffusion— label contains "Diffusion"
|
|
# all — no model filter (default)
|
|
#
|
|
# --skip-simple — skip steps in the "Simple Test" Buildkite group and labels
|
|
# starting with "Simple ·" (L1-style unit tests in the same YAML)
|
|
#
|
|
# Requirements: bash, python3, PyYAML (pip install pyyaml)
|
|
#
|
|
# Usage:
|
|
# bash tools/run_ready_jobs.sh
|
|
# REPO_ROOT=/path/to/vllm-omni bash tools/run_ready_jobs.sh --model-type omni --dry-run
|
|
# YML=/path/to/vllm-omni/.buildkite/cuda/test-ready.yml bash tools/run_ready_jobs.sh
|
|
#
|
|
# Repository / YAML (no dependency on where this script lives):
|
|
# • Set REPO_ROOT (or pass --repo-root) — default YAML is $REPO_ROOT/.buildkite/cuda/test-ready.yml
|
|
# • Or set YML (or --yaml) — repo root is inferred as parent of the .buildkite directory
|
|
# • Or run from inside the clone: git rev-parse --show-toplevel, else walk up from $PWD,
|
|
# then from the script's directory, until .buildkite/cuda/test-ready.yml exists
|
|
#
|
|
# Optional environment:
|
|
# REPO_ROOT - vllm-omni root (working directory for pytest); see above
|
|
# YML - path to test-ready.yml (default: $REPO_ROOT/.buildkite/cuda/test-ready.yml)
|
|
# LOG_DIR - logs + generated job scripts (default: $REPO_ROOT/logs/ready_jobs);
|
|
# per-job *.log plus timing_summary.log after the run
|
|
# MODEL_TYPE - comma-separated and/or repeated flags (default: all); see above
|
|
# LABEL_SUBSTR - substring of Buildkite step label
|
|
# DRY_RUN=1 - print extracted commands only; do not write scripts or run pytest
|
|
#
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
BUILDKITE_REL=".buildkite/cuda/test-ready.yml"
|
|
DEFAULT_LOG_SUBDIR="ready_jobs"
|
|
|
|
# shellcheck source=tools/run_jobs_common.sh
|
|
source "${SCRIPT_DIR}/run_jobs_common.sh"
|
|
run_yaml_ci_jobs_main "$@"
|