Skip to content

Commit 3f20b35

Browse files
jeejeeleeskyloevil
authored andcommitted
[Misc] Slight improve deepgemm print (vllm-project#24085)
Signed-off-by: Jee Jee Li <pandaleefree@gmail.com>
1 parent db22a93 commit 3f20b35

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

vllm/model_executor/layers/fused_moe/deep_gemm_moe.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,13 +57,14 @@ def _valid_deep_gemm(hidden_states: torch.Tensor, w1: torch.Tensor,
5757
if not _valid_deep_gemm_shape(M, N, K):
5858
logger.debug_once(
5959
"DeepGemm disabled due to unaligned problem size. "
60-
"M: %s, N: %s, K: %s. M should >= align size "
61-
"and N and K must be multiples of %s."
60+
"M: %s, N: %s, K: %s. M should >= %s "
61+
"and N and K must be multiples of %s. "
6262
"This is not an error and we will fall back to triton.",
6363
M,
6464
N,
6565
K,
6666
align,
67+
align,
6768
)
6869
return False
6970
elif N <= 512:

0 commit comments

Comments
 (0)