main
py 17 lines 419 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4 import pytest
5
6 from vllm.platforms import current_platform
7
8
9 @pytest.fixture(autouse=True)
10 def require_rocm_cdna4() -> None:
11 if not current_platform.is_rocm():
12 pytest.skip("requires ROCm")
13
14 from vllm.platforms.rocm import on_gfx950
15
16 if not on_gfx950():
17 pytest.skip("requires CDNA4")