[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):
|
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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user