[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 = (