"components/vscode:/vscode.git/clone" did not exist on "67d00b24d73f4a0067c73b624c81278321559bb4"
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