[AMD] Fix the diffusion perf fixture lookup (#38686)

This commit is contained in:
YC Yen-Ching Tseng
2026-09-10 00:07:08 -07:00
committed by GitHub
parent bd922bc39b
commit 6b2e13bcb0
3 changed files with 10 additions and 14 deletions
+10
View File
@@ -0,0 +1,10 @@
"""Make the diffusion perf fixture reachable from this directory.
``DiffusionServerBase`` (subclassed by the diffusion tests here) has an autouse
fixture that requests ``perf_results``, which is declared in
``sglang/multimodal_gen/test/server/conftest.py``. pytest resolves conftest.py
by the collected file's directory, so that declaration never reaches
``test/registered/``; re-exporting it here is what makes it resolvable.
"""
from sglang.multimodal_gen.test.server.conftest import perf_results # noqa: F401
@@ -101,13 +101,6 @@ FP8_MLA_CASES = [
class TestWan22FP8MLA(DiffusionServerBase): class TestWan22FP8MLA(DiffusionServerBase):
"""AMD test for FP8 MLA attention on Wan2.2-T2V-A14B.""" """AMD test for FP8 MLA attention on Wan2.2-T2V-A14B."""
@classmethod
def teardown_class(cls):
try:
super().teardown_class()
except AttributeError:
pass
@pytest.fixture(params=FP8_MLA_CASES, ids=lambda c: c.id) @pytest.fixture(params=FP8_MLA_CASES, ids=lambda c: c.id)
def case(self, request) -> DiffusionTestCase: def case(self, request) -> DiffusionTestCase:
return request.param return request.param
-7
View File
@@ -106,13 +106,6 @@ def _compute_clip_score(image_bytes: bytes, prompt: str) -> float | None:
class TestZImageTurboAMD(DiffusionServerBase): class TestZImageTurboAMD(DiffusionServerBase):
"""AMD nightly test for Z-Image-Turbo text-to-image generation.""" """AMD nightly test for Z-Image-Turbo text-to-image generation."""
@classmethod
def teardown_class(cls):
try:
super().teardown_class()
except AttributeError:
pass
@pytest.fixture(params=AMD_ZIMAGE_CASES, ids=lambda c: c.id) @pytest.fixture(params=AMD_ZIMAGE_CASES, ids=lambda c: c.id)
def case(self, request) -> DiffusionTestCase: def case(self, request) -> DiffusionTestCase:
return request.param return request.param