[AMD] [CI] Enable 3 nested unit tests needing harness stub fixes (#31843)

Co-authored-by: HAI <hixiao@gmail.com>
This commit is contained in:
Michael
2026-08-10 16:09:00 +08:00
committed by GitHub
co-authored by HAI
parent fb3d1419fd
commit f2a4c4c847
4 changed files with 21 additions and 0 deletions
@@ -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",
)
@@ -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={
@@ -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(