Skip to content

Commit 01220ce

Browse files
authored
[CI][CPU] Improve dummy Triton interfaces and fix the CPU CI (#19838)
Signed-off-by: jiang1.li <[email protected]>
1 parent 6f68c49 commit 01220ce

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

requirements/cpu.txt

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,6 @@ torchvision; platform_machine != "ppc64le" and platform_machine != "s390x"
2121
torchvision==0.22.0; platform_machine == "ppc64le"
2222
datasets # for benchmark scripts
2323

24-
# cpu cannot use triton 3.3.0
25-
triton==3.2.0; platform_machine == "x86_64"
26-
2724
# Intel Extension for PyTorch, only for x86_64 CPUs
2825
intel-openmp==2024.2.1; platform_machine == "x86_64"
2926
intel_extension_for_pytorch==2.7.0; platform_machine == "x86_64"

vllm/triton_utils/importing.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,9 +68,11 @@ class TritonPlaceholder(types.ModuleType):
6868

6969
def __init__(self):
7070
super().__init__("triton")
71+
self.__version__ = "3.3.0"
7172
self.jit = self._dummy_decorator("jit")
7273
self.autotune = self._dummy_decorator("autotune")
7374
self.heuristics = self._dummy_decorator("heuristics")
75+
self.Config = self._dummy_decorator("Config")
7476
self.language = TritonLanguagePlaceholder()
7577
logger.warning_once(
7678
"Triton is not installed. Using dummy decorators. "

0 commit comments

Comments
 (0)