[Bug] Include DSA variant in exact-bucket graph admission (#38992)

2026-09-10 21:03:03 -07:00
parent b3dc0388ed
commit 94ce940ff8
1 changed files with 5 additions and 0 deletions
@@ -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 = (