[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:
1 parent
5e34e1c97d
commit
2bcd1e449a
2 files changed
+30
-4
No files matched your search
@@ -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,
|
||||||
|
|||||||
Reference in new issue
Block a user