biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Bugifx] Remove TritonPlaceholder from sys.modules (#17317)

Browse Source 
Signed-off-by: Isotr0py <2037008807@qq.com>
This commit is contained in:
Isotr0py
2025-05-02 15:45:01 +08:00
committed by GitHub
parent 109e15a335
commit 9e2de9b9e9
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
vllm/triton_utils/importing.py
View File

@@ -1,6 +1,5 @@
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
import sys
import types import types
from importlib.util import find_spec from importlib.util import find_spec
@@ -45,9 +44,4 @@ if not HAS_TRITON:
super().__init__("triton.language") super().__init__("triton.language")
self.constexpr = None self.constexpr = None
self.dtype = None self.dtype = None
self.int64 = None
sys.modules['triton'] = TritonPlaceholder()
sys.modules['triton.language'] = TritonLanguagePlaceholder()
if 'triton' in sys.modules:
logger.info("Triton module has been replaced with a placeholder.")