biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Kernel] [Helion] [9/N] Canonicalize GPU variant names to base model names (#34928)

Browse Source 
Signed-off-by: Yanan Cao <gmagogsfm@gmail.com>
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Yanan Cao
2026-02-20 19:55:51 -08:00
committed by GitHub
parent e739c29ea4
commit 9d7577b2bd
4 changed files with 59 additions and 27733 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/kernels/helion/test_utils.py
View File

@@ -11,11 +11,13 @@ from vllm.kernels.helion.utils import canonicalize_gpu_name
"driver_reported_name,expected",
[
("NVIDIA H200", "nvidia_h200"),
("NVIDIA A100-SXM4-80GB", "nvidia_a100_sxm4_80gb"),
("NVIDIA H100 80GB HBM3", "nvidia_h100_80gb_hbm3"),
("NVIDIA A100-SXM4-80GB", "nvidia_a100"),
("NVIDIA H100 80GB HBM3", "nvidia_h100"),
("NVIDIA H100 PCIe", "nvidia_h100"),
("NVIDIA H100 SXM5", "nvidia_h100"),
("NVIDIA GeForce RTX 4090", "nvidia_geforce_rtx_4090"),
("AMD Instinct MI300X", "amd_instinct_mi300x"),
("Tesla V100-SXM2-32GB", "tesla_v100_sxm2_32gb"),
("Tesla V100-SXM2-32GB", "tesla_v100"),
],
)
def test_canonicalize_gpu_name(driver_reported_name, expected):