Skip to content

Commit 3157aeb

Browse files
authored
[Log] Add Warning for Deprecation of DeepGEMM old version (#22194)
Signed-off-by: yewentao256 <[email protected]>
1 parent 8a0ffd6 commit 3157aeb

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

vllm/utils/deep_gemm.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
import torch
1515

1616
import vllm.envs as envs
17+
from vllm.logger import logger
1718
from vllm.platforms import current_platform
1819
from vllm.utils import cdiv, has_deep_gemm
1920

@@ -57,6 +58,14 @@ def _resolve_symbol(module, new: str, old: str) -> Callable[..., Any] | None:
5758
if hasattr(module, new):
5859
return getattr(module, new)
5960
if hasattr(module, old):
61+
# TODO(wentao): deprecate old symbol in the future.
62+
logger.warning_once(
63+
"Found legacy DeepGEMM symbol `%s`. Please upgrade the `deep_gemm` "
64+
"package so that `%s` is available. Support for the legacy symbol "
65+
"will be removed in a future vLLM release.",
66+
old,
67+
new,
68+
)
6069
return getattr(module, old)
6170
return None
6271

0 commit comments

Comments
 (0)