Skip to content

Commit 354f325

Browse files
committed
vLLM CPU x86 smoke tests
Signed-off-by: Edward Quarm <equarmjnr@gmail.com>
1 parent 8814cf9 commit 354f325

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

utilities/constants.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ class RuntimeTemplates:
8080
VLLM_ROCM: str = "vllm-rocm-runtime-template"
8181
VLLM_GAUDI: str = "vllm-gaudi-runtime-template"
8282
VLLM_SPYRE: str = "vllm-spyre-x86-runtime-template"
83+
VLLM_CPU_x86: str = "vllm-cpu-x86-runtime-template"
8384
MLSERVER: str = f"{ModelFormat.MLSERVER}-runtime-template"
8485
TRITON_REST: str = "triton-rest-runtime-template"
8586
TRITON_GRPC: str = "triton-grpc-runtime-template"
@@ -129,7 +130,8 @@ class AcceleratorType:
129130
AMD: str = "amd"
130131
GAUDI: str = "gaudi"
131132
SPYRE: str = "spyre"
132-
SUPPORTED_LISTS: list[str] = [NVIDIA, AMD, GAUDI, SPYRE]
133+
CPU: str = "cpu"
134+
SUPPORTED_LISTS: list[str] = [NVIDIA, AMD, GAUDI, SPYRE, CPU]
133135

134136

135137
class ApiGroups:

0 commit comments

Comments
 (0)