forked from Karylab-cklius/vllm
8c13ee5735
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com> Co-authored-by: OpenAI Codex <noreply@openai.com>
24 lines
677 B
Python
24 lines
677 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
|
|
def test_exact_family_arch_precedes_generic_family_fallback(tmp_path: Path):
|
|
repo_root = Path(__file__).parents[1]
|
|
script = tmp_path / "test_cuda_archs.cmake"
|
|
script.write_text(
|
|
f"""
|
|
cmake_minimum_required(VERSION 3.26)
|
|
include("{repo_root / "cmake" / "utils.cmake"}")
|
|
cuda_archs_loose_intersection(
|
|
actual "10.0f;10.7f" "10.7")
|
|
if(NOT "${{actual}}" STREQUAL "10.7f")
|
|
message(FATAL_ERROR "Expected 10.7f, got '${{actual}}'")
|
|
endif()
|
|
"""
|
|
)
|
|
|
|
subprocess.run(["cmake", "-P", script], check=True)
|