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

[Bugfix] Migrate retired parallel accessors (#30653)

Browse Source
This commit is contained in:
Mohammad Miadh Angkad
2026-07-09 11:22:34 -07:00
committed by GitHub
parent 26ba3458d3
commit b0ecbceed9
4 changed files with 12 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/registered/unit/mem_cache/test_dsa_layer_split_broadcast.py
+3 -6
View File
@@ -45,10 +45,7 @@ def _run(rank: int, world: int, port: int):
init_distributed_environment,
initialize_model_parallel,
)
from sglang.srt.layers.dp_attention import (
get_attention_cp_rank,
get_attention_cp_size,
)
from sglang.srt.runtime_context import get_parallel
init_distributed_environment(
world_size=world,
@@ -66,8 +63,8 @@ def _run(rank: int, world: int, port: int):
LayerSplitDSATokenToKVPool,
)
cp_rank = get_attention_cp_rank()
cp_size = get_attention_cp_size()
cp_rank = get_parallel().attn_cp_rank
cp_size = get_parallel().attn_cp_size
assert cp_size == world
pool = LayerSplitDSATokenToKVPool(