xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 05:15:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

[CI][CPU] Improve dummy Triton interfaces and fix the CPU CI (#19838)

Browse Source 
Signed-off-by: jiang1.li <jiang1.li@intel.com>
This commit is contained in:
Li, Jiang 2025-06-19 23:46:09 +08:00 committed by GitHub
parent 6f68c49220
commit 01220ce89a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
requirements/cpu.txt
View File

@ -21,9 +21,6 @@ torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
torchvision==0.22.0; platform_machine == "ppc64le" torchvision==0.22.0; platform_machine == "ppc64le"
datasets # for benchmark scripts 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 Extension for PyTorch, only for x86_64 CPUs
intel-openmp==2024.2.1; platform_machine == "x86_64" intel-openmp==2024.2.1; platform_machine == "x86_64"
intel_extension_for_pytorch==2.7.0; platform_machine == "x86_64" intel_extension_for_pytorch==2.7.0; platform_machine == "x86_64"

2
vllm/triton_utils/importing.py
View File

@ -68,9 +68,11 @@ class TritonPlaceholder(types.ModuleType):
def __init__(self): def __init__(self):
super().__init__("triton") super().__init__("triton")
self.__version__ = "3.3.0"
self.jit = self._dummy_decorator("jit") self.jit = self._dummy_decorator("jit")
self.autotune = self._dummy_decorator("autotune") self.autotune = self._dummy_decorator("autotune")
self.heuristics = self._dummy_decorator("heuristics") self.heuristics = self._dummy_decorator("heuristics")
self.Config = self._dummy_decorator("Config")
self.language = TritonLanguagePlaceholder() self.language = TritonLanguagePlaceholder()
logger.warning_once( logger.warning_once(
"Triton is not installed. Using dummy decorators. " "Triton is not installed. Using dummy decorators. "