Skip to content

GitLab

Unverified Commit 9e2de9b9 authored by Isotr0py's avatar Isotr0py Committed by GitHub
Browse files

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


Signed-off-by: default avatarIsotr0py <2037008807@qq.com>
parent 109e15a3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 7 deletions
vllm/triton_utils/importing.py
View file @ 9e2de9b9
# 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: ...@@ -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.")
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment