diff --git a/.buildkite/amd/test-amd-nightly.yml b/.buildkite/amd/test-amd-nightly.yml index 4544a4f20..9e2b5bacc 100644 --- a/.buildkite/amd/test-amd-nightly.yml +++ b/.buildkite/amd/test-amd-nightly.yml @@ -82,3 +82,20 @@ steps: limit: 1 commands: - 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/" diff --git a/tests/e2e/online_serving/test_lingbot_video_expansion.py b/tests/e2e/online_serving/test_lingbot_video_expansion.py index 74c7c143b..caa278f08 100644 --- a/tests/e2e/online_serving/test_lingbot_video_expansion.py +++ b/tests/e2e/online_serving/test_lingbot_video_expansion.py @@ -27,9 +27,10 @@ PROMPT = "a robotic arm picks up a red block" NEGATIVE_PROMPT = "low quality, blurry, watermark, text" 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 [ pytest.param( OmniServerParams( @@ -37,12 +38,16 @@ def _get_diffusion_feature_cases(model: str): server_args=["--model-class-name", "LingBotVideoPipeline"], ), 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: request_config = { "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) -@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: request_config = { "model": omni_server.model,