diff --git a/python/sglang/srt/model_executor/runner/decode_cuda_graph_runner.py b/python/sglang/srt/model_executor/runner/decode_cuda_graph_runner.py index e7abb66e0..51463f6e4 100644 --- a/python/sglang/srt/model_executor/runner/decode_cuda_graph_runner.py +++ b/python/sglang/srt/model_executor/runner/decode_cuda_graph_runner.py @@ -705,6 +705,11 @@ class DecodeCudaGraphRunner(BaseCudaGraphRunner): cuda_graph_bs, stream_idx=get_current_stream_idx() if self.enable_pdmux else None, variant_label=self._resolve_lora_variant(forward_batch), + dsa_variant=( + self._resolve_dsa_variant(forward_batch) + if self.disable_padding + else None + ), ) is_bs_supported = (