Skip to content

Commit 0d255fd

Browse files
Isotr0pyYuqi Zhang
authored andcommitted
[Bugifx] Remove TritonPlaceholder from sys.modules (vllm-project#17317)
Signed-off-by: Isotr0py <2037008807@qq.com> Signed-off-by: Yuqi Zhang <yuqizhang@google.com>
1 parent 0890497 commit 0d255fd

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

vllm/triton_utils/importing.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
# SPDX-License-Identifier: Apache-2.0
22

3-
import sys
43
import types
54
from importlib.util import find_spec
65

@@ -45,9 +44,4 @@ def __init__(self):
4544
super().__init__("triton.language")
4645
self.constexpr = None
4746
self.dtype = None
48-
49-
sys.modules['triton'] = TritonPlaceholder()
50-
sys.modules['triton.language'] = TritonLanguagePlaceholder()
51-
52-
if 'triton' in sys.modules:
53-
logger.info("Triton module has been replaced with a placeholder.")
47+
self.int64 = None

0 commit comments

Comments
 (0)