[AMD] Fix the diffusion perf fixture lookup (#38686)
This commit is contained in:
@@ -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):
|
||||
"""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)
|
||||
def case(self, request) -> DiffusionTestCase:
|
||||
return request.param
|
||||
|
||||
@@ -106,13 +106,6 @@ def _compute_clip_score(image_bytes: bytes, prompt: str) -> float | None:
|
||||
class TestZImageTurboAMD(DiffusionServerBase):
|
||||
"""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)
|
||||
def case(self, request) -> DiffusionTestCase:
|
||||
return request.param
|
||||
|
||||
Reference in New Issue
Block a user