From 0294844f04e233d0968a319aa7c244a69f79f981 Mon Sep 17 00:00:00 2001 From: TZHelloWorld <40588411+TZHelloWorld@users.noreply.github.com> Date: Sun, 28 Dec 2025 13:47:39 +0800 Subject: [PATCH] [fix]deepgemm precompile when warmup (#15891) --- python/sglang/srt/layers/deep_gemm_wrapper/compile_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/layers/deep_gemm_wrapper/compile_utils.py b/python/sglang/srt/layers/deep_gemm_wrapper/compile_utils.py index dca005dd1..5e25e56a2 100644 --- 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