Unverified Commit 44d02f54 authored by Jee Jee Li's avatar Jee Jee Li Committed by GitHub
Browse files

[Misc] Restrict deep_gemm's log output (#20827)


Signed-off-by: default avatarJee Jee Li <pandaleefree@gmail.com>
parent a8593237
......@@ -43,7 +43,7 @@ def _valid_deep_gemm(hidden_states: torch.Tensor, w1: torch.Tensor,
aligned by `dg.get_m_alignment_for_contiguous_layout()`.
"""
if not has_deep_gemm():
logger.debug("DeepGemm disabled: deep_gemm not available.")
logger.debug_once("DeepGemm disabled: deep_gemm not available.")
return False
M = hidden_states.size(0)
......
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