mirror of
https://github.com/vllm-project/vllm.git
synced 2025-10-20 14:53:52 +08:00
[Bugifx] Remove TritonPlaceholder from sys.modules (#17317)
Signed-off-by: Isotr0py <2037008807@qq.com>
This commit is contained in:
@ -1,6 +1,5 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
import sys
|
||||
import types
|
||||
from importlib.util import find_spec
|
||||
|
||||
@ -45,9 +44,4 @@ if not HAS_TRITON:
|
||||
super().__init__("triton.language")
|
||||
self.constexpr = None
|
||||
self.dtype = 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.")
|
||||
self.int64 = None
|
||||
|
Reference in New Issue
Block a user