# tests/utils.py

def requires_platform(*platforms: str):
    """
    Skip test unless current_platform matches at least one of the given platforms.
    Accepted values: "cuda", "rocm", "cpu", "xpu", "tpu", "cuda_alike" (= cuda|rocm).
    """
    _CHECKS = {
        "cuda": current_platform.is_cuda,
        "rocm": current_platform.is_rocm,
        "cpu": current_platform.is_cpu,
        "xpu": current_platform.is_xpu,
        "tpu": current_platform.is_tpu,
        "cuda_alike": current_platform.is_cuda_alike,
    }
    match = any(_CHECKS[p]() for p in platforms)
    return pytest.mark.skipif(
        not match,
        reason=f"Requires platform(s) {platforms}, "
               f"got {current_platform._enum.name}",
    )


def requires_capability(min_capability: int):
    """Skip test unless device has compute capability >= min_capability."""
    return pytest.mark.skipif(
        not current_platform.has_device_capability(min_capability),
        reason=f"Requires compute capability >= {min_capability}",
    )

---

from tests.utils import requires_platform, requires_capability

@requires_platform("cuda")
def test_cutlass_scaled_mm():
    ...

@requires_platform("cuda", "rocm")
@requires_capability(90)
def test_fp8_marlin():
    ...

# Module-level
pytestmark = requires_platform("cuda")

---

@pytest.mark.target(platform="cuda", min_capability=90)
def test_fp8_marlin():
    ...

@pytest.mark.target(platform=["cuda", "rocm"])
def test_paged_attention():
    ...

# Exclude a platform
@pytest.mark.target(exclude_platform="rocm")
def test_cuda_only_feature():
    ...

---

# tests/plugins/platform_filter.py

def pytest_collection_modifyitems(config, items):
    for item in items:
        for m in item.iter_markers("target"):
            platform = m.kwargs.get("platform")
            exclude = m.kwargs.get("exclude_platform")
            min_cap = m.kwargs.get("min_capability")

            if platform:
                platforms = [platform] if isinstance(platform, str) else platform
                if not _matches_any(platforms):
                    item.add_marker(pytest.mark.skip(...))
            if exclude:
                excludes = [exclude] if isinstance(exclude, str) else exclude
                if _matches_any(excludes):
                    item.add_marker(pytest.mark.skip(...))
            if min_cap and not current_platform.has_device_capability(min_cap):
                item.add_marker(pytest.mark.skip(...))