mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-18 07:05:02 +08:00
[Misc] Restrict deep_gemm's log output (#20827)
Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
This commit is contained in:
parent
a8593237c0
commit
44d02f54db
@ -43,7 +43,7 @@ def _valid_deep_gemm(hidden_states: torch.Tensor, w1: torch.Tensor,
|
|||||||
aligned by `dg.get_m_alignment_for_contiguous_layout()`.
|
aligned by `dg.get_m_alignment_for_contiguous_layout()`.
|
||||||
"""
|
"""
|
||||||
if not has_deep_gemm():
|
if not has_deep_gemm():
|
||||||
logger.debug("DeepGemm disabled: deep_gemm not available.")
|
logger.debug_once("DeepGemm disabled: deep_gemm not available.")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
M = hidden_states.size(0)
|
M = hidden_states.size(0)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user