[CI][ROCm] Add LingBot nightly coverage (#7935)

Signed-off-by: haic0 <149741444+haic0@users.noreply.github.com>
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
haic0andCursor authored and GitHub committed 2026-10-06 05:31:51 +00:00
1 parent 5e34e1c97d
commit 2bcd1e449a
2 files changed
+30 -4

No files matched your search

+17
View File
@@ -82,3 +82,20 @@ steps:
limit: 1 limit: 1
commands: commands:
- pytest -s -v tests/e2e/online_serving/test_cosyvoice3_tts_expansion.py -m "slow" --run-level "core_model" - pytest -s -v tests/e2e/online_serving/test_cosyvoice3_tts_expansion.py -m "slow" --run-level "core_model"
- group: ":full_moon: LingBot-Video MI300 Nightly"
steps:
- label: "LingBot-Video Dense CFG-off Smoke"
agent_pool: mi300_1
mirror_hardwares: [amdproduction]
grade: NonBlocking
timeout_in_minutes: 90
artifact_paths:
- artifacts/rocm-lingbot-nightly/**/*
commands:
- pytest --collect-only -q tests/e2e/online_serving/test_lingbot_video_expansion.py::test_cfg_off -m "full_model and diffusion and rocm and MI325 and cards_1" --run-level full_model
- pytest -s -v tests/e2e/online_serving/test_lingbot_video_expansion.py::test_cfg_off -m "full_model and diffusion and rocm and MI325 and cards_1" --run-level full_model --junitxml=/tmp/lingbot-pytest.xml
- |
artifact_dir="$${BUILDKITE_BUILD_CHECKOUT_PATH:?}/artifacts/rocm-lingbot-nightly"
mkdir -p "$$artifact_dir"
cp -v /tmp/lingbot-pytest.xml "$$artifact_dir/"
@@ -27,9 +27,10 @@ PROMPT = "a robotic arm picks up a red block"
NEGATIVE_PROMPT = "low quality, blurry, watermark, text" NEGATIVE_PROMPT = "low quality, blurry, watermark, text"
SINGLE_CARD_FEATURE_MARKS = hardware_marks(res={"cuda": ["H100", "B200"]}) SINGLE_CARD_FEATURE_MARKS = hardware_marks(res={"cuda": ["H100", "B200"]})
CFG_OFF_SINGLE_CARD_FEATURE_MARKS = hardware_marks(res={"cuda": ["H100", "B200"], "rocm": "MI325"})
def _get_diffusion_feature_cases(model: str): def _get_diffusion_feature_cases(model: str, marks):
return [ return [
pytest.param( pytest.param(
OmniServerParams( OmniServerParams(
@@ -37,12 +38,16 @@ def _get_diffusion_feature_cases(model: str):
server_args=["--model-class-name", "LingBotVideoPipeline"], server_args=["--model-class-name", "LingBotVideoPipeline"],
), ),
id="default", id="default",
marks=SINGLE_CARD_FEATURE_MARKS, marks=marks,
), ),
] ]
@pytest.mark.parametrize("omni_server", _get_diffusion_feature_cases(MODEL), indirect=True) @pytest.mark.parametrize(
"omni_server",
_get_diffusion_feature_cases(MODEL, CFG_OFF_SINGLE_CARD_FEATURE_MARKS),
indirect=True,
)
def test_cfg_off(omni_server: OmniServer, openai_client: OpenAIClientHandler) -> None: def test_cfg_off(omni_server: OmniServer, openai_client: OpenAIClientHandler) -> None:
request_config = { request_config = {
"model": omni_server.model, "model": omni_server.model,
@@ -62,7 +67,11 @@ def test_cfg_off(omni_server: OmniServer, openai_client: OpenAIClientHandler) ->
openai_client.send_video_diffusion_request(request_config) openai_client.send_video_diffusion_request(request_config)
@pytest.mark.parametrize("omni_server", _get_diffusion_feature_cases(MODEL), indirect=True) @pytest.mark.parametrize(
"omni_server",
_get_diffusion_feature_cases(MODEL, SINGLE_CARD_FEATURE_MARKS),
indirect=True,
)
def test_batch_cfg_extra_params(omni_server: OmniServer, openai_client: OpenAIClientHandler) -> None: def test_batch_cfg_extra_params(omni_server: OmniServer, openai_client: OpenAIClientHandler) -> None:
request_config = { request_config = {
"model": omni_server.model, "model": omni_server.model,