[fix]deepgemm precompile when warmup (#15891)

2025-12-28 13:47:39 +08:00
parent 0e536600e8
commit 0294844f04
1 changed files with 2 additions and 1 deletions
--- a/python/sglang/srt/layers/deep_gemm_wrapper/compile_utils.py
+++ b/python/sglang/srt/layers/deep_gemm_wrapper/compile_utils.py
@@ -291,5 +291,6 @@ class _GroupedMaskedWarmupExecutor(_BaseWarmupExecutor):
 def deep_gemm_execution_hook(
    m: int, n: int, k: int, num_groups: int, kernel_type: DeepGemmKernelType
 ):
-    _maybe_compile_deep_gemm_one_type_all(kernel_type, n, k, num_groups)
+    if m > 0:
+        _maybe_compile_deep_gemm_one_type_all(kernel_type, n, k, num_groups)
    yield