"...ssh:/git@developer.sourcefind.cn:2222/OpenDAS/dynamo.git" did not exist on "3fd0ab3d74c9dc9e185580eb0a863f3c24465a67"
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, ...@@ -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)
......
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