43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
|
"""Tests for benchmarks/tts/hw_emulation.py case construction (CPU)."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
sys.path.insert(0, str(Path(__file__).parent.parent.parent / "benchmarks" / "tts"))
|
||
|
|
import hw_emulation as hw # noqa: E402
|
||
|
|
|
||
|
|
from vllm_omni.edge.hardware_probe import hardware_class # noqa: E402
|
||
|
|
|
||
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
||
|
|
|
||
|
|
|
||
|
|
def test_cases_map_to_expected_classes():
|
||
|
|
expected = {
|
||
|
|
"x86_cpu_avx2": "x86_cpu",
|
||
|
|
"x86_cpu_avx512": "x86_cpu",
|
||
|
|
"x86_cpu_amx": "x86_cpu",
|
||
|
|
"arm64_cpu_logic": "arm64_cpu",
|
||
|
|
"jetson_orin_8g": "jetson",
|
||
|
|
"jetson_orin_32g": "jetson",
|
||
|
|
"cuda_discrete_8g": "cuda_discrete",
|
||
|
|
}
|
||
|
|
for name, cls in expected.items():
|
||
|
|
prof = hw.build_profile(hw.CASES[name])
|
||
|
|
assert hardware_class(prof) == cls, name
|
||
|
|
assert prof.source == "emulation"
|
||
|
|
|
||
|
|
|
||
|
|
def test_avx2_case_disables_bf16_and_pins_cores():
|
||
|
|
case = hw.CASES["x86_cpu_avx2"]
|
||
|
|
prof = hw.build_profile(case)
|
||
|
|
assert prof.supports_bf16 is False and prof.n_logical == 4
|
||
|
|
assert case.env["ATEN_CPU_CAPABILITY"] == "avx2" and case.cpus == [0, 1, 2, 3]
|
||
|
|
|
||
|
|
|
||
|
|
def test_table_render_handles_missing_metrics():
|
||
|
|
table = hw.render_table([{"case": "x", "returncode": -1, "note": "timeout"}])
|
||
|
|
assert "| x | -1 |" in table and "timeout" in table
|