[Spec] Enable draft extend cuda graph for DeepSeek-V4 attention backend (#30853)

2026-07-11 04:29:49 -05:00
parent 79f096d43b
commit 7bac9c8cdb
1 changed files with 5 additions and 0 deletions
@@ -471,6 +471,11 @@ class EagleDraftWorker(EagleDraftWorkerBase):
            )

            graph_supported_backend_types.append(DeepseekSparseAttnBackend)
+            from sglang.srt.layers.attention.deepseek_v4_backend import (
+                DeepseekV4AttnBackend,
+            )
+
+            graph_supported_backend_types.append(DeepseekV4AttnBackend)

        graph_supported_backend = isinstance(
            self.draft_extend_attn_backend,