mirror of
https://github.com/vllm-project/vllm.git
synced 2026-08-02 11:58:01 +00:00
7acaea634c
Signed-off-by: Lalithnarayan C <[email protected]> Signed-off-by: Tyler Michael Smith <[email protected]> Co-authored-by: Chinmay-Kulkarni-AMD <[email protected]> Co-authored-by: Tyler Michael Smith <[email protected]> Co-authored-by: Tyler Michael Smith <[email protected]> Co-authored-by: Claude Sonnet 4.6 <[email protected]>
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
from unittest.mock import mock_open, patch
|
|
|
|
from vllm.platforms import _is_amd_zen_cpu
|
|
|
|
|
|
def test_is_amd_zen_cpu_detects_amd_with_avx512():
|
|
cpuinfo = "vendor_id: AuthenticAMD\nflags: avx avx2 avx512f avx512bw"
|
|
with (
|
|
patch("os.path.exists", return_value=True),
|
|
patch("builtins.open", mock_open(read_data=cpuinfo)),
|
|
):
|
|
assert _is_amd_zen_cpu()
|
|
|
|
|
|
def test_is_amd_zen_cpu_returns_false_for_amd_without_avx512():
|
|
cpuinfo = "vendor_id: AuthenticAMD\nflags: avx avx2"
|
|
with (
|
|
patch("os.path.exists", return_value=True),
|
|
patch("builtins.open", mock_open(read_data=cpuinfo)),
|
|
):
|
|
assert not _is_amd_zen_cpu()
|
|
|
|
|
|
def test_is_amd_zen_cpu_returns_false_for_intel_with_avx512():
|
|
cpuinfo = "vendor_id: GenuineIntel\nflags: avx avx2 avx512f"
|
|
with (
|
|
patch("os.path.exists", return_value=True),
|
|
patch("builtins.open", mock_open(read_data=cpuinfo)),
|
|
):
|
|
assert not _is_amd_zen_cpu()
|
|
|
|
|
|
def test_is_amd_zen_cpu_returns_false_when_cpuinfo_missing():
|
|
with patch("os.path.exists", return_value=False):
|
|
assert not _is_amd_zen_cpu()
|