diff --git a/python/sglang/srt/speculative/eagle_worker_v2.py b/python/sglang/srt/speculative/eagle_worker_v2.py index 7706f5013..9ad5e5275 100644 --- a/python/sglang/srt/speculative/eagle_worker_v2.py +++ b/python/sglang/srt/speculative/eagle_worker_v2.py @@ -416,10 +416,15 @@ class EagleDraftWorker(EagleDraftWorkerBase): from sglang.srt.layers.attention.deepseek_v4_backend_hip_radix import ( DeepseekV4HipRadixBackend, ) + from sglang.srt.layers.attention.dsa_backend import ( + DeepseekSparseAttnBackend, + ) - supports_hip_draft_extend_graph = isinstance( - self.draft_attn_backend, AiterMultiStepDraftBackend - ) or isinstance(self.draft_extend_attn_backend, DeepseekV4HipRadixBackend) + supports_hip_draft_extend_graph = ( + isinstance(self.draft_attn_backend, AiterMultiStepDraftBackend) + or isinstance(self.draft_extend_attn_backend, DeepseekV4HipRadixBackend) + or isinstance(self.draft_extend_attn_backend, DeepseekSparseAttnBackend) + ) graph_supported_backend_types = [ TritonAttnBackend,