[AMD] [CI] Enable 3 nested unit tests needing harness stub fixes (#31843)
Co-authored-by: HAI <hixiao@gmail.com>
This commit is contained in:
@@ -1091,6 +1091,11 @@ _AMD_READY_NESTED_UNIT_TESTS = (
|
|||||||
"sana_wm/test_streaming_cached.py",
|
"sana_wm/test_streaming_cached.py",
|
||||||
"sana_wm/test_streaming_stage.py",
|
"sana_wm/test_streaming_stage.py",
|
||||||
"sana_wm/test_streaming_vae.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
|
import torch
|
||||||
|
|
||||||
from sglang.multimodal_gen.configs.sample.sampling_params import SamplingParams
|
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 (
|
from sglang.multimodal_gen.runtime.pipelines_core.stages.progressive_resolution.denoising import (
|
||||||
ProgressiveDenoisingStage,
|
ProgressiveDenoisingStage,
|
||||||
ProgressiveDenoisingStageRouter,
|
ProgressiveDenoisingStageRouter,
|
||||||
@@ -500,6 +501,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase):
|
|||||||
ctx = SimpleNamespace(
|
ctx = SimpleNamespace(
|
||||||
latents=torch.zeros(B, new_num_img, self._IN_C),
|
latents=torch.zeros(B, new_num_img, self._IN_C),
|
||||||
extra=self._make_ctx_extra(B, old_num_img, max_text_tokens),
|
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(
|
batch = SimpleNamespace(
|
||||||
extra={
|
extra={
|
||||||
@@ -555,6 +559,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase):
|
|||||||
ctx = SimpleNamespace(
|
ctx = SimpleNamespace(
|
||||||
latents=torch.zeros(B, new_grid_h * new_grid_w, self._IN_C),
|
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),
|
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})
|
batch = SimpleNamespace(extra={"ideogram4": old_data})
|
||||||
|
|
||||||
@@ -594,6 +601,9 @@ class TestIdeogram4OnResolutionChange(unittest.TestCase):
|
|||||||
ctx = SimpleNamespace(
|
ctx = SimpleNamespace(
|
||||||
latents=torch.zeros(B, grid_h * grid_w, self._IN_C),
|
latents=torch.zeros(B, grid_h * grid_w, self._IN_C),
|
||||||
extra=self._make_ctx_extra(B, old_num_img, max_text_tokens),
|
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(
|
batch = SimpleNamespace(
|
||||||
extra={
|
extra={
|
||||||
|
|||||||
@@ -47,6 +47,7 @@ def _global_args():
|
|||||||
comfyui_mode=False,
|
comfyui_mode=False,
|
||||||
enable_cfg_parallel=False,
|
enable_cfg_parallel=False,
|
||||||
enable_torch_compile=False,
|
enable_torch_compile=False,
|
||||||
|
enable_breakable_cuda_graph=False,
|
||||||
attention_backend=None,
|
attention_backend=None,
|
||||||
# DenoisingStage.__init__ reads this for its CFG-parallel plumbing.
|
# DenoisingStage.__init__ reads this for its CFG-parallel plumbing.
|
||||||
pipeline_config=SimpleNamespace(
|
pipeline_config=SimpleNamespace(
|
||||||
|
|||||||
@@ -101,6 +101,11 @@ _MM_GEN_FILE_BACKENDS = {
|
|||||||
"unit/sana_wm/test_streaming_cached.py": ("AMD",),
|
"unit/sana_wm/test_streaming_cached.py": ("AMD",),
|
||||||
"unit/sana_wm/test_streaming_stage.py": ("AMD",),
|
"unit/sana_wm/test_streaming_stage.py": ("AMD",),
|
||||||
"unit/sana_wm/test_streaming_vae.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
|
# Filenames that match `test_*.py` by convention but contain no real tests
|
||||||
|
|||||||
Reference in New Issue
Block a user