Move DP-attention adapter methods to SchedulerDPAttnAdapter (#25612)

This commit is contained in:
fzyzcjy
2026-05-18 18:32:16 +08:00
committed by GitHub
parent 8f37a8a3f3
commit 4d6eec7b32
8 changed files with 272 additions and 289 deletions
+1 -1
View File
@@ -71,7 +71,7 @@ from sglang.srt.layers.moe import initialize_moe_config
from sglang.srt.layers.quantization.fp4_utils import initialize_fp4_gemm_config
from sglang.srt.layers.quantization.fp8_utils import initialize_fp8_gemm_config
from sglang.srt.managers.schedule_batch import Req, ScheduleBatch
from sglang.srt.managers.scheduler_dp_attn_mixin import prepare_mlp_sync_batch_raw
from sglang.srt.managers.scheduler_components.dp_attn import prepare_mlp_sync_batch_raw
from sglang.srt.mem_cache.base_prefix_cache import EvictParams
from sglang.srt.model_executor.forward_batch_info import ForwardBatch
from sglang.srt.model_executor.model_runner import ModelRunner