minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

Fix MockDSV4ModelRunner missing spec_algorithm (#31056)

Browse Source
This commit is contained in:
Jialin Ouyang
2026-07-13 15:24:28 -05:00
committed by GitHub
parent 2ab531cfcf
commit 47030b28be
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/test/kits/attention_unittest/attention_methods/dsv4_attention.py
+2
View File
@@ -35,6 +35,7 @@ from sglang.srt.model_executor.cuda_graph_config import (
from sglang.srt.model_executor.forward_batch_info import ForwardBatch, ForwardMode from sglang.srt.model_executor.forward_batch_info import ForwardBatch, ForwardMode
from sglang.srt.model_executor.forward_context import ForwardContext, forward_context from sglang.srt.model_executor.forward_context import ForwardContext, forward_context
from sglang.srt.runtime_context import get_context, get_parallel from sglang.srt.runtime_context import get_context, get_parallel
from sglang.srt.speculative.spec_info import SpeculativeAlgorithm
# DSV4 backend pre-resolves attention TP at construction; pin to single-rank. # DSV4 backend pre-resolves attention TP at construction; pin to single-rank.
_parallel_override = get_parallel().override( _parallel_override = get_parallel().override(
@@ -410,6 +411,7 @@ class MockDSV4ModelRunner:
self.sliding_window_size = DSV4_SWA_WINDOW self.sliding_window_size = DSV4_SWA_WINDOW
self.use_mla_backend = True self.use_mla_backend = True
self.is_draft_worker = False self.is_draft_worker = False
self.spec_algorithm = SpeculativeAlgorithm.NONE
self._kernel_warmed_up = True self._kernel_warmed_up = True
@property @property