From f2a4c4c847786df76a11382e9eb3bdde6f223a10 Mon Sep 17 00:00:00 2001 From: Michael <13900043+michaelzhang-ai@users.noreply.github.com> Date: Mon, 10 Aug 2026 01:09:00 -0700 Subject: [PATCH] [AMD] [CI] Enable 3 nested unit tests needing harness stub fixes (#31843) Co-authored-by: HAI --- python/sglang/multimodal_gen/test/server/gpu_cases.py | 5 +++++ .../unit/progressive_resolution/test_progressive.py | 10 ++++++++++ .../test/unit/sana_wm/test_streaming_realtime_path.py | 1 + scripts/ci/utils/ci_coverage_report.py | 5 +++++ 4 files changed, 21 insertions(+) diff --git a/python/sglang/multimodal_gen/test/server/gpu_cases.py b/python/sglang/multimodal_gen/test/server/gpu_cases.py index 0cc8af740..a5373340a 100644 --- a/python/sglang/multimodal_gen/test/server/gpu_cases.py +++ b/python/sglang/multimodal_gen/test/server/gpu_cases.py @@ -1091,6 +1091,11 @@ _AMD_READY_NESTED_UNIT_TESTS = ( "sana_wm/test_streaming_cached.py", "sana_wm/test_streaming_stage.py", "sana_wm/test_streaming_vae.py", + # Enabled with small test-harness stub fixes (see this PR's test edits). + "progressive_resolution/test_progressive.py", + "sana_wm/test_streaming_realtime_path.py", + # Stub gap already fixed upstream; only needs enabling here. + "realtime/test_lingbot_causal_denoising.py", ) diff --git a/python/sglang/multimodal_gen/test/unit/progressive_resolution/test_progressive.py b/python/sglang/multimodal_gen/test/unit/progressive_resolution/test_progressive.py index ade149a8e..64e2b9186 100644 --- a/python/sglang/multimodal_gen/test/unit/progressive_resolution/test_progressive.py +++ b/python/sglang/multimodal_gen/test/unit/progressive_resolution/test_progressive.py @@ -9,6 +9,7 @@ from types import SimpleNamespace import torch from sglang.multimodal_gen.configs.sample.sampling_params import SamplingParams +from sglang.multimodal_gen.runtime.distributed.cfg_policy import CFGPolicy from sglang.multimodal_gen.runtime.pipelines_core.stages.progressive_resolution.denoising import ( ProgressiveDenoisingStage, ProgressiveDenoisingStageRouter, @@ -500,6 +501,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase): ctx = SimpleNamespace( latents=torch.zeros(B, new_num_img, self._IN_C), extra=self._make_ctx_extra(B, old_num_img, max_text_tokens), + # Non-None so _on_resolution_change proceeds (it early-returns when + # cfg_policy is None); the value itself is not used further here. + cfg_policy=CFGPolicy(), ) batch = SimpleNamespace( extra={ @@ -555,6 +559,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase): ctx = SimpleNamespace( latents=torch.zeros(B, new_grid_h * new_grid_w, self._IN_C), extra=self._make_ctx_extra(B, old_grid_h * old_grid_w, max_text_tokens), + # Non-None so _on_resolution_change proceeds (it early-returns when + # cfg_policy is None); the value itself is not used further here. + cfg_policy=CFGPolicy(), ) batch = SimpleNamespace(extra={"ideogram4": old_data}) @@ -594,6 +601,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase): ctx = SimpleNamespace( latents=torch.zeros(B, grid_h * grid_w, self._IN_C), extra=self._make_ctx_extra(B, old_num_img, max_text_tokens), + # Non-None so _on_resolution_change proceeds (it early-returns when + # cfg_policy is None); the value itself is not used further here. + cfg_policy=CFGPolicy(), ) batch = SimpleNamespace( extra={ diff --git a/python/sglang/multimodal_gen/test/unit/sana_wm/test_streaming_realtime_path.py b/python/sglang/multimodal_gen/test/unit/sana_wm/test_streaming_realtime_path.py index 2e85089e1..d87d1d792 100644 --- a/python/sglang/multimodal_gen/test/unit/sana_wm/test_streaming_realtime_path.py +++ b/python/sglang/multimodal_gen/test/unit/sana_wm/test_streaming_realtime_path.py @@ -47,6 +47,7 @@ def _global_args(): comfyui_mode=False, enable_cfg_parallel=False, enable_torch_compile=False, + enable_breakable_cuda_graph=False, attention_backend=None, # DenoisingStage.__init__ reads this for its CFG-parallel plumbing. pipeline_config=SimpleNamespace( diff --git a/scripts/ci/utils/ci_coverage_report.py b/scripts/ci/utils/ci_coverage_report.py index 918fcd5a6..e8c802471 100755 --- a/scripts/ci/utils/ci_coverage_report.py +++ b/scripts/ci/utils/ci_coverage_report.py @@ -101,6 +101,11 @@ _MM_GEN_FILE_BACKENDS = { "unit/sana_wm/test_streaming_cached.py": ("AMD",), "unit/sana_wm/test_streaming_stage.py": ("AMD",), "unit/sana_wm/test_streaming_vae.py": ("AMD",), + # Enabled with small test-harness stub fixes. + "unit/progressive_resolution/test_progressive.py": ("AMD",), + "unit/sana_wm/test_streaming_realtime_path.py": ("AMD",), + # Stub gap already fixed upstream; only needs enabling here. + "unit/realtime/test_lingbot_causal_denoising.py": ("AMD",), } # Filenames that match `test_*.py` by convention but contain no real tests