From 01220ce89a332a4105c0031933c5079036ceefa0 Mon Sep 17 00:00:00 2001 From: "Li, Jiang" Date: Thu, 19 Jun 2025 23:46:09 +0800 Subject: [PATCH] [CI][CPU] Improve dummy Triton interfaces and fix the CPU CI (#19838) Signed-off-by: jiang1.li --- requirements/cpu.txt | 3 --- vllm/triton_utils/importing.py | 2 ++ 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/requirements/cpu.txt b/requirements/cpu.txt index d7b0fc6d80a7..8742898cff00 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -21,9 +21,6 @@ torchvision; platform_machine != "ppc64le" and platform_machine != "s390x" torchvision==0.22.0; platform_machine == "ppc64le" datasets # for benchmark scripts -# cpu cannot use triton 3.3.0 -triton==3.2.0; platform_machine == "x86_64" - # Intel Extension for PyTorch, only for x86_64 CPUs intel-openmp==2024.2.1; platform_machine == "x86_64" intel_extension_for_pytorch==2.7.0; platform_machine == "x86_64" diff --git a/vllm/triton_utils/importing.py b/vllm/triton_utils/importing.py index 21beb76f37cc..dd30b2bc5f07 100644 --- a/vllm/triton_utils/importing.py +++ b/vllm/triton_utils/importing.py @@ -68,9 +68,11 @@ class TritonPlaceholder(types.ModuleType): def __init__(self): super().__init__("triton") + self.__version__ = "3.3.0" self.jit = self._dummy_decorator("jit") self.autotune = self._dummy_decorator("autotune") self.heuristics = self._dummy_decorator("heuristics") + self.Config = self._dummy_decorator("Config") self.language = TritonLanguagePlaceholder() logger.warning_once( "Triton is not installed. Using dummy decorators. "